/*
 * (c) Copyright Ascensio System SIA 2010-2023
 *
 * This program is a free software product. You can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License (AGPL)
 * version 3 as published by the Free Software Foundation. In accordance with
 * Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
 * that Ascensio System SIA expressly excludes the warranty of non-infringement
 * of any third-party rights.
 *
 * This program is distributed WITHOUT ANY WARRANTY; without even the implied
 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR  PURPOSE. For
 * details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
 *
 * You can contact Ascensio System SIA at 20A-6 Ernesta Birznieka-Upish
 * street, Riga, Latvia, EU, LV-1050.
 *
 * The  interactive user interfaces in modified source and object code versions
 * of the Program must display Appropriate Legal Notices, as required under
 * Section 5 of the GNU AGPL version 3.
 *
 * Pursuant to Section 7(b) of the License you must retain the original Product
 * logo when distributing the program. Pursuant to Section 7(e) we decline to
 * grant you any rights under trademark law for use of our trademarks.
 *
 * All the Product's GUI elements, including illustrations and icon sets, as
 * well as technical writing content are licensed under the terms of the
 * Creative Commons Attribution-ShareAlike 4.0 International. See the License
 * terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
 *
 */

#include "OleObjectMapping.h"

namespace DocFileFormat
{	
	OleObjectMapping::OleObjectMapping(XMLTools::CStringXmlWriter* writer, ConversionContext* context, PictureDescriptor* pict, IMapping* caller, const std::wstring& shapeId)
		:
	OleObjectMapping::AbstractOpenXmlMapping(writer), m_context(NULL), _pict(NULL), _caller(NULL), _shapeId(shapeId)
	{
		m_context	=	context;
		_pict		=	pict;
		_caller		=	caller;
	}

	void OleObjectMapping::Apply(IVisitable* visited)
	{
		OleObject* ole = dynamic_cast<OleObject*>(visited);
		if (!ole) return;
		
		if (ole->isEmbedded || ole->isPackage)
		{
			if (ole->isEquation) ole->ClipboardFormat = L"Equation";
			else if (ole->ClipboardFormat.empty()) ole->ClipboardFormat = L"MSWordDocx";

			ole->Program = L"Word.Document";
		}
		m_pXmlWriter->WriteNodeBegin(L"o:OLEObject", TRUE);

		int relID = -1;

		if (ole->isLinked)
		{
			relID = m_context->_docx->RegisterExternalOLEObject(_caller, ole->ClipboardFormat, ole->Link);

			m_pXmlWriter->WriteAttribute(L"Type", L"Link");
			m_pXmlWriter->WriteAttribute(L"UpdateMode", ole->UpdateMode);
		}
		else
		{
			if (ole->isEmbedded || ole->isPackage)
				relID = m_context->_docx->RegisterPackage(_caller, ole->ClipboardFormat);
			else
				relID = m_context->_docx->RegisterOLEObject(_caller, ole->ClipboardFormat);

			m_pXmlWriter->WriteAttribute(L"Type", L"Embed");

			copyEmbeddedObject(ole);
		}

		m_pXmlWriter->WriteAttribute(L"ProgID", ole->Program);
		m_pXmlWriter->WriteAttribute(L"ShapeID", _shapeId);
		m_pXmlWriter->WriteAttribute(L"DrawAspect", L"Content");
		m_pXmlWriter->WriteAttribute(L"ObjectID", ole->ObjectId);
		m_pXmlWriter->WriteAttribute(L"r:id", L"rId" + FormatUtils::IntToWideString(relID));
		m_pXmlWriter->WriteNodeEnd(L"", TRUE, FALSE);

		m_pXmlWriter->WriteNodeEnd(L"o:OLEObject");
	}

	std::wstring OleObjectMapping::GetTargetExt(const std::wstring& objectType)
	{
		std::wstring objectExt = L".bin";

		if ( objectType == L"Biff8" )
		{
			objectExt = L".xls";
		}
		else if ( objectType == L"MSWordDoc" )
		{
			objectExt = L".doc";
		}
		else if ( objectType == L"MSPresentation" )
		{
			objectExt = L".ppt";
		}
		else if ( objectType == L"MSWordDocx" )
		{
			objectExt = L".docx";
		}
		else if ( objectType == L"Equation" )
		{
			objectExt = L".xml";
		}
		return objectExt;
	}

	std::wstring OleObjectMapping::GetContentType(const std::wstring& objectType)
	{
		std::wstring objectContentType = OpenXmlContentTypes::OleObject;

		if ( objectType == L"Biff8" )
		{
			objectContentType = OpenXmlContentTypes::MSExcel;
		}
		else if ( objectType == L"MSWordDoc" )
		{
			objectContentType = OpenXmlContentTypes::MSWord;
		}
		else if ( objectType == L"MSPresentation" )
		{
			objectContentType = OpenXmlContentTypes::MSPowerpoint;
		}
		else if ( objectType == L"MSWordDocx" )
		{
			objectContentType = OpenXmlContentTypes::MSWordDocx;
		}
		else if ( objectType == L"Equation" )
		{
			objectContentType = OpenXmlContentTypes::Xml;
		}
		return objectContentType;
	}

	void OleObjectMapping::copyEmbeddedObject( const OleObject* ole )
	{
		if ( ole == NULL ) return;

		std::wstring clsid;
		std::wstring exelChart = L"Excel.Chart";

		if ( std::search( ole->Program.begin(), ole->Program.end(), exelChart.begin(), exelChart.end() ) == ole->Program.end() )
		{
			clsid = ole->ClassId;
		}
		OleObjectFileStructure object_descr(OleObjectMapping::GetTargetExt( ole->ClipboardFormat ), ole->ObjectId, clsid);

		if (ole->nWordVersion == 2)
		{
			object_descr.clsid = ole->ClipboardFormat;
			object_descr.bNativeOnly = true;
		}
		if (ole->isEquation || ole->isEmbedded || ole->nWordVersion == 2 || ole->isPackage)
		{
			object_descr.data = ole->embeddedData;
		}

		m_context->_docx->OleObjectsList.push_back(object_descr);
	}
}