#include "uscxml/config.h"
#include "uscxml/Interpreter.h"
#include "uscxml/util/String.h"
#include "uscxml/transform/ChartToC.h"
#include "uscxml/transform/ChartToJava.h"
#include "uscxml/transform/ChartToVHDL.h"
#include "uscxml/transform/ChartToPromela.h"

#include <boost/algorithm/string.hpp>

#include <fstream>
#include <iostream>
#include <map>

#include "uscxml/plugins/Factory.h"
#include "uscxml/server/HTTPServer.h"
#include "getopt.h"

#include "uscxml/interpreter/Logging.h"

#define ANNOTATE(envKey, annotationParam) \
envVarIsTrue(envKey) || std::find(options.begin(), options.end(), annotationParam) != options.end()

void printUsageAndExit(const char* progName) {
	// remove path from program name
	std::string progStr(progName);
	if (progStr.find_last_of(PATH_SEPERATOR) != std::string::npos) {
		progStr = progStr.substr(progStr.find_last_of(PATH_SEPERATOR) + 1, progStr.length() - (progStr.find_last_of(PATH_SEPERATOR) + 1));
	}

	printf("%s version " USCXML_VERSION " (" CMAKE_BUILD_TYPE " build - " CMAKE_COMPILER_STRING ")\n", progStr.c_str());
	printf("Usage\n");
	printf("\t%s", progStr.c_str());
	printf(" [-t c|pml|flat|min] [-a {OPTIONS}] [-v] [-lN]");
#ifdef BUILD_AS_PLUGINS
	printf(" [-p pluginPath]");
#endif
	printf(" [-i URL] [-o FILE]");
	printf("\n");
	printf("Options\n");
	printf("\t-t c           : convert to C program\n");
	printf("\t-t pml         : convert to spin/promela program\n");
    printf("\t-t vhdl        : convert to VHDL hardware description\n");
    printf("\t-t java        : convert to Java classes\n");
	printf("\t-t flat        : flatten to SCXML state-machine\n");
	printf("\t-a FILE        : write annotated SCXML document for transformation\n");
	printf("\t-X {PARAMETER} : pass additional parameters to the transformation\n");
	printf("\t    prefix=ID    - prefix all symbols and identifiers with ID (-tc)\n");
	printf("\t-v             : be verbose\n");
	printf("\t-lN            : Set loglevel to N\n");
	printf("\t-i URL         : Input file (defaults to STDIN)\n");
	printf("\t-o FILE        : Output file (defaults to STDOUT)\n");
	printf("\n");
	exit(1);
}

