/*

 * 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 *uri;

    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.

     */

    uri=g_strconcat("file:../yum-repo-test-dir/Packages/",name,"-",version,

      ".noarch.rpm",NULL);

    rpm=razor_rpm_open(uri,&err);

    if (!rpm)

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

    g_free(uri);

    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;

}