#include <stdlib.h>

 #include <stdio.h>

 #include <string.h>

 #include <errno.h>

 #include <glib.h>

 #include <glib/gstdio.h>

 #include <bl/bl.h>

 #include "testcase.h"

 #include "testcaseinput.h"

 GQuark testcase_error_quark(void)

 {

     return g_quark_from_static_string("testcase-error-quark");

 }

 /*

  * Return the length (in bytes) of any common prefix between s1 and s2.

  * The returned length will always represent an exact number of characters.

  */

 size_t common_prefix_length(const char *s1,const char *s2)

 {

     gunichar c1,c2;

     const char *s=s1;

     while(*s1 && *s2)

     {

 	c1=g_utf8_get_char(s1);

 	c2=g_utf8_get_char(s2);

 	if (c1!=c2)

 	    break;

 	s1=g_utf8_next_char(s1);

 	s2=g_utf8_next_char(s2);

     }

     return s1-s;

 }

 void print_unexpected(const char *unexpected,gsize differs_at)

 {

     int col;

     gunichar c;

     const char *endp,*bol,*s;

     GString *string;

     endp=strchr(unexpected+differs_at,'\n');

     if (!endp)

 	endp=unexpected+strlen(unexpected);

     string=g_string_new_len(unexpected,endp-unexpected);

     bol=strrchr(string->str,'\n');

     if (bol)

 	bol++;

     else

 	bol=string->str;

     col=0;

     s=bol;

     endp=string->str+differs_at;

     while(s<endp)

     {

 	c=g_utf8_get_char(s);

 	s=g_utf8_next_char(s);

 	if (c=='\t')

 	    col=(col&~7)+8;

 	else if (g_unichar_iswide(c))

 	    col+=2;

 	else if (!g_unichar_iszerowidth(c))

 	    col++;

     }

     fprintf(stderr,"%s\n%*s^\n",string->str,col,"");

     g_string_free(string,TRUE);

 }

 /*

  * Create all the input files needed by a testcase and, if required,

  * a temporary directory in which to store them.

  */

 gboolean testcase_create_input_files(Testcase *testcase,GError **error)

 {

     GSList *link,*link2;

     if (testcase->flags&TESTCASE_TMP_DIR)

     {

 	testcase->tmpdir=g_strdup("TEST-XXXXXX");

 	if (!g_mkdtemp(testcase->tmpdir))

 	{

 	    g_set_error(error,G_FILE_ERROR,g_file_error_from_errno(errno),

 	      "Failed to create temporary directory: %s",g_strerror(errno));

 	    g_free(testcase->tmpdir);

 	    testcase->tmpdir=NULL;

 	    return FALSE;

 	}

     }

     for(link=testcase->inputs;link;link=link->next)

 	if (!testcase_input_create(testcase,link->data,error))

 	{

 	    for(link2=testcase->inputs;link2!=link;link2=link2->next)

 		(void)testcase_input_remove(testcase,link2->data,NULL);

 	    if (testcase->tmpdir)

 	    {

 		(void)g_rmdir(testcase->tmpdir);

 		g_free(testcase->tmpdir);

 		testcase->tmpdir=NULL;

 	    }

 	    return FALSE;

 	}

     return TRUE;

 }

 /*

  * Remove all the input files used by a testcase and, if created,

  * the temporary directory in which they are stored.

  */

 gboolean testcase_remove_input_files(Testcase *testcase,GError **error)

 {

     GSList *link;

     GError *tmp_err=NULL;

     gboolean retval=TRUE;

     for(link=testcase->inputs;link;link=link->next)

 	if (!testcase_input_remove(testcase,link->data,&tmp_err))

 	{

 	    if (error && !*error)

 		g_propagate_error(error,tmp_err);

 	    else

 		g_clear_error(&tmp_err);

 	    retval=FALSE;

 	}

     if (testcase->tmpdir)

     {

 	if (g_rmdir(testcase->tmpdir))

 	{

 	    if (error && !*error)

 		g_set_error(error,G_FILE_ERROR,g_file_error_from_errno(errno),

 		  "Failed to remove temporary directory: %s",g_strerror(errno));

 	    retval=FALSE;

 	}

 	g_free(testcase->tmpdir);

 	testcase->tmpdir=NULL;

     }

     return retval;

 }

 /*

  * Replace every occurance of an input file name in <str> with the

  * filename which holds that input. For input files with fixed names,

  * this is a noop. For input files which use the "XXXXXX" sequence

  * to create a unique filename, the XXXXXX will be replaced with the

  * 6 characters that were chosen to be unique.

  */

 char *testcase_resolve_input_files(Testcase *testcase,const char *str)

 {

     GSList *link;

     gsize offset,pos;

     char *s;

     TestcaseInput *input;

     GString *filename=g_string_new(str);

     for(link=testcase->inputs;link;link=link->next)

     {

 	input=link->data;

 	if (!input->name_used)

 	{

 	    g_warning("%s: Input file uninstantiated",input->name);

 	    continue;

 	}

 	offset=0;

 	do

 	{

 	    s=strstr(filename->str+offset,input->name);

 	    if (s)

 	    {

 		pos=s-filename->str;

 		g_string_overwrite(filename,pos,input->name_used);

 		offset=pos+strlen(input->name);

 	    }

 	} while(s);

     }

     return g_string_free(filename,FALSE);

 }

 gboolean testcase_spawn_bookloupe(Testcase *testcase,char **standard_output,

   GError **error)

 {

     gboolean r;

     int i,exit_status;

     char **argv;

     char *output,*s;

     GError *tmp_err=NULL;

     if (testcase->options)

 	argv=g_new(char *,g_strv_length(testcase->options)+3);

     else

 	argv=g_new(char *,3);

     s=getenv("BOOKLOUPE");

     if (!s)

 	s="bookloupe";

     argv[0]=path_to_absolute(s);

     for(i=0;testcase->options && testcase->options[i];i++)

 	argv[i+1]=testcase_resolve_input_files(testcase,testcase->options[i]);

     argv[i+1]=testcase_resolve_input_files(testcase,"TEST-XXXXXX");

     argv[i+2]=NULL;

     if (standard_output)

     {

 	r=spawn_sync(testcase->tmpdir,argv,&s,&exit_status,error);

 	if (r)

 	{

 	    if (testcase->encoding)

 	    {

 		output=g_convert(s,-1,"UTF-8",testcase->encoding,NULL,NULL,

 		  &tmp_err);

 		g_free(s);

 		if (!output)

 		{

 		    g_propagate_prefixed_error(error,tmp_err,

 		      "Conversion from %s failed: ",testcase->encoding);

 		    r=FALSE;

 		}

 	    }

 	    else

 	    {

 		output=s;

 		if (!g_utf8_validate(s,-1,NULL))

 		{

 		    g_set_error_literal(error,TESTCASE_ERROR,

 		      TESTCASE_ERROR_FAILED,

 		      "bookloupe output is not valid UTF-8");

 		    r=FALSE;

 		}

 	    }

 	}

     }

     else

     {

 	r=spawn_sync(testcase->tmpdir,argv,NULL,&exit_status,error);

 	output=NULL;

     }

     g_strfreev(argv);

     if (r && exit_status)

     {

 	g_set_error(error,TESTCASE_ERROR,TESTCASE_ERROR_FAILED,

 	  "bookloupe exited with code %d",exit_status);

 	r=FALSE;

     }

     if (r && standard_output)

 	*standard_output=output;

     return r;

 }

 /*

  * Run a testcase, returning FALSE on fail or error and

  * TRUE on pass or expected-fail.

  * Suitable message(s) will be printed in all cases.

  */

 gboolean testcase_run(Testcase *testcase)

 {

     gboolean r;

     size_t pos,offset;

     GString *header,*expected;

     char *output,*filename,*s;

     GError *error=NULL;

     if (!testcase_create_input_files(testcase,&error))

     {

 	fprintf(stderr,"%s: FAIL\n",testcase->basename);

 	fprintf(stderr,"%s\n",error->message);

 	g_error_free(error);

 	return FALSE;

     }

     if (testcase->expected)

 	r=testcase_spawn_bookloupe(testcase,&output,&error);

     else

     {

 	r=testcase_spawn_bookloupe(testcase,NULL,&error);

         output=NULL;

     }

     if (!r)

     {

 	fprintf(stderr,"%s: FAIL\n",testcase->basename);

 	fprintf(stderr,"%s\n",error->message);

 	g_error_free(error);

 	(void)testcase_remove_input_files(testcase,NULL);

 	return FALSE;

     }

     filename=testcase_resolve_input_files(testcase,"TEST-XXXXXX");

     if (!testcase_remove_input_files(testcase,&error))

     {

 	fprintf(stderr,"%s: FAIL\n",testcase->basename);

 	fprintf(stderr,"%s\n",error->message);

 	g_error_free(error);

 	return FALSE;

     }

     if (testcase->expected)

     {

 	header=g_string_new("\n\nFile: ");

 	g_string_append(header,filename);

 	g_string_append(header,"\n");

 	expected=g_string_new(testcase->expected);

 	if (!g_str_has_prefix(output,header->str))

 	{

 	    fprintf(stderr,"%s: FAIL\n",testcase->basename);

 	    offset=common_prefix_length(output,header->str);

 	    fprintf(stderr,"Unexpected header from bookloupe:\n");

 	    print_unexpected(output,offset);

 	    r=FALSE;

 	}

 	pos=header->len;

 	if (r)

 	{

 	    /* Skip the summary */

 	    s=strstr(output+pos,"\n\n");

 	    if (s)

 		pos=s-output+2;

 	    else

 	    {

 		fprintf(stderr,"%s: FAIL\n",testcase->basename);

 		offset=common_prefix_length(output,header->str);

 		fprintf(stderr,"Unterminated summary from bookloupe:\n%s\n",

 		  output+pos);

 		r=FALSE;

 	    }

 	}

 	if (r && strcmp(output+pos,expected->str))

 	{

 	    fprintf(stderr,"%s: FAIL\n",testcase->basename);

 	    offset=common_prefix_length(output+pos,expected->str);

 	    if (!offset && !output[pos+offset])

 		fprintf(stderr,"Unexpected zero warnings from bookloupe.\n");

 	    else

 	    {

 		fprintf(stderr,"Unexpected output from bookloupe:\n");

 		print_unexpected(output+pos,offset);

 	    }

 	    r=FALSE;

 	}

 	g_string_free(header,TRUE);

 	g_string_free(expected,TRUE);

     }

     g_free(filename);

     g_free(output);

     if (r)

 	fprintf(stderr,"%s: PASS\n",testcase->basename);

     return r;

 }

 /*

  * Free a testcase.

  */

 void testcase_free(Testcase *testcase)

 {

     g_free(testcase->basename);

     g_slist_foreach(testcase->inputs,(GFunc)testcase_input_free,NULL);

     g_slist_free(testcase->inputs);

     g_free(testcase->expected);

     g_free(testcase->encoding);

     g_strfreev(testcase->options);

     g_free(testcase);

 }