#include <plasp-app/OptionGroups.h>

////////////////////////////////////////////////////////////////////////////////////////////////////
//
// Option Groups
//
////////////////////////////////////////////////////////////////////////////////////////////////////

void OptionGroupBasic::addTo(cxxopts::Options &options)
{
	options.add_options("basic")
	("h,help", "Display this help message")
	("v,version", "Display version information")
	("warnings-as-errors", "Treat warnings as errors");
}

////////////////////////////////////////////////////////////////////////////////////////////////////

void OptionGroupBasic::parse(cxxopts::Options &options)
{
	help = options["help"].as<bool>();
	version = options["version"].as<bool>();
	warningsAsErrors = options["warnings-as-errors"].as<bool>();
}

////////////////////////////////////////////////////////////////////////////////////////////////////

void OptionGroupOutput::addTo(cxxopts::Options &options)
{
	options.add_options("output")
		("color", "Colorize output (always, never, auto)", cxxopts::value<std::string>()->default_value("auto"))
		("p,log-priority", "Log messages starting from this priority (debug, info, warning, error)", cxxopts::value<std::string>()->default_value("info"));
}

////////////////////////////////////////////////////////////////////////////////////////////////////

void OptionGroupOutput::parse(cxxopts::Options &options)
{
	const auto colorPolicyString = options["color"].as<std::string>();

	if (colorPolicyString == "auto")
		colorPolicy = colorlog::ColorStream::ColorPolicy::Auto;
	else if (colorPolicyString == "never")
		colorPolicy = colorlog::ColorStream::ColorPolicy::Never;
	else if (colorPolicyString == "always")
		colorPolicy = colorlog::ColorStream::ColorPolicy::Always;
	else
		throw OptionException("unknown color policy “" + colorPolicyString + "”");

	const auto logPriorityString = options["log-priority"].as<std::string>();

	try
	{
		logPriority = colorlog::priorityFromName(logPriorityString.c_str());
	}
	catch (const std::exception &e)
	{
		throw OptionException(e.what());
	}
}

////////////////////////////////////////////////////////////////////////////////////////////////////

void OptionGroupParser::addTo(cxxopts::Options &options)
{
	options.add_options("parser")
		("i,input", "Input files (in PDDL or SAS format)", cxxopts::value<std::vector<std::string>>())
		("parsing-mode", "Parsing mode (strict, compatibility)", cxxopts::value<std::string>()->default_value("strict"))
		("l,language", "Input language (pddl, sas, auto)", cxxopts::value<std::string>()->default_value("auto"));
	options.parse_positional("input");
}

////////////////////////////////////////////////////////////////////////////////////////////////////

void OptionGroupParser::parse(cxxopts::Options &options)
{
	const auto parsingModeString = options["parsing-mode"].as<std::string>();

	if (parsingModeString == "compatibility")
		parsingMode = pddl::Mode::Compatibility;
	else if (parsingModeString != "strict")
		throw OptionException("unknown parsing mode “" + parsingModeString + "”");

	if (options.count("input"))
		inputFiles = options["input"].as<std::vector<std::string>>();

	const auto languageName = options["language"].as<std::string>();
	language = plasp::Language::fromString(languageName);

	if (language == plasp::Language::Type::Unknown)
		throw OptionException("unknown input language “" + languageName + "”");
}