/*

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

}