StdInCommand.h
author František Kučera <franta-hg@frantovo.cz>
Sat, 28 Jul 2018 22:29:40 +0200
branchv_0
changeset 16 70af16946466
parent 11 ArgumentsCommand.h@3798b6bc9aea
child 17 f543468a7e49
permissions -rw-r--r--
StdInCommand: generate relational data from STDIN (and also CLI arguments)
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
11
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     1
#pragma once
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     2
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     3
#include <cstdlib>
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     4
#include <iostream>
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     5
#include <string>
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     6
#include <vector>
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     7
#include <algorithm>
16
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
     8
#include <string>
11
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     9
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    10
#include <typedefs.h>
16
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    11
#include <locale>
11
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    12
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    13
#include "Command.h"
16
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    14
#include "CLI.h"
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    15
#include "RelpipeCLIException.h"
11
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    16
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    17
namespace relpipe {
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    18
namespace in {
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    19
namespace cli {
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    20
16
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    21
/**
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    22
 * TODO: consider code merge with ArgumentsCommand
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    23
 */
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    24
class StdInCommand : public Command {
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    25
private:
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    26
	wstring_convert<codecvt_utf8<wchar_t>> convertor; // TODO: support also other encodings.
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    27
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    28
	/**
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    29
	 * Reads next value from arguments and (if no arguments left) from input
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    30
	 * @param input
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    31
	 * @param arguments
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    32
	 * @param i current index in the arguments vector
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    33
	 * @param required if true, exception is thrown when no data found
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    34
	 * @return 
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    35
	 */
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    36
	relpipe::writer::string_t readNext(std::istream& input, const std::vector<relpipe::writer::string_t>& arguments, size_t& i, relpipe::writer::boolean_t required) {
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    37
		using namespace relpipe::writer;
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    38
		using namespace relpipe::cli;
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    39
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    40
		if (i < arguments.size()) {
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    41
			return arguments[i++];
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    42
		} else {
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    43
			std::stringstream value;
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    44
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    45
			while (true) {
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    46
				char ch;
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    47
				input.get(ch);
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    48
				if (ch == 0 || input.eof() || input.fail()) break;
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    49
				else value << ch;
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    50
			}
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    51
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    52
			return convertor.from_bytes(value.str());
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    53
		}
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    54
	}
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    55
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    56
11
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    57
public:
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    58
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    59
	void process(std::istream& input, std::ostream& output, const relpipe::writer::string_t& command, const std::vector<relpipe::writer::string_t>& arguments) override {
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    60
		using namespace relpipe::writer;
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    61
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    62
		size_t i = 0;
16
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    63
		string_t relationName = readNext(input, arguments, i, true);
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    64
		integer_t attributeCount = std::stoul(readNext(input, arguments, i, true)); // TODO: use integer data type's method?
11
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    65
		boolean_t writeHeader = true; // TODO: add option for header omitting
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    66
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    67
		std::shared_ptr<RelationalWriter> writer(Factory::create(output));
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    68
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    69
		std::vector<std::pair<string_t, TypeId >> attributes(attributeCount);
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    70
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    71
		for (size_t j = 0; j < attributeCount; j++) {
16
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    72
			string_t attributeName = readNext(input, arguments, i, true);
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    73
			TypeId attributeType = writer->toTypeId(readNext(input, arguments, i, true));
11
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    74
			attributes[j] = {attributeName, attributeType};
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    75
		}
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    76
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    77
		writer->startRelation(relationName, attributes, writeHeader);
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    78
16
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    79
		while (true) {
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    80
			string_t value = readNext(input, arguments, i, false);
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    81
			if (value.empty() && (input.eof() || input.fail())) break;
70af16946466 StdInCommand: generate relational data from STDIN (and also CLI arguments)
František Kučera <franta-hg@frantovo.cz>
parents: 11
diff changeset
    82
			else writer->writeAttribute(value);
11
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    83
		}
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    84
	}
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    85
};
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    86
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    87
}
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    88
}
3798b6bc9aea ArgumentsCommand: generate relational data from CLI arguments
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    89
}