#include "./message_prompt_builder.hpp"

#include "./rpbot.hpp"

#include <solanaceae/contact/contact_store_i.hpp>
#include <solanaceae/contact/components.hpp>
#include <solanaceae/message3/components.hpp>

bool MessagePromptBuilder::buildNameLookup(void) {
	const auto& cr = _cs.registry();
	if (cr.all_of<Contact::Components::ParentOf>(_c)) { // group rpbot
		const auto& subs = cr.get<Contact::Components::ParentOf>(_c).subs;
		// should include self
		for (const auto sub_c : subs) {
			if (cr.all_of<Contact::Components::Name>(sub_c)) {
				names[sub_c] = cr.get<Contact::Components::Name>(sub_c).name;
			}
		}
	} else { // pm rpbot
		if (cr.all_of<Contact::Components::Name>(_c)) {
			names[_c] = cr.get<Contact::Components::Name>(_c).name;
		} else {
			std::cerr << "RPBot error: other missing name\n";
			return false;
		}

		if (cr.all_of<Contact::Components::Self>(_c)) {
			const auto self = cr.get<Contact::Components::Self>(_c).self;
			if (cr.all_of<Contact::Components::Name>(self)) {
				names[self] = cr.get<Contact::Components::Name>(self).name;
			} else {
				std::cerr << "RPBot error: self missing name\n";
				return false;
			}
		} else {
			std::cerr << "RPBot error: contact missing self\n";
			return false;
		}
	}

	return true;
}

std::string MessagePromptBuilder::buildPromptMessageHistory(void) {
	auto* mr =  _rmm.get(_c);
	assert(mr);

	std::string prompt;

	auto view = mr->view<Message::Components::Timestamp>();
	for (auto view_it = view.rbegin(), view_last = view.rend(); view_it != view_last; view_it++) {
		const Message3 e = *view_it;

		// manually filter ("reverse" iteration <.<)
		// TODO: add mesagetext?
		if (!mr->all_of<Message::Components::ContactFrom, Message::Components::ContactTo>(e)) {
			continue;
		}

		//Message::Components::ContactFrom& c_from = mr->get<Message::Components::ContactFrom>(e);
		//Message::Components::ContactTo& c_to = msg_reg.get<Message::Components::ContactTo>(e);
		//Message::Components::Timestamp ts = view.get<Message::Components::Timestamp>(e);

		prompt += "\n";
		prompt += buildPromptMessage({*mr, e});
	}

	return prompt;
}

std::string MessagePromptBuilder::buildPromptMessage(const Message3Handle m) {
	if (!m.all_of<Message::Components::ContactFrom, Message::Components::MessageText>()) {
		// TODO: case for transfers
		return "";
	}

	const auto line_prefix = promptMessagePrefixSimple(m) + ": ";

	// TODO: trim
	std::string message_lines = line_prefix + m.get<Message::Components::MessageText>().text;
	for (auto nlpos = message_lines.find('\n'); nlpos != std::string::npos; nlpos = message_lines.find('\n', nlpos+1)) {
		message_lines.insert(nlpos+1, line_prefix);
		nlpos += line_prefix.size();
	}

	// TODO: cache as comp

	return message_lines;
}

std::string MessagePromptBuilder::promptMessagePrefixSimple(const Message3Handle m) {
	const Contact4 from = m.get<Message::Components::ContactFrom>().c;
	if (names.count(from)) {
		return std::string{names[from]};
	} else {
		return "<unk-user>";
	}
}

std::string MessagePromptBuilder::promptMessagePrefixDirected(const Message3Handle m) {
	// with both contacts (eg: "Name1 to Name2"; or "Name1 to Everyone"

	const Contact4 from = m.get<Message::Components::ContactFrom>().c;
	const Contact4 to = m.get<Message::Components::ContactTo>().c;

	std::string res;

	if (names.count(from)) {
		res = std::string{names[from]};
	} else {
		res = "<unk-user>";
	}

	res += " to ";

	if (names.count(to)) {
		return std::string{names[to]};
	} else {
		return "<unk-user>";
	}

	return res;
}