/*

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

}

static void *alloc_lua(void *user_data,void *ptr,size_t osize,size_t nsize)

{

    if (!nsize)

    {

	free(ptr);

	return NULL;

    }

    else

	return realloc(ptr,nsize);

}

void setup()

{

    char path[PATH_MAX],*s,*t,*prefix;

    int changed;

    struct comps *comps;

    struct comps_group *group;

    struct comps_requirement *pkg;

    struct vector *packages=NULL;

#ifdef WIN32

    SHGetFolderPath(NULL,CSIDL_PROGRAM_FILES|CSIDL_FLAG_DONT_VERIFY,NULL,0,

      path);

    prefix=strdup(path);

    GetModuleFileName(NULL,path,sizeof(path));

    s=strrchr(path,'/');

    if (s)

    {

	t=strrchr(s,'\\');

	if (t)

	    s=t;

    }

    else

	s=strrchr(path,'\\');

    if (s)

	*s='\0';

#else

    strcpy(path,"/tmp");

    prefix=NULL;

#endif

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

    comps=plover_comps_new_from_file(s);

    if (!comps)

    {

	perror(s);

	exit(1);

    }

    free(s);

#ifdef WIN32

    s=plover_strconcat(prefix,"\\",comps->vendor?comps->vendor:"Plover",NULL);

    free(prefix);

    prefix=s;

#endif

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

    if (!packages->len)

    {

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

	exit(1);

    }

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

    vector_free(packages);

    plover_comps_free(comps);

#ifdef WIN32

    free(prefix);

#endif

}

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

}