#include "user.h"
#include "user_handler.h"
#include "mail.h"

#include <cstdio>
#include <fstream>
#include <mutex>
#include <ostream>
#include <string>
#include <vector>
#include <map>

using json = nlohmann::json;

user::user(fs::path user_data_json) : m()
{
	std::ifstream ifs(user_data_json);
	json user_data = json::parse(ifs);

	this->name = user_data["name"];
	for ( auto& mail_json : user_data["mails"]["received"] ) {
		mail* mail = new struct mail(
			mail_json["filename"],
			mail_json["timestamp"],
			mail_json["subject"]
		);
		mail->id = mail_json["id"];
		mail->sender = mail_json["sender"];
		mail->recipients = mail_json["recipients"].get<std::vector<std::string>>();
		mail->deleted = mail_json["deleted"];
		
		this->inbox.insert(mail);
	}

	for ( auto& mail_json : user_data["mails"]["sent"] ) {
		mail* mail = new struct mail(
			mail_json["filename"],
			mail_json["timestamp"],
			mail_json["subject"]
		);
		mail->id = mail_json["id"];
		mail->sender = mail_json["sender"];
		mail->recipients = mail_json["recipients"].get<std::vector<std::string>>();
		mail->deleted = mail_json["deleted"];
		
		this->sent.insert(mail);
	}

	this->user_data = user_data;
	this->file_location = user_data_json;
}

user::user(std::string name, fs::path user_dir)
	: name(name),
	  m()
{
	json user;
	user["mails"]["sent"] = json::object();
	user["mails"]["received"] = json::object();
	user["name"] = name;

	std::ofstream ofs(user_dir/(name+".json"));
	ofs << user;
	this->user_data = user;
	this->file_location = user_dir/(name+".json");
}

user::~user() {

}

void user::addMail(mail* mail) 
{
	std::lock_guard<std::mutex> guard(this->m);

	mail->id = this->inbox.size();

	this->inbox.insert(mail);
	this->user_data["mails"]["received"][std::to_string(mail->id)] = mail->mailToJson();
}

void user::sendMail(mail* mail, std::vector<std::string> recipients) 
{
	std::lock_guard<std::mutex> guard(this->m);

	std::vector<user*> users;
	for ( auto& name : recipients) {
		// TODO: error handling for non existing user
		users.push_back(user_handler::getInstance().getOrCreateUser(name));
	}

	mail->sender = this->name;
	mail->recipients = recipients;

	mail->id = this->sent.size();

	this->sent.insert(mail);
	this->user_data["mails"]["sent"][std::to_string(mail->id)] = mail->mailToJson();

	for ( auto& user : users ) {
		user->addMail(mail);
	}
}

mail* user::getMail(u_int id) 
{
	maillist::iterator it = std::find_if(this->inbox.begin(), this->inbox.end(), [id](auto& i){ return (*i)(id); });
	return it == this->inbox.end() ? nullptr : (*it)->filename.empty() ? nullptr : *it;
}

bool user::delMail(u_int id) 
{
	std::lock_guard<std::mutex> guard(this->m);

	maillist::iterator it = std::find_if(this->inbox.begin(), this->inbox.end(), [id](auto& i){ return (*i)(id); });

	bool success = true;

	if (it == this->inbox.end() ||
		(*it)->deleted)
		return false;

	if (!(*it)->filename.empty())
		success = fs::remove(user_handler::getInstance().getSpoolDir()/"messages"/(*it)->filename);

	if (success) {
		this->user_data["mails"]["received"][std::to_string((*it)->id)]["subject"] = "";
		this->user_data["mails"]["received"][std::to_string((*it)->id)]["filename"] = "";
		this->user_data["mails"]["received"][std::to_string((*it)->id)]["deleted"] = true;
		(*it)->deleted = true; // other info will be deleted on shutdown
	}

	return success;
}

void user::saveToFile()
{
	std::fstream fs(this->file_location);
	fs << this->user_data.dump();
}