/*
 * xed-plugins-engine.c
 * This file is part of xed
 *
 * Copyright (C) 2002-2005 Paolo Maggi
 *
 * 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 St, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

/*
 * Modified by the xed Team, 2002-2005. See the AUTHORS file for a
 * list of people on the xed Team.
 * See the ChangeLog files for a list of changes.
 *
 * $Id$
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <string.h>

#include <glib/gi18n.h>

#include "xed-plugins-engine.h"
#include "xed-plugin-info-priv.h"
#include "xed-plugin.h"
#include "xed-debug.h"
#include "xed-app.h"
#include "xed-prefs-manager.h"
#include "xed-plugin-loader.h"
#include "xed-object-module.h"
#include "xed-dirs.h"

#define XED_PLUGINS_ENGINE_BASE_KEY "/apps/xed/plugins"
#define XED_PLUGINS_ENGINE_KEY XED_PLUGINS_ENGINE_BASE_KEY "/active-plugins"

#define PLUGIN_EXT	".xed-plugin"
#define LOADER_EXT	G_MODULE_SUFFIX

typedef struct
{
	XedPluginLoader *loader;
	XedObjectModule *module;
} LoaderInfo;

/* Signals */
enum
{
	ACTIVATE_PLUGIN,
	DEACTIVATE_PLUGIN,
	LAST_SIGNAL
};

static guint signals[LAST_SIGNAL] = { 0 };

G_DEFINE_TYPE(XedPluginsEngine, xed_plugins_engine, G_TYPE_OBJECT)

struct _XedPluginsEnginePrivate
{
	GList *plugin_list;
	GHashTable *loaders;

	gboolean activate_from_prefs;
};

XedPluginsEngine *default_engine = NULL;

static void	xed_plugins_engine_activate_plugin_real (XedPluginsEngine *engine,
							   XedPluginInfo    *info);
static void	xed_plugins_engine_deactivate_plugin_real (XedPluginsEngine *engine,
							     XedPluginInfo    *info);

typedef gboolean (*LoadDirCallback)(XedPluginsEngine *engine, const gchar *filename, gpointer userdata);

static gboolean
load_dir_real (XedPluginsEngine *engine,
	       const gchar        *dir,
	       const gchar        *suffix,
	       LoadDirCallback     callback,
	       gpointer            userdata)
{
	GError *error = NULL;
	GDir *d;
	const gchar *dirent;
	gboolean ret = TRUE;

	g_return_val_if_fail (dir != NULL, TRUE);

	xed_debug_message (DEBUG_PLUGINS, "DIR: %s", dir);

	d = g_dir_open (dir, 0, &error);
	if (!d)
	{
		g_warning ("%s", error->message);
		g_error_free (error);
		return TRUE;
	}

	while ((dirent = g_dir_read_name (d)))
	{
		gchar *filename;

		if (!g_str_has_suffix (dirent, suffix))
			continue;

		filename = g_build_filename (dir, dirent, NULL);

		ret = callback (engine, filename, userdata);

		g_free (filename);

		if (!ret)
			break;
	}

	g_dir_close (d);
	return ret;
}

static gboolean
load_plugin_info (XedPluginsEngine *engine,
		  const gchar        *filename,
		  gpointer            userdata)
{
	XedPluginInfo *info;

	info = _xed_plugin_info_new (filename);

	if (info == NULL)
		return TRUE;

	/* If a plugin with this name has already been loaded
	 * drop this one (user plugins override system plugins) */
	if (xed_plugins_engine_get_plugin_info (engine, xed_plugin_info_get_module_name (info)) != NULL)
	{
		xed_debug_message (DEBUG_PLUGINS, "Two or more plugins named '%s'. "
				     "Only the first will be considered.\n",
				     xed_plugin_info_get_module_name (info));

		_xed_plugin_info_unref (info);

		return TRUE;
	}

	engine->priv->plugin_list = g_list_prepend (engine->priv->plugin_list, info);

	xed_debug_message (DEBUG_PLUGINS, "Plugin %s loaded", info->name);
	return TRUE;
}

