src/INILiteralHandler.h
branchv_0
changeset 0 1bb084f84eb9
child 1 e04e5bbc147b
--- /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 <http://www.gnu.org/licenses/>.
+ */
+#pragma once
+
+#include <memory>
+#include <string>
+#include <vector>
+#include <iostream>
+#include <sstream>
+#include <locale>
+#include <codecvt>
+
+#include <relpipe/common/type/typedefs.h>
+#include <relpipe/reader/TypeId.h>
+#include <relpipe/reader/handlers/RelationalReaderStringHandler.h>
+#include <relpipe/reader/handlers/AttributeMetadata.h>
+
+#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<relpipe::reader::handlers::AttributeMetadata> 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<relpipe::reader::handlers::AttributeMetadata> 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();
+	}
+
+};
+
+}
+}
+}