/*

 * Copyright (C) 2010  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 "config.h"

#include <stdlib.h>

#include <string.h>

#ifdef WIN32

#include <windows.h>

#endif	/* WIN32 */

#include <lua.h>

#include <glib.h>

#include <gio/gio.h>

#include <gtk/gtk.h>

#include <whelk/whelk.h>

#include <plover/plover.h>

#include <plover/packageset.h>

#include <plover-gtk/stockicons.h>

#include "app-manager.h"

#include "localmedia.h"

LUALIB_API int luaopen_posix(lua_State *L);

#define LOGO_NAME	"plover-applications"

GtkBuilder *ui;

GtkTreeModel *installed,*applications,*location,*local_media;

char *prefix=NULL;

struct razor_relocations *relocations=NULL;

void show_busy_cursor(gboolean busy)

{

    GList *list,*link,*remaining;

    GdkDisplay *display;

    GdkCursor *cursor;

    GtkWidget *w;

    list=gtk_window_list_toplevels();

    while(list)

    {

	w=GTK_WIDGET(list->data);

	if (!w->window)

	{

	    link=list;

	    list=g_list_remove_link(list,link);

	    g_list_free_1(link);

	}

	else

	{

	    display=gtk_widget_get_display(w);

	    cursor=busy?gdk_cursor_new_for_display(display,GDK_WATCH):NULL;

	    remaining=NULL;

	    for(link=list;link;link=link->next)

	    {

		w=GTK_WIDGET(link->data);

		if (w->window)

		{

		    if (gtk_widget_get_display(w)==display)

			gdk_window_set_cursor(w->window,cursor);

		    else

			remaining=g_list_prepend(remaining,w);

		}

	    }

	    gdk_display_flush(display);

	    if (cursor)

		gdk_cursor_unref(cursor);

	    g_list_free(list);

	    list=remaining;

	}

    }

}

/*

 * In Gtk+ 2.16.6, the default handler generates g_warnings on error.

 * It should display an error to the user. Do it ourselves.

 */

static void show_uri(GtkLinkButton *button,const gchar *uri,gpointer data)

{

    GdkScreen *screen;

    GtkWidget *dialog;

    GError *error=NULL;

    if (gtk_widget_has_screen(GTK_WIDGET(button)))

	screen=gtk_widget_get_screen(GTK_WIDGET(button));

    else

	screen=NULL;

    gtk_show_uri(screen,uri,GDK_CURRENT_TIME,&error);

    if (error)

    {

	dialog=gtk_message_dialog_new(

	  GTK_WINDOW(gtk_builder_get_object(ui,"MainWindow")),

	  GTK_DIALOG_DESTROY_WITH_PARENT,GTK_MESSAGE_ERROR,GTK_BUTTONS_CLOSE,

	  "Unable to show '%s'",uri);

	gtk_message_dialog_format_secondary_text(GTK_MESSAGE_DIALOG(dialog),

	  error->message);

	g_error_free(error);

	gtk_dialog_run(GTK_DIALOG(dialog));

	gtk_widget_destroy(dialog);

    }

}

#if 0

/* Checks whether a loader for SVG files has been registered

 * with GdkPixbuf.

 */

static gboolean pixbuf_supports_svg(void)

{

    GSList *formats;

    GSList *tmp_list;

    static gint found_svg=-1;

    gchar **mime_types,**mime_type;

    if (found_svg!=-1)

	return found_svg;

    formats=gdk_pixbuf_get_formats();

    found_svg=FALSE;

    for (tmp_list=formats;tmp_list && !found_svg;tmp_list=tmp_list->next)

    {

	mime_types=gdk_pixbuf_format_get_mime_types(tmp_list->data);

	for (mime_type=mime_types;*mime_type && !found_svg;mime_type++)

	    if (!strcmp(*mime_type,"image/svg"))

		found_svg=TRUE;

	g_strfreev(mime_types);

    }

    g_slist_free(formats);

    return found_svg;

}

static void install_icon_at_size(const char *icon_name,GtkIconSet *icon_set,

  GtkIconSize size,const char *filename)

{

    int w,h;

    GdkPixbuf *pixbuf;

    GtkIconSource *source;

    if (gtk_icon_size_lookup(size,&w,&h))

    {

	pixbuf=gdk_pixbuf_new_from_file_at_size(filename,w,h,NULL);

	if (pixbuf)

	{

	    source=gtk_icon_source_new();

	    gtk_icon_source_set_size_wildcarded(source,FALSE);

	    gtk_icon_source_set_size(source,size);

	    gtk_icon_source_set_pixbuf(source,pixbuf);

	    gtk_icon_set_add_source(icon_set,source);

	    gtk_icon_source_free(source);

	    g_object_unref(pixbuf);

	}

    }

}

