/*
 * Copyright (C) 2009, 2016  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 <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_uri, *srcdir, *suffix, *cwd;
	FILE *fp;
	struct razor_error *error = NULL;
	struct razor_uri base_ru, test_ru, ru;

	if (argc > 2) {
		fprintf(stderr, "usage: %s [TESTS-URI]\n", argv[0]);
		exit(1);
	}
	if (argc == 2)
		test_uri = argv[1];
	else
		test_uri = "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");
	fclose(fp);
	chmod(s, S_IRUSR | S_IWUSR | S_IXUSR);
	free(s);

	for (len = 32;; len *= 2) {
	    cwd = malloc(len);
	    if (getcwd(cwd, len))
		    break;
	    free(cwd);
	}
	cwd = realloc(cwd, strlen(cwd) + 1);

	s = razor_concat("file:", cwd, NULL);
	if (razor_uri_parse(&test_ru, test_uri, &error) ||
	    razor_uri_parse(&base_ru, s, &error)) {
		fprintf(stderr, "%s\n", razor_error_get_msg(error));
		razor_error_free(error);
		exit(1);
	}
	free(s);

	razor_uri_resolve(&ru, &base_ru, &test_ru);
	razor_uri_destroy(&base_ru);
	s = razor_uri_recompose(&ru);
	razor_uri_destroy(&ru);
	script = razor_uri_get_contents(s, &len, 0, &error);
	free(s);
	if (!script) {
		srcdir = getenv("srcdir");
		if (srcdir && errno == ENOENT) {
			razor_error_free(error);
			error = NULL;
			suffix = srcdir[strlen(srcdir) - 1] == '/' ? NULL : "/";
			if (*srcdir == '/')
				s = razor_concat("file:", srcdir, suffix, NULL);
			else
				s = razor_concat("file:", cwd, "/", srcdir,
						 suffix, NULL);
			razor_uri_parse(&base_ru, s, &error);
			free(s);
			if (!error) {
				razor_uri_resolve(&ru, &base_ru, &test_ru);
				razor_uri_destroy(&base_ru);
				s = razor_uri_recompose(&ru);
				razor_uri_destroy(&ru);
				script = razor_uri_get_contents(s, &len, 0,
								&error);
				free(s);
			}
			if (error) {
				fprintf(stderr, "%s\n",
					razor_error_get_msg(error));
				razor_error_free(error);
				exit(1);
			}
		} else {
			fprintf(stderr, "%s\n", razor_error_get_msg(error));
			razor_error_free(error);
			exit(1);
		}
	}
	razor_uri_destroy(&test_ru);
	free(cwd);

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

	recursive_remove(root);
	exit(r ? 1 : 0);
}