From e0af8874c027cbc1205d42ad3a6f7ace8a57a2d5 Mon Sep 17 00:00:00 2001
From: Nikola Petrov <nikolape7@gmail.com>
Date: Fri, 19 Jul 2024 01:49:34 +0200
Subject: [PATCH] Fix for windows

---
 source/arg_func.cpp         |  2 +-
 main.cpp => source/main.cpp |  0
 source/sys.cpp              | 43 +++++++++++++++----------------------
 3 files changed, 18 insertions(+), 27 deletions(-)
 rename main.cpp => source/main.cpp (100%)

diff --git a/source/arg_func.cpp b/source/arg_func.cpp
index 68a3b5f..5569b14 100644
--- a/source/arg_func.cpp
+++ b/source/arg_func.cpp
@@ -357,5 +357,5 @@ void arg_file(Buffer &decrypted_buffer, Buffer &encrypted_buffer, const char *la
 
 void arg_size(Buffer &encrypted_buffer)
 {
-	printf("File size is: %ldB\n", encrypted_buffer.taken);
+	printf("File size is: %zuB\n", encrypted_buffer.taken);
 }
\ No newline at end of file
diff --git a/main.cpp b/source/main.cpp
similarity index 100%
rename from main.cpp
rename to source/main.cpp
diff --git a/source/sys.cpp b/source/sys.cpp
index 0bd31e3..3e5ef68 100644
--- a/source/sys.cpp
+++ b/source/sys.cpp
@@ -1,5 +1,6 @@
 #include <iostream>
 #include <string>
+#include <filesystem>
 
 #include "sys.hpp"
 #include "func.hpp"
@@ -79,42 +80,33 @@ std::string get_user_password()
 
 std::optional<std::string> get_save_path()
 {
-	PWSTR path = nullptr;
-
-	HRESULT result = SHGetKnownFolderPath(FOLDERID_ProgramData, 0, nullptr, &path);
-
-	if (!SUCCEEDED(result))
+	namespace fs = std::filesystem;
+	const char *path = std::getenv("PROGRAMDATA");
+	if (path == nullptr)
 	{
-		CoTaskMemFree(path); // Free the allocated memory
 		return {};
 	}
 
-	std::wstring userDirectory(path);
-	CoTaskMemFree(path); // Free the allocated memory
+	fs::path userDirectory(path);
+	userDirectory /= "password_manager";
 
-	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 (!fs::exists(userDirectory))
 	{
-		if (!CreateDirectory(userDirectory.c_str(), nullptr))
+		// Directory doesn't exist, create it
+		if (!fs::create_directory(userDirectory))
+		{
 			return {};
+		}
 	}
-
-	userDirectory.append(L"\\save.txt");
-
-	int size = WideCharToMultiByte(CP_UTF8, 0, userDirectory.c_str(), -1, nullptr, 0, nullptr, nullptr);
-	if (size > 0)
+	else if (!fs::is_directory(userDirectory))
 	{
-		std::string str(size, 0);
-		WideCharToMultiByte(CP_UTF8, 0, userDirectory.c_str(), -1, &str[0], size, nullptr, nullptr);
-
-		return str;
+		// Path exists but is not a directory
+		return {};
 	}
 
-	return {};
+	userDirectory /= "save.txt";
+
+	return userDirectory.string();
 }
 
 #else
@@ -123,7 +115,6 @@ std::optional<std::string> get_save_path()
 #include <iostream>
 #include <termios.h>
 #include <unistd.h>
-#include <filesystem>
 
 bool put_data_on_clipboard(const char *text)
 {