#include <stdlib.h>

#include <string.h>

#include <errno.h>

#include <glib.h>

#include <bl/bl.h>

#include "testcase.h"

#include "testcaseoutput.h"

/*

 * Replace \r\n with \n, \n with U+240A (visible symbol for LF)

 * and \r with U+240D (visible symbol for CR).

 */

static char *dos2unix(const char *text)

{

    gunichar c;

    gboolean cr=FALSE;

    const gunichar visible_lf=0x240A;

    const gunichar visible_cr=0x240D;

    GString *string;

    string=g_string_new(NULL);

    while(*text)

    {

	c=g_utf8_get_char(text);

	text=g_utf8_next_char(text);

	if (cr)

	{

	    cr=FALSE;

	    if (c=='\n')

	    {

		g_string_append_c(string,'\n');

		continue;

	    }

	    else

		g_string_append_unichar(string,visible_cr);

	}

	if (c=='\r')

	    cr=TRUE;

	else if (c=='\n')

	    g_string_append_unichar(string,visible_lf);

	else

	    g_string_append_unichar(string,c);

    }

    if (cr)

	g_string_append_unichar(string,visible_cr);

    return g_string_free(string,FALSE);

}

/*

 * Read an output file needed for a testcase (as specified in <output>).

 * The file is read in the encoding specified for communicating with

 * bookloupe.

 */

gboolean testcase_output_read(Testcase *testcase,TestcaseOutput *output,

  gchar **contents,gsize *length,GError **error)

{

    char *filename,*s,*t;

    gboolean retval;

    GError *tmp_err=NULL;

    if (!strcmp(output->name,"stdout"))

    {

	*contents=g_strdup(testcase->test_output);

	if (length)

	    *length=strlen(testcase->test_output);

    }

    else

    {

	if (testcase->tmpdir)

	    filename=g_build_filename(testcase->tmpdir,output->name,NULL);

	else

	    filename=g_strdup(output->name);

	if (!g_file_get_contents(filename,&s,NULL,error))

	{

	    g_free(filename);

	    return FALSE;

	}

	g_free(filename);

	if (testcase->encoding)

	{

	    t=dos2unix(s);

	    g_free(s);

	    s=g_convert(t,-1,"UTF-8",testcase->encoding,NULL,length,&tmp_err);

	    g_free(t);

	    if (!s)

	    {

		g_propagate_prefixed_error(error,tmp_err,

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

		return FALSE;

	    }

	    *contents=s;

	}

	else

	{

	    *contents=dos2unix(s);

	    if (length)

		*length=strlen(*contents);

	}

    }

    return TRUE;

}

/*

 * Remove an output file created by program under test.

 */

gboolean testcase_output_remove(Testcase *testcase,TestcaseOutput *output,

  GError **error)

{

    char *filename;

    if (!strcmp(output->name,"stdout"))

	return TRUE;

    if (testcase->tmpdir)

	filename=g_build_filename(testcase->tmpdir,output->name,NULL);

    else

	filename=g_strdup(output->name);

    if (g_unlink(filename)<0)

    {

	g_set_error(error,G_FILE_ERROR,g_file_error_from_errno(errno),

	  "%s: %s",filename,g_strerror(errno));

	return FALSE;

    }

    g_free(filename);

    return TRUE;

}

/* Create a new description of an output file expected by a testcase */

TestcaseOutput *testcase_output_new(const char *name,const char *contents)

{

    TestcaseOutput *output;

    output=g_new0(TestcaseOutput,1);

    output->name=g_strdup(name);

    output->contents=g_strdup(contents);

    return output;

}

/* Free the description of a testcase output file */

void testcase_output_free(TestcaseOutput *output)

{

    g_free(output->name);

    g_free(output->contents);

    g_free(output);

}