#include "argz.hpp"

#include "scanner.hpp"

#include <cstdlib>

#include <fstream>

#include <iostream>

#include <memory>

#include <sstream>

#include <string>


extern int html_main(const char *filename, std::ostream &out);

int scanFile(const std::string &contents) {
    try {
        std::unique_ptr<scan::Scanner> scan(new scan::Scanner(scan::TString(contents)));
        uint64_t tokens = scan->scan();
        std::cout << "Sucessfull scan returned: " << tokens << " token(s)...\n";

        for (size_t i = 0; i < scan->size(); ++i) {
            scan->operator[](i).print(std::cout);
        }
        return static_cast<int>(tokens);
    } catch (scan::ScanExcept &e) {
        std::cerr << "Fatal error: " << e.why() << "\n";
    }
    return 0;
}

int main(int argc, char **argv) {
    Argz<std::string> argz(argc, argv);
    argz.addOptionSingleValue('i', "input text").addOptionSingleValue('o', "output file").addOptionSingle('h', "help").addOptionSingle('v', "help");
    std::string in_file, out_file;
    int value = 0;

    Argument<std::string> arg;
    try {
        while ((value = argz.proc(arg)) != -1) {
            switch (value) {
            case 'h':
            case 'v':
                argz.help(std::cout);
                exit(EXIT_SUCCESS);
                break;
            case 'i':
                in_file = arg.arg_value;
                break;
            case 'o':
                out_file = arg.arg_value;
                break;
            }
        }
    } catch (const ArgException<std::string> &e) {
        std::cerr << "Syntax Error: " << e.text() << "\n";
    }

    if (in_file.length() == 0) {
        std::cerr << "Input file must be provided...use -i \n";
        exit(EXIT_FAILURE);
    }

    std::fstream file;
    file.open(in_file, std::ios::in);
    std::ostringstream stream;
    stream << file.rdbuf();
    file.close();

    if (in_file.length() > 0 && out_file.length() == 0) {
        if (stream.str().length() > 0) {
            return scanFile(stream.str());
        }
    } else if (out_file.length() > 0) {
        std::fstream file;
        file.open(out_file, std::ios::out);
        if (!file.is_open()) {
            std::cerr << "Error could not open output file: " << out_file << "\n";
            exit(EXIT_FAILURE);
        }
        html_main(in_file.c_str(), file);
    }
    return 0;
}