#include "arg_func.h"
#include "func.h"
#include <string>
#include "win.h"
#include "func.h"
#include "buffer.h"
#include "cryptography.h"


Pass* arg_get(Buffer& decrypted_buffer, const char* label)
{
	Pass* pass = find_password(&decrypted_buffer, label);
	if (!pass) printf_s("Password not found\n");
	return pass;
}

Pass* arg_generate(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto)
{
	printf_s("Generating password for %s\n", label);
	Pass* pass = find_password(&decrypted_buffer, label);
	Pass new_pass = { 0 };
	if (pass != nullptr) generate_password(pass->password);
	else
	{
		generate_password(new_pass.password);
		strcpy_s(new_pass.label, MAX_STRING_SIZE, label);
		int position = decrypted_buffer.add((uint8_t*)&new_pass, sizeof(Pass));
		if (position < 0)
		{
			printf_s("Error adding password\n");
			return nullptr;
		}
		pass = (Pass*)(decrypted_buffer.buffer + position);
	}
	crypto.encrypt(&decrypted_buffer, &encrypted_buffer);
	save_buffer_to_file(&encrypted_buffer);
	return pass;
}

void arg_list(Buffer& decrypted_buffer)
{
	Pass* passwords = (Pass*)decrypted_buffer.buffer;
	for (int i = 0; i < decrypted_buffer.taken / sizeof(Pass); i++)
	{
		printf_s("%s\n", passwords[i].label);
	}
}

void arg_delete(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto)
{
	printf_s("Deleting password for %s\n", label);
	Pass* pass = find_password(&decrypted_buffer, label);
	if (!pass)
	{
		printf_s("Password not found\n");
		return;
	}
	else
	{
		decrypted_buffer.remove_fast((uint8_t*)pass, sizeof(Pass));
	}
	crypto.encrypt(&decrypted_buffer, &encrypted_buffer);
	save_buffer_to_file(&encrypted_buffer);
	printf_s("Password deleted\n");
}

void arg_print_all_p(Buffer& decrypted_buffer, std::string& user_pass)
{
	printf_s("Input main password for confirmation:\n");
	std::string new_string = get_passwd();
	if (new_string.empty())
	{
		printf_s("Error getting password\n");
		return;
	}
	if (new_string != user_pass)
	{
		printf_s("Wrong password\n");
		return;
	}
	Pass* passwords = (Pass*)decrypted_buffer.buffer;
	for (int i = 0; i < decrypted_buffer.taken / sizeof(Pass); i++)
	{
		printf_s("%s: %s\n", passwords[i].label, passwords[i].password);
	}
}

Pass* arg_input(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto)
{
	printf_s("Input password for %s:\n", label);

	std::string new_string = get_passwd();


	if (new_string.empty())
	{
		printf_s("Error getting password\n");
		return nullptr;
	}

	Pass* pass = find_password(&decrypted_buffer, label);
	Pass new_pass = { 0 };


	if (pass != nullptr)
	{
		strcpy_s(pass->password, MAX_STRING_SIZE, new_string.c_str());
	}
	else
	{
		strcpy_s(new_pass.password, MAX_STRING_SIZE, new_string.c_str());
		strcpy_s(new_pass.label, MAX_STRING_SIZE, label);
		int position = decrypted_buffer.add((uint8_t*)&new_pass, sizeof(Pass));
		if (position < 0)
		{
			printf_s("Error adding password\n");
			return nullptr;
		}
		pass = (Pass*)(decrypted_buffer.buffer + position);
	}

	crypto.encrypt(&decrypted_buffer, &encrypted_buffer);
	save_buffer_to_file(&encrypted_buffer);
	return pass;
}

void arg_change(Buffer& decrypted_buffer, Buffer& encrypted_buffer, std::string& user_pass, Cryptography& crypto)
{
	printf_s("Input main password for confirmation:");

	std::string new_string = get_passwd();
	if (new_string.empty())
	{
		printf_s("Error getting password\n");
		return;
	}

	if (new_string != user_pass)
	{
		printf_s("Passwords don't match\n");
		return;
	}

	printf_s("Input new password:");
	new_string = get_passwd();
	if (new_string.empty())
	{
		printf_s("Error getting password\n");
		return;
	}

	printf_s("Input new password again:");
	user_pass = get_passwd();
	if (user_pass.empty())
	{
		printf_s("Error getting password\n");
		return;
	}

	if (new_string != user_pass)
	{
		printf_s("Passwords don't match\n");
		return;
	}

	crypto.generate_key_and_iv_from_password(new_string.c_str(), new_string.size());
	crypto.encrypt(&decrypted_buffer, &encrypted_buffer);
	save_buffer_to_file(&encrypted_buffer);
	printf_s("Password changed\n");
}

void arg_show(Buffer& decrypted_buffer, const char* label) {

	Pass* pass = find_password(&decrypted_buffer, label);
	if (!pass)
		printf_s("Password not found\n");
	else
		printf_s("Password: %s\n", pass->password);
}