configuration: --write-ddl, --write-dml, --write-column-names, --insert-mode, --type-cast
/**
* Relational pipes
* Copyright © 2022 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 <iostream>
#include <relpipe/reader/typedefs.h>
#include <relpipe/cli/CLI.h>
#include <relpipe/cli/RelpipeCLIException.h>
#include "Configuration.h"
namespace relpipe {
namespace out {
namespace sql {
class CLIParser {
private:
relpipe::reader::string_t readNext(const std::vector<relpipe::reader::string_t>& 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::reader::string_t& 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);
}
Configuration::InsertMode parseInsertMode(const relpipe::reader::string_t& value) {
if (value == L"single") return Configuration::InsertMode::SINGLE;
else if (value == L"multi") return Configuration::InsertMode::MULTI;
else throw relpipe::cli::RelpipeCLIException(L"Unable to parse InsertMode value: " + value + L" (expecting single or multi)", relpipe::cli::CLI::EXIT_CODE_BAD_CLI_ARGUMENTS);
}
void updateRelationName(Configuration::RelationConfiguration& currentRelation) {
if (currentRelation.relation.size() == 0) currentRelation.relation = L".*";
}
void addRelation(Configuration& c, Configuration::RelationConfiguration& currentRelation) {
if (currentRelation.relation.size()) {
currentRelation.relationPattern = std::wregex(currentRelation.relation);
c.relationConfigurations.push_back(currentRelation);
currentRelation = Configuration::RelationConfiguration();
}
}
public:
static const relpipe::reader::string_t OPTION_RELATION;
static const relpipe::reader::string_t OPTION_WRITE_DDL;
static const relpipe::reader::string_t OPTION_WRITE_DML;
static const relpipe::reader::string_t OPTION_WRITE_COLUMN_NAMES;
static const relpipe::reader::string_t OPTION_INSERT_MODE;
static const relpipe::reader::string_t OPTION_TYPE_CAST;
Configuration parse(const std::vector<relpipe::reader::string_t>& arguments) {
Configuration c;
Configuration::RelationConfiguration currentRelation;
for (int i = 0; i < arguments.size();) {
relpipe::reader::string_t option = readNext(arguments, i);
if (option == OPTION_RELATION) {
addRelation(c, currentRelation); // previous relation
currentRelation.relation = readNext(arguments, i);
} else if (option == OPTION_WRITE_DDL) {
updateRelationName(currentRelation);
currentRelation.writeDDL = parseBoolean(readNext(arguments, i));
} else if (option == OPTION_WRITE_DML) {
updateRelationName(currentRelation);
currentRelation.writeDML = parseBoolean(readNext(arguments, i));
} else if (option == OPTION_WRITE_COLUMN_NAMES) {
updateRelationName(currentRelation);
currentRelation.writeColumnNames = parseBoolean(readNext(arguments, i));
} else if (option == OPTION_INSERT_MODE) {
updateRelationName(currentRelation);
currentRelation.insertMode = parseInsertMode(readNext(arguments, i));
} else if (option == OPTION_TYPE_CAST) {
updateRelationName(currentRelation);
Configuration::TypeCastRule tcr;
tcr.attribute = std::wregex(readNext(arguments, i));
tcr.type = std::wregex(readNext(arguments, i));
tcr.sqlType = readNext(arguments, i);
currentRelation.typeCastRules.push_back(tcr);
} 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 relpipe::common::type::StringX CLIParser::OPTION_RELATION = L"--relation";
const relpipe::common::type::StringX CLIParser::OPTION_WRITE_DDL = L"--write-ddl";
const relpipe::common::type::StringX CLIParser::OPTION_WRITE_DML = L"--write-dml";
const relpipe::common::type::StringX CLIParser::OPTION_WRITE_COLUMN_NAMES = L"--write-column-names";
const relpipe::common::type::StringX CLIParser::OPTION_INSERT_MODE = L"--insert-mode";
const relpipe::common::type::StringX CLIParser::OPTION_TYPE_CAST = L"--type-cast";
}
}
}