static void
load_all_plugins (XedPluginsEngine *engine)
{
	gchar *plugin_dir;
	const gchar *pdirs_env = NULL;

	/* load user plugins */
	plugin_dir = xed_dirs_get_user_plugins_dir ();
	if (g_file_test (plugin_dir, G_FILE_TEST_IS_DIR))
	{
		load_dir_real (engine,
			       plugin_dir,
			       PLUGIN_EXT,
			       load_plugin_info,
			       NULL);

	}
	g_free (plugin_dir);

	/* load system plugins */
	pdirs_env = g_getenv ("XED_PLUGINS_PATH");

	xed_debug_message (DEBUG_PLUGINS, "XED_PLUGINS_PATH=%s", pdirs_env);

	if (pdirs_env != NULL)
	{
		gchar **pdirs;
		gint i;

		pdirs = g_strsplit (pdirs_env, G_SEARCHPATH_SEPARATOR_S, 0);

		for (i = 0; pdirs[i] != NULL; i++)
		{
			if (!load_dir_real (engine,
					    pdirs[i],
					    PLUGIN_EXT,
					    load_plugin_info,
					    NULL))
			{
				break;
			}
		}

		g_strfreev (pdirs);
	}
	else
	{
		plugin_dir = xed_dirs_get_xed_plugins_dir ();

		load_dir_real (engine,
			       plugin_dir,
			       PLUGIN_EXT,
			       load_plugin_info,
			       NULL);

		g_free (plugin_dir);
	}
}

static guint
hash_lowercase (gconstpointer data)
{
	gchar *lowercase;
	guint ret;

	lowercase = g_ascii_strdown ((const gchar *)data, -1);
	ret = g_str_hash (lowercase);
	g_free (lowercase);

	return ret;
}

static gboolean
equal_lowercase (gconstpointer a, gconstpointer b)
{
	return g_ascii_strcasecmp ((const gchar *)a, (const gchar *)b) == 0;
}

static void
loader_destroy (LoaderInfo *info)
{
	if (!info)
		return;

	if (info->loader)
		g_object_unref (info->loader);

	g_free (info);
}

static void
add_loader (XedPluginsEngine *engine,
	    const gchar        *loader_id,
	    XedObjectModule  *module)
{
	LoaderInfo *info;

	info = g_new (LoaderInfo, 1);
	info->loader = NULL;
	info->module = module;

	g_hash_table_insert (engine->priv->loaders, g_strdup (loader_id), info);
}

static void
xed_plugins_engine_init (XedPluginsEngine *engine)
{
	xed_debug (DEBUG_PLUGINS);

	if (!g_module_supported ())
	{
		g_warning ("xed is not able to initialize the plugins engine.");
		return;
	}

	engine->priv = G_TYPE_INSTANCE_GET_PRIVATE (engine,
						    XED_TYPE_PLUGINS_ENGINE,
						    XedPluginsEnginePrivate);

	load_all_plugins (engine);

	/* make sure that the first reactivation will read active plugins
	   from the prefs */
	engine->priv->activate_from_prefs = TRUE;

	/* mapping from loadername -> loader object */
	engine->priv->loaders = g_hash_table_new_full (hash_lowercase,
						       equal_lowercase,
						       (GDestroyNotify)g_free,
						       (GDestroyNotify)loader_destroy);
}

static void
loader_garbage_collect (const char *id, LoaderInfo *info)
{
	if (info->loader)
		xed_plugin_loader_garbage_collect (info->loader);
}

void
xed_plugins_engine_garbage_collect (XedPluginsEngine *engine)
{
	g_hash_table_foreach (engine->priv->loaders,
			      (GHFunc) loader_garbage_collect,
			      NULL);
}

