/*
 * Copyright (C) Ascensio System SIA, 2009-2026
 *
 * 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, together with the
 * additional terms provided in the LICENSE file.
 *
 * 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: https://www.gnu.org/licenses/agpl-3.0.html
 *
 * You can contact Ascensio System SIA by email at info@onlyoffice.com
 * or by postal mail at 20A-6 Ernesta Birznieka-Upisha Street, Riga,
 * LV-1050, Latvia, European Union.
 *
 * The interactive user interfaces in modified versions of the Program
 * are required to display Appropriate Legal Notices in accordance with
 * Section 5 of the GNU AGPL version 3.
 *
 * No trademark rights are granted under this License.
 *
 * All non-code elements of the Product, including illustrations,
 * icon sets, and technical writing content, are licensed under the
 * Creative Commons Attribution-ShareAlike 4.0 International License:
 * https://creativecommons.org/licenses/by-sa/4.0/legalcode
 *
 * This license applies only to such non-code elements and does not
 * modify or replace the licensing terms applicable to the Program's
 * source code, which remains licensed under the GNU Affero General
 * Public License v3.
 *
 * SPDX-License-Identifier: AGPL-3.0-only
 */

#include "IWork.h"
#include "../DesktopEditor/common/File.h"
#include "../DesktopEditor/common/Directory.h"

#include <libetonyek/libetonyek.h>
#include <libodfgen/OdtGenerator.hxx>
#include <libodfgen/OdsGenerator.hxx>
#include <libodfgen/OdpGenerator.hxx>
#include <libodfgen/test/StringDocumentHandler.hxx>

#include <memory>
#include <fstream>

class CIWorkFile_Private
{
public:
	std::wstring m_sTempDirectory;

public:
	CIWorkFile_Private()
	{
	}
	~CIWorkFile_Private()
	{
	}
};

CIWorkFile::CIWorkFile()
{
	m_internal = new CIWorkFile_Private();
}

CIWorkFile::~CIWorkFile()
{
	delete m_internal;
}

#if !defined(_WIN32) && !defined(_WIN64)
	#define DATA_TYPE_INPUTFILE std::string
#else
	#define DATA_TYPE_INPUTFILE std::wstring
#endif

bool GetRVNGInputStream(const DATA_TYPE_INPUTFILE& sFile, std::shared_ptr<librevenge::RVNGInputStream>& oRVNGInputStream, libetonyek::EtonyekDocument::Type& oDocumentType)
{
	oRVNGInputStream.reset(new librevenge::RVNGFileStream(sFile.c_str()));

	oDocumentType = libetonyek::EtonyekDocument::TYPE_UNKNOWN;
	const libetonyek::EtonyekDocument::Confidence confidence = libetonyek::EtonyekDocument::isSupported(oRVNGInputStream.get(), &oDocumentType);

	return libetonyek::EtonyekDocument::CONFIDENCE_NONE != confidence;
}

IWorkFileType CIWorkFile::GetType(const std::wstring& sFile) const
{
	//TODO:: since currently we work directly with file only, directory handling is not needed yet
	if (NSDirectory::PathIsDirectory(sFile))
		return IWorkFileType::None;

	std::shared_ptr<librevenge::RVNGInputStream> input;
	libetonyek::EtonyekDocument::Type oDocumentType;

	#if !defined(_WIN32) && !defined(_WIN64)
	std::string sFileA = U_TO_UTF8(sFile);
	if (!GetRVNGInputStream(sFileA, input, oDocumentType))
		return IWorkFileType::None;
	#else
	if (!GetRVNGInputStream(sFile, input, oDocumentType))
		return IWorkFileType::None;
	#endif

	switch (oDocumentType)
	{
	case libetonyek::EtonyekDocument::TYPE_PAGES:
		return IWorkFileType::Pages;
	case libetonyek::EtonyekDocument::TYPE_NUMBERS:
		return IWorkFileType::Numbers;
	case libetonyek::EtonyekDocument::TYPE_KEYNOTE:
		return IWorkFileType::Keynote;
	default:
		break;
	}

	return IWorkFileType::None;
}

template<class Generator>
int Convert(const std::wstring& wsOutputFile, std::shared_ptr<librevenge::RVNGInputStream>& ptrInput, const std::wstring& wsPassword = L"", const std::wstring& wsTempDirectory = L"")
{
	StringDocumentHandler content;
	Generator generator;
	generator.addDocumentHandler(&content, ODF_FLAT_XML);

	bool bRes = libetonyek::EtonyekDocument::parse(ptrInput.get(), &generator);
	if (!bRes)
		return 1;

	const std::string sOutputFileA = U_TO_UTF8(wsOutputFile);
	std::ofstream output(sOutputFileA.c_str());
	output << content.cstr();

	if (output.bad())
		return -1;

	return 0;
}

int CIWorkFile::Convert2Odf(const std::wstring& sFile, const std::wstring& sOutputFile) const
{
	//TODO:: since currently we work directly with file only, directory handling is not needed yet
	if (NSDirectory::PathIsDirectory(sFile))
		return -1;

	std::shared_ptr<librevenge::RVNGInputStream> input;
	libetonyek::EtonyekDocument::Type oDocumentType;

	#if !defined(_WIN32) && !defined(_WIN64)
		std::string sFileA = U_TO_UTF8(sFile);
		if (!GetRVNGInputStream(sFileA, input, oDocumentType))
			return -1;
	#else
		if (!GetRVNGInputStream(sFile, input, oDocumentType))
			return -1;
	#endif

	switch (oDocumentType)
	{
	case libetonyek::EtonyekDocument::TYPE_PAGES:   return Convert<OdtGenerator>(sOutputFile, input);
	case libetonyek::EtonyekDocument::TYPE_NUMBERS: return Convert<OdsGenerator>(sOutputFile, input);
	case libetonyek::EtonyekDocument::TYPE_KEYNOTE: return Convert<OdpGenerator>(sOutputFile, input);
	default:
		break;
	}

	return -1;
}

void CIWorkFile::SetTmpDirectory(const std::wstring& sFolder)
{
	m_internal->m_sTempDirectory = sFolder;
}