/*
 * Copyright (C) 2014, 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>
#ifdef MSWIN_API
#include <windows.h>
#endif
#include "razor.h"
#include "razor-internal.h"

#ifdef MSWIN_API
static int is_ascii_letter(char c)
{
	return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z');
}

static int is_slash(char c)
{
	return c == '/' || c == '\\';
}

static const char *mswin_path(const char *path)
{
	if (path == NULL)
		return NULL;

	if (path[0] == '/' && is_ascii_letter(path[1]) && path[2] == ':' &&
	    path[3] == '/')
		path++;

	return path;
}
#endif

static int path_cmp(const char *p1, const char *p2)
{
#ifdef MSWIN_API
	while(*p1 && *p2) {
		if (*p1 == *p2 || is_slash(*p1) && is_slash(*p2)) {
			p1++;
			p2++;
		} else
			break;
	}

	return *p1 || *p2;
#else
	return strcmp(p1, p2);
#endif
}

static int test_pfu(const char *uri, const char *path)
{
	char *s;
	int r;
	struct razor_error *error = NULL;

	s = razor_path_from_uri(uri, &error);

#ifdef MSWIN_API
	path = mswin_path(path);
#endif

	if (s && path)
		r = path_cmp(s, path);
	else
		r = (s != path);

	if (r) {
		fprintf(stderr, "Fail: razor_path_from_uri(\"%s\")", uri);
		if (s)
			fprintf(stderr, " returns \"%s\", expected", s);
		else
			fprintf(stderr, " fails (%s), expected",
				razor_error_get_msg(error));
		if (path)
			fprintf(stderr, " \"%s\"\n", path);
		else
			fprintf(stderr, " failure\n");
	}

	free(s);
	if (error)
		razor_error_free(error);

	return r;
}

static int test_abs(const char *abspath, const char *path)
{
	char *s;
	int r;

	s = razor_abspath(path);

	if (s && abspath)
		r = strcmp(s, abspath);
	else
		r = (s != abspath);

	if (r) {
		fprintf(stderr, "Fail: razor_abspath(\"%s\")", path);
		if (s)
			fprintf(stderr, " returns \"%s\", expected", s);
		else
			fprintf(stderr, " fails, expected");
		if (abspath)
			fprintf(stderr, " \"%s\"\n", abspath);
		else
			fprintf(stderr, " failure\n");
	}

	free(s);

	return r;
}

static int test_ptu(const char *uri, const char *path)
{
	char *s;
	int r;

	s = razor_path_to_uri(path);

	if (s && uri)
		r = strcmp(s, uri);
	else
		r = (s != uri);

	if (r) {
		fprintf(stderr, "Fail: razor_path_to_uri(\"%s\")", path);
		if (s)
			fprintf(stderr, " returns \"%s\", expected", s);
		else
			fprintf(stderr, " fails, expected");
		if (uri)
			fprintf(stderr, " \"%s\"\n", uri);
		else
			fprintf(stderr, " failure\n");
	}

	free(s);

	return r;
}

#ifdef MSWIN_API
UINT saved_cp;

static void cleanup_on_exit(void)
{
    SetConsoleOutputCP(saved_cp);
}
#endif

int main(int argc, char *argv[])
{
	int r = 0;
	const char *tmpdir;
	char *tempdir, *s;

	tmpdir = getenv("TMPDIR");
	if (!tmpdir || !*tmpdir)
		tmpdir = "/tmp";

	tempdir = razor_concat(tmpdir, "/test-pfu-XXXXXX", NULL);

	if (!mkdtemp(tempdir) || chdir(tempdir) < 0) {
		perror(tempdir);
		free(tempdir);
		exit(1);
	}

#ifdef MSWIN_API
	atexit(cleanup_on_exit);
	saved_cp = GetConsoleOutputCP();
	SetConsoleOutputCP(CP_UTF8);
#endif

	r |= test_pfu("file://localhost/etc/fstab", "/etc/fstab");
	r |= test_pfu("file:///etc/fstab", "/etc/fstab");
	r |= test_pfu("file://localhost/c:/WINDOWS/clock.avi",
		      "/c:/WINDOWS/clock.avi");
	r |= test_pfu("file:///c:/WINDOWS/clock.avi",
		      "/c:/WINDOWS/clock.avi");
	r |= test_pfu("file:///path/to/the%20file.txt",
		      "/path/to/the file.txt");
	r |= test_pfu("file:///home/s%C3%A9bastien", "/home/sébastien");
	r |= test_pfu("file:///home/luk%C3%A1%C5%A1", "/home/lukáš");
	r |= test_pfu("file:///var/log/22%20%e0%b8%aa%e0%b8%b4%e0%b8%87%e0%b8%ab%e0%b8%b2%e0%b8%84%e0%b8%a1%202014",
		      "/var/log/22 สิงหาคม 2014");

	s = razor_concat(tempdir, "/file.txt", NULL);
	r |= test_abs(s, "file.txt");
	free(s);

	s = razor_concat(tempdir, "/dir/../file.txt", NULL);
	r |= test_abs(s, "dir/../file.txt");
	free(s);

	s = razor_concat(tempdir, "/../file.txt", NULL);
	r |= test_abs(s, "../file.txt");
	free(s);

	s = razor_concat(tempdir, "/dir/../../file.txt", NULL);
	r |= test_abs(s, "dir/../../file.txt");
	free(s);

	r |= test_ptu("file:file.txt", "file.txt");
	r |= test_ptu("file:file.txt", "dir/../file.txt");

	s = razor_concat("file:", tmpdir, "/file.txt", NULL);
	r |= test_ptu(s, "../file.txt");
	r |= test_ptu(s, "dir/../../file.txt");
	free(s);

	free(tempdir);

	exit(r ? 1 : 0);
}