/*
 *   Copyright 2003-2004 The Apache Software Foundation.
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 */

/* Deploy.cpp : Defines the entry point for the console application. */


#include <stdio.h>
#include <stdlib.h>

#ifdef WIN32
#include <direct.h>
#else
#include <unistd.h>
#endif

#include "TranslationUnit.h"
#include "File.h"
#include <string>
#include <map>
#include <iostream>
using namespace std;

extern TranslationUnit* g_pTranslationUnit;
extern int yyparse();
extern void init_keyword_map();
int parse_header_file(const char *filename);
extern FILE *yyin;
extern list<string> g_classesfound;
/* this map is populated with the type mapping and the services informaiion */
/* by the WSDD parser */
map<string, string> g_ClassNamespaces; 

string g_sServiceFile = "";
string g_sWsFileName = "";
string g_sWebServiceClass = "";
list<string> g_sLibraryPaths;
list<string> g_sAxisIncludePaths;
list<string> g_sCompiledLibs;
bool bDontDeleteSourceFiles;

void usage();
bool got_all_options();
bool is_parameters_valid();
bool is_webservice_header_compilable();

#ifdef WIN32
const char* COMPILECOMMAND = "cl *.cpp /c /ML /W3 /GX /O2 /D \"WIN32\" ";
const char* LINKERCOMMAND = "link *.obj /dll /pdb:none ";
#else
const char* COMPILECOMMAND = "gcc *.cpp -c -Wall -Wshadow -fPIC -O2 ";
const char* LINKERCOMMAND = "gcc -lstdc++ -shared -Wl,-soname,";
#endif

#ifdef WIN32
#define OPTIONTAG '/'
#define DELETECOMMAND "del "
#else
#define OPTIONTAG '-'
#define DELETECOMMAND "rm -f "
#endif

#define COMPLILER_FAILED 2

#define DIRPATHLENGTH 256

int main(int argc, char* argv[])
{
    char currentdir[DIRPATHLENGTH];
    string arg;
    char opt;
    string optval = "";
    if (argc == 1) usage();
    bDontDeleteSourceFiles = false;
    for (int n=1; n<argc; n++)
    {
        arg = argv[n];
        if (arg[0] == OPTIONTAG)
        {
            if (!arg[1]) usage();
            opt = arg[1];
            optval = arg.substr(2);
            if ((opt != 'D') && (optval == "")) usage();
            switch (opt)
            {
            case 'I':
                g_sAxisIncludePaths.push_back(optval);
                break;
            case 'L':
                g_sLibraryPaths.push_back(optval);
                break;
            case 'l':
                g_sCompiledLibs.push_back(optval);
                break;
            case 'o':
                g_sServiceFile = optval;
                break;
            case 'D':
                bDontDeleteSourceFiles = true;
                break;
            case 'S':
                g_sWebServiceClass = optval;
                break;
            default:
                usage();
            }
        }
        else if ((n+1) == argc)
        {
            if (g_sWsFileName != "") usage();
            g_sWsFileName = arg;    
        }
        else
        {
            cout << "too many parameters " << arg << endl << endl;     
            usage();
        }
    }

    if (!got_all_options()) usage();
    cout << "Checking for validity of the parameters ..." ;
    if (!is_parameters_valid()) 
    {
        cout << "Invalid" << endl;
        exit(0); /* at least one parameter is invalid */
    }
    cout << "Valid" << endl;
    cout << "Checking web service header file ..." ;
    if (!is_webservice_header_compilable())
    {
        cout << "Failed" << endl;
        exit(0);
    }
    cout << "Success" << endl;

    cout << "Parsing web service header file ..." ; 
    init_keyword_map();
    /* add predefined classes to the list so that parser recognizes them */
    /* Change directory to the directory where the initial web service header 
     * file is
     */
    getcwd(currentdir, DIRPATHLENGTH);
    chdir(g_sWsFileName.substr(0, g_sWsFileName.find_last_of('/')).c_str());
    /* parse wsdd files */
    /* parse web service header files */
    if (parse_header_file(g_sWsFileName.substr
        (g_sWsFileName.find_last_of('/')+1, string::npos).c_str()))
    {
        cout << "Failed" << endl;
        exit(0);
    }
    chdir(currentdir);
    /* here we have to continue parsing wsdd file */
    cout << "Done" << endl;
    g_pTranslationUnit->SetWsFileName(g_sWsFileName.c_str());
    cout << "Generating Wrapper class declaration file ...";
    if (g_pTranslationUnit->GenerateWrapperClassDef()) 
    {
        cout << "Failed" << endl;
        exit(0);
    }
    cout << "Done" << endl;
    cout << "Generating Wrapper class implimentation file ...";
    if (g_pTranslationUnit->GenerateWrapperClassImpl())
    {
        cout << "Failed" << endl;
        exit(0);
    }
    cout << "Done" << endl;
    cout << "Generating Export functions file ...";
    if (g_pTranslationUnit->GenerateServiceFile(g_sServiceFile))
    {
        cout << "Failed" << endl;
        exit(0);
    }
    cout << "Done" << endl;

    cout << "Generating WSDL file for the service ...";
    string sURI = "www.opensource.lk";
    /* this should be taken from the WSDD file */
    if (g_pTranslationUnit->GenerateWSDL(g_sServiceFile,sURI))
    {
        cout << "Failed" << endl;
        exit(0);
    }
    cout << "Done" << endl;

    string command;
    list<string>::iterator sit;

    command = COMPILECOMMAND;
    for (sit = g_sAxisIncludePaths.begin(); sit != g_sAxisIncludePaths.end();
    sit++)
    {
        command += "-I" + (*sit) + " ";
    }    

    cout << "Compiling ...";
    if (system(command.c_str()) == -1)
    {
        cout << "Failed" << endl;
        exit(0);
    }
    cout << "Done" << endl;
    
    command = LINKERCOMMAND;
#ifdef WIN32
    command += "/out:\"" + g_sServiceFile + ".dll\" ";
#else
    command += g_sServiceFile + ".so -o " + g_sServiceFile + ".so *.o ";
#endif
    for (sit = g_sLibraryPaths.begin(); sit != g_sLibraryPaths.end(); sit++)
    {
#ifdef WIN32
        command += "/libpath:\"" + (*sit) + "\" ";
#else
        command += "-L " + (*sit) + " ";
#endif
    }
    for (sit = g_sCompiledLibs.begin(); sit != g_sCompiledLibs.end(); sit++)
    {
        command += (*sit) + " ";
    }
    cout << "Linking ...";
    if (system(command.c_str()) == -1)
    {
        cout << "Failed" << endl;
        exit(0);
    }
    cout << "Done" << endl;

    cout << "Deleting temporary files ...";
    command = DELETECOMMAND;
#ifdef WIN32
    command += "*.obj *.exp *.lib ";
#else
    command += "*.o *.a ";
#endif
    if (!bDontDeleteSourceFiles)
    {
        command += "*.cpp *.hpp";
    }
    system(command.c_str());
    cout << "Done" << endl;
    return 0;
}

