#include <Windows.h>
#include <iostream>
#include <string>
#include <shlobj.h>
#include "win.h"
#include "func.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;

	// Get the user's documents directory
	HRESULT result = SHGetKnownFolderPath(FOLDERID_Profile, 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"\\passwords.bin");

	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 {};
}