/*
 *  
 * This is only a simple script-launcher utility that calls Python interpreter
 * on win32 systems, so that running the tools is simpler.
 * 
 * $Id: win32scriptwrapper.c,v 1.2 2005/02/06 01:50:20 kollivier Exp $
 *  
 */

#include <string.h>
#include <stdio.h>
#include <malloc.h>
#include <stdlib.h>
#include <Python.h>
#include <windows.h>

/* ----------------------------------------------------------------------- */
/*       main() - sets PYTHONHOME and calls PyRun_SimpleFile on            */
/*               script with same name but .py extension                   */
/* ----------------------------------------------------------------------- */

int main(int argc, char** argv)
{
    int ret;
    char filename[2048];
    char dirname[2048];
    char envvar[2048];
    FILE *fp;
    char *lastp = NULL;

    strncpy(filename, argv[0], 2048);
    if (GetModuleFileName(NULL, (char*)filename, 2048) != 0)
    {
        lastp = strrchr(filename, '\\');
        if (lastp == NULL)
        {
            strncpy(dirname, ".", 2048);
        }
        else
        {
            strncpy(dirname, filename, lastp - filename);
            dirname[lastp - filename] = 0;
        }
        
        lastp = strrchr(filename, '.');
    }
    if (lastp == NULL)
    {
        fprintf(stderr, "Incorrect executable name!\n");
        return 1;
    }

    strncpy(lastp, ".py", 2048-(2+lastp-filename));

    fp = fopen(filename, "rb");
    if (fp == NULL)
    {
        fprintf(stderr, "Cannot open script file '%s'!\n", filename);
        return 2;
    }

    /* set PYTHONHOME so that system-wide installed copy of Python is
       never used; we want to use embedded one: */
    SetEnvironmentVariable("PYTHONHOME", dirname);
    sprintf(envvar, "PYTHONHOME=%s", dirname);
    _putenv(envvar);
    
    Py_Initialize();
    argv[0] = (char*)filename;
    PySys_SetArgv(argc, argv);
    ret = PyRun_SimpleFile(fp, filename);
    Py_Finalize();
    fclose(fp);

    if (ret != 0)
        fprintf(stderr, "Error executing the script!\n");
    return ret;
}


/* ----------------------------------------------------------------------- */
/*               WinMain -> main bridge, argv translation                  */
/* ----------------------------------------------------------------------- */

/*
    adapted from SDL (www.libsdl.org) sources, original notice follows:
   
    SDL_main.c, placed in the public domain by Sam Lantinga  4/13/98

    The WinMain function -- calls your program's main() function
*/


/* Parse a command line buffer into arguments */
static int ParseCommandLine(char *cmdline, char **argv)
{
        char *bufp;
        int argc;
		int quoted_arg = 0;

        argc = 0;
        for ( bufp = cmdline; *bufp; ) {
                /* Skip leading whitespace */
                while ( isspace(*bufp) ) {
                        ++bufp;
                }
                /* Skip over argument */
                if ( *bufp == '"' ) {
                        ++bufp;
                        if ( *bufp ) {
                                if ( argv ) {
                                        argv[argc] = bufp;
                                }
                                ++argc;
                        }
                        /* Skip over word */
                        while ( *bufp && (*bufp != '"') ) {
                                ++bufp;
                        }
                } else {
                        if ( *bufp ) {
                                if ( argv ) {
                                        argv[argc] = bufp;
                                }
                                ++argc;
                        }
                        /* Skip over word */
						/* quoted_arg deals with cases where an arg may have
						   a portion of the argument in quotes, like 
						   arg="value" */

                        while ( *bufp && (! isspace(*bufp) || quoted_arg) ) {
							if ( *bufp == '"'){
								if (!quoted_arg)
									quoted_arg = 1;
								else
									quoted_arg = 0;
							}

                                ++bufp;
                        }
                }
                if ( *bufp ) {
                        if ( argv ) {
                                *bufp = '\0';
                        }
                        ++bufp;
                }
        }
        if ( argv ) {
                argv[argc] = NULL;
        }
        return(argc);
}


int WINAPI WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPSTR szCmdLine, int sw)
{
        HINSTANCE handle;
        char **argv;
        int argc;
        char *cmdline;
        char *bufp;

        /* Grab the command line (use alloca() on Windows) */
        bufp = GetCommandLine();
        cmdline = (char *)alloca(strlen(bufp)+1);
        strcpy(cmdline, bufp);

        /* Parse it into argv and argc */
        argc = ParseCommandLine(cmdline, NULL);
        argv = (char **)alloca((argc+1)*(sizeof *argv));
        ParseCommandLine(cmdline, argv);

        return (main(argc, argv));
}