/*

* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of the License "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description: 

*

*/

#include <stdlib.h>

#include <unistd.h>

#include <string.h>

#include <stdio.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <fcntl.h>

#include <stdarg.h>

#include "talon_process.h"

#include "sema.h"

#include "buffer.h"

#include "../config.h"

/* The output semaphore. */

sbs_semaphore talon_sem;

#define TALON_ATTEMPT_STRMAX 32

#define RECIPETAG_STRMAX 2048

#define STATUS_STRMAX 120

#define TALONDELIMITER '|'

#define VARNAMEMAX 100

#define VARVALMAX 1024

#include "log.h"

#ifdef HAS_MSVCRT

/* Make all output handling binary */

unsigned int _CRT_fmode = _O_BINARY;

#endif

double getseconds(void)

{

	struct timeval tp;

	gettimeofday(&tp, NULL);

	return (double)tp.tv_sec + ((double)tp.tv_usec)/1000000.0L;

}

void talon_setenv(char name[], char val[])

{

#if defined(HAS_GETENVIRONMENTVARIABLE)

	SetEnvironmentVariableA(name,val); 

#elif defined(HAS_GETENV)

	setenv(name,val, 1);

#else

#	error "Need a function for setting environment variables"

#endif

}

#define TALON_MAXENV 4096

char * talon_getenv(char name[])

{

#if defined(HAS_SETENV)

	char *val = getenv(name);

	char *dest = NULL;

	if (val)

	{

		dest = malloc(strlen(val) + 1);

		if (dest)

		{

			strcpy(dest,val);

		}

	}

	return dest;

#elif defined(HAS_SETENVIRONMENTVARIABLE)

	char *val = malloc(TALON_MAXENV);

	if (0 != GetEnvironmentVariableA(name,val,TALON_MAXENV-1))

		return val;

	else

		return NULL;

#else

#	error "Need a function for setting environment variables"

#endif

}

void prependattributes(buffer *b, char *attributes)

{

	char recipetag[RECIPETAG_STRMAX];

	char *rt;

	char envvarname[VARNAMEMAX];

	char *att;

        strcpy(recipetag, "<recipe ");

	rt = recipetag + 8;

	att = attributes;

	while (*att != '\0' && rt < &recipetag[RECIPETAG_STRMAX-1])

	{

		if ( *att == '$' )

		{

			int e;

			char *v;

			/* insert the value of an environent variable */

			att++;

			e = 0;	

			do {

				envvarname[e++] = *att;

				att++;

			} while ( e < (VARNAMEMAX-1) && (isalnum(*att) || *att == '_'));

			envvarname[e] = '\0';

/* DEBUG(("envvarname: %s\n", envvarname)); */

			v = talon_getenv(envvarname);

			if (v)

			{

				/* DEBUG(("     value: %s\n", v)); */

				char *oldv=v;

				while (*v != '\0' && rt < &recipetag[RECIPETAG_STRMAX-1])

				{

					*rt = *v;

					rt++; v++;

				}

				free(oldv);

			}

		} else {

			*rt = *att;

			rt++; att++;

		}

	}

	char *finish = ">\n<![CDATA[\n";

	while (*finish != '\0' && rt < &recipetag[RECIPETAG_STRMAX-1])

	{

		*rt = *finish;

		*rt++; finish++;

	}

	*rt = '\0';

        buffer_prepend(b, recipetag, strlen(recipetag));

}

/* read a recipe string from a temporary file.

 *

 * We only expect to do this for very long command lines on

 * Windows. So allocate the maximum size for CreateProcess on

 * Win32 (32768 bytes) and error if the file is longer than that. 

 *

 */

char *read_recipe_from_file(const char *filename)

{

	FILE *fp = fopen(filename, "r");

	if (!fp)

	{

		error("talon: error: could not read '%s'\n", filename);

		return NULL;

	}

	int max_length = 32768;

	char *recipe = (char*)malloc(max_length);

	if (!recipe)

	{

		error("talon: error: could not allocate memory to read '%s'\n", filename);

		return NULL;

	}

	int position = 0;

	/* read the file one character at a time */

	int c;

	while ((c = getc(fp)) != EOF)

	{

		switch (c)

		{

			case '\r':

			case '\n':

				/* ignore newlines */

				break;

			case '"':

				/* double quotes have to be escaped so they aren't lost later */

				if (position < max_length)

					recipe[position++] = '\\';

				if (position < max_length)

					recipe[position++] = c;

				break;

			default:

				if (position < max_length)

					recipe[position++] = c;

				break;

		}

	}

	fclose(fp);

	/* add a terminating \0 */

	if (position < max_length)

		recipe[position] = '\0';

	else

	{

		error("talon: error: command longer than 32768 in '%s'\n", filename);

		return NULL;

	}

	return recipe;

}

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

{

	/* find the argument to -c then strip the talon related front section */

	char *recipe = NULL;

	int talon_returncode = 0;

#ifdef HAS_GETCOMMANDLINE

	char *commandline= GetCommandLine();

	/*

	 * The command line should be either,

	 * talon -c "some shell commands"

	 * or

	 * talon shell_script_file

	 *

	 * talon could be an absolute path and may have a .exe extension.

	 */

	if (recipe)

	{

		/* there was a -c so extract the quoted commands */

		int recipelen = strlen(recipe);

		if (recipelen > 0 && recipe[recipelen - 1] == '"')

			recipe[recipelen - 1] = '\0'; /* remove trailing quote */

	}

	else

	{

		/* there was no -c so extract the argument as a filename */

		recipe = strstr(commandline, "talon");

		if (recipe)

		{

			/* find the first space */

			while (!isspace(*recipe) && *recipe != '\0')

				recipe++;

			/* skip past the spaces */

			while (isspace(*recipe))

				recipe++;

			recipe = read_recipe_from_file(recipe);

			if (!recipe)

			{

			    error("talon: error: bad script file in shell call '%s'\n", commandline);

				return 1;

			}

		}

		else

		{

			error("talon: error: no 'talon' in shell call '%s'\n", commandline);

			return 1;

		}

	}

#else

	/*

	 * The command line should be either,

	 * talon -c "some shell commands"

	 * or

	 * talon shell_script_file

	 *

	 * talon could be an absolute path and may have a .exe extension.

	 */

	switch (argc)

	{

		case 2:

			recipe = read_recipe_from_file(argv[1]);

			break;

		case 3:

			if (strcmp("-c", argv[1]) != 0)

			{

				error("talon: error: %s\n", "usage is 'talon -c command' or 'talon script_filename'");

				return 1;

			}

			recipe = argv[2];

			break;

		default:

			error("talon: error: %s\n", "usage is 'talon -c command' or 'talon script_filename'");

			return 1;

	}

#endif

	/* did we get a recipe at all? */

	if (!recipe)

	{

		error("talon: error: %s", "no recipe supplied to the shell.\n");

		return 1;

	}

	/* remove any leading white space on the recipe */

	while (isspace(*recipe))

		recipe++;

	/* turn debugging on? */

	char *debugstr=talon_getenv("TALON_DEBUG");

	if (debugstr)

	{

		loglevel=LOGDEBUG;

		free(debugstr); debugstr=NULL;

	}

	DEBUG(("talon: recipe: %s\n", recipe));