#include "InfoCommand.hh"

#include "CommandException.hh"
#include "TclObject.hh"
#include "join.hh"

#include <cassert>
#include <iostream>
#include <ranges>

namespace openmsx {

InfoCommand::InfoCommand(CommandController& commandController_, const std::string& name_)
	: Command(commandController_, name_)
{
}

InfoCommand::~InfoCommand()
{
	assert(infoTopics.empty());
}

void InfoCommand::registerTopic(const InfoTopic& topic)
{
#ifndef NDEBUG
	if (infoTopics.contains(topic.getName())) {
		std::cerr << "INTERNAL ERROR: already have an info topic with "
		             "name " << topic.getName() << '\n';
		assert(false);
	}
#endif
	infoTopics.insert_noDuplicateCheck(&topic);
}

void InfoCommand::unregisterTopic(const InfoTopic& topic)
{
	if (!infoTopics.contains(topic.getName())) {
		std::cerr << "INTERNAL ERROR: can't unregister topic with name "
		          << topic.getName() << ", not found!\n";
		assert(false);
	}
	infoTopics.erase(topic.getName());
}

// Command

void InfoCommand::execute(std::span<const TclObject> tokens,
                          TclObject& result)
{
	switch (tokens.size()) {
	case 1:
		// list topics
		result.addListElements(std::views::transform(
			infoTopics, [](auto* t) { return t->getName(); }));
		break;
	default:
		// show info about topic
		assert(tokens.size() >= 2);
		const auto& topic = tokens[1].getString();
		auto it = infoTopics.find(topic);
		if (it == end(infoTopics)) {
			throw CommandException("No info on: ", topic);
		}
		(*it)->execute(tokens, result);
		break;
	}
}

std::string InfoCommand::help(std::span<const TclObject> tokens) const
{
	std::string result;
	switch (tokens.size()) {
	case 1: {
		// show help on info cmd
		auto topics = to_vector(std::views::transform(infoTopics, &InfoTopic::getName));
		std::ranges::sort(topics);
		result = strCat(
			"Show info on a certain topic\n"
			"  ", getName(), " <topic> [...]\n",
			"Possible topics are:\n  ", join(topics, "\n  "), "\n",
			"Use \"help ", getName(), " <topic>\" for more info on that specific topic.");
		break;
	}
	default:
		// show help on a certain topic
		assert(tokens.size() >= 2);
		auto topic = tokens[1].getString();
		auto it = infoTopics.find(topic);
		if (it == end(infoTopics)) {
			throw CommandException("No info on: ", topic);
		}
		result = (*it)->help(tokens);
		break;
	}
	return result;
}

void InfoCommand::tabCompletion(std::vector<std::string>& tokens) const
{
	switch (tokens.size()) {
	case 2: {
		// complete topic
		completeString(tokens, std::views::transform(infoTopics,
			[](auto* t) -> std::string_view { return t->getName(); }));
		break;
	}
	default:
		// show help on a certain topic
		assert(tokens.size() >= 3);
		if (auto it = infoTopics.find(tokens[1]); it != end(infoTopics)) {
			(*it)->tabCompletion(tokens);
		}
		break;
	}
}

} // namespace openmsx