/**
 * Relational pipes
 * Copyright © 2021 František Kučera (Frantovo.cz, GlobalCode.info)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
#pragma once

#include <vector>

#include <relpipe/common/type/typedefs.h>
#include <relpipe/cli/CLI.h>
#include <relpipe/cli/RelpipeCLIException.h>

#include "Configuration.h"

namespace relpipe {
namespace tr {
namespace sed {

class CLIParser {
private:

	relpipe::common::type::StringX readNext(std::vector<relpipe::common::type::StringX> arguments, int& i) {
		if (i < arguments.size()) return arguments[i++];
		else throw relpipe::cli::RelpipeCLIException(L"Missing CLI argument" + (i > 0 ? (L" after " + arguments[i - 1]) : L""), relpipe::cli::CLI::EXIT_CODE_BAD_CLI_ARGUMENTS);
	}

	/**
	 * TODO: use a common method
	 */
	bool parseBoolean(const relpipe::common::type::StringX& value) {
		if (value == L"true") return true;
		else if (value == L"false") return false;
		else throw relpipe::cli::RelpipeCLIException(L"Unable to parse boolean value: " + value + L" (expecting true or false)", relpipe::cli::CLI::EXIT_CODE_BAD_CLI_ARGUMENTS);
	}

	ENTITY parseEntity(const relpipe::common::type::StringX& value) {
		if (value == L"relation") return ENTITY::RELATION;
		else if (value == L"attribute") return ENTITY::ATTRIBUTE;
		else if (value == L"value") return ENTITY::VALUE;
		else throw relpipe::cli::RelpipeCLIException(L"Unable to parse entity value: " + value + L" (expecting „relation“, „attribute“ or „value“)", relpipe::cli::CLI::EXIT_CODE_BAD_CLI_ARGUMENTS);
	}

	ENTITY parseModifyTarget(const relpipe::common::type::StringX& value) {
		if (value == L"relation-name") return ENTITY::RELATION;
		else if (value == L"attribute-name") return ENTITY::ATTRIBUTE;
		else if (value == L"value") return ENTITY::VALUE;
		else throw relpipe::cli::RelpipeCLIException(L"Unable to parse entity value: " + value + L" (expecting „relation-name“, „attribute-name“ or „value“)", relpipe::cli::CLI::EXIT_CODE_BAD_CLI_ARGUMENTS);
	}

	void addRule(RelationConfiguration& currentRelation, RewriteRule& currentRule) {
		if (currentRule.modify != ENTITY::DEFAULT) {
			// the --value is optional → set defaults if missing:
			if (currentRule.modify == ENTITY::RELATION && currentRule.value.size() == 0) currentRule.value = currentRelation.relation;
			if (currentRule.modify == ENTITY::ATTRIBUTE && currentRule.value.size() == 0) currentRule.value = currentRule.attribute;

			currentRule.attributePattern = currentRule.caseSensitive[ENTITY::ATTRIBUTE] ? std::wregex(currentRule.attribute) : std::wregex(currentRule.attribute, std::regex_constants::icase);
			currentRule.valuePattern = currentRule.caseSensitive[ENTITY::VALUE] ? std::wregex(currentRule.value) : std::wregex(currentRule.value, std::regex_constants::icase);
			currentRelation.rules.push_back(currentRule);
			currentRule = RewriteRule(currentRelation.caseSensitive, currentRelation.invertMatch);
		}
	}

	void addRelation(Configuration& c, RelationConfiguration& currentRelation, RewriteRule& currentRule) {
		if (currentRelation.relation.size()) {
			currentRelation.relationPattern = currentRelation.caseSensitive[ENTITY::RELATION] ? std::wregex(currentRelation.relation) : std::wregex(currentRelation.relation, std::regex_constants::icase);
			addRule(currentRelation, currentRule); // last rule
			c.relationConfigurations.push_back(currentRelation);
			currentRelation = RelationConfiguration();
		}
	}

public:

	static const relpipe::common::type::StringX OPTION_RELATION;
	static const relpipe::common::type::StringX OPTION_MODIFY;
	static const relpipe::common::type::StringX OPTION_ATTRIBUTE;
	static const relpipe::common::type::StringX OPTION_VALUE;
	static const relpipe::common::type::StringX OPTION_REPLACEMENT;
	static const relpipe::common::type::StringX OPTION_CASE_SENSITIVE;
	static const relpipe::common::type::StringX OPTION_INVERT_MATCH;

	Configuration parse(const std::vector<relpipe::common::type::StringX>& arguments) {
		Configuration c;
		RelationConfiguration currentRelation;
		RewriteRule currentRule(currentRelation.caseSensitive, currentRelation.invertMatch);

		for (int i = 0; i < arguments.size();) {
			relpipe::common::type::StringX option = readNext(arguments, i);

			if (option == OPTION_RELATION) {
				addRelation(c, currentRelation, currentRule); // previous relation
				currentRelation.relation = readNext(arguments, i);
			} else if (option == OPTION_MODIFY) {
				addRule(currentRelation, currentRule); // previous rule
				currentRule.modify = parseModifyTarget(readNext(arguments, i));
				if (currentRule.modify == ENTITY::VALUE) currentRule.modify = ENTITY::DEFAULT;
			} else if (option == OPTION_ATTRIBUTE) {
				if (currentRule.modify == ENTITY::DEFAULT) {
					currentRule.modify = ENTITY::VALUE;
				} else if (currentRule.modify == ENTITY::VALUE) {
					addRule(currentRelation, currentRule); // previous rule
					currentRule.modify = ENTITY::VALUE;
				}
				currentRule.attribute = readNext(arguments, i);
			} else if (option == OPTION_VALUE) {
				currentRule.value = readNext(arguments, i);
			} else if (option == OPTION_REPLACEMENT) {
				currentRule.replacement = readNext(arguments, i);
			} else if (option == OPTION_CASE_SENSITIVE) {
				ENTITY entity = parseEntity(readNext(arguments, i));
				bool value = parseBoolean(readNext(arguments, i));
				if (currentRule.modify == ENTITY::DEFAULT) currentRelation.caseSensitive[entity] = value;
				currentRule.caseSensitive[entity] = value;
			} else if (option == OPTION_INVERT_MATCH) {
				ENTITY entity = parseEntity(readNext(arguments, i));
				bool value = parseBoolean(readNext(arguments, i));
				if (currentRule.modify == ENTITY::DEFAULT) currentRelation.invertMatch[entity] = value;
				currentRule.invertMatch[entity] = value;
			} else throw relpipe::cli::RelpipeCLIException(L"Unsupported CLI option: " + option, relpipe::cli::CLI::EXIT_CODE_BAD_CLI_ARGUMENTS);
		}
		addRelation(c, currentRelation, currentRule); // last relation

		return c;
	}

	virtual ~CLIParser() {
	}
};

const relpipe::common::type::StringX CLIParser::OPTION_RELATION = L"--relation";
const relpipe::common::type::StringX CLIParser::OPTION_MODIFY = L"--modify";
const relpipe::common::type::StringX CLIParser::OPTION_ATTRIBUTE = L"--attribute";
const relpipe::common::type::StringX CLIParser::OPTION_VALUE = L"--value";
const relpipe::common::type::StringX CLIParser::OPTION_REPLACEMENT = L"--replacement";
const relpipe::common::type::StringX CLIParser::OPTION_CASE_SENSITIVE = L"--case-sensitive";
const relpipe::common::type::StringX CLIParser::OPTION_INVERT_MATCH = L"--invert-match";

}
}
}