/*
 * (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 "Presentation.h"
#include "FileContainer.h"
#include "FileTypes.h"

#include "CommentAuthors.h"

#include "Limit/Conformance.h"
#include "Logic/TextListStyle.h"
#include "Logic/ClrMap.h"
#include "Logic/ExtP.h"
#include "Theme/ClrScheme.h"
#include "Comments.h"

#include "../DocxFormat/Media/VbaProject.h"
#include "../DocxFormat/Media/JsaProject.h"

#include "../../MsBinaryFile/Common/Vba/VbaReader.h"

namespace PPTX
{
	Presentation::Presentation(OOX::Document *pMain) : WrapperFile(pMain), PPTX::FileContainer(pMain)
	{
		m_bMacroEnabled = false;
	}
	Presentation::Presentation(OOX::Document *pMain, const OOX::CPath& filename, FileMap& map) : WrapperFile(pMain), PPTX::FileContainer(pMain)
	{
		m_bMacroEnabled = false;
		read(filename, map);
	}
	Presentation::~Presentation()
	{
	}
	void Presentation::read(const OOX::CPath& filename, FileMap& map)
	{
		embeddedFontLst.clear();
		handoutMasterIdLst.clear();
		sldIdLst.clear();
		sldMasterIdLst.clear();

		XmlUtils::CXmlNode oNodeRoot;
		oNodeRoot.FromXmlFile(filename.m_strFilename);

		XmlMacroReadAttributeBase(oNodeRoot, L"autoCompressPictures", attrAutoCompressPictures);
		XmlMacroReadAttributeBase(oNodeRoot, L"bookmarkIdSeed", attrBookmarkIdSeed);
		XmlMacroReadAttributeBase(oNodeRoot, L"compatMode", attrCompatMode);
		XmlMacroReadAttributeBase(oNodeRoot, L"conformance", attrConformance);
		XmlMacroReadAttributeBase(oNodeRoot, L"embedTrueTypeFonts", attrEmbedTrueTypeFonts);
		XmlMacroReadAttributeBase(oNodeRoot, L"firstSlideNum", attrFirstSlideNum);
		XmlMacroReadAttributeBase(oNodeRoot, L"removePersonalInfoOnSave", attrRemovePersonalInfoOnSave);
		XmlMacroReadAttributeBase(oNodeRoot, L"rtl", attrRtl);
		XmlMacroReadAttributeBase(oNodeRoot, L"saveSubsetFonts", attrSaveSubsetFonts);
		XmlMacroReadAttributeBase(oNodeRoot, L"serverZoom", attrServerZoom);
		XmlMacroReadAttributeBase(oNodeRoot, L"showSpecialPlsOnTitleSld", attrShowSpecialPlsOnTitleSld);
		XmlMacroReadAttributeBase(oNodeRoot, L"strictFirstAndLastChars", attrStrictFirstAndLastChars);

		//smartTags (Smart Tags)
		//modifyVerifier (Modification Verifier)

		std::vector<XmlUtils::CXmlNode> oNodes;
		if (oNodeRoot.GetNodes(L"*", oNodes))
		{
			for (size_t i = 0; i < oNodes.size(); ++i)
			{
				XmlUtils::CXmlNode& oNode = oNodes[i];

				std::wstring strName = XmlUtils::GetNameNoNS(oNode.GetName());

				if (L"custDataLst" == strName)
				{
					std::vector<XmlUtils::CXmlNode> oCustDataLstNodes;
					if (oNode.GetNodes(L"custData", oCustDataLstNodes))
					{
						for (auto n : oCustDataLstNodes)
						{
							custDataLst.emplace_back();
							XmlMacroReadAttributeBase(n, L"r:id", custDataLst.back());
						}
					}
				}
				else if (L"custShowLst" == strName)
				{
					custShowLst = oNode;
					if (custShowLst.is_init()) custShowLst->SetParentFilePointer(this);
				}
				else if (L"defaultTextStyle" == strName)
				{
					defaultTextStyle = oNode;
					if (defaultTextStyle.is_init()) defaultTextStyle->SetParentFilePointer(this);
				}
				else if (L"embeddedFontLst" == strName)
				{
					XmlMacroLoadArray(oNode, L"p:embeddedFont", embeddedFontLst, nsPresentation::EmbeddedFont);

					for (size_t i = 0; i < embeddedFontLst.size(); ++i)
						embeddedFontLst[i].SetParentFilePointer(this);
				}
				else if (L"handoutMasterIdLst" == strName)
				{
					XmlMacroLoadArray(oNode, L"p:handoutMasterId", handoutMasterIdLst, Logic::XmlId);

					for (size_t i = 0; i < handoutMasterIdLst.size(); ++i)
						handoutMasterIdLst[i].SetParentFilePointer(this);
				}
				else if (L"kinsoku" == strName)
				{
					kinsoku = oNode;
					if (kinsoku.is_init()) kinsoku->SetParentFilePointer(this);
				}
				else if (L"notesMasterIdLst" == strName)
				{
					XmlMacroLoadArray(oNode, L"p:notesMasterId", notesMasterIdLst, Logic::XmlId);

					for (size_t i = 0; i < notesMasterIdLst.size(); ++i)
						notesMasterIdLst[i].SetParentFilePointer(this);
				}
				else if (L"notesSz" == strName)
				{
					notesSz = oNode;
					if (notesSz.is_init()) notesSz->SetParentFilePointer(this);
				}
				else if (L"photoAlbum" == strName)
				{
					photoAlbum = oNode;
					if (photoAlbum.is_init()) photoAlbum->SetParentFilePointer(this);
				}
				else if (L"sldIdLst" == strName)
				{
					XmlMacroLoadArray(oNode, L"p:sldId", sldIdLst, Logic::XmlId);

					for (size_t i = 0; i < sldIdLst.size(); ++i)
						sldIdLst[i].SetParentFilePointer(this);
				}
				else if (L"sldMasterIdLst" == strName)
				{
					XmlMacroLoadArray(oNode, L"p:sldMasterId", sldMasterIdLst, Logic::XmlId);

					for (size_t i = 0; i < sldMasterIdLst.size(); ++i)
						sldMasterIdLst[i].SetParentFilePointer(this);
				}
				else if (L"sldSz" == strName)
				{
					sldSz = oNode;
					if (sldSz.is_init()) sldSz->SetParentFilePointer(this);
				}
				else if (L"extLst" == strName)
				{
					if (oNode.IsValid())
					{
						std::vector<XmlUtils::CXmlNode> oExtLstNodes;
						if (oNode.GetNodes(L"*", oExtLstNodes))
						{
							for (auto n : oExtLstNodes)
							{
								PPTX::Logic::Ext ext;
								ext.fromXML(n);
								if (ext.sectionLst.IsInit())
								{
									sectionLst = ext.sectionLst;
								}
							}
						}
					}
				}
			}
		}
		Normalize();
	}
	void Presentation::write(const OOX::CPath& filename, const OOX::CPath& directory, OOX::CContentTypes& content)const
	{
		WrapperFile::write(filename, directory, content);
		FileContainer::write(filename, directory, content);
	}

	void Presentation::toPPTY(NSBinPptxRW::CBinaryFileWriter* pWriter) const
	{
		pWriter->StartRecord(NSBinPptxRW::NSMainTables::Presentation);

		pWriter->WriteBYTE(NSBinPptxRW::g_nodeAttributeStart);

		pWriter->WriteBool2(0, attrAutoCompressPictures);
		pWriter->WriteInt2(1, attrBookmarkIdSeed);
		pWriter->WriteBool2(2, attrCompatMode);
		pWriter->WriteLimit2(3, attrConformance);
		pWriter->WriteBool2(4, attrEmbedTrueTypeFonts);
		pWriter->WriteInt2(5, attrFirstSlideNum);
		pWriter->WriteBool2(6, attrRemovePersonalInfoOnSave);
		pWriter->WriteBool2(7, attrRtl);
		pWriter->WriteBool2(8, attrSaveSubsetFonts);
		pWriter->WriteString2(9, attrServerZoom);
		pWriter->WriteBool2(10, attrShowSpecialPlsOnTitleSld);
		pWriter->WriteBool2(11, attrStrictFirstAndLastChars);

		pWriter->WriteBYTE(NSBinPptxRW::g_nodeAttributeEnd);

		pWriter->WriteRecord2(0, defaultTextStyle);
		pWriter->WriteRecordArray(1, 0, embeddedFontLst);
		pWriter->WriteRecord2(2, kinsoku);
		pWriter->WriteRecord2(3, notesSz);
		pWriter->WriteRecord2(4, photoAlbum);
		pWriter->WriteRecord2(5, sldSz);

		pWriter->WriteRecord2(6, commentAuthors);
		pWriter->WriteRecord2(7, sectionLst);

		if (m_pVbaProject.IsInit())
		{
			pWriter->StartRecord(8);
			{
				m_pVbaProject->toPPTY(pWriter);
			}
			pWriter->EndRecord();
		}
		pWriter->WriteRecord2(9, m_pJsaProject);
		pWriter->WriteRecord2(10, comments);

		pWriter->EndRecord();
	}
	void Presentation::fromPPTY(NSBinPptxRW::CBinaryFileReader* pReader)
	{
		BYTE _type = pReader->GetUChar();
		LONG _len = pReader->GetULong();
		LONG _start_pos = pReader->GetPos();
		LONG _end_pos = _len + _start_pos;

		// attributes
		BYTE _sa = pReader->GetUChar();

		while (true)
		{
			BYTE _at = pReader->GetUChar_TypeNode();

			if (_at == NSBinPptxRW::g_nodeAttributeEnd)
				break;

			switch (_at)
			{
			case 0: { attrAutoCompressPictures = pReader->GetBool(); break; }
			case 1: { attrBookmarkIdSeed = pReader->GetLong(); break; }
			case 2: { attrCompatMode = pReader->GetBool(); break; }
			case 3: { attrConformance = pReader->GetUChar(); break; }
			case 4: { attrEmbedTrueTypeFonts = pReader->GetBool(); break; }
			case 5: { attrFirstSlideNum = pReader->GetLong(); break; }
			case 6: { attrRemovePersonalInfoOnSave = pReader->GetBool(); break; }
			case 7: { attrRtl = pReader->GetBool(); break; }
			case 8: { attrSaveSubsetFonts = pReader->GetBool(); break; }
			case 9: { attrServerZoom = pReader->GetString2(); break; }
			case 10: { attrShowSpecialPlsOnTitleSld = pReader->GetBool(); break; }
			case 11: { attrStrictFirstAndLastChars = pReader->GetBool(); break; }
			default:
				return;
			}
		}

		while (pReader->GetPos() < _end_pos)
		{
			_type = pReader->GetUChar();

			switch (_type)
			{
			case 0:
			{
				defaultTextStyle = PPTX::Logic::TextListStyle();
				defaultTextStyle->m_name = L"p:defaultTextStyle";
				defaultTextStyle->fromPPTY(pReader);
				break;
			}
			case 1: { pReader->SkipRecord(); break; }
			case 2: { pReader->SkipRecord(); break; }
			case 3:
			{
				notesSz = new nsPresentation::NotesSz();
				pReader->Skip(5); // len + start attributes

				while (true)
				{
					BYTE _at = pReader->GetUChar_TypeNode();

					if (_at == NSBinPptxRW::g_nodeAttributeEnd)
						break;

					switch (_at)
					{
					case 0: { notesSz->cx = pReader->GetLong(); break; }
					case 1: { notesSz->cy = pReader->GetLong(); break; }
					default:
						return;
					}
				}
				break;
			}
			case 4: { pReader->SkipRecord(); break; }
			case 5:
			{
				sldSz = new nsPresentation::SldSz();
				pReader->Skip(5); // len + start attributes

				while (true)
				{
					BYTE _at = pReader->GetUChar_TypeNode();

					if (_at == NSBinPptxRW::g_nodeAttributeEnd)
						break;

					switch (_at)
					{
					case 0: { sldSz->cx = pReader->GetLong(); break; }
					case 1: { sldSz->cy = pReader->GetLong(); break; }
					case 2: { sldSz->type = pReader->GetUChar(); break; }
					default:
						return;
					}
				}
			}break;
			case 6:
			{
				if (false == commentAuthors.IsInit())
				{
					commentAuthors = new PPTX::Authors(File::m_pMainDocument);
				}
				commentAuthors->fromPPTY(pReader);
			}break;
			case 7:
			{
				sectionLst = new nsPresentation::SectionLst();
				sectionLst->fromPPTY(pReader);
			}break;
			case 8:
			{
				if (m_bMacroEnabled)
				{
					m_pVbaProject = new OOX::VbaProject(File::m_pMainDocument);
					m_pVbaProject->fromPPTY(pReader);

					smart_ptr<OOX::File> file = m_pVbaProject.smart_dynamic_cast<OOX::File>();
					FileContainer::Add(file);
				}
				else
					pReader->SkipRecord();
			}break;
			case 9:
			{
				m_pJsaProject = new OOX::JsaProject(File::m_pMainDocument);
				m_pJsaProject->fromPPTY(pReader);

				smart_ptr<OOX::File> file = m_pJsaProject.smart_dynamic_cast<OOX::File>();
				FileContainer::Add(file);
			}break;
			case 10:
			{
				comments = new PPTX::Comments(OOX::File::m_pMainDocument);
				comments->fromPPTY(pReader);
			}break;
			default:
			{
				pReader->SkipRecord();
				return;
			}
			}
		}
		pReader->Seek(_end_pos);
		
		m_bMacroEnabled = m_pVbaProject.IsInit();
	}

	void Presentation::toXmlWriter(NSBinPptxRW::CXmlWriter* pWriter) const
	{
		pWriter->StartNode(L"p:presentation");

		pWriter->StartAttributes();

		pWriter->WriteAttribute(L"xmlns:a", PPTX::g_Namespaces.a.m_strLink);
		pWriter->WriteAttribute(L"xmlns:r", PPTX::g_Namespaces.r.m_strLink);
		pWriter->WriteAttribute(L"xmlns:p", PPTX::g_Namespaces.p.m_strLink);
		pWriter->WriteAttribute(L"xmlns:m", PPTX::g_Namespaces.m.m_strLink);
		pWriter->WriteAttribute(L"xmlns:w", PPTX::g_Namespaces.w.m_strLink);

		pWriter->WriteAttribute(L"autoCompressPictures", attrAutoCompressPictures);
		pWriter->WriteAttribute(L"bookmarkIdSeed", attrBookmarkIdSeed);
		pWriter->WriteAttribute(L"compatMode", attrCompatMode);
		pWriter->WriteAttribute(L"conformance", attrConformance);
		pWriter->WriteAttribute(L"embedTrueTypeFonts", attrEmbedTrueTypeFonts);
		pWriter->WriteAttribute(L"firstSlideNum", attrFirstSlideNum);
		pWriter->WriteAttribute(L"removePersonalInfoOnSave", attrRemovePersonalInfoOnSave);
		pWriter->WriteAttribute(L"rtl", attrRtl);
		pWriter->WriteAttribute(L"saveSubsetFonts", attrSaveSubsetFonts);
		pWriter->WriteAttribute(L"serverZoom", attrServerZoom);
		pWriter->WriteAttribute(L"showSpecialPlsOnTitleSld", attrShowSpecialPlsOnTitleSld);
		pWriter->WriteAttribute(L"strictFirstAndLastChars", attrStrictFirstAndLastChars);

		pWriter->EndAttributes();

		pWriter->WriteArray(L"p:sldMasterIdLst", sldMasterIdLst);
		pWriter->WriteArray(L"p:notesMasterIdLst", notesMasterIdLst);
		pWriter->WriteArray(L"p:handoutMasterIdLst", handoutMasterIdLst);
		pWriter->WriteArray(L"p:embeddedFontLst", embeddedFontLst);
		pWriter->WriteArray(L"p:sldIdLst", sldIdLst);

		pWriter->Write(sldSz);
		pWriter->Write(notesSz);
		pWriter->Write(photoAlbum);
		pWriter->Write(kinsoku);

		if (false == custDataLst.empty())
		{
			pWriter->StartNode(L"p:custDataLst");
			pWriter->EndAttributes();

			for (auto cust : custDataLst)
			{
				pWriter->StartNode(L"p:custData");
				pWriter->StartAttributes();
				pWriter->WriteAttribute(L"r:id", cust.ToString());

				pWriter->EndAttributes();
				pWriter->EndNode(L"p:custData");
			}

			pWriter->EndNode(L"p:custDataLst");
		}
		pWriter->Write(custShowLst);

		pWriter->Write(defaultTextStyle);

		std::vector<Logic::Ext> extLst;

		if (sectionLst.IsInit())
		{
			Logic::Ext exp;
			exp.sectionLst = sectionLst;
			extLst.push_back(exp);
		}
		pWriter->WriteArray(L"p:extLst", extLst);

		pWriter->EndNode(L"p:presentation");
	}

	void Presentation::Normalize()
	{
		attrBookmarkIdSeed.normalize(1, 2147483647);
		attrFirstSlideNum.normalize_positive();
	}
} // namespace PPTX