/*

    SDL_main.c, placed in the public domain by Sam Lantinga  4/13/98

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

*/

#include <stdio.h>

#include <stdlib.h>

#define WIN32_LEAN_AND_MEAN

#include <windows.h>

#ifdef _WIN32_WCE

# define DIR_SEPERATOR TEXT("\\")

# undef _getcwd

# define _getcwd(str,len)	wcscpy(str,TEXT(""))

# define setbuf(f,b)

# define setvbuf(w,x,y,z)

# define fopen		_wfopen

# define freopen	_wfreopen

# define remove(x)	DeleteFile(x)

#else

# define DIR_SEPERATOR TEXT("/")

# include <direct.h>

#endif

/* Include the SDL main definition header */

#include "SDL.h"

#include "SDL_main.h"

#ifdef main

# ifndef _WIN32_WCE_EMULATION

#  undef main

# endif /* _WIN32_WCE_EMULATION */

#endif /* main */

/* The standard output files */

#define STDOUT_FILE	TEXT("stdout.txt")

#define STDERR_FILE	TEXT("stderr.txt")

#ifndef NO_STDIO_REDIRECT

# ifdef _WIN32_WCE

  static wchar_t stdoutPath[MAX_PATH];

  static wchar_t stderrPath[MAX_PATH];

# else

  static char stdoutPath[MAX_PATH];

  static char stderrPath[MAX_PATH];

# endif

#endif

#if defined(_WIN32_WCE) && _WIN32_WCE < 300

/* seems to be undefined in Win CE although in online help */

#define isspace(a) (((CHAR)a == ' ') || ((CHAR)a == '\t'))

#endif /* _WIN32_WCE < 300 */

static void UnEscapeQuotes( char *arg )

{

	char *last = NULL;

	while( *arg ) {

		if( *arg == '"' && *last == '\\' ) {

			char *c_curr = arg;

			char *c_last = last;

			while( *c_curr ) {

				*c_last = *c_curr;

				c_last = c_curr;

				c_curr++;

			}

			*c_last = '\0';

		}

		last = arg;

		arg++;

	}

}

/* Parse a command line buffer into arguments */

static int ParseCommandLine(char *cmdline, char **argv)

{

	char *bufp;

	char *lastp = NULL;

	int argc, last_argc;

	argc = last_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 != '"' || *lastp == '\\' ) ) {

				lastp = bufp;

				++bufp;

			}

		} else {

			if ( *bufp ) {

				if ( argv ) {

					argv[argc] = bufp;

				}

				++argc;

			}

			/* Skip over word */

			while ( *bufp && ! isspace(*bufp) ) {

				++bufp;

			}

		}

		if ( *bufp ) {

			if ( argv ) {

				*bufp = '\0';

			}

			++bufp;

		}

		/* Strip out \ from \" sequences */

		if( argv && last_argc != argc ) {

			UnEscapeQuotes( argv[last_argc] );	

		}

		last_argc = argc;	

	}

	if ( argv ) {

		argv[argc] = NULL;

	}

	return(argc);

}

/* Show an error message */

static void ShowError(const char *title, const char *message)

{

/* If USE_MESSAGEBOX is defined, you need to link with user32.lib */

#ifdef USE_MESSAGEBOX

	MessageBox(NULL, message, title, MB_ICONEXCLAMATION|MB_OK);

#else

	fprintf(stderr, "%s: %s\n", title, message);

#endif

}

/* Pop up an out of memory message, returns to Windows */

static BOOL OutOfMemory(void)

{

	ShowError("Fatal Error", "Out of memory - aborting");

	return FALSE;

}

/* SDL_Quit() shouldn't be used with atexit() directly because

   calling conventions may differ... */

static void cleanup(void)

{

	SDL_Quit();

}

/* Remove the output files if there was no output written */

static void cleanup_output(void)

{

#ifndef NO_STDIO_REDIRECT

	FILE *file;

	int empty;

#endif

	/* Flush the output in case anything is queued */

	fclose(stdout);

	fclose(stderr);

#ifndef NO_STDIO_REDIRECT

	/* See if the files have any output in them */

	if ( stdoutPath[0] ) {

		file = fopen(stdoutPath, TEXT("rb"));

		if ( file ) {

			empty = (fgetc(file) == EOF) ? 1 : 0;

			fclose(file);

			if ( empty ) {

				remove(stdoutPath);

			}

		}

	}

	if ( stderrPath[0] ) {

		file = fopen(stderrPath, TEXT("rb"));

		if ( file ) {

			empty = (fgetc(file) == EOF) ? 1 : 0;

			fclose(file);

			if ( empty ) {

				remove(stderrPath);

			}

		}

	}

#endif

}

#if defined(_MSC_VER) && !defined(_WIN32_WCE)

/* The VC++ compiler needs main defined */

#define console_main main

#endif

/* This is where execution begins [console apps] */

int console_main(int argc, char *argv[])

{

	size_t n;

	char *bufp, *appname;

	int status;

	/* Get the class name from argv[0] */

	appname = argv[0];

	if ( (bufp=SDL_strrchr(argv[0], '\\')) != NULL ) {

		appname = bufp+1;

	} else

	if ( (bufp=SDL_strrchr(argv[0], '/')) != NULL ) {

		appname = bufp+1;

	}

	if ( (bufp=SDL_strrchr(appname, '.')) == NULL )

		n = SDL_strlen(appname);

	else

		n = (bufp-appname);

	bufp = SDL_stack_alloc(char, n+1);

	if ( bufp == NULL ) {

		return OutOfMemory();

	}

	SDL_strlcpy(bufp, appname, n+1);

	appname = bufp;

	/* Load SDL dynamic link library */

	if ( SDL_Init(SDL_INIT_NOPARACHUTE) < 0 ) {

		ShowError("WinMain() error", SDL_GetError());

		return(FALSE);

	}

	atexit(cleanup_output);

	atexit(cleanup);

	/* Sam:

	   We still need to pass in the application handle so that

	   DirectInput will initialize properly when SDL_RegisterApp()

	   is called later in the video initialization.

	 */

	SDL_SetModuleHandle(GetModuleHandle(NULL));

	/* Run the application main() code */

	status = SDL_main(argc, argv);

	/* Exit cleanly, calling atexit() functions */

	exit(status);

	/* Hush little compiler, don't you cry... */

	return 0;

}

