author | František Kučera <franta-hg@frantovo.cz> |
Sat, 28 Jul 2018 22:29:40 +0200 | |
branch | v_0 |
changeset 16 | 70af16946466 |
parent 11 | ArgumentsCommand.h@3798b6bc9aea |
child 17 | f543468a7e49 |
permissions | -rw-r--r-- |
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 |
} |