/*

 * Copyright (C) 2009  J. Ali Harlow <ali@juiblex.co.uk>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License along

 * with this program; if not, write to the Free Software Foundation, Inc.,

 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 */

#include "config.h"

#include <stdlib.h>

#include <stdio.h>

#include <string.h>

#include <errno.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <dirent.h>

#include <unistd.h>

#include <math.h>

#include "razor-internal.h"

/*

 * This kludge is to work around an apparent bug in Fedora 10's lua package

 * (it appears to require libm but not to include a DT_NEEDED).

 */

void (*kludge)() = (void (*)())sin;

static void recursive_remove(const char *directory)

{

	DIR *dp;

	struct dirent *dirp;

	char *buf;

	dp = opendir(directory);

	while((dirp = readdir(dp))) {

		if (strcmp(dirp->d_name, ".") && strcmp(dirp->d_name, "..")) {

			buf = malloc(strlen(directory) + strlen(dirp->d_name)

				     + 2);

			sprintf(buf, "%s/%s", directory, dirp->d_name);

			if (remove(buf) < 0)

				recursive_remove(buf);

			free(buf);

		}

	}

	rmdir(directory);

}

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

{

	char root[] = "/tmp/razor.XXXXXX";

	int r;

	void *script;

	size_t len;

	char *s, *test_file, *srcdir;

	FILE *fp;

	if (argc > 2) {

		fprintf(stderr, "usage: %s [TESTS-FILE]\n", argv[0]);

		exit(1);

	}

	if (argc == 2)

		test_file = argv[1];

	else

		test_file = "test.lua";

	if (!mkdtemp(root)) {

		perror(root);

		exit(1);

	}

	s = malloc(strlen(root) + strlen("/testfile") + 1);

	strcpy(s, root);

	strcat(s, "/testfile");

	fp = fopen(s, "w");

	if (!fp) {

		perror(s);

		exit(1);

	}

	fprintf(fp, "#!" LUA_BINARY "\n"

	  "print('Abracadabra!')\n");

	fchmod(fileno(fp), S_IRUSR | S_IWUSR | S_IXUSR);

	fclose(fp);

	free(s);

	script = razor_file_get_contents(test_file, &len);

	if (!script) {

		srcdir = getenv("srcdir");

		if (srcdir && errno == ENOENT && *test_file != '/') {

			s = malloc(strlen(srcdir) + strlen(test_file) + 2);

			strcpy(s, srcdir);

			strcat(s, "/");

			strcat(s, test_file);

			script = razor_file_get_contents(s, &len);

			if (!script) {

				perror(s);

				exit(1);

			}

			free(s);

		} else {

			perror(test_file);

			exit(1);

		}

	}

	r = run_lua_script(root, test_file, script, len, -1);

	razor_file_free_contents(script, len);

	recursive_remove(root);

	exit(r ? 1 : 0);

}