/*

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

 #include <plover/plover.h>

 #include <plover/package.h>

 struct razor_set *test_set=NULL;

 struct razor_package *test_pkg=NULL;

 static void finalize(void)

 {

     if (test_set)

 	razor_set_unref(test_set);

 }

 void import_properties(struct razor_importer *importer,

   struct razor_set *set,struct razor_package *pkg)

 {

     struct razor_property_iterator *iter;

     struct razor_property *prop;

     uint32_t flags;

     const char *name,*version;

     iter=razor_property_iterator_create(set,pkg);

     while(razor_property_iterator_next(iter,&prop,&name,&flags,&version))

 	razor_importer_add_property(importer,name,flags,version);

     razor_property_iterator_destroy(iter);

 }

 void import_files(struct razor_importer *importer,

   struct razor_set *set,struct razor_package *pkg)

 {

     struct razor_file_iterator *iter;

     const char *name;

     iter=razor_file_iterator_create(set,pkg,FALSE);

     while(razor_file_iterator_next(iter,&name))

 	razor_importer_add_file(importer,name);

     razor_file_iterator_destroy(iter);

 }

 gboolean import_rpm(struct razor_rpm *rpm)

 {

     int i;

     struct razor_importer *importer;

     struct razor_package_iterator *iter;

     struct razor_set *set;

     struct razor_package *pkg=NULL;

     const char *name,*version,*arch,*summary,*description,*url,*license;

     const char *s,*pkg_name;

     const char *const *prefixes;

     /*

      * First pass: get a razor_set and razor_package with everything

      * except the prefixes.

      */

     importer=razor_importer_create();

     if (razor_importer_add_rpm(importer,rpm))

 	g_error("../yum-repo-test-dir: Failed to add rpm");

     set=razor_importer_finish(importer);

     if (!set)

 	g_error("../yum-repo-test-dir: Failed to import");

     iter=razor_package_iterator_create(set);

     pkg=NULL;

     razor_rpm_get_details(rpm,RAZOR_DETAIL_NAME,&pkg_name,RAZOR_DETAIL_LAST);

     while(razor_package_iterator_next(iter,&pkg,RAZOR_DETAIL_NAME,&s,

       RAZOR_DETAIL_LAST))

 	if (!strcmp(s,pkg_name))

 	    break;

     razor_package_iterator_destroy(iter);

     if (!pkg)

     {

 	razor_set_unref(set);

 	return FALSE;

     }

     /*

      * Second pass: use the information from the first pass plus the

      * prefixes to create a final razor_set and razor_package.

      */

     importer=razor_importer_create();

     razor_package_get_details(set,pkg,RAZOR_DETAIL_NAME,&name,

       RAZOR_DETAIL_VERSION,&version,RAZOR_DETAIL_ARCH,&arch,

       RAZOR_DETAIL_SUMMARY,&summary,RAZOR_DETAIL_DESCRIPTION,&description,

       RAZOR_DETAIL_URL,&url,RAZOR_DETAIL_LICENSE,&license,RAZOR_DETAIL_LAST);

     razor_importer_begin_package(importer,name,version,arch);

     razor_importer_add_details(importer,summary,description,url,license);

     import_properties(importer,set,pkg);

     import_files(importer,set,pkg);

     razor_rpm_get_details(rpm,RAZOR_DETAIL_PREFIXES,&prefixes,

       RAZOR_DETAIL_LAST);

     for(i=0;prefixes && prefixes[i];i++)

 	razor_importer_add_install_prefix(importer,prefixes[i]);

     razor_importer_finish_package(importer);

     razor_set_unref(set);

     test_set=razor_importer_finish(importer);

     g_assert(test_set != NULL);

     iter=razor_package_iterator_create(test_set);

     test_pkg=NULL;

     while(razor_package_iterator_next(iter,&test_pkg,RAZOR_DETAIL_NAME,&s,

       RAZOR_DETAIL_LAST))

 	if (!strcmp(s,pkg_name))

 	    break;

     razor_package_iterator_destroy(iter);

     return !!test_pkg;

 }

 static PloverPackage *package_getref(const char *name,const char *version)

 {

     struct razor_rpm *rpm;

     PloverPackage *package=NULL;

     struct razor_error *err=NULL;

     gchar *filename;

     if (test_set)

     {

 	razor_set_unref(test_set);

 	test_set=NULL;

     }

     /*

      * yum metadata doesn't include all the information included in

      * a rpm file, so do this the hard way. Unfortunately,

      * razor_importer_add_rpm() doesn't support prefixes so we have

      * to do it the really hard way.

      */

     filename=g_strconcat("../yum-repo-test-dir/rpms/",name,"-",version,

       ".noarch.rpm",NULL);

     rpm=razor_rpm_open(filename,&err);

     if (!rpm)

 	g_error("%s: %s",filename,razor_error_get_msg(err));

     g_free(filename);

     if (import_rpm(rpm))

 	package=plover_package_new(test_set,test_pkg);

     else

 	package=NULL;

     razor_rpm_close(rpm);

     return package;

 }

 static void test_basic_properties(void)

 {

     PloverPackage *package;

     package=package_getref("zsh","1-1");

     g_assert(package != NULL);

     g_assert(plover_package_get_razor_set(package) == test_set);

     g_assert(plover_package_get_razor_package(package) == test_pkg);

     g_assert_cmpstr(plover_package_get_name(package),==,"zsh");

     g_assert_cmpstr(plover_package_get_summary(package),==,"Test package");

     g_assert_cmpstr(plover_package_get_version(package),==,"1-1");

     g_assert_cmpstr(plover_package_get_license(package),==,"GPL");

     g_assert_cmpstr(plover_package_get_arch(package),==,"noarch");

     g_assert_cmpstr(plover_package_get_description(package),==,"Test package");

     g_assert_cmpstr(plover_package_get_URL(package),==,

       "http://www.juiblex.co.uk/beach");

     g_object_unref(package);

 }

 static void test_package_properties(void)

 {

     PloverPackage *package;

     struct razor_property_iterator *iter;

     struct razor_property *prop;

     uint32_t flags;

     const char *s,*name,*version;

     GString *str;

     GList *expected,*lnk;

     package=package_getref("zsh","1-1");

     g_assert(package != NULL);

     expected=g_list_prepend(NULL,

       (gpointer)g_intern_string("requires(pre,postun): zip"));

     expected=g_list_prepend(expected,

       (gpointer)g_intern_string("requires: zip"));

     expected=g_list_prepend(expected,

       (gpointer)g_intern_string("provides: zsh = 1-1"));

     iter=plover_package_property_iterator_create(package);

     g_assert(iter != NULL);

     while(razor_property_iterator_next(iter,&prop,&name,&flags,&version))

     {

 	if (g_str_has_prefix(name,"rpmlib("))

 	    continue;

 	s=razor_property_type_to_string(prop);

 	g_assert(s != NULL);

 	str=g_string_new(s);

 	if (flags&RAZOR_PROPERTY_SCRIPT_MASK)

 	{

 	    g_string_append_c(str,'(');

 	    if (flags&RAZOR_PROPERTY_PRE)

 		g_string_append(str,"pre");

 	    if (flags&RAZOR_PROPERTY_POST)

 	    {

 		if (str->str[str->len-1]!='(' && str->str[str->len-1]!=',')

 		    g_string_append_c(str,',');

 		g_string_append(str,"post");

 	    }

 	    if (flags&RAZOR_PROPERTY_PREUN)

 	    {

 		if (str->str[str->len-1]!='(' && str->str[str->len-1]!=',')

 		    g_string_append_c(str,',');

 		g_string_append(str,"preun");

 	    }

 	    if (flags&RAZOR_PROPERTY_POSTUN)

 	    {

 		if (str->str[str->len-1]!='(' && str->str[str->len-1]!=',')

 		    g_string_append_c(str,',');

 		g_string_append(str,"postun");

 	    }

 	    g_string_append_c(str,')');

 	}

 	g_string_append(str,": ");

 	g_string_append(str,name);

 	if (*version)

 	{

 	    g_string_append_c(str,' ');

 	    g_string_append(str,razor_property_relation_to_string(prop));

 	    g_string_append_c(str,' ');

 	    g_string_append(str,version);

 	}

 	lnk=g_list_find(expected,g_intern_string(str->str));

 	if (!lnk)

 	    g_warning("Unexpected property in zsh package: %s",str->str);

 	else

 	    expected=g_list_delete_link(expected,lnk);

 	g_string_free(str,TRUE);

     }

     razor_property_iterator_destroy(iter);

     if (expected)

 	g_warning("%d missing propert%s in set, including %s",

 	  g_list_length(expected),g_list_length(expected)==1?"y":"ies",

 	  expected->data);

     g_object_unref(package);

 }

 static void test_package_files(void)

 {

     PloverPackage *package;

     struct razor_file_iterator *iter;

     const char *name;

     GList *expected,*lnk;

     package=package_getref("zsh","1-1");

     g_assert(package != NULL);

     expected=g_list_prepend(NULL,(gpointer)g_intern_string("/etc/zsh.conf"));

     expected=g_list_prepend(expected,(gpointer)g_intern_string("/usr/bin/zsh"));

     iter=plover_package_file_iterator_create(package,FALSE);

     g_assert(iter != NULL);

     while(razor_file_iterator_next(iter,&name))

     {

 	lnk=g_list_find(expected,g_intern_string(name));

 	if (!lnk)

 	    g_warning("Unexpected file in zsh package: %s",name);

 	else

 	    expected=g_list_delete_link(expected,lnk);

     }

     razor_file_iterator_destroy(iter);

     if (expected)

 	g_warning("%d missing file%s in set, including %s",

 	  g_list_length(expected),g_list_length(expected)==1?"":"s",

 	  expected->data);

     g_object_unref(package);

 }

 static void test_package_icon(void)

 {

     /*

      * icons aren't supported yet.

      */

     PloverPackage *package;

     GInputStream *stream;

     GError *err=NULL;

     package=package_getref("zsh","1-1");

     g_assert(package != NULL);

     stream=plover_package_read_icon(package,&err);

     if (stream)

     {

 	/* Unexpected, but hardly an error */

 	g_object_unref(stream);

     }

     else

     {

 	g_assert(err != NULL);

 	g_assert(err->message != NULL);

 	g_error_free(err);

     }

     g_object_unref(package);

 }

 static void test_package_prefixes(void)

 {

     PloverPackage *package;

     const char *const *prefixes;

     package=package_getref("zsh","1-1");

     g_assert(package != NULL);

     prefixes=plover_package_get_prefixes(package);

     g_assert_cmpstr(prefixes[0], ==, "/usr");

     g_assert(prefixes[1] == NULL);

     g_object_unref(package);

 }

 int main(int argc,char **argv)

 {

     int retval;

     g_test_init(&argc,&argv,NULL);

     g_test_bug_base("mailto:ali@juiblex.co.uk");

     g_test_add_func("/package/basic",test_basic_properties);

     g_test_add_func("/package/icons",test_package_icon);

     g_test_add_func("/package/properties",test_package_properties);

     g_test_add_func("/package/files",test_package_files);

     g_test_add_func("/package/prefixes",test_package_prefixes);

     g_test_message("PloverPackage::changed signal unused and thus untestable");

     retval=g_test_run();

     finalize();

     return retval;

 }