/* This is where execution begins [windowed apps] */

#ifdef _WIN32_WCE

int WINAPI WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPWSTR szCmdLine, int sw)

#else

int WINAPI WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPSTR szCmdLine, int sw)

#endif

{

	HINSTANCE handle;

	char **argv;

	int argc;

	char *cmdline;

#ifdef _WIN32_WCE

	wchar_t *bufp;

	int nLen;

#else

	char *bufp;

	size_t nLen;

#endif

#ifndef NO_STDIO_REDIRECT

	DWORD pathlen;

#ifdef _WIN32_WCE

	wchar_t path[MAX_PATH];

#else

	char path[MAX_PATH];

#endif

	FILE *newfp;

#endif

	/* Start up DDHELP.EXE before opening any files, so DDHELP doesn't

	   keep them open.  This is a hack.. hopefully it will be fixed 

	   someday.  DDHELP.EXE starts up the first time DDRAW.DLL is loaded.

	 */

	handle = LoadLibrary(TEXT("DDRAW.DLL"));

	if ( handle != NULL ) {

		FreeLibrary(handle);

	}

#ifndef NO_STDIO_REDIRECT

	pathlen = GetModuleFileName(NULL, path, SDL_arraysize(path));

	while ( pathlen > 0 && path[pathlen] != '\\' ) {

		--pathlen;

	}

	path[pathlen] = '\0';

#ifdef _WIN32_WCE

	wcsncpy( stdoutPath, path, SDL_arraysize(stdoutPath) );

	wcsncat( stdoutPath, DIR_SEPERATOR STDOUT_FILE, SDL_arraysize(stdoutPath) );

#else

	SDL_strlcpy( stdoutPath, path, SDL_arraysize(stdoutPath) );

	SDL_strlcat( stdoutPath, DIR_SEPERATOR STDOUT_FILE, SDL_arraysize(stdoutPath) );

#endif

	/* Redirect standard input and standard output */

	newfp = freopen(stdoutPath, TEXT("w"), stdout);

#ifndef _WIN32_WCE

	if ( newfp == NULL ) {	/* This happens on NT */

#if !defined(stdout)

		stdout = fopen(stdoutPath, TEXT("w"));

#else

		newfp = fopen(stdoutPath, TEXT("w"));

		if ( newfp ) {

			*stdout = *newfp;

		}

#endif

	}

#endif /* _WIN32_WCE */

#ifdef _WIN32_WCE

	wcsncpy( stderrPath, path, SDL_arraysize(stdoutPath) );

	wcsncat( stderrPath, DIR_SEPERATOR STDOUT_FILE, SDL_arraysize(stdoutPath) );

#else

	SDL_strlcpy( stderrPath, path, SDL_arraysize(stderrPath) );

	SDL_strlcat( stderrPath, DIR_SEPERATOR STDERR_FILE, SDL_arraysize(stderrPath) );

#endif

	newfp = freopen(stderrPath, TEXT("w"), stderr);

#ifndef _WIN32_WCE

	if ( newfp == NULL ) {	/* This happens on NT */

#if !defined(stderr)

		stderr = fopen(stderrPath, TEXT("w"));

#else

		newfp = fopen(stderrPath, TEXT("w"));

		if ( newfp ) {

			*stderr = *newfp;

		}

#endif

	}

#endif /* _WIN32_WCE */

	setvbuf(stdout, NULL, _IOLBF, BUFSIZ);	/* Line buffered */

	setbuf(stderr, NULL);			/* No buffering */

#endif /* !NO_STDIO_REDIRECT */

#ifdef _WIN32_WCE

	nLen = wcslen(szCmdLine)+128+1;

	bufp = SDL_stack_alloc(wchar_t, nLen*2);

	wcscpy (bufp, TEXT("\""));

	GetModuleFileName(NULL, bufp+1, 128-3);

	wcscpy (bufp+wcslen(bufp), TEXT("\" "));

	wcsncpy(bufp+wcslen(bufp), szCmdLine,nLen-wcslen(bufp));

	nLen = wcslen(bufp)+1;

	cmdline = SDL_stack_alloc(char, nLen);

	if ( cmdline == NULL ) {

		return OutOfMemory();

	}

	WideCharToMultiByte(CP_ACP, 0, bufp, -1, cmdline, nLen, NULL, NULL);

#else

	/* Grab the command line */

	bufp = GetCommandLine();

	nLen = SDL_strlen(bufp)+1;

	cmdline = SDL_stack_alloc(char, nLen);

	if ( cmdline == NULL ) {

		return OutOfMemory();

	}

	SDL_strlcpy(cmdline, bufp, nLen);

#endif

	/* Parse it into argv and argc */

	argc = ParseCommandLine(cmdline, NULL);

	argv = SDL_stack_alloc(char*, argc+1);

	if ( argv == NULL ) {

		return OutOfMemory();

	}

	ParseCommandLine(cmdline, argv);

	/* Run the main program (after a little SDL initialization) */

	console_main(argc, argv);

	/* Hush little compiler, don't you cry... */

	return 0;

}