/*

 * 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

#include <glib.h>

#include <gtk/gtk.h>

#include <plover-gtk/packagestore.h>

#include <plover-gtk/packagefilestore.h>

#include "app-manager.h"

#ifdef WIN32

static BOOL CALLBACK plover_applications_visible_callback(HMODULE module,

  const char *type,char *name,LONG_PTR param)

{

    gboolean *visible=(void *)param;

    if (!IS_INTRESOURCE(name) && !strcmp(name,"MAINICON"))

	*visible=TRUE;

    return !*visible;

}

#endif

static gboolean plover_applications_visible_func(GtkTreeModel *model,

  GtkTreeIter *iter,gpointer data)

{

    /* Visible if row is non-empty and package contains a .desktop file

     * in /usr/share/applications (UNIX) or package contains a .exe file

     * which has a default application icon (MS-Windows).

     */

    PloverPackage *package;

    GtkTreeModel *file_store;

    GtkTreeIter fi;

    gchar *name;

#ifdef WIN32

    HMODULE module;

    DWORD flags=

#ifdef LOAD_LIBRARY_AS_IMAGE_RESOURCE

      LOAD_LIBRARY_AS_IMAGE_RESOURCE|

#endif

      LOAD_LIBRARY_AS_DATAFILE;

#else

    gchar *dir;

#endif

    gboolean visible=FALSE;

    gtk_tree_model_get(model,iter,PLOVER_PACKAGE_STORE_OBJ_COLUMN,&package,-1);

    if (package)

    {

	file_store=

	  GTK_TREE_MODEL(plover_package_file_store_new_from_package(package));

	if (gtk_tree_model_get_iter_first(file_store,&fi))

	{

	    do

	    {

		gtk_tree_model_get(file_store,&fi,

		  PLOVER_PACKAGE_FILE_STORE_NAME_COLUMN,&name,-1);

#ifdef WIN32

		if (g_str_has_suffix(name,".exe"))

		{

		    module=LoadLibraryExA(name,NULL,flags);

		    if (module)

		    {

			(void)EnumResourceNamesA(module,RT_ICON,

			  plover_applications_visible_callback,

			  (LONG_PTR)&visible);

			if (!visible)

			    (void)EnumResourceNamesA(module,RT_GROUP_ICON,

			      plover_applications_visible_callback,

			      (LONG_PTR)&visible);

			FreeLibrary(module);

		    }

		}

#else

		dir=g_path_get_dirname(name);

		if (!strcmp(dir,"/usr/share/applications") &&

		  g_str_has_suffix(name,".desktop"))

		    visible=TRUE;

		g_free(dir);

#endif

		g_free(name);

	    } while(!visible && gtk_tree_model_iter_next(file_store,&fi));

	}

	g_object_unref(file_store);

    }

    g_object_unref(package);

    return visible;

}

GtkTreeModel *plover_applications_model_new(GtkTreeModel *installed)

{

    GtkTreeModel *model;

    model=gtk_tree_model_filter_new(installed,NULL);

    gtk_tree_model_filter_set_visible_func(GTK_TREE_MODEL_FILTER(model),

      plover_applications_visible_func,NULL,NULL);

    return model;

}