int main(int argc, char** argv) {
	using namespace uscxml;

	bool verbose = false;
	std::string outType;
	std::string pluginPath;
    std::string inputFile;
    std::string annotatedFile;
	std::string outputFile;
	std::list<std::string> options;
	std::multimap<std::string, std::string> extensions;

#if defined(HAS_SIGNAL_H) && !defined(WIN32)
	signal(SIGPIPE, SIG_IGN);
#endif

	optind = 0;
	opterr = 0;

	struct option longOptions[] = {
		{"verbose",       no_argument,       0, 'v'},
		{"type",          required_argument, 0, 't'},
		{"annotate",      required_argument, 0, 'a'},
		{"param",         required_argument, 0, 'X'},
		{"plugin-path",   required_argument, 0, 'p'},
		{"input-file",    required_argument, 0, 'i'},
		{"output-file",   required_argument, 0, 'o'},
		{"loglevel",      required_argument, 0, 'l'},
		{0, 0, 0, 0}
	};

	// parse global options
	int optionInd = 0;
	int option;
	for (;;) {
		option = getopt_long_only(argc, argv, "+vp:X:t:i:o:l:a:", longOptions, &optionInd);
		if (option == -1) {
			break;
		}
		switch(option) {
		// cases without short option
		case 0: {
			break;
		}
		// cases with short-hand options
		case 'v':
			verbose = true;
			break;
		case 't':
			outType = optarg;
			break;
		case 'p':
			pluginPath = optarg;
			break;
		case 'i':
			inputFile = optarg;
			break;
		case 'a':
            annotatedFile = optarg;
			break;
		case 'X': {
			std::list<std::string> extension = tokenize(optarg, '=');
			if (extension.size() != 2)
				printUsageAndExit(argv[0]);
			std::string key = boost::trim_copy(*(extension.begin()));
			std::string value = boost::trim_copy(*(++extension.begin()));
			extensions.insert(std::pair<std::string, std::string>(key, value));
		}
		break;
		case 'o':
			outputFile = optarg;
            extensions.insert(std::pair<std::string, std::string>("outputFile", outputFile));
			break;
		case 'l':
			break;
		case '?': {
			break;
		}
		default:
			break;
		}
	}

	// make sure given annotation options are available in the environment
	if(ANNOTATE("USCXML_ANNOTATE_GLOBAL_STATE_STEP", "step"))
		setenv("USCXML_ANNOTATE_GLOBAL_STATE_STEP", "YES", 1);

	if (ANNOTATE("USCXML_ANNOTATE_GLOBAL_TRANS_PRIO", "priority"))
		setenv("USCXML_ANNOTATE_GLOBAL_TRANS_PRIO", "YES", 1);

	if (ANNOTATE("USCXML_ANNOTATE_VERBOSE_COMMENTS", "verbose"))
		setenv("USCXML_ANNOTATE_VERBOSE_COMMENTS", "YES", 1);

	if (ANNOTATE("USCXML_ANNOTATE_TRANS_EXITSET", "exitset"))
		setenv("USCXML_ANNOTATE_TRANS_EXITSET", "YES", 1);

	if (ANNOTATE("USCXML_ANNOTATE_TRANS_DOMAIN", "domain"))
		setenv("USCXML_ANNOTATE_TRANS_DOMAIN", "YES", 1);

	if (ANNOTATE("USCXML_ANNOTATE_TRANS_CONFLICTS", "conflicts"))
		setenv("USCXML_ANNOTATE_TRANS_CONFLICTS", "YES", 1);

	if (ANNOTATE("USCXML_ANNOTATE_TRANS_ENTRYSET", "entryset"))
		setenv("USCXML_ANNOTATE_TRANS_ENTRYSET", "YES", 1);

	if(ANNOTATE("USCXML_ANNOTATE_GLOBAL_TRANS_MEMBERS", "members"))
		setenv("USCXML_ANNOTATE_GLOBAL_TRANS_MEMBERS", "YES", 1);

	if(ANNOTATE("USCXML_ANNOTATE_GLOBAL_TRANS_SENDS", "sends"))
		setenv("USCXML_ANNOTATE_GLOBAL_TRANS_SENDS", "YES", 1);

	if(ANNOTATE("USCXML_ANNOTATE_GLOBAL_TRANS_RAISES", "raises"))
		setenv("USCXML_ANNOTATE_GLOBAL_TRANS_RAISES", "YES", 1);

	if(ANNOTATE("USCXML_ANNOTATE_PROGRESS", "progress"))
		setenv("USCXML_ANNOTATE_PROGRESS", "YES", 1);

	if(ANNOTATE("USCXML_ANNOTATE_NOCOMMENT", "nocomment"))
		setenv("USCXML_ANNOTATE_NOCOMMENT", "YES", 1);


//	if (outType.length() == 0 && outputFile.length() > 0) {
//		// try to get type from outfile extension
//		size_t dotPos = outputFile.find_last_of(".");
//		if (dotPos != std::string::npos) {
//			outType= outputFile.substr(dotPos + 1);
//		}
//	}

//	if (outType.length() == 0)
//		printUsageAndExit(argv[0]);

	if (outType != "flat" &&
	        outType != "scxml" &&
	        outType != "pml" &&
	        outType != "c" &&
            outType != "vhdl" &&
            outType != "java" &&
	        outType != "min" &&
	        std::find(options.begin(), options.end(), "priority") == options.end() &&
	        std::find(options.begin(), options.end(), "domain") == options.end() &&
	        std::find(options.begin(), options.end(), "conflicts") == options.end() &&
	        std::find(options.begin(), options.end(), "exitset") == options.end() &&
	        std::find(options.begin(), options.end(), "entryset") == options.end())
		printUsageAndExit(argv[0]);

	// register plugins
	if (pluginPath.length() > 0) {
		Factory::setDefaultPluginPath(pluginPath);
	}

	// start HTTP server
	HTTPServer::getInstance(30444, 30445, NULL);

	Interpreter interpreter;
	try {
		if (inputFile.size() == 0 || inputFile == "-") {
			LOGD(USCXML_INFO) << "Reading SCXML from STDIN" << std::endl;
			std::stringstream ss;
			std::string line;
			while (std::getline(std::cin, line)) {
				ss << line;
			}
			URL tmp("anonymous.scxml");
            tmp = URL::resolveWithCWD(tmp);
			interpreter = Interpreter::fromXML(ss.str(), tmp);
		} else {
			interpreter = Interpreter::fromURL(inputFile);
		}
	} catch (Event e) {
		// we will reattempt below, not yet a fatal error
	} catch (const std::exception &e) {
		std::cout << e.what() << std::endl;
	}

	if (!interpreter) {
		URL tmp(inputFile);
        tmp = URL::resolveWithCWD(tmp);
        std::string content = tmp.getInContent();

		std::string inlineBeginMarker = "INLINE SCXML BEGIN\n";
		std::string inlineEndMarker = "\nINLINE SCXML END";

		size_t inlineSCXMLBegin = content.find(inlineBeginMarker);
		if (inlineSCXMLBegin != std::string::npos) {
			inlineSCXMLBegin += inlineBeginMarker.size();
			size_t inlineSCXMLEnd = content.find(inlineEndMarker);
			std::string inlineSCXMLContent = content.substr(inlineSCXMLBegin, inlineSCXMLEnd - inlineSCXMLBegin);
			try {
				interpreter = Interpreter::fromXML(inlineSCXMLContent, tmp);
			} catch (Event e) {
				std::cout << e << std::endl;
			} catch (const std::exception &e) {
				std::cout << e.what() << std::endl;
			}
		}

	}

	if (!interpreter) {
		LOGD(USCXML_ERROR) << "Cannot create interpreter from " << inputFile << std::endl;
		exit(EXIT_FAILURE);

	}

	try {
		if (verbose) {
			std::list<InterpreterIssue> issues = interpreter.validate();
			for (std::list<InterpreterIssue>::iterator issueIter = issues.begin(); issueIter != issues.end(); issueIter++) {
				std::cerr << *issueIter << std::endl;
			}
		}

        Transformer transformer;
		if (outType == "c") {
			transformer = ChartToC::transform(interpreter);
			transformer.setExtensions(extensions);
			transformer.setOptions(options);
            
			if (outputFile.size() == 0 || outputFile == "-") {
				transformer.writeTo(std::cout);
			} else {
				std::ofstream outStream;
				outStream.open(outputFile.c_str());
				transformer.writeTo(outStream);
				outStream.close();
			}
		}

        if (outType == "java") {
            transformer = ChartToJava::transform(interpreter);
            transformer.setExtensions(extensions);
            transformer.setOptions(options);
            
            if (outputFile.size() == 0 || outputFile == "-") {
                transformer.writeTo(std::cout);
            } else {
                std::ofstream outStream;
                outStream.open(outputFile.c_str());
                transformer.writeTo(outStream);
                outStream.close();
            }
        }

		if (outType == "vhdl") {
            transformer = ChartToVHDL::transform(interpreter);
            transformer.setExtensions(extensions);
            transformer.setOptions(options);

			if (outputFile.size() == 0 || outputFile == "-") {
				transformer.writeTo(std::cout);
			} else {
				std::ofstream outStream;
				outStream.open(outputFile.c_str());
				transformer.writeTo(outStream);
				outStream.close();
			}
		}

		if (outType == "pml") {
            transformer = ChartToPromela::transform(interpreter);
            transformer.setExtensions(extensions);
            transformer.setOptions(options);
            
            if (outputFile.size() == 0 || outputFile == "-") {
                transformer.writeTo(std::cout);
            } else {
                std::ofstream outStream;
                outStream.open(outputFile.c_str());
                transformer.writeTo(outStream);
                outStream.close();
            }
		}

//		if (outType == "tex") {
//			if (outputFile.size() == 0 || outputFile == "-") {
//				ChartToTex::transform(interpreter).writeTo(std::cout);
//			} else {
//				std::ofstream outStream;
//				outStream.open(outputFile.c_str());
//				ChartToTex::transform(interpreter).writeTo(outStream);
//				outStream.close();
//			}
//			exit(EXIT_SUCCESS);
//		}

//		if (outType == "flat") {
//			if (outputFile.size() == 0 || outputFile == "-") {
//				ChartToFlatSCXML::transform(interpreter).writeTo(std::cout);
//			} else {
//				std::ofstream outStream;
//				outStream.open(outputFile.c_str());
//				ChartToFlatSCXML::transform(interpreter).writeTo(outStream);
//				outStream.close();
//			}
//			exit(EXIT_SUCCESS);
//		}

//		if (outType == "min") {
//			if (outputFile.size() == 0 || outputFile == "-") {
//				ChartToMinimalSCXML::transform(interpreter).writeTo(std::cout);
//			} else {
//				std::ofstream outStream;
//				outStream.open(outputFile.c_str());
//				ChartToMinimalSCXML::transform(interpreter).writeTo(outStream);
//				outStream.close();
//			}
//			exit(EXIT_SUCCESS);
//		}

        if (annotatedFile.size() > 0) {
            std::ofstream outStream;
            outStream.open(annotatedFile.c_str());
            outStream << (*transformer.getImpl()->getDocument());
            outStream.close();

        }


	} catch (Event e) {
		std::cout << e << std::endl;
        return EXIT_FAILURE;
	} catch (const std::exception &e) {
		std::cout << e.what() << std::endl;
        return EXIT_FAILURE;
	}

    return EXIT_SUCCESS;
}