#include <ctime>
#include "func.h"
#include "glob.h"
#include "Buffer.h"

int find_logininfo_in_buffer(Buffer& buffer, const char* label)
{
	Index* index = (Index*)buffer.buffer;
	LoginInfo* pass = (LoginInfo*)(buffer.buffer + sizeof(Index));

	for (size_t i = 0; i < index->count; i++)
	{
		if (glob(label, (const char*)buffer.buffer + pass[i].label + index->offset) == Glob_Result::MATCHED)
			return i;
	}

	return -1;
}

void delete_logininfo_from_buffer(Buffer& buffer, int index_of_pass)
{
	Index* in_index = (Index*)buffer.buffer;
	LoginInfo* in_pass = (LoginInfo*)(buffer.buffer + sizeof(Index));

	int size_of_label = strlen((char*)buffer.buffer + in_pass[index_of_pass].label + in_index->offset) + 1;
	int size_of_username = strlen((char*)buffer.buffer + in_pass[index_of_pass].username + in_index->offset) + 1;
	int size_of_password = strlen((char*)buffer.buffer + in_pass[index_of_pass].password + in_index->offset) + 1;
	int size_pass = size_of_label + size_of_username + size_of_password;

	uint8_t* start_of_pass = buffer.buffer + in_pass[index_of_pass].label + in_index->offset;
	buffer.remove_fast(start_of_pass, size_pass);

	for (size_t i = index_of_pass; i < in_index->count; i++)
	{
		in_pass[i].label -= size_pass;
		in_pass[i].password -= size_pass;
	}

	buffer.remove_fast((uint8_t*)&in_pass[index_of_pass], sizeof(LoginInfo));
	in_index->count -= 1;
	in_index->offset -= sizeof(LoginInfo);
}

void add_logininfo_to_buffer(Buffer& buffer, const char* label, const char* username, const char* password)
{
	int label_start = buffer.add_end((uint8_t*)label, strlen(label) + 1);
	int username_start = buffer.add_end((uint8_t*)username, strlen(username) + 1);
	int password_start = buffer.add_end((uint8_t*)password, strlen(password) + 1);

	Index* index = (Index*)buffer.buffer;

	label_start = label_start - index->offset;
	username_start = username_start - index->offset;
	password_start = password_start - index->offset;
	LoginInfo pass = { label_start, username_start, password_start };
	index->count++;
	index->offset += sizeof(LoginInfo);

	buffer.add_middle((uint8_t*)&pass, sizeof(LoginInfo), index->offset - sizeof(LoginInfo));

}

LoginInfoPointer get_logininfo_pointer_from_buffer(Buffer& buffer, int index_of_pass)
{
	Index* index = (Index*)buffer.buffer;
	LoginInfo* pass = (LoginInfo*)(buffer.buffer + sizeof(Index));
	LoginInfoPointer ret;
	ret.label = (const char*)buffer.buffer + pass[index_of_pass].label + index->offset;
	ret.username = (const char*)buffer.buffer + pass[index_of_pass].username + index->offset;
	ret.password = (const char*)buffer.buffer + pass[index_of_pass].password + index->offset;
	return ret;
}

void generate_password(std::string& password, int len)
{
	srand(time(NULL));
	char characters[] = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789~!@#$%^&*()_-+={[}]|:;<,>.?";
	for (int i = 0; i < len; i++)
	{
		password += characters[rand() % (sizeof(characters) - 1)];
	}
}