diff -r abf1d424f2a0 -r 92d85e02b276 src/SedHandler.h --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/SedHandler.h Mon Dec 10 17:28:53 2018 +0100 @@ -0,0 +1,111 @@ +/** + * Relational pipes + * Copyright © 2018 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 +#include +#include +#include + +#include +#include +#include +#include + +#include + +#include + +namespace relpipe { +namespace tr { +namespace sed { + +using namespace std; +using namespace relpipe; +using namespace relpipe::reader; +using namespace relpipe::reader::handlers; + +class SedHandler : public RelationalReaderStringHadler { +private: + shared_ptr relationalWriter; + + wregex relationNameRegEx; + wregex attributeNameRegEx; + wregex searchRegEx; + wregex replacementRegEx; + + vector currentReplacableAttributes; + integer_t currentAttributeIndex = 0; + +public: + + SedHandler(ostream& output, const vector& arguments) { + relationalWriter.reset(writer::Factory::create(output)); + + if (arguments.size() == 4) { + relationNameRegEx = wregex(arguments[0]); + attributeNameRegEx = wregex(arguments[1]); + searchRegEx = wregex(arguments[2]); + replacementRegEx = wregex(arguments[3]); + } else { + throw cli::RelpipeCLIException(L"Usage: relpipe-tr-sed ", cli::CLI::EXIT_CODE_UNKNOWN_COMMAND); + } + } + + void startRelation(string_t name, vector attributes) override { + // TODO: move to a reusable method (or use same metadata on both reader and writer side?) + vector writerMetadata; + for (AttributeMetadata readerMetadata : attributes) { + writerMetadata.push_back({readerMetadata.getAttributeName(), relationalWriter->toTypeId(readerMetadata.getTypeName())}); + } + + + currentReplacableAttributes.resize(attributes.size(), false); + if (regex_match(name, relationNameRegEx)) { + for (int i = 0; i < currentReplacableAttributes.size(); i++) { + currentReplacableAttributes[i] = regex_match(attributes[i].getAttributeName(), attributeNameRegEx); + } + } + + relationalWriter->startRelation(name, writerMetadata, true); + } + + void attribute(const string_t& value) override { + if (currentReplacableAttributes[currentAttributeIndex]) { + relationalWriter->writeAttribute(L"1234"); + } else { + relationalWriter->writeAttribute(value); + } + + currentAttributeIndex++; + currentAttributeIndex = currentAttributeIndex % currentReplacableAttributes.size(); + } + + void endOfPipe() { + + } + +}; + +} +} +}