/*

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

 #include <string.h>

 #include <limits.h>

 #ifdef WIN32

 #include <windows.h>

 #include <shlobj.h>

 #endif

 #include <lua.h>

 #include "config.h"

 #include "plover/plover.h"

 #include "whelk/whelk.h"

 LUALIB_API int luaopen_posix(lua_State *L);

 void setup(const char *argv0)

 {

     char *yum_uri,*local_database;

     char *active_database=NULL,*alternate_database=NULL;

     gchar *s,*prefix,*distribution,*vendor_prefix;

     int ch,changed;

     struct comps *comps;

     struct comps_group *group;

     struct comps_requirement *pkg;

     struct plover_vector *packages=NULL;

     GError *error=NULL;

     s=plover_get_program(argv0);

     yum_uri=razor_path_to_uri(s);

     g_free(s);

     s=g_strconcat(yum_uri,"/repodata/comps.xml",NULL);

     comps=plover_comps_new_from_uri(s,&error);

     g_free(s);

     if (g_error_matches(error,PLOVER_RAZOR_ERROR,

       RAZOR_GENERAL_ERROR_UNSUPPORTED_ARCHIVE))

     {

 	g_clear_error(&error);

 	free(yum_uri);

 	s=plover_get_program_directory(argv0);

 	yum_uri=razor_path_to_uri(s);

 	g_free(s);

 	s=g_strconcat(yum_uri,"/repodata/comps.xml",NULL);

 	comps=plover_comps_new_from_uri(s,&error);

     }

     if (!comps)

     {

 	fprintf(stderr,"%s\n",error->message);

 	g_error_free(error);

 	exit(1);

     }

     prefix=plover_comps_get_default_prefix(comps);

     if (prefix)

     {

 	s=g_strconcat(prefix,"/var/lib/razor",NULL);

 	local_database=razor_path_to_uri(s);

 	g_free(s);

     }

     else

 	local_database=NULL;

     switch(comps->database)

     {

 	case COMPS_DATABASE_DISTRIBUTION_LOCAL:

 	    active_database=local_database;

 	    break;

 	case COMPS_DATABASE_GLOBAL:

 	    alternate_database=local_database;

 	    break;

     }

     if (prefix)

     {

 	distribution=g_strdup(comps->distribution);

 	plover_comps_set_distribution(comps,NULL);

 	vendor_prefix=plover_comps_get_default_prefix(comps);

 	plover_comps_set_distribution(comps,distribution);

 	g_free(distribution);

 	razor_set_database_uri(alternate_database);

 	if (plover_installed_files_match_prefix(vendor_prefix)==1)

 	{

 	    printf("There is an existing installation under %s\n"

 	      "which is not compatible with this distribution. In order\n"

 	      "to continue, the existing installation must be uninstalled.\n"

 	      "Do you want to remove all packages in the existing installion? ",

 	      prefix);

 	    ch=getchar();

 	    if (ch!='y' && ch!='Y' && ch!=EOF && ch!='\n')

 		exit(1);

 	    while(ch!='\n' && ch!=EOF)

 		ch=getchar();

 	    if (plover_remove(NULL,&error))

 	    {

 		fprintf(stderr,"%s\n",error->message);

 		g_error_free(error);

 		exit(1);

 	    }

 	}

 	g_free(vendor_prefix);

     }

     razor_set_database_uri(active_database);

     free(local_database);

     if (prefix && !plover_installed_files_match_prefix(prefix))

     {

 	printf("The existing installation is not under %s\n"

 	  "In order to continue, all the existing packages must be removed.\n"

 	  "Do you want to remove all existing packages? ",prefix);

 	ch=getchar();

 	if (ch!='y' && ch!='Y' && ch!=EOF && ch!='\n')

 	    exit(1);

 	while(ch!='\n' && ch!=EOF)

 	    ch=getchar();

 	if (plover_remove(NULL,&error))

 	{

 	    fprintf(stderr,"%s\n",error->message);

 	    g_error_free(error);

 	    exit(1);

 	}

     }

     group=plover_comps_lookup_group(comps,"base");

     if (!group)

     {

 	fprintf(stderr,"No base group found in comps.xml\n");

 	exit(1);

     }

     packages=plover_vector_new();

     do

     {

 	changed=0;

 	for(pkg=group->packages;pkg;pkg=pkg->next)

 	{

 	    if (plover_vector_contains(packages,pkg->name))

 		continue;

 	    if (pkg->type==COMPS_REQUIREMENT_DEFAULT ||

 	      pkg->type==COMPS_REQUIREMENT_MANDATORY ||

 	      pkg->type==COMPS_REQUIREMENT_CONDITIONAL &&

 	      plover_vector_contains(packages,pkg->requires))

 	    {

 		changed++;

 		plover_vector_append(packages,pkg->name);

 	    }

 	}

     } while(changed);

     plover_comps_free(comps);

     if (!packages->len)

     {

 	fprintf(stderr,"No packages to install\n");

 	exit(1);

     }

     if (!plover_install_uri(yum_uri,prefix,packages->strings,&error))

     {

 	fprintf(stderr,"%s\n",error->message);

 	g_error_free(error);

 	exit(1);

     }

     plover_vector_free(packages);

     g_free(prefix);

     free(yum_uri);

 }

 int main(int argc,char **argv)

 {

     GError *error=NULL;

     plover_exception_handler_init();

     razor_set_lua_loader("posix",(void (*)())luaopen_posix);

     razor_set_lua_loader("whelk",(void (*)())luaopen_whelk);

     if (argc>1 && !strcmp(argv[1],"-u"))

     {

 	if (!plover_remove(NULL,&error))

 	{

 	    fprintf(stderr,"%s\n",error->message);

 	    g_error_free(error);

 	    exit(1);

 	}

     }

     else

 	setup(argv[0]);

     exit(0);

 }