#include <iostream>
#include <string>

#include "sys.hpp"
#include "func.hpp"

#ifdef _WIN32

#include <shlobj.h>
#include <Windows.h>

bool put_data_on_clipboard(const char *text)
{
	size_t len = strlen(text);
	if (len == 0)
	{
		printf_s("Text is empty\n");
		return false;
	}
	// Open the clipboard
	if (!OpenClipboard(nullptr))
	{
		printf_s("Failed to open clipboard");
		return false;
	}
	// Allocate global memory for the text
	HGLOBAL hMem = GlobalAlloc(GMEM_MOVEABLE, strlen(text) + 1);
	if (!hMem)
	{
		CloseClipboard();
		printf_s("Failed to allocate memory for text");
		return false;
	}
	// Copy the text to the allocated memory
	char *memData = static_cast<char *>(GlobalLock(hMem));
	if (!memData)
	{
		CloseClipboard();
		printf_s("Failed to lock memory for text");
		return false;
	}
	strcpy_s(memData, strlen(text) + 1, text);
	GlobalUnlock(hMem);
	// Set the data to the clipboard
	EmptyClipboard();
	SetClipboardData(CF_TEXT, hMem);
	// Clean up and close the clipboard
	CloseClipboard();
	return true;
}

std::string get_user_password()
{

	HANDLE hStdInput = GetStdHandle(STD_INPUT_HANDLE);
	DWORD mode = 0;

	// Create a restore point Mode
	// is know 503
	GetConsoleMode(hStdInput, &mode);

	// Enable echo input
	// set to 499
	SetConsoleMode(hStdInput, mode & (~ENABLE_ECHO_INPUT));

	// Take input
	std::string ipt;
	std::getline(std::cin, ipt);

	// Otherwise next cout will print
	// into the same line
	std::cout << std::endl;

	// Restore the mode
	SetConsoleMode(hStdInput, mode);

	return ipt;
}

std::optional<std::string> get_save_path()
{
	PWSTR path = nullptr;

	HRESULT result = SHGetKnownFolderPath(FOLDERID_ProgramData, 0, nullptr, &path);

	if (!SUCCEEDED(result))
	{
		CoTaskMemFree(path); // Free the allocated memory
		return {};
	}

	std::wstring userDirectory(path);
	CoTaskMemFree(path); // Free the allocated memory

	userDirectory.append(L"\\password_manager");

	DWORD attrib = GetFileAttributes(userDirectory.c_str());

	// check if it exists
	if (!(attrib != INVALID_FILE_ATTRIBUTES && (attrib & FILE_ATTRIBUTE_DIRECTORY)))
	{
		if (!CreateDirectory(userDirectory.c_str(), nullptr))
			return {};
	}

	userDirectory.append(L"\\save.txt");

	int size = WideCharToMultiByte(CP_UTF8, 0, userDirectory.c_str(), -1, nullptr, 0, nullptr, nullptr);
	if (size > 0)
	{
		std::string str(size, 0);
		WideCharToMultiByte(CP_UTF8, 0, userDirectory.c_str(), -1, &str[0], size, nullptr, nullptr);

		return str;
	}

	return {};
}

#else

bool put_data_on_clipboard(const char *text)
{
	// TODO
	return true;
}

std::string get_user_password()
{
	// TODO
	return "123";
}

std::optional<std::string> get_save_path()
{
	// TODO
	return "save.bin";
}

#endif