diff -r 000000000000 -r 1bb084f84eb9 src/INILiteralHandler.h --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/INILiteralHandler.h Wed Dec 09 17:55:03 2020 +0100 @@ -0,0 +1,93 @@ +/** + * 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 +#include +#include +#include + +#include +#include +#include +#include + +#include "Configuration.h" +#include "INIWriter.h" + +namespace relpipe { +namespace out { +namespace ini { + +class INILiteralHandler : public relpipe::reader::handlers::RelationalReaderStringHandler { +private: + INIWriter& writer; + relpipe::common::type::StringX sectionAttributeName; + relpipe::common::type::StringX currentSection; + std::vector currentAttributes; + relpipe::common::type::Integer relationCount = 0; + relpipe::common::type::Integer currentAttributeIndex = 0; +public: + + INILiteralHandler(INIWriter& writer, relpipe::common::type::StringX sectionAttributeName = L"") : writer(writer), sectionAttributeName(sectionAttributeName) { + } + + virtual ~INILiteralHandler() { + } + + void startRelation(relpipe::common::type::StringX name, std::vector attributes) override { + if (relationCount) writer.endSection(); + currentSection = name; + currentAttributes = attributes; + INIWriter::SectionStartEvent e; + e.name = name; + writer.startSection(e); + relationCount++; + } + + void attribute(const relpipe::common::type::StringX& value) override { + if (currentAttributeIndex && currentAttributeIndex % currentAttributes.size() == 0) { + currentAttributeIndex = 0; + writer.endSection(); + // Usually there should be only one record in each relation. + // But if there are more of them, we will rather create multiple sections with same name (makes more sense) + // than duplicate keys in a single section. + INIWriter::SectionStartEvent e; + e.name = currentSection; + writer.startSection(e); + } + + INIWriter::EntryEvent e; + e.key = currentAttributes[currentAttributeIndex].getAttributeName(); + e.value = value; + writer.entry(e); + + currentAttributeIndex++; + } + + void endOfPipe() { + writer.endSection(); + } + +}; + +} +} +}