/*

 * Copyright (C) 2008  Kristian Høgsberg <krh@redhat.com>

 * Copyright (C) 2008  Red Hat, Inc

 * 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 <stdint.h>

#include <stdio.h>

#include <string.h>

#include <sys/stat.h>

#include <dirent.h>

#include <unistd.h>

#include <fcntl.h>

#include <limits.h>

#include <assert.h>

#ifdef MSWIN_API

#include <shlobj.h>

#endif

#include "razor.h"

#include "razor-internal.h"

#ifndef O_BINARY

#define O_BINARY	0

#endif

static const char system_repo_filename[] = "system.rzdb";

/*

 * system_lock_filename is chosen to be the same as the pre v0.3

 * next_repo_filename. This means that once a system has been

 * updated by a v0.3+ copy of razor all pre v0.3 versions of razor

 * will see the system as permenantly locked.

 */

static const char system_lock_filename[] = "system-next.rzdb";

static const char system_tmp_filename[] = "system.tmp";

#ifdef MSWIN_API

#define RAZOR_ROOT_PATH	NULL

#else

#define RAZOR_ROOT_PATH	"/var/lib/razor"

#endif

static const char *razor_root_path = RAZOR_ROOT_PATH;

struct razor_root {

	struct razor_set *system;

	struct razor_set *next;

	int fd;

	char path[PATH_MAX];

	char new_path[PATH_MAX];

};

static void

razor_root_init(void)

{

#ifdef MSWIN_API

	static char root_path[MAX_PATH];

	if (!razor_root_path) {

		SHGetFolderPath(NULL,

			CSIDL_COMMON_APPDATA | CSIDL_FLAG_DONT_VERIFY, NULL, 0,

			root_path);

		strcat(root_path, "\\Razor");

		razor_root_path = root_path;

	}

#endif

}

RAZOR_EXPORT int

razor_root_create(const char *root)

{

	struct stat buf;

	struct razor_set *set;

	char path[PATH_MAX];

	assert (root != NULL);

	razor_root_init();

	if (root[0] == '\0') {

		/* root is file system root */

	} else if (stat(root, &buf) < 0) {

		if (mkdir(root, 0777) < 0) {

			fprintf(stderr,

				"could not create install root \"%s\"\n",

				root);

			return -1;

		}

		fprintf(stderr, "created install root \"%s\"\n", root);

	} else if (!S_ISDIR(buf.st_mode)) {

		fprintf(stderr,

			"install root \"%s\" exists, but is not a directory\n",

			root);

		return -1;

	}

	snprintf(path, sizeof path, "%s/%s",

		 razor_root_path, system_repo_filename);

	if (razor_create_dir(root, path) < 0) {

		fprintf(stderr, "could not create %s%s\n",

			root, razor_root_path);

		return -1;

	}

	set = razor_set_create();

	snprintf(path, sizeof path, "%s%s/%s",

		 root, razor_root_path, system_repo_filename);

	if (stat(path, &buf) == 0) {

		fprintf(stderr,

			"a razor install root is already initialized\n");

		return -1;

	}

	if (razor_set_write(set, path, RAZOR_SECTION_ALL) < 0) {

		fprintf(stderr, "could not write initial package set\n");

		return -1;

	}

	razor_set_destroy(set);

	return 0;

}

RAZOR_EXPORT struct razor_root *

razor_root_open(const char *root)

{

	struct razor_root *image;

	char lock_path[PATH_MAX];

	assert (root != NULL);

	razor_root_init();

	image = malloc(sizeof *image);

	if (image == NULL)

		return NULL;

	image->system = razor_set_create_without_root();

	if (image->system == NULL) {

		free(image);

		return NULL;

	}

	snprintf(lock_path, sizeof lock_path,

		 "%s%s/%s", root, razor_root_path, system_lock_filename);

	if (razor_set_aquire_lock(image->system, lock_path, 1) < 0) {

		razor_set_destroy(image->system);

		free(image);

		return NULL;

	}

	snprintf(image->new_path, sizeof image->new_path,

		 "%s%s/%s", root, razor_root_path, system_tmp_filename);

	image->fd = open(image->new_path,

			 O_CREAT | O_WRONLY | O_TRUNC | O_BINARY,

			 0666);

	if (image->fd < 0) {

		razor_set_destroy(image->system);

		free(image);

		return NULL;

	}

	snprintf(image->path, sizeof image->path,

		 "%s%s/%s", root, razor_root_path, system_repo_filename);

	if (razor_set_bind_sections(image->system, image->path)) {

		close(image->fd);

		unlink(image->new_path);

		razor_set_destroy(image->system);

		free(image);

		return NULL;

	}

	return image;

}

RAZOR_EXPORT struct razor_set *

razor_root_open_read_only(const char *root)

{

	char path[PATH_MAX];

	struct razor_set *set;

	assert (root != NULL);

	razor_root_init();

	set = razor_set_create_without_root();

	if (set == NULL)

		return NULL;

	snprintf(path, sizeof path,

		 "%s%s/%s", root, razor_root_path, system_lock_filename);

	if (razor_set_aquire_lock(set, path, 0) < 0) {

		razor_set_destroy(set);

		return NULL;

	}

	snprintf(path, sizeof path, "%s%s/%s",

		 root, razor_root_path, system_repo_filename);

	if (razor_set_bind_sections(set, path)) {

		razor_set_destroy(set);

		return NULL;

	}

	return set;

}

RAZOR_EXPORT struct razor_set *

razor_root_get_system_set(struct razor_root *root)

{

	assert (root != NULL);

	return root->system;

}

RAZOR_EXPORT int

razor_root_close(struct razor_root *root)

{

	assert (root != NULL);

	razor_set_destroy(root->system);

	close(root->fd);

	unlink(root->new_path);

	free(root);

	return 0;

}

RAZOR_EXPORT void

razor_root_update(struct razor_root *root, struct razor_set *next)

{

	assert (root != NULL);

	assert (next != NULL);

	razor_root_init();

	razor_set_write_to_fd(next, root->fd, RAZOR_SECTION_ALL);

	root->next = next;

	/* Sync the new repo file so the new package set is on disk

	 * before we start upgrading. */

	fsync(root->fd);

	printf("wrote %s\n", root->new_path);

}

RAZOR_EXPORT int

razor_root_commit(struct razor_root *root)

{

	int retval;

	assert (root != NULL);

	/* Make it so. */

	close(root->fd);

#ifdef MSWIN_API

	/* Rename is not atomic under MS-Windows */

	remove(root->path);

#endif

	retval = rename(root->new_path, root->path);

	if (retval)

		perror(root->path);

	else

		printf("renamed %s to %s\n", root->new_path, root->path);

	razor_set_destroy(root->system);

	free(root);

	return retval;

}