static void
xed_plugins_engine_finalize (GObject *object)
{
	XedPluginsEngine *engine = XED_PLUGINS_ENGINE (object);
	GList *item;

	xed_debug (DEBUG_PLUGINS);

	/* Firs deactivate all plugins */
	for (item = engine->priv->plugin_list; item; item = item->next)
	{
		XedPluginInfo *info = XED_PLUGIN_INFO (item->data);

		if (xed_plugin_info_is_active (info))
			xed_plugins_engine_deactivate_plugin_real (engine, info);
	}

	/* unref the loaders */
	g_hash_table_destroy (engine->priv->loaders);

	/* and finally free the infos */
	for (item = engine->priv->plugin_list; item; item = item->next)
	{
		XedPluginInfo *info = XED_PLUGIN_INFO (item->data);

		_xed_plugin_info_unref (info);
	}

	g_list_free (engine->priv->plugin_list);

	G_OBJECT_CLASS (xed_plugins_engine_parent_class)->finalize (object);
}

static void
xed_plugins_engine_class_init (XedPluginsEngineClass *klass)
{
	GType the_type = G_TYPE_FROM_CLASS (klass);
	GObjectClass *object_class = G_OBJECT_CLASS (klass);

	object_class->finalize = xed_plugins_engine_finalize;
	klass->activate_plugin = xed_plugins_engine_activate_plugin_real;
	klass->deactivate_plugin = xed_plugins_engine_deactivate_plugin_real;

	signals[ACTIVATE_PLUGIN] =
		g_signal_new ("activate-plugin",
			      the_type,
			      G_SIGNAL_RUN_LAST,
			      G_STRUCT_OFFSET (XedPluginsEngineClass, activate_plugin),
			      NULL, NULL,
			      g_cclosure_marshal_VOID__BOXED,
			      G_TYPE_NONE,
			      1,
			      XED_TYPE_PLUGIN_INFO | G_SIGNAL_TYPE_STATIC_SCOPE);

	signals[DEACTIVATE_PLUGIN] =
		g_signal_new ("deactivate-plugin",
			      the_type,
			      G_SIGNAL_RUN_LAST,
			      G_STRUCT_OFFSET (XedPluginsEngineClass, deactivate_plugin),
			      NULL, NULL,
			      g_cclosure_marshal_VOID__BOXED,
			      G_TYPE_NONE,
			      1,
			      XED_TYPE_PLUGIN_INFO | G_SIGNAL_TYPE_STATIC_SCOPE);

	g_type_class_add_private (klass, sizeof (XedPluginsEnginePrivate));
}

static gboolean
load_loader (XedPluginsEngine *engine,
	     const gchar        *filename,
	     gpointer		 data)
{
	XedObjectModule *module;
	gchar *base;
	gchar *path;
	const gchar *id;
	GType type;

	/* try to load in the module */
	path = g_path_get_dirname (filename);
	base = g_path_get_basename (filename);

	/* for now they are all resident */
	module = xed_object_module_new (base,
					  path,
					  "register_xed_plugin_loader",
					  TRUE);

	g_free (base);
	g_free (path);

	/* make sure to load the type definition */
	if (!g_type_module_use (G_TYPE_MODULE (module)))
	{
		g_object_unref (module);
		g_warning ("Plugin loader module `%s' could not be loaded", filename);

		return TRUE;
	}

	/* get the exported type and check the name as exported by the
	 * loader interface */
	type = xed_object_module_get_object_type (module);
	id = xed_plugin_loader_type_get_id (type);

	add_loader (engine, id, module);
	g_type_module_unuse (G_TYPE_MODULE (module));

	return TRUE;
}

static void
ensure_loader (LoaderInfo *info)
{
	if (info->loader == NULL && info->module != NULL)
	{
		/* create a new loader object */
		XedPluginLoader *loader;
		loader = (XedPluginLoader *)xed_object_module_new_object (info->module, NULL);

		if (loader == NULL || !XED_IS_PLUGIN_LOADER (loader))
		{
			g_warning ("Loader object is not a valid XedPluginLoader instance");

			if (loader != NULL && G_IS_OBJECT (loader))
				g_object_unref (loader);
		}
		else
		{
			info->loader = loader;
		}
	}
}

