/*
 * Copyright (C) 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 <stdlib.h>
#include <string.h>
#include <glib.h>
#include <gdk/gdk.h>
#include <gtk/gtk.h>
#include <razor/razor.h>
#include <plover/package.h>
#include <plover/packageset.h>
#include <plover-gtk/packagestore.h>
#include "treemodel.h"

static void verify_package_store(PloverPackageStore *store,
  GSList *expected_packages)
{
    int i;
    const char *s;
    gchar *nosummary;
    PloverPackage *package;
    PloverPackageSet *package_set;
    GtkTreeModel *model;
    GtkTreeIter iter;
    GValue value={0,};
    GSList *expected,*lnk;
    GdkPixbuf *icon;
    model=GTK_TREE_MODEL(store);
    test_tree_model(model);
    expected=g_slist_copy(expected_packages);
    g_assert_cmpint(gtk_tree_model_get_n_columns(model),==,PLOVER_PACKAGE_STORE_NO_COLUMNS);
    g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_OBJ_COLUMN),==,PLOVER_TYPE_PACKAGE);
    g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_INSTALLED_COLUMN),==,G_TYPE_BOOLEAN);
    g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_ICON_COLUMN),==,GDK_TYPE_PIXBUF);
    g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_NAME_COLUMN),==,G_TYPE_STRING);
    g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_VERSION_COLUMN),==,G_TYPE_STRING);
    g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_SUMMARY_COLUMN),==,G_TYPE_STRING);
    for(i=0;;i++)
    {
	if (!gtk_tree_model_iter_nth_child(model,&iter,NULL,i))
	    break;
	gtk_tree_model_get_value(model,&iter,PLOVER_PACKAGE_STORE_OBJ_COLUMN,
	  &value);
	package=g_value_dup_object(&value);
	lnk=g_slist_find(expected,package);
	if (!lnk)
	    g_warning("Unexpected package in store: %s",
	      plover_package_get_name(package));
	else
	{
	    g_object_unref(lnk->data);
	    expected=g_slist_delete_link(expected,lnk);
	}
	g_value_unset(&value);
	gtk_tree_model_get_value(model,&iter,
	  PLOVER_PACKAGE_STORE_INSTALLED_COLUMN,&value);
	g_assert_cmpint(G_VALUE_TYPE(&value),==,G_TYPE_BOOLEAN);
	/* Can't check value of installed yet, it
	 * isn't set (or even properly defined).
	 */
	g_value_unset(&value);
	gtk_tree_model_get_value(model,&iter,
	  PLOVER_PACKAGE_STORE_ICON_COLUMN,&value);
	icon=g_value_get_object(&value);
	if (icon)
	    g_assert(GDK_IS_PIXBUF(icon));
	g_value_unset(&value);
	gtk_tree_model_get_value(model,&iter,PLOVER_PACKAGE_STORE_NAME_COLUMN,
	  &value);
	s=g_value_get_string(&value);
	g_assert_cmpstr(plover_package_get_name(package),==,s);
	g_value_unset(&value);
	gtk_tree_model_get_value(model,&iter,
	  PLOVER_PACKAGE_STORE_VERSION_COLUMN,&value);
	s=g_value_get_string(&value);
	g_assert_cmpstr(plover_package_get_version(package),==,s);
	g_value_unset(&value);
	gtk_tree_model_get_value(model,&iter,
	  PLOVER_PACKAGE_STORE_SUMMARY_COLUMN,&value);
	s=g_value_get_string(&value);
	if (*plover_package_get_summary(package))
	    g_assert_cmpstr(plover_package_get_summary(package),==,s);
	else
	{
	    nosummary=g_strconcat("The ",plover_package_get_name(package),
	      " package",NULL);
	    g_assert_cmpstr(nosummary,==,s);
	    g_free(nosummary);
	}
	g_value_unset(&value);
    }
    if (expected)
    {
	package=PLOVER_PACKAGE(expected->data);
	g_warning("%d missing package%s in store, including %s",
	  g_slist_length(expected),g_slist_length(expected)==1?"":"s",
	  plover_package_get_name(package));
    }
}

