/**

 * 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);

	}

	void addRule(RelationConfiguration& currentRelation, RewriteRule& currentRule) {

		if (currentRule.attribute.size()) {

			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_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_ATTRIBUTE) {

				addRule(currentRelation, currentRule); // previous rule

				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.attribute.size()) currentRule.caseSensitive[entity] = value;

				else currentRelation.caseSensitive[entity] = value;

			} else if (option == OPTION_INVERT_MATCH) {

				ENTITY entity = parseEntity(readNext(arguments, i));

				bool value = parseBoolean(readNext(arguments, i));

				if (currentRule.attribute.size()) currentRule.invertMatch[entity] = value;

				else currentRelation.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_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";

}

}

}