static XedPluginLoader *
get_plugin_loader (XedPluginsEngine *engine, XedPluginInfo *info)
{
	const gchar *loader_id;
	LoaderInfo *loader_info;

	loader_id = info->loader;

	loader_info = (LoaderInfo *)g_hash_table_lookup (
			engine->priv->loaders,
			loader_id);

	if (loader_info == NULL)
	{
		gchar *loader_dir;

		loader_dir = xed_dirs_get_xed_plugin_loaders_dir ();

		/* loader could not be found in the hash, try to find it by
		   scanning */
		load_dir_real (engine,
			       loader_dir,
			       LOADER_EXT,
			       (LoadDirCallback)load_loader,
			       NULL);
		g_free (loader_dir);

		loader_info = (LoaderInfo *)g_hash_table_lookup (
				engine->priv->loaders,
				loader_id);
	}

	if (loader_info == NULL)
	{
		/* cache non-existent so we don't scan again */
		add_loader (engine, loader_id, NULL);
		return NULL;
	}

	ensure_loader (loader_info);
	return loader_info->loader;
}

XedPluginsEngine *
xed_plugins_engine_get_default (void)
{
	if (default_engine != NULL)
		return default_engine;

	default_engine = XED_PLUGINS_ENGINE (g_object_new (XED_TYPE_PLUGINS_ENGINE, NULL));
	g_object_add_weak_pointer (G_OBJECT (default_engine),
				   (gpointer) &default_engine);
	return default_engine;
}

const GList *
xed_plugins_engine_get_plugin_list (XedPluginsEngine *engine)
{
	xed_debug (DEBUG_PLUGINS);

	return engine->priv->plugin_list;
}

static gint
compare_plugin_info_and_name (XedPluginInfo *info,
			      const gchar *module_name)
{
	return strcmp (xed_plugin_info_get_module_name (info), module_name);
}

XedPluginInfo *
xed_plugins_engine_get_plugin_info (XedPluginsEngine *engine,
				      const gchar        *name)
{
	GList *l = g_list_find_custom (engine->priv->plugin_list,
				       name,
				       (GCompareFunc) compare_plugin_info_and_name);

	return l == NULL ? NULL : (XedPluginInfo *) l->data;
}

static void
save_active_plugin_list (XedPluginsEngine *engine)
{
	GSList *active_plugins = NULL;
	GList *l;

	for (l = engine->priv->plugin_list; l != NULL; l = l->next)
	{
		XedPluginInfo *info = (XedPluginInfo *) l->data;

		if (xed_plugin_info_is_active (info))
		{
			active_plugins = g_slist_prepend (active_plugins,
							  (gpointer)xed_plugin_info_get_module_name (info));
		}
	}

	xed_prefs_manager_set_active_plugins (active_plugins);

	g_slist_free (active_plugins);
}

static gboolean
load_plugin (XedPluginsEngine *engine,
	     XedPluginInfo    *info)
{
	XedPluginLoader *loader;
	gchar *path;

	if (xed_plugin_info_is_active (info))
		return TRUE;

	if (!xed_plugin_info_is_available (info))
		return FALSE;

	loader = get_plugin_loader (engine, info);

	if (loader == NULL)
	{
		g_warning ("Could not find loader `%s' for plugin `%s'", info->loader, info->name);
		info->available = FALSE;
		return FALSE;
	}

	path = g_path_get_dirname (info->file);
	g_return_val_if_fail (path != NULL, FALSE);

	info->plugin = xed_plugin_loader_load (loader, info, path);

	g_free (path);

	if (info->plugin == NULL)
	{
		g_warning ("Error loading plugin '%s'", info->name);
		info->available = FALSE;
		return FALSE;
	}

	return TRUE;
}

static void
xed_plugins_engine_activate_plugin_real (XedPluginsEngine *engine,
					   XedPluginInfo *info)
{
	const GList *wins;

	if (!load_plugin (engine, info))
		return;

	for (wins = xed_app_get_windows (xed_app_get_default ());
	     wins != NULL;
	     wins = wins->next)
	{
		xed_plugin_activate (info->plugin, XED_WINDOW (wins->data));
	}
}

