diff -r 09cd32a65709 -r 3c8ea5dcf793 src/CLICommand.h --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/CLICommand.h Tue Sep 22 21:00:30 2020 +0200 @@ -0,0 +1,74 @@ +/** + * Relational pipes + * Copyright © 2020 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 . + */ +#pragma once + +#include +#include +#include + +#include "Configuration.h" + +namespace relpipe { +namespace in { +namespace cli { + +class CLICommand { +private: + + std::wstring_convert> convertor; // TODO: support also other encodings and use platform encoding as default + + void processRelation(RelationConfiguration& configuration, std::shared_ptr writer) { + // Write header / metadata: + std::vector attributesMetadata; + for (AttributeRecipe ar : configuration.attributes) attributesMetadata.push_back({ar.name, ar.type}); + writer->startRelation(configuration.relation, attributesMetadata, true); + + // Write records from CLI: + for (auto value : configuration.values) writer->writeAttribute(value); + + // Write records from STDIN: + if (configuration.valueStream) { + std::stringstream rawValue; + while (true) { + char ch = 0; + configuration.valueStream->get(ch); + if (ch == 0 && configuration.valueStream->good()) { + writer->writeAttribute(convertor.from_bytes(rawValue.str())); + rawValue.str(""); + rawValue.clear(); + } else if (configuration.valueStream->good()) { + rawValue << ch; + } else if (configuration.valueStream->eof()) { + return; + } else { + throw relpipe::cli::RelpipeCLIException(L"Error while reading values from the input stream.", relpipe::cli::CLI::EXIT_CODE_UNEXPECTED_ERROR); // TODO: better exception + } + } + } + + } + +public: + + void process(Configuration& configuration, std::shared_ptr writer) { + for (RelationConfiguration& relationConfiguration : configuration.relationConfigurations) processRelation(relationConfiguration, writer); + } +}; + +} +} +}