/**

 * 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

 *

 * 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/>.

 */

#include <cstdlib>

#include <memory>

#include <relpipe/writer/RelationalWriter.h>

#include <relpipe/writer/RelpipeWriterException.h>

#include <relpipe/writer/Factory.h>

#include <relpipe/writer/TypeId.h>

#include <relpipe/cli/CLI.h>

#include <relpipe/cli/RelpipeCLIException.h>

#include "Command.h"

#include "ArgumentsCommand.h"

#include "DemoCommand.h"

#include "StdInCommand.h"

using namespace relpipe::cli;

using namespace relpipe::in::cli;

using namespace relpipe::writer;

Command* findCommand(string_t commandName) {

	// TODO: better command names

	// TODO: help command

	if (commandName == L"demo") return new DemoCommand();

	else if (commandName == L"generate") return new ArgumentsCommand();

	else if (commandName == L"generate-without-stdin") return new StdInCommand(false); // TODO: just for testing, StdInCommand(false) should work same as ArgumentsCommand()

	else if (commandName == L"generate-from-stdin") return new StdInCommand(true);

	else throw RelpipeCLIException(L"Unknown command: " + commandName, CLI::EXIT_CODE_UNKNOWN_COMMAND);

}

int main(int argc, char** argv) {

	setlocale(LC_ALL, "");

	CLI::untieStdIO();

	CLI cli(argc, argv);

	int resultCode = CLI::EXIT_CODE_UNEXPECTED_ERROR;

	try {

		if (cli.arguments().size() > 0) {

			const wstring commandName = cli.arguments()[0];

			vector<wstring> arguments(cli.arguments().size() - 1);

			for (int i = 1; i < cli.arguments().size(); i++) {

				arguments[i - 1] = cli.arguments()[i];

			}

			std::shared_ptr<Command> command(findCommand(commandName));

			command->process(cin, cout, commandName, arguments);

			resultCode = CLI::EXIT_CODE_SUCCESS;

		} else {

			throw RelpipeCLIException(L"Missing command…", CLI::EXIT_CODE_BAD_SYNTAX);

		}

	} catch (RelpipeCLIException e) {

		fwprintf(stderr, L"Caught CLI exception: %ls\n", e.getMessge().c_str());

		fwprintf(stderr, L"Debug: Input stream: eof=%ls, lastRead=%d\n", (cin.eof() ? L"true" : L"false"), cin.gcount());

		resultCode = e.getExitCode();

	} catch (RelpipeWriterException e) {

		fwprintf(stderr, L"Caught Writer exception: %ls\n", e.getMessge().c_str());

		fwprintf(stderr, L"Debug: Input stream: eof=%ls, lastRead=%d\n", (cin.eof() ? L"true" : L"false"), cin.gcount());

		resultCode = CLI::EXIT_CODE_DATA_ERROR;

	}

	return resultCode;

}