140 lines
2.6 KiB
C++
140 lines
2.6 KiB
C++
#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 |