static void install_icons(void)

{

    gchar *s;

    GtkIconSource *source;

    GtkIconSet *icon_set;

    GtkIconFactory *factory;

    factory=gtk_icon_factory_new();

    icon_set=gtk_icon_set_new();

    if (pixbuf_supports_svg())

    {

	source=gtk_icon_source_new();

	s=g_build_filename(prefix?prefix:"/usr",

	  "share/icons/hicolor/scalable/apps/plover-applications.svg",NULL);

	gtk_icon_source_set_filename(source,s);

	g_free(s);

	gtk_icon_set_add_source(icon_set,source);

	gtk_icon_source_free(source);

    }

    else

    {

	s=g_build_filename(prefix?prefix:"/usr",

	  "share/icons/hicolor/24x24/apps/plover-applications.png",NULL);

	install_icon_at_size(LOGO_NAME,icon_set,GTK_ICON_SIZE_MENU,s);

	install_icon_at_size(LOGO_NAME,icon_set,GTK_ICON_SIZE_BUTTON,s);

	install_icon_at_size(LOGO_NAME,icon_set,GTK_ICON_SIZE_SMALL_TOOLBAR,s);

	install_icon_at_size(LOGO_NAME,icon_set,GTK_ICON_SIZE_LARGE_TOOLBAR,s);

	g_free(s);

	s=g_build_filename(prefix?prefix:"/usr",

	  "share/icons/hicolor/48x48/apps/plover-applications.png",NULL);

	install_icon_at_size(LOGO_NAME,icon_set,GTK_ICON_SIZE_DND,s);

	install_icon_at_size(LOGO_NAME,icon_set,GTK_ICON_SIZE_DIALOG,s);

	g_free(s);

    }

    gtk_icon_factory_add(factory,LOGO_NAME,icon_set);

    gtk_icon_set_unref(icon_set);

    icon_set=gtk_icon_factory_lookup(factory,LOGO_NAME);

    gtk_icon_factory_add_default(factory);

    g_object_unref(factory);

    icon_set=gtk_icon_factory_lookup_default(LOGO_NAME);

    gtk_window_set_default_icon_name(LOGO_NAME);

}

#endif

static void install_icons(void)

{

    GtkIconSet *icon_set;

    plover_icons_add_to_stock("apps",LOGO_NAME);

    icon_set=gtk_icon_factory_lookup_default(LOGO_NAME);

    gtk_window_set_default_icon_name(LOGO_NAME);

}

int main(int argc,char **argv)