gboolean
xed_plugins_engine_activate_plugin (XedPluginsEngine *engine,
				      XedPluginInfo    *info)
{
	xed_debug (DEBUG_PLUGINS);

	g_return_val_if_fail (info != NULL, FALSE);

	if (!xed_plugin_info_is_available (info))
		return FALSE;

	if (xed_plugin_info_is_active (info))
		return TRUE;

	g_signal_emit (engine, signals[ACTIVATE_PLUGIN], 0, info);

	if (xed_plugin_info_is_active (info))
		save_active_plugin_list (engine);

	return xed_plugin_info_is_active (info);
}

static void
call_plugin_deactivate (XedPlugin *plugin,
			XedWindow *window)
{
	xed_plugin_deactivate (plugin, window);

	/* ensure update of ui manager, because we suspect it does something
	   with expected static strings in the type module (when unloaded the
	   strings don't exist anymore, and ui manager updates in an idle
	   func) */
	gtk_ui_manager_ensure_update (xed_window_get_ui_manager (window));
}

static void
xed_plugins_engine_deactivate_plugin_real (XedPluginsEngine *engine,
					     XedPluginInfo *info)
{
	const GList *wins;
	XedPluginLoader *loader;

	if (!xed_plugin_info_is_active (info) ||
	    !xed_plugin_info_is_available (info))
		return;

	for (wins = xed_app_get_windows (xed_app_get_default ());
	     wins != NULL;
	     wins = wins->next)
	{
		call_plugin_deactivate (info->plugin, XED_WINDOW (wins->data));
	}

	/* first unref the plugin (the loader still has one) */
	g_object_unref (info->plugin);

	/* find the loader and tell it to gc and unload the plugin */
	loader = get_plugin_loader (engine, info);

	xed_plugin_loader_garbage_collect (loader);
	xed_plugin_loader_unload (loader, info);

	info->plugin = NULL;
}

gboolean
xed_plugins_engine_deactivate_plugin (XedPluginsEngine *engine,
					XedPluginInfo    *info)
{
	xed_debug (DEBUG_PLUGINS);

	g_return_val_if_fail (info != NULL, FALSE);

	if (!xed_plugin_info_is_active (info))
		return TRUE;

	g_signal_emit (engine, signals[DEACTIVATE_PLUGIN], 0, info);
	if (!xed_plugin_info_is_active (info))
		save_active_plugin_list (engine);

	return !xed_plugin_info_is_active (info);
}

void
xed_plugins_engine_activate_plugins (XedPluginsEngine *engine,
					XedWindow        *window)
{
	GSList *active_plugins = NULL;
	GList *pl;

	xed_debug (DEBUG_PLUGINS);

	g_return_if_fail (XED_IS_PLUGINS_ENGINE (engine));
	g_return_if_fail (XED_IS_WINDOW (window));

	/* the first time, we get the 'active' plugins from GSettings */
	if (engine->priv->activate_from_prefs)
	{
		active_plugins = xed_prefs_manager_get_active_plugins ();
	}

	for (pl = engine->priv->plugin_list; pl; pl = pl->next)
	{
		XedPluginInfo *info = (XedPluginInfo*)pl->data;

		if (engine->priv->activate_from_prefs &&
		    g_slist_find_custom (active_plugins,
					 xed_plugin_info_get_module_name (info),
					 (GCompareFunc)strcmp) == NULL)
			continue;

		/* If plugin is not active, don't try to activate/load it */
		if (!engine->priv->activate_from_prefs &&
		    !xed_plugin_info_is_active (info))
			continue;

		if (load_plugin (engine, info))
			xed_plugin_activate (info->plugin,
					       window);
	}

	if (engine->priv->activate_from_prefs)
	{
		g_slist_foreach (active_plugins, (GFunc) g_free, NULL);
		g_slist_free (active_plugins);
		engine->priv->activate_from_prefs = FALSE;
	}

	xed_debug_message (DEBUG_PLUGINS, "End");

	/* also call update_ui after activation */
	xed_plugins_engine_update_plugins_ui (engine, window);
}

