/*

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

 struct vector {

     int len,alloc;

     char **strings;

 };

 struct vector *vector_new(void)

 {

     struct vector *vector;

     vector=malloc(sizeof(*vector));

     vector->len=0;

     vector->alloc=16;

     vector->strings=calloc(vector->alloc,sizeof(char *));

     return vector;

 }

 void vector_append(struct vector *vector,const char *str)

 {

     if (++(vector->len)>=vector->alloc)

     {

 	vector->alloc*=2;

 	vector->strings=realloc(vector->strings,vector->alloc*sizeof(char *));

     }

     vector->strings[vector->len-1]=strdup(str);

     vector->strings[vector->len]=NULL;

 }

 int vector_contains(struct vector *vector,const char *str)

 {

     int i;

     for(i=0;i<vector->len;i++)

 	if (!strcmp(vector->strings[i],str))

 	    return 1;

     return 0;

 }

 void vector_free(struct vector *vector)

 {

     int i;

     for(i=0;i<vector->len;i++)

 	free(vector->strings[i]);

     free(vector->strings);

     free(vector);

 }

 void setup(const char *argv0)

 {

     char *path,*s,*prefix;

     int ch,changed;

     struct comps *comps;

     struct comps_group *group;

     struct comps_requirement *pkg;

     struct vector *packages=NULL;

     path=plover_get_program_directory(argv0);

     s=plover_strconcat(path,"/repodata/comps.xml",NULL);

     comps=plover_comps_new_from_file(s);

     if (!comps)

     {

 	perror(s);

 	exit(1);

     }

     free(s);

     prefix=plover_default_prefix_for_vendor(comps->vendor);

     if (!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();

 	plover_remove(NULL);

     }

     group=plover_comps_lookup_group(comps,"base");

     if (!group)

     {

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

 	exit(1);

     }

     packages=vector_new();

     do

     {

 	changed=0;

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

 	{

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

 		continue;

 	    if (pkg->type==COMPS_REQUIREMENT_DEFAULT ||

 	      pkg->type==COMPS_REQUIREMENT_MANDATORY ||

 	      pkg->type==COMPS_REQUIREMENT_CONDITIONAL &&

 	      vector_contains(packages,pkg->requires))

 	    {

 		changed++;

 		vector_append(packages,pkg->name);

 	    }

 	}

     } while(changed);

     plover_comps_free(comps);

     if (!packages->len)

     {

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

 	exit(1);

     }

     plover_install(path,prefix,packages->strings);

     vector_free(packages);

     free(prefix);

     free(path);

 }

 int main(int argc,char **argv)

 {

     razor_set_lua_loader("posix",luaopen_posix);

     razor_set_lua_loader("whelk",luaopen_whelk);

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

 	plover_remove(NULL);

     else

 	setup(argv[0]);

     exit(0);

 }