/*
 * 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 "SystemUtils.h"
#include "Directory.h"
#include <stdlib.h>

#ifdef _WIN32
#include "ShlObj.h"
#ifdef CreateDirectory
#undef CreateDirectory
#endif
#ifdef CreateFile
#undef CreateFile
#endif
#ifdef GetSystemDirectory
#undef GetSystemDirectory
#endif
#endif

#if defined(_MAC) && !defined(_IOS)
#include <pwd.h>
#include <unistd.h>

#if 0
#include <CoreFoundation/CoreFoundation.h>
std::wstring getDirectoryPathMac(const NSSystemUtils::SystemDirectoryType& type)
{
	// Determine directory type
	CFSearchPathDirectory searchPath = kCFDocumentDirectory;
	switch (type) {
	case 1:
		searchPath = kCFDownloadsDirectory;
		break;
	case 2:
		searchPath = kCFDesktopDirectory;
		break;
	default:
		break;
	}

	// Get array of paths
	CFArrayRef pathsArray = CFCopySearchPathForDirectoriesInDomains(searchPath, kCFUserDomainMask, true);

	std::wstring result = L"";
	if (pathsArray && CFArrayGetCount(pathsArray) > 0)
	{
		CFStringRef pathString = (CFStringRef)CFArrayGetValueAtIndex(pathsArray, 0);

		if (pathString)
		{
			CFIndex length = CFStringGetMaximumSizeForEncoding(CFStringGetLength(pathString), kCFStringEncodingUTF8) + 1;
			char* buffer = new char[length];
			if (CFStringGetCString(pathString, buffer, length, kCFStringEncodingUTF8))
			{
				std::string resultA = std::string(buffer);
				result = UTF8_TO_U(resultA);
			}
			delete[] buffer;
		}
	}

	if (pathsArray)
		CFRelease(pathsArray);

	return result;
}
#endif

#endif

#if defined(_LINUX) && !defined(_MAC) && !defined(__ANDROID__)
#include <pwd.h>
#include <unistd.h>
#endif

namespace NSSystemUtils
{
	std::string GetEnvVariableA(const std::wstring& strName)
	{
		std::wstring sTmp = GetEnvVariable(strName);
		return U_TO_UTF8(sTmp);
	}

	std::wstring GetEnvVariable(const std::wstring& strName)
	{
#if defined(_WIN32) || defined(_WIN32_WCE) || defined(_WIN64)
		wchar_t* pVal = _wgetenv(strName.c_str());
		if (NULL != pVal)
		{
			std::wstring sRes(pVal);
			return sRes;
		}
		else
		{
			return L"";
		}
#else
		std::string strNameA = NSFile::CUtf8Converter::GetUtf8StringFromUnicode(strName);
		char* pValA = getenv(strNameA.c_str());
		if (NULL != pValA)
		{
			return NSFile::CUtf8Converter::GetUnicodeStringFromUTF8((BYTE*)pValA, strlen(pValA));
		}
		else
		{
			return L"";
		}
#endif
	}

	std::wstring GetAppDataDir()
	{
		std::wstring sBranding = GetBuildBranding();
		std::wstring sAppDataPath;

#ifdef _WIN32
		wchar_t sAppDataLocal[65535];

		if (SUCCEEDED(SHGetFolderPathW(NULL, CSIDL_LOCAL_APPDATA, NULL, 0, sAppDataLocal)))
		{
			sAppDataPath = std::wstring(sAppDataLocal);
		}
#else
		std::wstring sHomeDir = NSSystemUtils::GetEnvVariable(L"HOME");

		if (!sHomeDir.empty())
		{
			if (NSDirectory::Exists(sHomeDir + L"/.local/share"))
				sHomeDir = sHomeDir + L"/.local/share";
			else if (NSDirectory::Exists(sHomeDir + L"/.local"))
				sHomeDir = sHomeDir + L"/.local";
		}

		sAppDataPath = sHomeDir;
#endif

		if (!NSDirectory::Exists(sAppDataPath))
			return L"";

		sAppDataPath += (L"/" + sBranding);

		if (!NSDirectory::Exists(sAppDataPath))
			NSDirectory::CreateDirectory(sAppDataPath);

		return sAppDataPath;
	}

	std::wstring GetBuildBranding()
	{
		std::string sBrandingA = "ONLYOFFICE";
#ifdef BUILD_BRANDING_NAME
		sBrandingA = BUILD_BRANDING_NAME;
#endif

		return UTF8_TO_U(sBrandingA);
	}

	std::wstring GetSystemDirectory(const SystemDirectoryType& type)
	{
		std::wstring result = L"";
		if (type == SystemDirectoryType::Undefined)
			return result;

#if defined(_WIN32) && !defined(WIN_XP_OR_VISTA)

		KNOWNFOLDERID folderId = FOLDERID_Documents;
		switch (type)
		{
		case SystemDirectoryType::Downloads:
			folderId = FOLDERID_Downloads;
			break;
		case SystemDirectoryType::Desktop:
			folderId = FOLDERID_Desktop;
			break;
		default:
			break;
		}

		PWSTR dirPath = NULL;
		HRESULT hr = SHGetKnownFolderPath(folderId, 0, NULL, &dirPath);

		if (SUCCEEDED(hr))
			result = std::wstring(dirPath);

		if (dirPath)
			CoTaskMemFree(dirPath);

#endif

#if defined(_LINUX) && !defined(_MAC) && !defined(__ANDROID__)

		std::wstring home_dir = NSSystemUtils::GetEnvVariable(L"HOME");
		result = home_dir + L"/Documents";

		switch (type)
		{
		case SystemDirectoryType::Downloads:
			result = home_dir + L"/Downloads";
			break;
		case SystemDirectoryType::Desktop:
			result = home_dir + L"/Desktop";
			break;
		default:
			break;
		}

#endif

#if defined(_MAC) && !defined(_IOS)

#if 0
		result = getDirectoryPathMac(type);
#endif
		if (!result.empty())
			return result;

		std::wstring home_dir = NSSystemUtils::GetEnvVariable(L"HOME");
		result = home_dir + L"/Documents";

		switch (type)
		{
		case SystemDirectoryType::Downloads:
			result = home_dir + L"/Downloads";
			break;
		case SystemDirectoryType::Desktop:
			result = home_dir + L"/Desktop";
			break;
		default:
			break;
		}

#endif

		return result;
	}
} // namespace NSSystemUtils