void
xed_plugins_engine_deactivate_plugins (XedPluginsEngine *engine,
					  XedWindow        *window)
{
	GList *pl;

	xed_debug (DEBUG_PLUGINS);

	g_return_if_fail (XED_IS_PLUGINS_ENGINE (engine));
	g_return_if_fail (XED_IS_WINDOW (window));

	for (pl = engine->priv->plugin_list; pl; pl = pl->next)
	{
		XedPluginInfo *info = (XedPluginInfo*)pl->data;

		/* check if the plugin is actually active */
		if (!xed_plugin_info_is_active (info))
			continue;

		/* call deactivate for the plugin for this window */
		xed_plugin_deactivate (info->plugin, window);
	}

	xed_debug_message (DEBUG_PLUGINS, "End");
}

void
xed_plugins_engine_update_plugins_ui (XedPluginsEngine *engine,
					 XedWindow        *window)
{
	GList *pl;

	xed_debug (DEBUG_PLUGINS);

	g_return_if_fail (XED_IS_PLUGINS_ENGINE (engine));
	g_return_if_fail (XED_IS_WINDOW (window));

	/* call update_ui for all active plugins */
	for (pl = engine->priv->plugin_list; pl; pl = pl->next)
	{
		XedPluginInfo *info = (XedPluginInfo*)pl->data;

		if (!xed_plugin_info_is_active (info))
			continue;

	       	xed_debug_message (DEBUG_PLUGINS, "Updating UI of %s", info->name);
		xed_plugin_update_ui (info->plugin, window);
	}
}

void
xed_plugins_engine_configure_plugin (XedPluginsEngine *engine,
				       XedPluginInfo    *info,
				       GtkWindow          *parent)
{
	GtkWidget *conf_dlg;

	GtkWindowGroup *wg;

	xed_debug (DEBUG_PLUGINS);

	g_return_if_fail (info != NULL);

	conf_dlg = xed_plugin_create_configure_dialog (info->plugin);
	g_return_if_fail (conf_dlg != NULL);
	gtk_window_set_transient_for (GTK_WINDOW (conf_dlg),
				      parent);

	wg = gtk_window_get_group (parent);
	if (wg == NULL)
	{
		wg = gtk_window_group_new ();
		gtk_window_group_add_window (wg, parent);
	}

	gtk_window_group_add_window (wg,
				     GTK_WINDOW (conf_dlg));

	gtk_window_set_modal (GTK_WINDOW (conf_dlg), TRUE);
	gtk_widget_show (conf_dlg);
}

void
xed_plugins_engine_active_plugins_changed (XedPluginsEngine *engine)
{
	gboolean to_activate;
	GSList *active_plugins;
	GList *pl;

	xed_debug (DEBUG_PLUGINS);

	active_plugins = xed_prefs_manager_get_active_plugins ();

	for (pl = engine->priv->plugin_list; pl; pl = pl->next)
	{
		XedPluginInfo *info = (XedPluginInfo*)pl->data;

		if (!xed_plugin_info_is_available (info))
			continue;

		to_activate = (g_slist_find_custom (active_plugins,
						    xed_plugin_info_get_module_name (info),
						    (GCompareFunc)strcmp) != NULL);

		if (!xed_plugin_info_is_active (info) && to_activate)
			g_signal_emit (engine, signals[ACTIVATE_PLUGIN], 0, info);
		else if (xed_plugin_info_is_active (info) && !to_activate)
			g_signal_emit (engine, signals[DEACTIVATE_PLUGIN], 0, info);
	}

	g_slist_foreach (active_plugins, (GFunc) g_free, NULL);
	g_slist_free (active_plugins);
}

void
xed_plugins_engine_rescan_plugins (XedPluginsEngine *engine)
{
	xed_debug (DEBUG_PLUGINS);

	load_all_plugins (engine);
}