src/YAMLHandler.h
branchv_0
changeset 0 c52ca92aa593
child 1 af15c47f77ae
equal deleted inserted replaced
-1:000000000000 0:c52ca92aa593
       
     1 /**
       
     2  * Relational pipes
       
     3  * Copyright © 2018 František Kučera (Frantovo.cz, GlobalCode.info)
       
     4  *
       
     5  * This program is free software: you can redistribute it and/or modify
       
     6  * it under the terms of the GNU General Public License as published by
       
     7  * the Free Software Foundation, version 3 of the License.
       
     8  *
       
     9  * This program is distributed in the hope that it will be useful,
       
    10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
       
    11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
       
    12  * GNU General Public License for more details.
       
    13  *
       
    14  * You should have received a copy of the GNU General Public License
       
    15  * along with this program. If not, see <http://www.gnu.org/licenses/>.
       
    16  */
       
    17 #pragma once
       
    18 
       
    19 #include <memory>
       
    20 #include <string>
       
    21 #include <vector>
       
    22 #include <iostream>
       
    23 #include <sstream>
       
    24 #include <locale>
       
    25 #include <codecvt>
       
    26 
       
    27 #include <relpipe/common/type/typedefs.h>
       
    28 #include <relpipe/reader/TypeId.h>
       
    29 #include <relpipe/reader/handlers/RelationalReaderStringHandler.h>
       
    30 #include <relpipe/reader/handlers/AttributeMetadata.h>
       
    31 
       
    32 namespace relpipe {
       
    33 namespace out {
       
    34 namespace yaml {
       
    35 
       
    36 class YAMLHandler : public relpipe::reader::handlers::RelationalReaderStringHandler {
       
    37 private:
       
    38 	std::ostream& output;
       
    39 	std::wstring_convert<std::codecvt_utf8<wchar_t>> convertor; // we generate YAML always in UTF-8 like XML?
       
    40 	std::vector<relpipe::reader::handlers::AttributeMetadata> currentAttributes;
       
    41 	size_t currentAttributeIndex = 0;
       
    42 
       
    43 	void writeYamlValue(const relpipe::common::type::StringX& value) {
       
    44 		// TODO: escaping
       
    45 		output << convertor.to_bytes(value);
       
    46 	}
       
    47 public:
       
    48 
       
    49 	YAMLHandler(std::ostream& output) : output(output) {
       
    50 	}
       
    51 
       
    52 	void startRelation(relpipe::common::type::StringX name, std::vector<relpipe::reader::handlers::AttributeMetadata> attributes) override {
       
    53 		currentAttributes = attributes;
       
    54 	}
       
    55 
       
    56 	void attribute(const relpipe::common::type::StringX& value) override {
       
    57 		if (currentAttributeIndex % currentAttributes.size() == 0) {
       
    58 			currentAttributeIndex = 0;
       
    59 			
       
    60 		} else {
       
    61 			
       
    62 		}
       
    63 
       
    64 		currentAttributeIndex++;
       
    65 	}
       
    66 
       
    67 	void endOfPipe() {
       
    68 		output.flush();
       
    69 	}
       
    70 
       
    71 };
       
    72 
       
    73 }
       
    74 }
       
    75 }