/*

 * Copyright (C) 2010, 2011  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 <string.h>

#include <fcntl.h>

#include <errno.h>

#include <unistd.h>

#include <glib-object.h>

#include <razor.h>

#include "plover/plover.h"

#include "plover-gtk/error.h"

#include "plover-gtk/packageset.h"

#include "plover-gtk/package.h"

G_DEFINE_TYPE(PloverPackageSet,plover_package_set,G_TYPE_OBJECT);

typedef struct _PloverPackageSetPrivate {

    struct razor_atomic *atomic;

    struct razor_root *root;

    struct razor_set *set;

    GSList *packages;

    int no_details;

} PloverPackageSetPrivate;

#define PLOVER_PACKAGE_SET_GET_PRIVATE(obj)\

				G_TYPE_INSTANCE_GET_PRIVATE(obj,\

				  PLOVER_TYPE_PACKAGE_SET,\

				  PloverPackageSetPrivate)

enum {

    CHANGED=0,

    N_SIGNALS

};

static guint signals[N_SIGNALS];

static void plover_package_set_finalize(GObject *obj)

{

    PloverPackageSetPrivate *priv=PLOVER_PACKAGE_SET_GET_PRIVATE(obj);

    if (priv->root)

	razor_root_close(priv->root);

    if (priv->atomic)

	razor_atomic_destroy(priv->atomic);

    if (G_OBJECT_CLASS(plover_package_set_parent_class)->finalize)

	G_OBJECT_CLASS(plover_package_set_parent_class)->finalize(obj);

}

static void plover_package_set_dispose(GObject *obj)

{

    PloverPackageSetPrivate *priv=PLOVER_PACKAGE_SET_GET_PRIVATE(obj);

    if (priv->set)

    {

	razor_set_unref(priv->set);

	priv->set=NULL;

    }

    if (G_OBJECT_CLASS(plover_package_set_parent_class)->dispose)

	G_OBJECT_CLASS(plover_package_set_parent_class)->dispose(obj);

}

static void plover_package_set_class_init(PloverPackageSetClass *klass)

{

    GObjectClass *oclass=G_OBJECT_CLASS(klass);

    oclass->finalize=plover_package_set_finalize;

    oclass->dispose=plover_package_set_dispose;

    g_type_class_add_private(klass,sizeof(PloverPackageSetPrivate));

    signals[CHANGED]=g_signal_newv("changed",

      G_TYPE_FROM_CLASS(klass),G_SIGNAL_RUN_LAST,NULL,NULL,NULL,

      g_cclosure_marshal_VOID__VOID,G_TYPE_NONE,0,NULL);

}

static void plover_package_set_init(PloverPackageSet *set)

{

    PloverPackageSetPrivate *priv;

    priv=PLOVER_PACKAGE_SET_GET_PRIVATE(set);

    priv->no_details=-1;

}

PloverPackageSet *plover_package_set_new(void)

{

    return g_object_new(PLOVER_TYPE_PACKAGE_SET,NULL);

}

PloverPackageSet *plover_package_set_new_from_installed(const char *root,

  GError **err)

{

    PloverPackageSet *set;

    PloverPackageSetPrivate *priv;

    set=plover_package_set_new();

    priv=PLOVER_PACKAGE_SET_GET_PRIVATE(set);

    priv->atomic=razor_atomic_open("Read root package set");

    priv->root=razor_root_open(root,priv->atomic);

    if (!priv->root)

    {

	g_set_error(err,PLOVER_RAZOR_ERROR,PLOVER_RAZOR_ERROR_FAILED,

	  "Failed to open %s as razor root",root);

	g_object_unref(set);

	return NULL;

    }

    priv->set=razor_set_ref(razor_root_get_system_set(priv->root));

    if (!priv->set)

    {

	g_set_error(err,PLOVER_RAZOR_ERROR,PLOVER_RAZOR_ERROR_FAILED,

	  "Failed to get system set from %s",root);

	g_object_unref(set);

	return NULL;

    }

    return set;

}

PloverPackageSet *plover_package_set_new_from_repository(const char *base,

  struct razor_relocations *relocations,GError **err)

{

#if HAVE_FCHDIR

    int fd;

#else

    size_t wd_len;

    char *wd;

#endif

    gchar *s;

    struct razor_set *reloc;

    struct razor_atomic *atomic;

    PloverPackageSet *set;

    PloverPackageSetPrivate *priv;

    set=plover_package_set_new();

    priv=PLOVER_PACKAGE_SET_GET_PRIVATE(set);

#if HAVE_FCHDIR

    fd=open(".",O_RDONLY);

#else

    wd_len=32;

    wd=malloc(wd_len);

    while (!getcwd(wd,wd_len) && errno==ERANGE)

    {

	free(wd);

	wd_len*=2;

	wd=malloc(wd_len);

    }

#endif

    s=g_build_filename(base,"repodata",NULL);

    if (chdir(s)<0)

    {

	g_set_error(err,G_FILE_ERROR,g_file_error_from_errno(errno),

	  "%s: %s",s,g_strerror(errno));

	g_object_unref(set);

#if HAVE_FCHDIR

	close(fd);

#else

	free(wd);

#endif

	return NULL;

    }

    g_free(s);

    priv->set=plover_razor_set_create_from_yum("..");

#if HAVE_FCHDIR

    (void)fchdir(fd);

    close(fd);

#else

    chdir(wd);

    free(wd);

#endif

    if (priv->set && relocations)

    {

	atomic=razor_atomic_open("Relocate packages");

	reloc=plover_relocate_packages(priv->set,atomic,base,relocations);

	if (!reloc)

	    g_set_error(err,PLOVER_RAZOR_ERROR,PLOVER_RAZOR_ERROR_FAILED,

	      razor_atomic_get_error_msg(atomic));

	razor_atomic_destroy(atomic);

	if (!reloc)

	{

	    g_object_unref(set);

	    return NULL;

	}

	razor_set_unref(priv->set);

	priv->set=reloc;

    }

    if (!priv->set)

    {

	g_set_error(err,PLOVER_RAZOR_ERROR,PLOVER_RAZOR_ERROR_FAILED,

	  "Failed to create package set from repository %s",base);

	g_object_unref(set);

	return NULL;

    }

    return set;

}

GSList *plover_package_set_get_packages(PloverPackageSet *set)

{

    struct razor_package_iterator *iter;

    struct razor_package *pkg;

    PloverPackageSetPrivate *priv;

    PloverPackage *package;

    g_return_val_if_fail(PLOVER_IS_PACKAGE_SET(set),NULL);

    priv=PLOVER_PACKAGE_SET_GET_PRIVATE(set);

    if (priv->set && !priv->packages)

    {

	iter=razor_package_iterator_create(priv->set);

	while(razor_package_iterator_next(iter,&pkg,RAZOR_DETAIL_LAST))

	{

	    package=plover_package_new(priv->set,pkg);

	    priv->packages=g_slist_prepend(priv->packages,package);

	}

	razor_package_iterator_destroy(iter);

    }

    return priv->packages;

}

/*

 * Some versions of razor have a bug which causes all detail strings

 * to be discarded. If such a version of razor is used to install or

 * update a package, then all the detail strings for the installed

 * set will be lost. This function tests for this condition and can

 * be used to present something more useful than blank details.

 */

gboolean plover_package_set_get_no_details(PloverPackageSet *set)

{

    PloverPackageSetPrivate *priv;

    PloverPackage *package;

    GSList *packages,*link;

    g_return_val_if_fail(PLOVER_IS_PACKAGE_SET(set),FALSE);

    priv=PLOVER_PACKAGE_SET_GET_PRIVATE(set);

    if (priv->no_details<0)

    {

	packages=plover_package_set_get_packages(set);

	if (packages)

	{

	    priv->no_details=0;

	    for(link=packages;link;link=link->next)

	    {

		package=link->data;

		priv->no_details+=2;

		if (*plover_package_get_summary(package))

		    priv->no_details--;

		if (*plover_package_get_license(package))

		    priv->no_details--;

		if (*plover_package_get_description(package))

		    priv->no_details--;

		if (*plover_package_get_URL(package))

		    priv->no_details--;

	    }

	    if (priv->no_details<0)	/* More than 50% of strings present */

		priv->no_details=0;

	}

    }

    return priv->no_details>0;

}