#include <stdlib.h>

#include <stdio.h>

#include <string.h>

#include <glib.h>

#include <bl/bl.h>

#include "testcaseparser.h"

#include "testcaseinput.h"

#include "testcaseoutput.h"

#include "testcaseio.h"

#include "warningsparser.h"

/*

 * Read a testcase in from a file.

 * On error, print a suitable message using g_printerr and return NULL.

 * The returned testcase should be freed with testcase_free().

 */

Testcase *testcase_parse_file(const char *filename)

{

    Testcase *testcase;

    TestcaseParser *parser;

    TestcaseInput *input=NULL;

    GMarkupParseContext *context;

    GError *err=NULL;

    char *s,*arg;

    const char *tag,*text;

    gboolean found_tag=FALSE,newlines_set=FALSE;

    parser=testcase_parser_new_from_file(filename);

    if (!parser)

	return NULL;

    if (!*testcase_parser_get_flag(parser))

    {

	g_printerr("%s: Not a valid testcase (flag)\n",filename);

	testcase_parser_free(parser);

	return NULL;

    }

    testcase=g_new0(Testcase,1);

    testcase->basename=g_path_get_basename(filename);

    s=strrchr(testcase->basename,'.');

    if (s)

	*s='\0';

    while(testcase_parser_get_next_tag(parser,&tag,&text))

    {

	if (!input && !strcmp(tag,"INPUT"))

	    input=testcase_input_new("TEST-XXXXXX",text);

	else if (g_str_has_prefix(tag,"INPUT(") && tag[strlen(tag)-1]==')')

	{

	    arg=g_strndup(tag+6,strlen(tag)-7);

	    s=g_path_get_dirname(arg);

	    if (strcmp(s,"."))

	    {

		/*

		 * Otherwise it would be possible to overwrite an arbitary

		 * file on somebody's computer by getting them to run a

		 * testcase!

		 */

		g_printerr(

		  "%s: Input files may not have a directory component\n",arg);

		g_free(s);

		g_free(arg);

		testcase_free(testcase);

		testcase_parser_free(parser);

		return NULL;

	    }

	    g_free(s);

	    testcase->inputs=g_slist_prepend(testcase->inputs,

	      testcase_input_new(arg,text));

	    if (!strstr(arg,"XXXXXX"))

		testcase->flags|=TESTCASE_TMP_DIR;

	    g_free(arg);

	}

	else if (!testcase->expected && !testcase->warnings &&

	  !strcmp(tag,"EXPECTED"))

	    testcase->expected=g_strdup(text);

	else if (g_str_has_prefix(tag,"EXPECTED(") && tag[strlen(tag)-1]==')')

	{

	    arg=g_strndup(tag+9,strlen(tag)-10);

	    s=g_path_get_dirname(arg);

	    if (strcmp(s,"."))

	    {

		g_printerr("%s: Expected files may not have a "

		  "directory component\n",arg);

		g_free(s);

		g_free(arg);

		testcase_free(testcase);

		testcase_parser_free(parser);

		return NULL;

	    }

	    g_free(s);

	    testcase->outputs=g_slist_prepend(testcase->outputs,

	      testcase_output_new(arg,text));

	    g_free(arg);

	}

	else if (!testcase->expected && !testcase->warnings &&

	  !strcmp(tag,"WARNINGS"))

	{

	    context=warnings_parse_context_new(testcase);

	    if (!g_markup_parse_context_parse(context,text,-1,&err) ||

	      !g_markup_parse_context_end_parse(context,&err))

	    {

		g_markup_parse_context_free(context);

		g_printerr("%s\n",err->message);

		g_clear_error(&err);

		testcase_free(testcase);

		testcase_parser_free(parser);

		return NULL;

	    }

	    g_markup_parse_context_free(context);

	}

	else if (!testcase->encoding && !strcmp(tag,"ENCODING"))

	    testcase->encoding=g_strchomp(g_strdup(text));

	else if (!newlines_set && !strcmp(tag,"NEWLINES"))

	{

	    newlines_set=TRUE;

	    s=g_strdup(text);

	    g_strchomp(s);

	    if (!strcmp(s,"LF"))

		testcase->flags|=TESTCASE_UNIX_NEWLINES;

	    else if (!strcmp(s,"CR"))

		testcase->flags|=TESTCASE_OS9_NEWLINES;

	    else if (strcmp(s,"CRLF"))

	    {

		g_printerr(

		  "%s: Unrecognised style for newlines. Try CR or LF.\n",s);

		g_free(s);

		testcase_free(testcase);

		testcase_parser_free(parser);

		return NULL;

	    }

	    g_free(s);

	}

	else if (!testcase->options && !strcmp(tag,"OPTIONS"))

	{

	    testcase->options=g_strsplit(text,"\n",0);

	    if (testcase->options && g_strv_length(testcase->options)>0)

	    {

		g_free(testcase->options[g_strv_length(testcase->options)-1]);

		testcase->options[g_strv_length(testcase->options)-1]=NULL;

	    }

	}

	else

	{

	    g_printerr("%s: Not a valid testcase (%s)\n",filename,tag);

	    testcase_free(testcase);

	    testcase_parser_free(parser);

	    return NULL;

	}

	found_tag=TRUE;

    }

    if (!testcase_parser_at_eof(parser))

    {

	if (found_tag)

	    g_printerr("%s: Not a valid testcase (garbage at end)\n",

	      filename);

	else

	    g_printerr("%s: Not a valid testcase (no valid tags)\n",

	      filename);

	testcase_free(testcase);

	testcase_parser_free(parser);

	return NULL;

    }

    if (!input)

	input=testcase_input_new("TEST-XXXXXX",NULL);

    testcase->inputs=g_slist_prepend(testcase->inputs,input);

    testcase_parser_free(parser);

    return testcase;

}