/**
* Relational pipes
* Copyright © 2018 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, either version 3 of the License, or
* (at your option) any later version.
*
* 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 <cstdlib>
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <relpipe/writer/typedefs.h>
#include <relpipe/writer/AttributeMetadata.h>
#include "Command.h"
namespace relpipe {
namespace in {
namespace cli {
class ArgumentsCommand : public Command {
public:
void process(std::istream& input, std::ostream& output, const relpipe::writer::string_t& command, const std::vector<relpipe::writer::string_t>& arguments) override {
using namespace relpipe::writer;
size_t i = 0;
string_t relationName = arguments[i++];
integer_t attributeCount = std::stoul(arguments[i++]); // TODO: use integer data type's method?
boolean_t writeHeader = true; // TODO: add option for header omitting
// TODO: check argument count
std::shared_ptr<RelationalWriter> writer(Factory::create(output));
std::vector<AttributeMetadata> attributes(attributeCount);
for (size_t j = 0; j < attributeCount; j++) {
string_t attributeName = arguments[i++];
TypeId attributeType = writer->toTypeId(arguments[i++]);
attributes[j] = {attributeName, attributeType};
}
writer->startRelation(relationName, attributes, writeHeader);
for (; i < arguments.size(); i++) {
writer->writeAttribute(arguments[i]);
}
}
};
}
}
}