password_manager/source/sys.cpp

#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

#include <cstring>
#include <iostream>
#include <termios.h>
#include <unistd.h>
#include <filesystem>

bool put_data_on_clipboard(const char *text)
{
	size_t len = strlen(text);
	if (len == 0)
	{
		printf("Text is empty\n");
		return false;
	}

	// Open a pipe to xclip command
	FILE *pipe = popen("xclip -selection clipboard", "w");
	if (!pipe)
	{
		printf("Failed to open pipe to xclip\n");
		return false;
	}

	// Write text to xclip
	if (fwrite(text, 1, len, pipe) != len)
	{
		printf("Failed to write text to xclip\n");
		pclose(pipe);
		return false;
	}

	// Close the pipe
	pclose(pipe);
	return true;
}

std::string get_user_password()
{
	struct termios oldt, newt;

	// Save the current terminal attributes
	tcgetattr(STDIN_FILENO, &oldt);

	// Copy the old settings to the new settings
	newt = oldt;

	// Disable echo input
	newt.c_lflag &= ~ECHO;

	// Apply the new settings
	tcsetattr(STDIN_FILENO, TCSANOW, &newt);

	// 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 terminal attributes
	tcsetattr(STDIN_FILENO, TCSANOW, &oldt);

	return ipt;
}

std::optional<std::string> get_save_path()
{
	namespace fs = std::filesystem;

	const char *homeDir = getenv("HOME");
	if (homeDir == nullptr)
	{
		return {};
	}

	fs::path userDirectory(homeDir);
	userDirectory /= ".config";
	userDirectory /= "password_manager";

	if (!fs::exists(userDirectory))
	{
		// Directory doesn't exist, create it
		if (!fs::create_directory(userDirectory))
		{
			return {};
		}
	}
	else if (!fs::is_directory(userDirectory))
	{
		// Path exists but is not a directory
		return {};
	}

	userDirectory /= "save.txt";

	return userDirectory.string();
}

#endif