src/YAMLCommand.h
branchv_0
changeset 1 8d2d8f4077af
parent 0 66a454290286
child 2 d68192f0e960
--- a/src/YAMLCommand.h	Tue Dec 01 23:30:00 2020 +0100
+++ b/src/YAMLCommand.h	Thu Dec 03 17:16:09 2020 +0100
@@ -1,6 +1,6 @@
 /**
  * Relational pipes
- * Copyright © 2019 František Kučera (Frantovo.cz, GlobalCode.info)
+ * 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
@@ -21,6 +21,8 @@
 #include <memory>
 #include <vector>
 
+#include <yaml.h>
+
 #include <relpipe/writer/RelationalWriter.h>
 #include <relpipe/writer/typedefs.h>
 
@@ -32,12 +34,123 @@
 private:
 	std::wstring_convert<codecvt_utf8<wchar_t>> convertor; // TODO: support also other encodings.
 
+	class YAMLEvent {
+	private:
+		yaml_event_t event;
+	public:
+
+		YAMLEvent(yaml_event_t event) : event(event) {
+		}
+
+		virtual ~YAMLEvent() {
+			yaml_event_delete(&event);
+		}
+
+		const yaml_event_type_t getType() const {
+			return event.type;
+		}
+
+		const yaml_event_t* getEvent() const {
+			return &event;
+		}
+
+		YAMLEvent(const YAMLEvent&) = delete;
+		YAMLEvent& operator=(const YAMLEvent&) = delete;
+	};
+
+	class YAMLParser {
+	private:
+		yaml_parser_t yamlParser;
+		std::istream* input = nullptr;
+
+		static int readFromInput(void* instance, unsigned char* buffer, size_t size, size_t* length) {
+			std::istream* input = ((YAMLParser*) instance)->input;
+			input->read((char*) buffer, size);
+			*length = input->gcount();
+			return (input->good() || input->eof()) ? 1 : 0;
+		}
+	public:
+
+		YAMLParser() {
+			yaml_parser_initialize(&yamlParser);
+		}
+
+		virtual ~YAMLParser() {
+			yaml_parser_delete(&yamlParser);
+		}
+
+		YAMLParser(const YAMLParser&) = delete;
+		YAMLParser& operator=(const YAMLParser&) = delete;
+
+		void setInput(std::istream* input) {
+			this->input = input;
+			yaml_parser_set_input(&yamlParser, readFromInput, (void*) this);
+		}
+
+		YAMLEvent* next() {
+			yaml_event_t event;
+			int result = yaml_parser_parse(&yamlParser, &event);
+			return result == 1 && event.type != yaml_event_type_e::YAML_NO_EVENT ? new YAMLEvent(event) : nullptr; // 1 = OK in yaml.h; YAML_NO_EVENT = end
+		}
+	};
+
+	YAMLParser parser;
+
+	using YAMLEvent_p = std::shared_ptr<YAMLEvent>;
+
+	enum class Mode {
+		ROOT,
+		RELATIONS_SEQUENCE,
+		MAPPING,
+		MAP_KEY
+	};
+
+	std::vector<Mode> mode;
+
+	/**
+	 * Both YAML and XML strings are in UTF-8.
+	 */
+	const char* y2x(yaml_char_t* value) {
+		return value ? (const char*) value : "";
+	}
+
 public:
 
+	YAMLCommand() {
+	}
+
+	virtual ~YAMLCommand() {
+	}
+
 	void process(std::istream& input, std::shared_ptr<relpipe::writer::RelationalWriter> writer) {
+		parser.setInput(&input);
+
 		std::vector<relpipe::writer::AttributeMetadata> attributesMetadata;
 		attributesMetadata.push_back(relpipe::writer::AttributeMetadata{L"todo", relpipe::writer::TypeId::STRING});
 		writer->startRelation(L"YAML", attributesMetadata, true);
+
+		mode.push_back(Mode::ROOT);
+		std::string itemName;
+
+		writer->writeAttribute(L"before cycle");
+		for (YAMLEvent_p event = YAMLEvent_p(parser.next()); event; event = YAMLEvent_p(parser.next())) {
+
+			if (event->getType() == YAML_NO_EVENT) { writer->writeAttribute(L"YAML_NO_EVENT");
+			} else if (event->getType() == YAML_STREAM_START_EVENT) { writer->writeAttribute(L"YAML_STREAM_START_EVENT");
+			} else if (event->getType() == YAML_STREAM_END_EVENT) { writer->writeAttribute(L"YAML_STREAM_END_EVENT");
+			} else if (event->getType() == YAML_DOCUMENT_START_EVENT) { writer->writeAttribute(L"YAML_DOCUMENT_START_EVENT");
+			} else if (event->getType() == YAML_DOCUMENT_END_EVENT) { writer->writeAttribute(L"YAML_DOCUMENT_END_EVENT");
+			} else if (event->getType() == YAML_ALIAS_EVENT) { writer->writeAttribute(L"YAML_ALIAS_EVENT");
+			} else if (event->getType() == YAML_SCALAR_EVENT) { writer->writeAttribute(L"YAML_SCALAR_EVENT");
+			} else if (event->getType() == YAML_SEQUENCE_START_EVENT) { writer->writeAttribute(L"YAML_SEQUENCE_START_EVENT");
+			} else if (event->getType() == YAML_SEQUENCE_END_EVENT) { writer->writeAttribute(L"YAML_SEQUENCE_END_EVENT");
+			} else if (event->getType() == YAML_MAPPING_START_EVENT) { writer->writeAttribute(L"YAML_MAPPING_START_EVENT");
+			} else if (event->getType() == YAML_MAPPING_END_EVENT) { writer->writeAttribute(L"YAML_MAPPING_END_EVENT");
+			} else { writer->writeAttribute(L"else???");
+				// TODO: unsupported type?
+			}
+		}
+		writer->writeAttribute(L"after cycle");
 	}
 };