int parse_header_file(const char *filename)
{
    int error = 0;
    if (filename) 
    {
        if ((yyin = fopen( filename, "r" ))) 
        {
            error = yyparse();
            if ( error )
            {
                fclose(yyin); return 1;
            }
            fclose(yyin);
            return 0;
        }
        else 
        {
            return 2;
        }
    }
    else 
    {
        return 3;
    }
}

void usage()
{
    cout << "Generates the depoyable web services for Axis C++ using \
        webservice's header file(s) and the static libraries" << endl;
    cout << "Usage: wcg " << endl;
    /* cout << "\t/c<Compiler to use>" << endl; */
    cout << "\t" << OPTIONTAG << "I<path to include directory>" << endl;
    cout << "\t" << OPTIONTAG << "L<path to web service static library \
        directory" << endl;
    cout << "\t" << OPTIONTAG << "l<web service static library file>" << endl;
    cout << "\t" << OPTIONTAG << "D<do not delete intermediate source files>" 
        << endl;
    cout << "\t" << OPTIONTAG << "S<web service class name>" << endl;
    cout << "\t" << OPTIONTAG << "o<output file (service name)>" << endl;
    cout << "\t<web service header file>" << endl;
    cout << endl;
#ifdef WIN32
    cout << "Make sure you have setup environment for Visual C++ compiler \
        and linker" << endl; 
#else
    cout << "Make sure you have setup environment for GNU compiler gcc" 
        << endl;     
#endif
    cout << "Axis Wrapper Class Generator (www.opensource.lk)" << endl;
    exit(0);
}

bool got_all_options()
{
    if (g_sServiceFile == "") return false;
    if (g_sWsFileName == "") return false;
    if (g_sWebServiceClass == "") return false;
    if (g_sLibraryPaths.empty()) return false;
    if (g_sAxisIncludePaths.empty()) return false;
    if (g_sCompiledLibs.empty()) return false;
    return true;
}

/* we have to check the validity of the parameters here such as whether
 * the include path exists, libraries exists etc. also should print the
 * appropriate error message here
*/
bool is_parameters_valid()
{
    cout << "did not check - assuming they are correct .. :)" << endl;
    return true;
}

bool is_webservice_header_compilable()
{
    /* create a dummy cpp file that includes the given header file */
    string filename = "Opensource_lk";
    File file(filename + ".cpp");
    file << "#include \"" << g_sWsFileName << "\"" << endl;
    file << "class Opensource { public: int axis;};" << endl;
    file.close();

    string command="";
#ifdef WIN32
    command = "cl /c /w ";
#else
    command = "gcc -c ";
#endif

    for (list<string>::iterator sit = g_sAxisIncludePaths.begin(); 
    sit != g_sAxisIncludePaths.end(); sit++)
    {
        command += "-I" + (*sit) + " ";
    }    

    command += filename + ".cpp";
    cout << "Command :" << command << endl;
    if (system(command.c_str()) == COMPLILER_FAILED)
    {
        command = DELETECOMMAND;
        command += filename + ".*";
        system(command.c_str());
        return false;
    }
    command = DELETECOMMAND;
    command += filename + ".*";
    system(command.c_str());
    return true;
}


/* cl *.cpp -I../../ -I../../../ -I../../../common /c /ML /W3 /GX /O2 */

/* link *.obj Ws.lib /dll /pdb:none /machine:I386 /out:
   "Service.dll" /libpath:"" */

/* gcc *.cpp -I../includepath -I/incpath2 -c -Wall -Wshadow -fPIC -O2 */

/* gcc -shared -Wl,-soname,libmyservice.so -o libmyservice.so *.o 
Ws.a -lstdc++ */