{

    GError *err=0;

    GtkWidget *w;

    gchar *s,*database_uri,*contents;

    gchar *database=NULL,*setup_base=NULL,*update_base=NULL;

    gsize len;

    struct comps *comps;

    PloverPackageSet *set=NULL;

    GSList *objects,*lnk;

    gboolean started;

    GOptionEntry options[]={

	{"database",0,0,G_OPTION_ARG_STRING,&database,

	  "Operate on a distribution-local database","vendor/distribution"},

	{"setup",0,0,G_OPTION_ARG_STRING,&setup_base,

	  "Setup from installation media","uri"},

	{"update",0,0,G_OPTION_ARG_STRING,&update_base,

	  "Update from upgrade media","uri"},

	{NULL}

    };

#ifdef WIN32

    /*

     * app-manager is normally a GUI application, but rpm scripts may well

     * call console applications and it looks ugly if console windows keep

     * popping up. Avoid this by allocating our own console and hiding it.

     * Note:

     *	- If app-manager is a console application (typically for debugging),

     *    then skip this step.

     *  - Call ShowWindow twice to negate special handling on first call.

     */

    if (!GetConsoleWindow())

    {

	AllocConsole();

	ShowWindow(GetConsoleWindow(),SW_HIDE);

	ShowWindow(GetConsoleWindow(),SW_HIDE);

    }

#endif

    plover_exception_handler_init();

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

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

    if (!gtk_init_with_args(&argc,&argv,NULL,options,NULL,&err))

    {

	g_printerr("%s\n",err->message);

	exit(1);

    }

    if (setup_base && update_base)

    {

	g_printerr("--setup and --update are mutually exclusive\n");

	exit(1);

    }

#ifdef WIN32

    prefix=g_win32_get_package_installation_directory_of_module(NULL);

#endif

    install_icons();

    ui=gtk_builder_new();

    if (!g_file_get_contents("app-manager.ui",&contents,&len,&err) &&

      g_error_matches(err,G_FILE_ERROR,G_FILE_ERROR_NOENT))

    {

#ifdef WIN32

	s=g_build_filename(prefix,"share","plover","app-manager.ui",NULL);

#else

	s=g_build_filename(PLOVER_DATADIR,"app-manager.ui",NULL);

#endif

	g_clear_error(&err);

	(void)g_file_get_contents(s,&contents,&len,&err);

	g_free(s);

    }

    if (!err)

    {

	(void)gtk_builder_add_from_string(ui,contents,len,&err);

	g_free(contents);

    }

    if (err)

    {

	g_error("%s",err->message);

	exit(0);

    }

    gtk_builder_connect_signals(ui,NULL);

    gtk_link_button_set_uri_hook(show_uri,NULL,NULL);

    if (setup_base)

	started=setup(setup_base);

    else if (update_base)

	started=update(update_base);

    else

    {

	if (database)

	{

	    g_free(prefix);

	    prefix=NULL;

	    s=strchr(database,'/');

	    if (*s)

		*s++='\0';

	    comps=plover_comps_new();

	    plover_comps_set_vendor(comps,database);

	    if (s)

	    {

		plover_comps_set_distribution(comps,s);

		*--s='/';

	    }

	    prefix=plover_comps_get_default_prefix(comps);

	    plover_comps_free(comps);

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

	    database_uri=razor_path_to_uri(s);

	    g_free(s);

	    razor_set_database_uri(database_uri);

	    g_free(database_uri);

	}

	if (prefix)

	{

	    relocations=razor_relocations_create();

	    razor_relocations_add(relocations,"/usr",prefix);

	}

	installed=GTK_TREE_MODEL(plover_package_store_new());

	set=plover_package_set_new();

	(void)plover_package_set_open(set,"",TRUE,NULL);

	plover_package_store_add_set(PLOVER_PACKAGE_STORE(installed),set);

	if (plover_package_set_get_no_details(set))

	{

	    w=GTK_WIDGET(gtk_builder_get_object(ui,"ViewFiles"));

	    gtk_check_menu_item_set_active(GTK_CHECK_MENU_ITEM(w),TRUE);

	}

	applications=plover_applications_model_new(installed);

	set_package_model(applications);

	w=GTK_WIDGET(gtk_builder_get_object(ui,"MainWindow"));

	gtk_widget_show(w);

	started=TRUE;

    }

    if (started)

	gtk_main();

    g_clear_object(&set);

    objects=gtk_builder_get_objects(ui);

    for(lnk=objects;lnk;lnk=lnk->next)

	if (GTK_IS_WIDGET(lnk->data) &&

	  gtk_widget_is_toplevel(GTK_WIDGET(lnk->data)))

	    gtk_widget_destroy(GTK_WIDGET(lnk->data));

    g_slist_free(objects);

    g_clear_object(&ui);

    g_clear_object(&installed);

    g_clear_object(&applications);

    g_clear_object(&location);

    g_clear_object(&local_media);

    if (relocations)

	razor_relocations_destroy(relocations);

    g_free(prefix);

    g_free(setup_base);

    g_free(update_base);

    g_free(database);

    exit(0);

}

G_MODULE_EXPORT void

  on_applications_toggled(GtkToggleToolButton *button,gpointer data)

{

    if (gtk_toggle_tool_button_get_active(button))

    {

	if (!applications)

	    applications=plover_applications_model_new(installed);

	set_package_model(applications);

    }

}

G_MODULE_EXPORT void

  on_all_packages_toggled(GtkToggleToolButton *button,gpointer data)

{

    if (gtk_toggle_tool_button_get_active(button))

	set_package_model(installed);

}

G_MODULE_EXPORT void

  on_local_media_toggled(GtkToggleToolButton *button,gpointer data)

{

    if (gtk_toggle_tool_button_get_active(button))

    {

	if (!local_media)

	{

	    show_busy_cursor(TRUE);

	    local_media=plover_local_media_store_new();

	    show_busy_cursor(FALSE);

	}

	set_package_model(local_media);

    }

}

G_MODULE_EXPORT void

  on_location_toggled(GtkToggleToolButton *button,gpointer data)

{

    if (gtk_toggle_tool_button_get_active(button))

	set_package_model(location);

}

G_MODULE_EXPORT void on_open_location(GtkWidget *widget)

