// **********************************************************************
//
// Copyright (c) 2001
// MutableRealms, Inc.
// Huntsville, AL, USA
//
// All Rights Reserved
//
// **********************************************************************

#include <Gen.h>
#include <fstream>

using namespace std;
using namespace Slice;

void
usage(const char* n)
{
    cerr << "Usage: " << n << " [options] slice-files...\n";
    cerr <<        
        "Options:\n"
        "-h, --help              Show this message.\n"
        "-v, --version           Display the Ice version.\n"
        "-DNAME                  Define NAME as 1.\n"
        "-DNAME=DEF              Define NAME as DEF.\n"
        "-UNAME                  Remove any definition for NAME.\n"
        "-IDIR                   Put DIR in the include file search path.\n"
        "--output-dir DIR        Create files in the directory DIR.\n"
        "--package PKG           Generate everything in package PKG.\n"
        "--tie                   Generate TIE classes.\n"
        "--impl                  Generate sample implementations.\n"
        "--impl-tie              Generate sample TIE implementations.\n"
        "-d, --debug             Print debug messages.\n"
        ;
}

int
main(int argc, char* argv[])
{
    string cpp("cpp");
    vector<string> includePaths;
    string output;
    string package;
    bool tie = false;
    bool impl = false;
    bool implTie = false;
    bool debug = false;

    int idx = 1;
    while (idx < argc)
    {
        if (strncmp(argv[idx], "-I", 2) == 0)
        {
            cpp += ' ';
            cpp += argv[idx];

            string path = argv[idx] + 2;
            if (path.length())
            {
                includePaths.push_back(path);
            }

            for (int i = idx ; i + 1 < argc ; ++i)
            {
                argv[i] = argv[i + 1];
            }
            --argc;
        }
        else if (strncmp(argv[idx], "-D", 2) == 0 ||
                 strncmp(argv[idx], "-U", 2) == 0)
        {
            cpp += ' ';
            cpp += argv[idx];

            for (int i = idx ; i + 1 < argc ; ++i)
            {
                argv[i] = argv[i + 1];
            }
            --argc;
        }
        else if (strcmp(argv[idx], "-h") == 0 ||
                 strcmp(argv[idx], "--help") == 0)
        {
            usage(argv[0]);
            return EXIT_SUCCESS;
        }
        else if (strcmp(argv[idx], "-v") == 0 ||
                 strcmp(argv[idx], "--version") == 0)
        {
            cout << ICE_STRING_VERSION << endl;
            return EXIT_SUCCESS;
        }
        else if (strcmp(argv[idx], "-d") == 0 ||
                 strcmp(argv[idx], "--debug") == 0)
        {
            debug = true;
            for (int i = idx ; i + 1 < argc ; ++i)
            {
                argv[i] = argv[i + 1];
            }
            --argc;
        }
        else if (strcmp(argv[idx], "--output-dir") == 0)
        {
            if (idx + 1 >= argc)
            {
                cerr << argv[0] << ": argument expected for`" << argv[idx]
                     << "'" << endl;
                usage(argv[0]);
                return EXIT_FAILURE;
            }

            output = argv[idx + 1];
            for (int i = idx ; i + 2 < argc ; ++i)
            {
                argv[i] = argv[i + 2];
            }
            argc -= 2;
        }
        else if (strcmp(argv[idx], "--package") == 0)
        {
            if (idx + 1 >= argc)
            {
                cerr << argv[0] << ": argument expected for`" << argv[idx]
                     << "'" << endl;
                usage(argv[0]);
                return EXIT_FAILURE;
            }

            package = argv[idx + 1];
            for (int i = idx ; i + 2 < argc ; ++i)
            {
                argv[i] = argv[i + 2];
            }
            argc -= 2;
        }
        else if (strcmp(argv[idx], "--tie") == 0)
        {
            tie = true;
            for (int i = idx ; i + 1 < argc ; ++i)
            {
                argv[i] = argv[i + 1];
            }
            --argc;
        }
        else if (strcmp(argv[idx], "--impl") == 0)
        {
            impl = true;
            for (int i = idx ; i + 1 < argc ; ++i)
            {
                argv[i] = argv[i + 1];
            }
            --argc;
        }
        else if (strcmp(argv[idx], "--impl-tie") == 0)
        {
            implTie = true;
            for (int i = idx ; i + 1 < argc ; ++i)
            {
                argv[i] = argv[i + 1];
            }
            --argc;
        }
        else if (argv[idx][0] == '-')
        {
            cerr << argv[0] << ": unknown option `" << argv[idx] << "'"
                 << endl;
            usage(argv[0]);
            return EXIT_FAILURE;
        }
        else
        {
            ++idx;
        }
    }

    if (argc < 2)
    {
        cerr << argv[0] << ": no input file" << endl;
        usage(argv[0]);
        return EXIT_FAILURE;
    }

    if (impl && implTie)
    {
        cerr << argv[0] << ": cannot specify both --impl and --impl-tie" << endl;
        usage(argv[0]);
        return EXIT_FAILURE;
    }

    int status = EXIT_SUCCESS;

    for (idx = 1 ; idx < argc ; ++idx)
    {
        string base(argv[idx]);
        string suffix;
        string::size_type pos = base.rfind('.');
        if (pos != string::npos)
        {
            suffix = base.substr(pos);
            transform(suffix.begin(), suffix.end(), suffix.begin(), tolower);
        }
        if (suffix != ".ice")
        {
            cerr << argv[0] << ": input files must end with `.ice'" << endl;
            return EXIT_FAILURE;
        }
        base.erase(pos);

        ifstream test(argv[idx]);
        if (!test)
        {
            cerr << argv[0] << ": can't open `" << argv[idx]
                 << "' for reading: " << strerror(errno) << endl;
            return EXIT_FAILURE;
        }
        test.close();

        string cmd = cpp + " " + argv[idx];
#ifdef WIN32
        FILE* cppHandle = _popen(cmd.c_str(), "r");
#else
        FILE* cppHandle = popen(cmd.c_str(), "r");
#endif
        if (cppHandle == 0)
        {
            cerr << argv[0] << ": can't run C++ preprocessor: "
                 << strerror(errno) << endl;
            return EXIT_FAILURE;
        }

        UnitPtr unit = Unit::createUnit(false, false);
        int parseStatus = unit->parse(cppHandle, debug);

#ifdef WIN32
        _pclose(cppHandle);
#else
        pclose(cppHandle);
#endif

        if (parseStatus == EXIT_FAILURE)
        {
            status = EXIT_FAILURE;
        }
        else
        {
            Gen gen(argv[0], base, includePaths, package, output);
            if (!gen)
            {
                unit->destroy();
                return EXIT_FAILURE;
            }
            gen.generate(unit);
            if (tie)
            {
                gen.generateTie(unit);
            }
            if (impl)
            {
                gen.generateImpl(unit);
            }
            if (implTie)
            {
                gen.generateImplTie(unit);
            }
        }

        unit->destroy();
    }

    return status;
}