/*
 * (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 "TxtXmlFile.h"

#include "ConvertDocx2Txt.h"
#include "ConvertTxt2Docx.h"

#include "../../OOXML/Base/Unit.h"

#include "../../DesktopEditor/common/SystemUtils.h"
#include "../../Common/OfficeFileErrorDescription.h"

#include "../../OOXML/DocxFormat/Docx.h"
#include "../../OOXML/DocxFormat/App.h"
#include "../../OOXML/DocxFormat/Core.h"

namespace NSBinPptxRW
{
	class  CDrawingConverter;
}

#include "../../Common/OfficeDefines.h"
#include "../../OOXML/Binary/Document/BinReader/FileWriter.h"

CTxtXmlFile::CTxtXmlFile()
{
}

static int ParseTxtOptions(const std::wstring & sXmlOptions)
{
	int encoding = -1;
	
	XmlUtils::CXmlLiteReader xmlReader;
	
	if (xmlReader.FromString(sXmlOptions))
	{
		xmlReader.ReadNextNode();//root - <Options>

		int nCurDepth = xmlReader.GetDepth();
		while ( xmlReader.ReadNextSiblingNode( nCurDepth ) )
		{
			std::wstring sName = xmlReader.GetName();

			if (sName == _T("TXTOptions"))
			{
				int nCurDepth1 = xmlReader.GetDepth();
				while ( xmlReader.ReadNextSiblingNode( nCurDepth1 ) )
				{
					std::wstring sName1 = xmlReader.GetName();
					if (sName1 == _T("Encoding"))
					{
						std::wstring strValue = xmlReader.GetText2();
						encoding = XmlUtils::GetInteger(strValue);
					}
				}
			}
		}
	}
	return encoding;
}


_UINT32 CTxtXmlFile::txt_LoadFromFile(const std::wstring & sSrcFileName, const std::wstring & sDstPath, const std::wstring & sXMLOptions)
{
	Writers::FileWriter *pDocxWriter =  new Writers::FileWriter(sDstPath, L"", true, 1, NULL, L"");
	if (pDocxWriter == NULL) return AVS_FILEUTILS_ERROR_CONVERT;

	CreateDocxEmpty(sDstPath, pDocxWriter);

	try
	{
		int encoding  = ParseTxtOptions(sXMLOptions);

		Txt2Docx::Converter converter( encoding);
		converter.read(sSrcFileName);
		converter.convert();
		converter.write(pDocxWriter->get_document_writer().m_oContent);
	}
	catch(...)
	{
		return AVS_FILEUTILS_ERROR_CONVERT;
	}

	pDocxWriter->get_document_writer().Write(); //overwrite document.xml

	delete pDocxWriter;
	pDocxWriter = NULL;

	return 0;
}


_UINT32 CTxtXmlFile::txt_SaveToFile(const std::wstring & sDstFileName, const std::wstring & sSrcPath, const std::wstring & sXMLOptions)
{
	bool result = true;

	try
	{
		Docx2Txt::Converter converter;
		result = converter.read(sSrcPath);
		if (result)
		{
			converter.convert();

			int encoding = ParseTxtOptions(sXMLOptions);

			if (encoding == EncodingType::Utf8)
				result = converter.writeUtf8(sDstFileName);
			else if (encoding == EncodingType::Unicode)
				result = converter.writeUnicode(sDstFileName);
			else if (encoding == EncodingType::Ansi)
				result = converter.writeAnsi(sDstFileName);
			else if (encoding == EncodingType::BigEndian)
				result = converter.writeBigEndian(sDstFileName);
			else if (encoding > 0) //code page
			{
				result = converter.write(sDstFileName);
			}
			else //auto define
				result = converter.write(sDstFileName);
		}
	}
	catch(...)
	{
		result = false;
	}

	return result ? 0 : AVS_FILEUTILS_ERROR_CONVERT;
}


void CTxtXmlFile::CreateDocxEmpty(const std::wstring & _strDirectory, Writers::FileWriter * pDocxWriter)
{
	std::wstring strDirectory = _strDirectory;
	// rels
	OOX::CPath pathRels = strDirectory + FILE_SEPARATOR_STR +L"_rels";
	NSDirectory::CreateDirectory(pathRels.GetPath());

	// word
	OOX::CPath pathWord = strDirectory + FILE_SEPARATOR_STR + L"word";
	NSDirectory::CreateDirectory(pathWord.GetPath());

	// documentRels
	OOX::CPath pathWordRels = pathWord + FILE_SEPARATOR_STR + _T("_rels");
	NSDirectory::CreateDirectory(pathWordRels.GetPath());

	//media
	OOX::CPath pathMedia = pathWord + FILE_SEPARATOR_STR + _T("media");
	std::wstring sMediaPath = pathMedia.GetPath();

	// theme
	OOX::CPath pathTheme = pathWord + FILE_SEPARATOR_STR + _T("theme");
	NSDirectory::CreateDirectory(pathTheme.GetPath());

	OOX::CPath pathThemeRels = pathTheme + FILE_SEPARATOR_STR + _T("_rels");
	NSDirectory::CreateDirectory(pathThemeRels.GetPath());
	
	pathTheme = pathTheme + FILE_SEPARATOR_STR + _T("theme1.xml");

	//default files

	pDocxWriter->m_oTheme.Write(pathTheme.GetPath());

	OOX::CContentTypes oContentTypes;
	//docProps
	OOX::CPath pathDocProps = strDirectory + FILE_SEPARATOR_STR + _T("docProps");
	NSDirectory::CreateDirectory(pathDocProps.GetPath());
	
	OOX::CPath DocProps = std::wstring(_T("docProps"));

	OOX::CApp* pApp = new OOX::CApp(NULL);
	if (pApp)
	{
		std::wstring sApplication = NSSystemUtils::GetEnvVariable(NSSystemUtils::gc_EnvApplicationName);
		if (sApplication.empty())
			sApplication = NSSystemUtils::gc_EnvApplicationNameDefault;
#if defined(INTVER)
		std::string s = VALUE2STR(INTVER);
		sApplication += L"/" + std::wstring(s.begin(), s.end());
#endif
		pApp->m_sApplication = sApplication;
		pApp->SetDefaults();
		
		pApp->write(pathDocProps + FILE_SEPARATOR_STR + _T("app.xml"), DocProps, oContentTypes);
		delete pApp;
	}
	OOX::CCore* pCore = new OOX::CCore(NULL);
	if (pCore)
	{
		pCore->SetCreator(_T(""));
		pCore->SetLastModifiedBy(_T(""));
		pCore->write(pathDocProps + FILE_SEPARATOR_STR + _T("core.xml"), DocProps, oContentTypes);
		delete pCore;
	}
	/////////////////////////////////////////////////////////////////////////////////////
	pDocxWriter->m_oTheme.Write(strDirectory);
	pDocxWriter->get_style_writers().Write();
	pDocxWriter->get_font_table_writer().Write();
	
	pDocxWriter->get_settings_writer().Write();
	pDocxWriter->get_web_settings_writer().Write();
	
	pDocxWriter->get_document_writer().Write();
	
	pDocxWriter->m_oDocumentRelsWriter.Write();

	oContentTypes.Registration(L"application/vnd.openxmlformats-officedocument.wordprocessingml.document.main+xml",	OOX::CPath(L"/word"),		OOX::CPath(L"document.xml"));
	oContentTypes.Registration(L"application/vnd.openxmlformats-officedocument.wordprocessingml.styles+xml",		OOX::CPath(L"/word"),		OOX::CPath(L"styles.xml"));
	oContentTypes.Registration(L"application/vnd.openxmlformats-officedocument.wordprocessingml.settings+xml",		OOX::CPath(L"/word"),		OOX::CPath(L"settings.xml"));
	oContentTypes.Registration(L"application/vnd.openxmlformats-officedocument.wordprocessingml.webSettings+xml",	OOX::CPath(L"/word"),		OOX::CPath(L"webSettings.xml"));
	oContentTypes.Registration(L"application/vnd.openxmlformats-officedocument.wordprocessingml.fontTable+xml",		OOX::CPath(L"/word"),		OOX::CPath(L"fontTable.xml"));
	oContentTypes.Registration(L"application/vnd.openxmlformats-officedocument.theme+xml",							OOX::CPath(L"/word/theme"), OOX::CPath(L"theme1.xml"));

	oContentTypes.Write(strDirectory);
}