{

    GtkWidget *w=GTK_WIDGET(gtk_builder_get_object(ui,"MainWindow"));

    GtkWidget *dialog;

    GFile *file,*parent;

    GFileInfo *fi;

    GMount *mount;

    gchar *path,*name;

    PloverPackageSet *set;

    GSList *sets;

    GError *err=NULL;

    dialog=gtk_file_chooser_dialog_new("Open Location",GTK_WINDOW(w),

      GTK_FILE_CHOOSER_ACTION_SELECT_FOLDER,GTK_STOCK_CANCEL,

      GTK_RESPONSE_CANCEL,GTK_STOCK_OPEN,GTK_RESPONSE_ACCEPT,NULL);

#if GTK_CHECK_VERSION(2,18,0)

    gtk_file_chooser_set_create_folders(GTK_FILE_CHOOSER(dialog),FALSE);

#endif

    if (gtk_dialog_run(GTK_DIALOG(dialog))==GTK_RESPONSE_ACCEPT)

    {

	show_busy_cursor(TRUE);

	path=gtk_file_chooser_get_filename(GTK_FILE_CHOOSER(dialog));

	set=plover_package_set_new_from_yum(path,relocations,&err);

	if (set)

	{

	    if (!location)

		location=GTK_TREE_MODEL(plover_package_store_new());

	    while((sets=

	      plover_package_store_get_sets(PLOVER_PACKAGE_STORE(location))))

		plover_package_store_remove_set(PLOVER_PACKAGE_STORE(location),

		  PLOVER_PACKAGE_SET(sets->data));

	    plover_package_store_add_set(PLOVER_PACKAGE_STORE(location),set);

	    g_object_unref(set);

	    w=GTK_WIDGET(gtk_builder_get_object(ui,"LocationButton"));

	    file=g_file_new_for_path(path);

	    parent=g_file_get_parent(file);

	    if (parent)

	    {

		g_object_unref(parent);

		mount=NULL;

	    }

	    else

		mount=g_file_find_enclosing_mount(file,NULL,NULL);

	    if (mount)

	    {

		name=g_mount_get_name(mount);

		g_object_unref(mount);

	    }

	    else

	    {

		fi=g_file_query_info(file,

		  G_FILE_ATTRIBUTE_STANDARD_DISPLAY_NAME,

		  G_FILE_QUERY_INFO_NOFOLLOW_SYMLINKS,NULL,NULL);

		if (fi)

		{

		    name=g_strdup(g_file_info_get_display_name(fi));

		    g_object_unref(fi);

		}

		else

		    name=g_filename_display_basename(path);

		g_object_unref(file);

	    }

	    gtk_tool_button_set_label(GTK_TOOL_BUTTON(w),name);

	    g_free(name);

	    gtk_widget_show(w);

	    gtk_toggle_tool_button_set_active(GTK_TOGGLE_TOOL_BUTTON(w),TRUE);

	}

	else

	{

	    gtk_widget_destroy(dialog);

	    dialog=gtk_message_dialog_new(GTK_WINDOW(w),

	      GTK_DIALOG_DESTROY_WITH_PARENT,GTK_MESSAGE_ERROR,

	      GTK_BUTTONS_CLOSE,"Error loading repository '%s'",path);

	    gtk_message_dialog_format_secondary_text(GTK_MESSAGE_DIALOG(dialog),

	      "%s",err->message);

	    gtk_dialog_run(GTK_DIALOG(dialog));

	    g_error_free(err);

	}

	g_free(path);

	show_busy_cursor(FALSE);

    }

    gtk_widget_destroy(dialog);

}

G_MODULE_EXPORT void on_scan_local_media(GtkWidget *widget)

{

    show_busy_cursor(TRUE);

    if (!local_media)

	local_media=plover_local_media_store_new();

    plover_local_media_store_scan(PLOVER_LOCAL_MEDIA_STORE(local_media));

    show_busy_cursor(FALSE);

}

G_MODULE_EXPORT void on_help_about(GtkWidget *widget)

{

    GtkWidget *w=GTK_WIDGET(gtk_builder_get_object(ui,"MainWindow"));

    gtk_show_about_dialog(GTK_WINDOW(w),"name",PACKAGE_NAME,

      "version",PACKAGE_VERSION,"comments","Application Manager",

      "copyright","Copyright © 2010, 2014 J. Ali Harlow",

      "logo-icon-name",LOGO_NAME,

      NULL);

}

G_MODULE_EXPORT void on_find_clicked(GtkButton *button)

{

    gchar *text;

    GtkWidget *w=GTK_WIDGET(gtk_builder_get_object(ui,"SearchEntry"));

    text=g_strdup(gtk_entry_get_text(GTK_ENTRY(w)));

    gtk_entry_set_text(GTK_ENTRY(w),"");

    gtk_entry_set_text(GTK_ENTRY(w),text);

    g_free(text);

}