/*
 * (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 "nativecontrol.h"
#include "../../DesktopEditor/raster/ImageFileFormatChecker.h"
#include "../../DesktopEditor/raster/BgraFrame.h"
#include "../../Common/Network/FileTransporter/include/FileTransporter.h"
#include "server.h"

CImagesWorker::CImagesWorker(const std::wstring& sFolder)
{
	m_sFolder = sFolder;
	std::vector<std::wstring> files = NSDirectory::GetFiles(sFolder);
	m_nIndex = (int)files.size() + 1;
}
std::wstring CImagesWorker::GetImageLocal(const std::wstring& sUrl)
{
	if (CServerInstance::getInstance().IsEnable())
	{
		if (!CServerInstance::getInstance().CheckTmpDirectory(sUrl))
			return L"error";
	}
	std::wstring sExt = NSFile::GetFileExtention(sUrl);
	std::wstring sRet = L"image" + std::to_wstring(m_nIndex++) + L"." + sExt;
	m_mapImages.insert(std::make_pair(sUrl, sRet));
	NSFile::CFileBinary::Copy(sUrl, m_sFolder + L"/" + sRet);
	return sRet;
}
std::wstring CImagesWorker::GetImage(const std::wstring& sUrl)
{
	std::wstring sUrlFile = sUrl;
	if (sUrlFile.find(L"file://") == 0)
	{
		sUrlFile = sUrlFile.substr(7);

		// MS Word copy image with url "file://localhost/..." on mac
		if (sUrlFile.find(L"localhost") == 0)
			sUrlFile = sUrlFile.substr(9);

		NSStringUtils::string_replace(sUrlFile, L"%20", L" ");

		if (!NSFile::CFileBinary::Exists(sUrlFile))
			sUrlFile = sUrlFile.substr(1);
	}

	std::map<std::wstring, std::wstring>::iterator find = m_mapImages.find(sUrlFile);
	if (find != m_mapImages.end())
		return find->second;

	if (NSFile::CFileBinary::Exists(sUrlFile))
		return GetImageLocal(sUrlFile);

	bool bIsNeedDownload = false;
	if (true)
	{
		if (sUrlFile.find(L"www.") != std::wstring::npos)
			bIsNeedDownload = true;
		else if (sUrlFile.find(L"http://") != std::wstring::npos)
			bIsNeedDownload = true;
		else if (sUrlFile.find(L"ftp://") != std::wstring::npos)
			bIsNeedDownload = true;
		else if (sUrlFile.find(L"https://") != std::wstring::npos)
			bIsNeedDownload = true;
	}

	if (bIsNeedDownload)
	{
		NSNetwork::NSFileTransport::CFileDownloader oDownloader(sUrl, false);

		std::wstring sTmpFile = NSFile::CFileBinary::CreateTempFileWithUniqueName(NSFile::CFileBinary::GetTempPath(), L"IMG");
		if (NSFile::CFileBinary::Exists(sTmpFile))
			NSFile::CFileBinary::Remove(sTmpFile);
		sTmpFile = sTmpFile + L".png";

		oDownloader.SetFilePath(sTmpFile);
		oDownloader.Start(0);
		while ( oDownloader.IsRunned() )
		{
			NSThreads::Sleep( 10 );
		}
		bool bIsDownloaded = oDownloader.IsFileDownloaded();

		if (bIsDownloaded)
		{
			CBgraFrame oFrame;
			oFrame.OpenFile(sTmpFile);

			std::wstring sRet = L"image" + std::to_wstring(m_nIndex++) + L".png";
			m_mapImages.insert(std::make_pair(sUrlFile, sRet));

			oFrame.SaveFile(m_sFolder + L"/media/" + sRet, 4);

			NSFile::CFileBinary::Remove(sTmpFile);
			return sRet;
		}
	}

	if (sUrlFile.find(L"image") == 0 || sUrlFile.find(L"display") == 0)
		return sUrlFile;
	return L"error";
}