/**

 * Relational pipes

 * Copyright © 2021 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, version 3 of the License.

 *

 * 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 <codecvt>

#include <memory>

#include <iostream>

#include "Configuration.h"

namespace relpipe {

namespace in {

namespace x11 {

class X11Command {

private:

	std::wstring_convert<codecvt_utf8<wchar_t>> convertor; // TODO: support also other encodings and use platform encoding as default

	void processRelation(RelationConfiguration& configuration, std::shared_ptr<writer::RelationalWriter> writer) {

		// Write header / metadata:

		std::vector<relpipe::writer::AttributeMetadata> attributesMetadata;

		for (AttributeRecipe ar : configuration.attributes) attributesMetadata.push_back({ar.name, ar.type});

		writer->startRelation(configuration.relation, attributesMetadata, configuration.writeHeader);

		// Write records from CLI:

		for (auto value : configuration.values) writer->writeAttribute(value);

		// Write records from STDIN:

		if (configuration.valueStream) {

			std::stringstream rawValue;

			while (true) {

				char ch = 0;

				configuration.valueStream->get(ch);

				if (ch == 0 && configuration.valueStream->good()) {

					writer->writeAttribute(convertor.from_bytes(rawValue.str()));

					rawValue.str("");

					rawValue.clear();

				} else if (configuration.valueStream->good()) {

					rawValue << ch;

				} else if (configuration.valueStream->eof()) {

					return;

				} else {

					throw relpipe::cli::RelpipeCLIException(L"Error while reading values from the input stream.", relpipe::cli::CLI::EXIT_CODE_UNEXPECTED_ERROR); // TODO: better exception

				}

			}

		}

	}

public:

	void process(Configuration& configuration, std::shared_ptr<writer::RelationalWriter> writer) {

		for (RelationConfiguration& relationConfiguration : configuration.relationConfigurations) processRelation(relationConfiguration, writer);

	}

};

}

}

}