diff -r 000000000000 -r c205f5d06418 src/CLIParser.h --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/CLIParser.h Sat Jul 27 18:41:13 2019 +0200 @@ -0,0 +1,127 @@ +/** + * Relational pipes + * Copyright © 2019 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, either version 3 of the License, or + * (at your option) any later version. + * + * 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 . + */ +#pragma once + +#include + +#include +#include +#include + +#include "Configuration.h" + +namespace relpipe { +namespace tr { +namespace sql { + +/** + * This tr-sql CLI options are inspired by tr-guile options. + * These configurations are independent and might diverge, but when possible, they should share same option names and same logic for common parts. + */ +class CLIParser { +private: + + // FIXME: move common methods/classes to relpipe-lib-cli or relpipe-lib-helper + + string_t readNext(std::vector 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); + } + + void addRelation(Configuration& c, RelationConfiguration& currentRelation) { + if (currentRelation.relation.size()) { + c.relationConfigurations.push_back(currentRelation); + currentRelation = RelationConfiguration(); + } + } + + relpipe::writer::TypeId parseTypeId(const string_t& value) { + using t = relpipe::writer::TypeId; + if (value == L"string") return t::STRING; + else if (value == L"integer") return t::INTEGER; + else if (value == L"boolean") return t::BOOLEAN; + else throw relpipe::cli::RelpipeCLIException(L"Unable to parse TypeId: " + value, relpipe::cli::CLI::EXIT_CODE_BAD_CLI_ARGUMENTS); + } + +public: + + static const string_t OPTION_RELATION; + static const string_t OPTION_OUTPUT_ATTRIBUTE; + static const string_t OPTION_INPUT_ATTRIBUTES_APPEND; + static const string_t OPTION_INPUT_ATTRIBUTES_PREPEND; + static const string_t OPTION_DEBUG_VARIABLE_MAPPING; + static const string_t OPTION_BEFORE_RECORDS; + static const string_t OPTION_AFTER_RECORDS; + static const string_t OPTION_FOR_EACH; + static const string_t OPTION_DROP; + static const string_t OPTION_DEFINE; + + Configuration parse(const std::vector& arguments) { + Configuration c; + RelationConfiguration currentRelation; + + for (int i = 0; i < arguments.size();) { + string_t option = readNext(arguments, i); + + if (option == OPTION_BEFORE_RECORDS) currentRelation.sqlBeforeRecords = readNext(arguments, i); + else if (option == OPTION_AFTER_RECORDS) currentRelation.sqlAfterRecords = readNext(arguments, i); + else if (option == OPTION_FOR_EACH) currentRelation.sqlForEach = readNext(arguments, i); + else if (option == OPTION_DROP) currentRelation.drop = true; + else if (option == OPTION_INPUT_ATTRIBUTES_APPEND) currentRelation.inputAttributesAppend = true; + else if (option == OPTION_INPUT_ATTRIBUTES_PREPEND) currentRelation.inputAttributesPrepend = true; + else if (option == OPTION_DEBUG_VARIABLE_MAPPING) currentRelation.debugVariableMapping = true; + else if (option == OPTION_RELATION) { + addRelation(c, currentRelation); // previous relation + currentRelation.relation = readNext(arguments, i); + } else if (option == OPTION_OUTPUT_ATTRIBUTE) { + relpipe::writer::AttributeMetadata attribute; + attribute.attributeName = readNext(arguments, i); + attribute.typeId = parseTypeId(readNext(arguments, i)); + currentRelation.writerMetadata.push_back(attribute); + } else if (option == OPTION_DEFINE) { + DefinitionRecipe definition; + definition.name = readNext(arguments, i); + definition.type = readNext(arguments, i); + definition.value = readNext(arguments, i); + if (currentRelation.relation.size()) currentRelation.definitions.push_back(definition); + else c.definitions.push_back(definition); + } else throw relpipe::cli::RelpipeCLIException(L"Unsupported CLI option: " + option, relpipe::cli::CLI::EXIT_CODE_BAD_CLI_ARGUMENTS); + } + addRelation(c, currentRelation); // last relation + + return c; + } + + virtual ~CLIParser() { + } +}; + +const string_t CLIParser::OPTION_RELATION = L"--relation"; +const string_t CLIParser::OPTION_OUTPUT_ATTRIBUTE = L"--output-attribute"; +const string_t CLIParser::OPTION_INPUT_ATTRIBUTES_APPEND = L"--input-attributes-append"; +const string_t CLIParser::OPTION_INPUT_ATTRIBUTES_PREPEND = L"--input-attributes-prepend"; +const string_t CLIParser::OPTION_DEBUG_VARIABLE_MAPPING = L"--debug-variable-mapping"; +const string_t CLIParser::OPTION_BEFORE_RECORDS = L"--before-records"; +const string_t CLIParser::OPTION_AFTER_RECORDS = L"--after-records"; +const string_t CLIParser::OPTION_FOR_EACH = L"--for-each"; +const string_t CLIParser::OPTION_DROP = L"--drop"; +const string_t CLIParser::OPTION_DEFINE = L"--define"; + +} +} +}