static void test_empty(void)
{
    PloverPackageStore *store;
    store=plover_package_store_new();
    g_assert(!plover_package_store_get_sets(store));
    verify_package_store(store,NULL);
    g_object_unref(store);
}

static void test_basic(void)
{
    PloverPackageSet *package_set;
    PloverPackageStore *store;
    GError *err=NULL;
    GSList *expected;
    package_set=plover_package_set_new_from_yum("../yum-repo-test-dir",NULL,
      &err);
    if (!package_set)
	g_error("../yum-repo-test-dir: %s",err->message);
    store=plover_package_store_new();
    plover_package_store_add_set(store,package_set);
    expected=g_slist_copy(plover_package_set_get_packages(package_set));
    g_slist_foreach(expected,(GFunc)g_object_ref,NULL);
    g_object_unref(package_set);
    verify_package_store(store,expected);
    g_slist_foreach(expected,(GFunc)g_object_unref,NULL);
    g_slist_free(expected);
    g_object_unref(store);
}

static PloverPackageSet *no_details_new(void)
{
    struct razor_set *set;
    struct razor_importer *importer;
    PloverPackageSet *package_set;
    importer=razor_importer_create();
    razor_importer_begin_package(importer,"no-details","1-1","noarch");
    razor_importer_add_details(importer,"","","","");
    razor_importer_add_property(importer,"no-details",
      RAZOR_PROPERTY_PROVIDES|RAZOR_PROPERTY_EQUAL,"1-1");
    razor_importer_finish_package(importer);
    set=razor_importer_finish(importer);
    package_set=plover_package_set_new_from_razor(set);
    razor_set_unref(set);
    return package_set;
}

static void test_no_details(void)
{
    PloverPackageSet *package_set;
    PloverPackageStore *store;
    GSList *expected;
    package_set=no_details_new();
    store=plover_package_store_new();
    plover_package_store_add_set(store,package_set);
    expected=g_slist_copy(plover_package_set_get_packages(package_set));
    g_slist_foreach(expected,(GFunc)g_object_ref,NULL);
    g_object_unref(package_set);
    verify_package_store(store,expected);
    g_slist_foreach(expected,(GFunc)g_object_unref,NULL);
    g_slist_free(expected);
    g_object_unref(store);
}

static void test_remove(void)
{
    PloverPackageSet *yum_set,*nodetails_set;
    PloverPackageStore *store;
    GSList *expected;
    GError *err=NULL;
    store=plover_package_store_new();
    yum_set=plover_package_set_new_from_yum("../yum-repo-test-dir",NULL,&err);
    if (!yum_set)
	g_error("../yum-repo-test-dir: %s",err->message);
    plover_package_store_add_set(store,yum_set);
    nodetails_set=no_details_new();
    expected=g_slist_copy(plover_package_set_get_packages(nodetails_set));
    g_slist_foreach(expected,(GFunc)g_object_ref,NULL);
    plover_package_store_add_set(store,nodetails_set);
    plover_package_store_remove_set(store,yum_set);
    g_object_unref(nodetails_set);
    g_object_unref(yum_set);
    verify_package_store(store,expected);
    g_slist_foreach(expected,(GFunc)g_object_unref,NULL);
    g_slist_free(expected);
    g_object_unref(store);
}

int main(int argc,char **argv)
{
    int retval;
    gtk_test_init(&argc,&argv,NULL);
    g_test_bug_base("mailto:ali@juiblex.co.uk");
    g_test_add_func("/packagestore/empty",test_empty);
    g_test_add_func("/packagestore/basic",test_basic);
    g_test_add_func("/packagestore/no-details",test_no_details);
    g_test_add_func("/packagestore/remove",test_remove);
    retval=g_test_run();
    return retval;
}