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

#include "../../OOXML/PPTXFormat/FileTypes.h"
#include "../../OOXML/PPTXFormat/Folder.h"
#include "../../OOXML/PPTXFormat/Slide.h"
#include "../../OOXML/PPTXFormat/Presentation.h"
#include "../../OOXML/PPTXFormat/Logic/Shape.h"
#include "../../OOXML/PPTXFormat/Logic/CSld.h"
#include "../../OOXML/PPTXFormat/Logic/SpTree.h"
#include "../../OOXML/PPTXFormat/Logic/SpTreeElem.h"

#include "../../OOXML/DocxFormat/WritingElement.h"

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

class CPptxTxtConverter::CPptxTxtConverterImpl
{
public:
	CPptxTxtConverterImpl() = default;
	CPptxTxtConverterImpl(const CPptxTxtConverterImpl& other) = delete;
	CPptxTxtConverterImpl(CPptxTxtConverterImpl&& other) = delete;
	virtual ~CPptxTxtConverterImpl();

	void AddSlide(const PPTX::Slide& oSlide);

	// returns true if ok
	void Save(const std::wstring& wsDstTxtFile);

private:
	std::wstring GetTextFromShape(const PPTX::Logic::Shape& oShape) const;
	std::vector<std::wstring> m_arTxtData;

	const size_t knStringReserve = 1000;
};

CPptxTxtConverter::CPptxTxtConverter()
    : m_pImpl(std::unique_ptr<CPptxTxtConverterImpl>(new CPptxTxtConverterImpl()))
{
}
CPptxTxtConverter::~CPptxTxtConverter()
{
}
int CPptxTxtConverter::Convert(const std::wstring& wsSrcPptxDir, const std::wstring& wsDstTxtFile)
{
	std::wstring norm_src_pptx_dir = wsSrcPptxDir;
	if (norm_src_pptx_dir.find(L"./") == 0)
		norm_src_pptx_dir.erase(0, 2);

	norm_src_pptx_dir = CorrectPathW(norm_src_pptx_dir);
	if (norm_src_pptx_dir.find(L"\\\\?\\") == 0)
		norm_src_pptx_dir.erase(0, 4);

	if (norm_src_pptx_dir.back() != FILE_SEPARATOR_WCHAR)
		norm_src_pptx_dir += FILE_SEPARATOR_WCHAR;

	std::unique_ptr<PPTX::Document> pptx_document(new PPTX::Document());
	bool is_read_ok = pptx_document->read(norm_src_pptx_dir);
	if (!is_read_ok)
		return AVS_FILEUTILS_ERROR_CONVERT;

	const auto presentation_file_type = OOX::Presentation::FileTypes::Presentation;
	smart_ptr<PPTX::Presentation> presentation = pptx_document->Get(presentation_file_type).smart_dynamic_cast<PPTX::Presentation>();
	if (!presentation.is_init())
		return AVS_FILEUTILS_ERROR_CONVERT;

	for (size_t i = 0; i < presentation->sldIdLst.size(); ++i)
	{
		std::wstring rid = presentation->sldIdLst[i].rid.get();
		smart_ptr<PPTX::Slide> slide = ((*presentation)[rid]).smart_dynamic_cast<PPTX::Slide>();

		if (slide.IsInit() == false)
			continue;

		m_pImpl->AddSlide(*slide);
	}
	m_pImpl->Save(wsDstTxtFile);
	return S_OK;
}

CPptxTxtConverter::CPptxTxtConverterImpl::~CPptxTxtConverterImpl()
{
}

void CPptxTxtConverter::CPptxTxtConverterImpl::AddSlide(const PPTX::Slide& oSlide)
{
	std::wstring slide_text;
	slide_text.reserve(knStringReserve);

	auto& sp_tree_elems = oSlide.cSld->spTree.SpTreeElems;
	for (auto& elem : sp_tree_elems)
	{
		if (elem.getType() != OOX::et_a_Shape)
			continue;

		smart_ptr<PPTX::Logic::Shape> shape = elem.GetElem().smart_dynamic_cast<PPTX::Logic::Shape>();
		slide_text += GetTextFromShape(*shape);
	}
	m_arTxtData.push_back(std::move(slide_text));
}
void CPptxTxtConverter::CPptxTxtConverterImpl::Save(const std::wstring& wsDstTxtFile)
{
	NSStringUtils::CStringBuilder str_builder;
	str_builder.AddSize(1000);
	for (const auto& txt_str : m_arTxtData)
	{
		str_builder.WriteString(txt_str);
		str_builder.WriteString(L"\n");
	}
	NSFile::CFileBinary::SaveToFile(wsDstTxtFile, str_builder.GetData());
	m_arTxtData.clear();
}

std::wstring CPptxTxtConverter::CPptxTxtConverterImpl::GetTextFromShape(const PPTX::Logic::Shape& oShape) const
{
	std::wstring text;
	text.reserve(knStringReserve);

	if (oShape.txBody.IsInit())
		for (const auto& paragraph : oShape.txBody->Paragrs)
			text += paragraph.GetText(true);

	return text; // RVO
}