/*-
 * Copyright (c) 2009-2012 Juan Romero Pardines.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>

#include "xbps_api_impl.h"

/**
 * @file lib/package_configure.c
 * @brief Package configuration routines
 * @defgroup configure Package configuration functions
 *
 * Configure a package or all packages. Only packages in XBPS_PKG_STATE_UNPACKED
 * state will be processed (unless overriden). Package configuration steps:
 *  - Its <b>post-install</b> target in the INSTALL script will be executed.
 *  - Its state will be changed to XBPS_PKG_STATE_INSTALLED if previous step
 *    ran successful.
 *
 * @note
 * If the \a XBPS_FLAG_FORCE_CONFIGURE is set through xbps_init() in the flags
 * member, the package (or packages) will be reconfigured even if its
 * state is XBPS_PKG_STATE_INSTALLED.
 */
static int
configure_pkgs_cb(prop_object_t obj, void *arg, bool *done)
{
	const char *pkgname, *version;

	(void)arg;
	(void)done;

	prop_dictionary_get_cstring_nocopy(obj, "pkgname", &pkgname);
	prop_dictionary_get_cstring_nocopy(obj, "version", &version);
	return xbps_configure_pkg(pkgname, version, true, false, false);
}

int
xbps_configure_packages(bool flush)
{
	struct xbps_handle *xhp = xbps_handle_get();
	int rv;

	rv = xbps_regpkgdb_foreach_pkg_cb(configure_pkgs_cb, NULL);
	if (rv == 0 && flush)
		rv = xbps_regpkgdb_update(xhp, true);

	return rv;
}

int
xbps_configure_pkg(const char *pkgname,
		   const char *version,
		   bool check_state,
		   bool update,
		   bool flush)
{
	struct xbps_handle *xhp;
	prop_dictionary_t pkgd;
	const char *lver;
	char *buf, *pkgver;
	int rv = 0;
	pkg_state_t state = 0;

	assert(pkgname != NULL);
	xhp = xbps_handle_get();

	if (check_state) {
		rv = xbps_pkg_state_installed(pkgname, &state);
		xbps_dbg_printf("%s-%s: state %d\n", pkgname, version, state);
		if (rv == ENOENT) {
			/*
			 * package not installed or has been removed.
			 */
			return 0;
		} else if (rv != 0) {
			xbps_dbg_printf("%s: [configure] failed to get "
			    "pkg state: %s\n", pkgname, strerror(rv));
			return EINVAL;
		}

		if (state == XBPS_PKG_STATE_INSTALLED) {
			if ((xhp->flags & XBPS_FLAG_FORCE_CONFIGURE) == 0)
				return 0;
		} else if (state != XBPS_PKG_STATE_UNPACKED)
			return EINVAL;
	
		pkgd = xbps_regpkgdb_get_pkgd(pkgname, false);
		prop_dictionary_get_cstring_nocopy(pkgd, "version", &lver);
		prop_object_release(pkgd);
	} else {
		lver = version;
	}

	pkgver = xbps_xasprintf("%s-%s", pkgname, lver);
	if (pkgver == NULL)
		return ENOMEM;

	xbps_set_cb_state(XBPS_STATE_CONFIGURE, 0, pkgname, lver, NULL);

	buf = xbps_xasprintf("%s/metadata/%s/INSTALL",
	    XBPS_META_PATH, pkgname);
	if (buf == NULL) {
		free(pkgver);
		return ENOMEM;
	}

	if (chdir(xhp->rootdir) == -1) {
		xbps_set_cb_state(XBPS_STATE_CONFIGURE_FAIL, errno,
		    pkgname, lver,
		    "%s: [configure] failed to chdir to rootdir `%s': %s",
		    pkgver, xhp->rootdir, strerror(errno));
		free(buf);
		free(pkgver);
		return EINVAL;
	}

	if (access(buf, X_OK) == 0) {
		if (xbps_file_exec(buf, "post",
		    pkgname, lver, update ? "yes" : "no",
		    xhp->conffile, NULL) != 0) {
			xbps_set_cb_state(XBPS_STATE_CONFIGURE_FAIL, errno,
			    pkgname, lver,
			    "%s: [configure] INSTALL script failed to execute "
			    "the post ACTION: %s", pkgver, strerror(errno));
			free(buf);
			free(pkgver);
			return errno;
		}
	} else {
		if (errno != ENOENT) {
			free(buf);
			free(pkgver);
			return errno;
		}
	}
	free(buf);
	rv = xbps_set_pkg_state_installed(pkgname, lver, pkgver,
	    XBPS_PKG_STATE_INSTALLED);
	if (rv != 0) {
		xbps_set_cb_state(XBPS_STATE_CONFIGURE_FAIL, rv,
		    pkgname, lver,
		    "%s: [configure] failed to set state to installed: %s",
		    pkgver, strerror(rv));
	}
	free(pkgver);
	if (flush)
		rv = xbps_regpkgdb_update(xhp, true);

	return rv;
}