#include "xed-file-browser-messages.h"
#include "xed-file-browser-store.h"
#include <xed/xed-message.h>

#define MESSAGE_OBJECT_PATH "/plugins/filebrowser"
#define WINDOW_DATA_KEY     "XedFileBrowserMessagesWindowData"

#define BUS_CONNECT(bus, name, data) xed_message_bus_connect(bus, MESSAGE_OBJECT_PATH, #name, (XedMessageCallback)  message_##name##_cb, data, NULL)

typedef struct
{
    XedWindow *window;
    XedMessage *message;
} MessageCacheData;

typedef struct
{
    guint row_inserted_id;
    guint row_deleted_id;
    guint root_changed_id;
    guint begin_loading_id;
    guint end_loading_id;

    GList *merge_ids;
    GtkActionGroup *merged_actions;

    XedMessageBus *bus;
    XedFileBrowserWidget *widget;
    GHashTable *row_tracking;

    GHashTable *filters;
} WindowData;

typedef struct
{
    gulong id;

    XedWindow *window;
    XedMessage *message;
} FilterData;

static WindowData *
window_data_new (XedWindow            *window,
                 XedFileBrowserWidget *widget)
{
    WindowData *data = g_slice_new (WindowData);
    GtkUIManager *manager;
    GList *groups;

    data->bus = xed_window_get_message_bus (window);
    data->widget = widget;
    data->row_tracking = g_hash_table_new_full (g_str_hash,
                                                g_str_equal,
                                                (GDestroyNotify)g_free,
                                                (GDestroyNotify)gtk_tree_row_reference_free);

    data->filters = g_hash_table_new_full (g_str_hash,
                                           g_str_equal,
                                           (GDestroyNotify)g_free,
                                           NULL);

    manager = xed_file_browser_widget_get_ui_manager (widget);

    data->merge_ids = NULL;
    data->merged_actions = gtk_action_group_new ("MessageMergedActions");

    groups = gtk_ui_manager_get_action_groups (manager);
    gtk_ui_manager_insert_action_group (manager, data->merged_actions, g_list_length (groups));

    g_object_set_data (G_OBJECT (window), WINDOW_DATA_KEY, data);

    return data;
}

static WindowData *
get_window_data (XedWindow * window)
{
    return (WindowData *) (g_object_get_data (G_OBJECT (window), WINDOW_DATA_KEY));
}

static void
window_data_free (XedWindow *window)
{
    WindowData *data = get_window_data (window);
    GtkUIManager *manager;
    GList *item;

    g_hash_table_destroy (data->row_tracking);
    g_hash_table_destroy (data->filters);

    manager = xed_file_browser_widget_get_ui_manager (data->widget);
    gtk_ui_manager_remove_action_group (manager, data->merged_actions);

    for (item = data->merge_ids; item; item = item->next)
    {
        gtk_ui_manager_remove_ui (manager, GPOINTER_TO_INT (item->data));
    }

    g_list_free (data->merge_ids);
    g_object_unref (data->merged_actions);

    g_slice_free (WindowData, data);

    g_object_set_data (G_OBJECT (window), WINDOW_DATA_KEY, NULL);
}

static FilterData *
filter_data_new (XedWindow  *window,
                 XedMessage *message)
{
    FilterData *data = g_slice_new (FilterData);
    WindowData *wdata;

    data->window = window;
    data->id = 0;
    data->message = message;

    wdata = get_window_data (window);

    g_hash_table_insert (wdata->filters, xed_message_type_identifier (xed_message_get_object_path (message),
                                                                      xed_message_get_method (message)), data);

    return data;
}

static void
filter_data_free (FilterData *data)
{
    WindowData *wdata = get_window_data (data->window);
    gchar *identifier;

    identifier = xed_message_type_identifier (xed_message_get_object_path (data->message),
                                              xed_message_get_method (data->message));

    g_hash_table_remove (wdata->filters, identifier);
    g_free (identifier);

    g_object_unref (data->message);
    g_slice_free (FilterData, data);
}

static GtkTreePath *
track_row_lookup (WindowData  *data,
                  const gchar *id)
{
    GtkTreeRowReference *ref;

    ref = (GtkTreeRowReference *)g_hash_table_lookup (data->row_tracking, id);

    if (!ref)
    {
        return NULL;
    }

    return gtk_tree_row_reference_get_path (ref);
}

static void
message_cache_data_free (MessageCacheData *data)
{
    g_object_unref (data->message);
    g_slice_free (MessageCacheData, data);
}

static MessageCacheData *
message_cache_data_new (XedWindow  *window,
                        XedMessage *message)
{
    MessageCacheData *data = g_slice_new (MessageCacheData);

    data->window = window;
    data->message = message;

    return data;
}

static void
message_get_root_cb (XedMessageBus *bus,
                     XedMessage    *message,
                     WindowData    *data)
{
    XedFileBrowserStore *store;
    GFile *location;

    store = xed_file_browser_widget_get_browser_store (data->widget);
    location = xed_file_browser_store_get_virtual_root (store);

    if (location)
    {
        xed_message_set (message, "location", location, NULL);
        g_object_unref (location);
    }
}

static void
message_set_root_cb (XedMessageBus *bus,
                     XedMessage    *message,
                     WindowData    *data)
{
    GFile *root;
    GFile *virtual = NULL;

    xed_message_get (message, "location", &root, NULL);

    if (!root)
    {
        return;
    }

    if (xed_message_has_key (message, "virtual"))
    {
        xed_message_get (message, "virtual", &virtual, NULL);
    }

    if (virtual)
    {
        xed_file_browser_widget_set_root_and_virtual_root (data->widget, root, virtual);
    }
    else
    {
        xed_file_browser_widget_set_root (data->widget, root, TRUE);
    }
}

static void
message_set_emblem_cb (XedMessageBus *bus,
                       XedMessage    *message,
                       WindowData    *data)
{
    gchar *id = NULL;
    gchar *emblem = NULL;
    GtkTreePath *path;
    XedFileBrowserStore *store;

    xed_message_get (message, "id", &id, "emblem", &emblem, NULL);

    if (!id || !emblem)
    {
        g_free (id);
        g_free (emblem);

        return;
    }

    path = track_row_lookup (data, id);

    if (path != NULL)
    {
        GError *error = NULL;
        GdkPixbuf *pixbuf;

        pixbuf = gtk_icon_theme_load_icon (gtk_icon_theme_get_default (), emblem, 10, 0, &error);

        if (pixbuf)
        {
            GValue value = { 0, };
            GtkTreeIter iter;

            store = xed_file_browser_widget_get_browser_store (data->widget);

            if (gtk_tree_model_get_iter (GTK_TREE_MODEL (store), &iter, path))
            {
                g_value_init (&value, GDK_TYPE_PIXBUF);
                g_value_set_object (&value, pixbuf);

                xed_file_browser_store_set_value (store, &iter, XED_FILE_BROWSER_STORE_COLUMN_EMBLEM, &value);

                g_value_unset (&value);
            }

            g_object_unref (pixbuf);
        }

        if (error)
            g_error_free (error);
    }

    g_free (id);
    g_free (emblem);
}

static gchar *
item_id (const gchar *path,
         GFile       *location)
{
    gchar *uri;
    gchar *id;

    uri = g_file_get_uri (location);
    id = g_strconcat (path, "::", uri, NULL);
    g_free (uri);

    return id;
}

static gchar *
track_row (WindowData          *data,
           XedFileBrowserStore *store,
           GtkTreePath         *path,
           GFile               *location)
{
    GtkTreeRowReference *ref;
    gchar *id;
    gchar *pathstr;

    pathstr = gtk_tree_path_to_string (path);
    id = item_id (pathstr, location);

    ref = gtk_tree_row_reference_new (GTK_TREE_MODEL (store), path);
    g_hash_table_insert (data->row_tracking, g_strdup (id), ref);

    g_free (pathstr);

    return id;
}

static void
set_item_message (WindowData  *data,
                  GtkTreeIter *iter,
                  GtkTreePath *path,
                  XedMessage  *message)
{
    XedFileBrowserStore *store;
    GFile *location;
    guint flags = 0;
    gchar *track_id;

    store = xed_file_browser_widget_get_browser_store (data->widget);

    gtk_tree_model_get (GTK_TREE_MODEL (store), iter,
                        XED_FILE_BROWSER_STORE_COLUMN_LOCATION, &location,
                        XED_FILE_BROWSER_STORE_COLUMN_FLAGS, &flags,
                        -1);

    if (!location)
    {
        return;
    }

    if (path && gtk_tree_path_get_depth (path) != 0)
    {
        track_id = track_row (data, store, path, location);
    }
    else
    {
        track_id = NULL;
    }

    xed_message_set (message, "id", track_id, "location", location, NULL);

    if (xed_message_has_key (message, "is_directory"))
    {
        xed_message_set (message, "is_directory", FILE_IS_DIR (flags), NULL);
    }

    g_free (track_id);
}

static gboolean
custom_message_filter_func (XedFileBrowserWidget *widget,
                            XedFileBrowserStore  *store,
                            GtkTreeIter          *iter,
                            FilterData           *data)
{
    WindowData *wdata = get_window_data (data->window);
    GFile *location;
    guint flags = 0;
    gboolean filter = FALSE;
    GtkTreePath *path;

    gtk_tree_model_get (GTK_TREE_MODEL (store), iter,
                        XED_FILE_BROWSER_STORE_COLUMN_LOCATION, &location,
                        XED_FILE_BROWSER_STORE_COLUMN_FLAGS, &flags,
                        -1);

    if (!location || FILE_IS_DUMMY (flags))
    {
        return FALSE;
    }

    path = gtk_tree_model_get_path (GTK_TREE_MODEL (store), iter);
    set_item_message (wdata, iter, path, data->message);
    gtk_tree_path_free (path);

    xed_message_set (data->message, "filter", filter, NULL);

    xed_message_bus_send_message_sync (wdata->bus, data->message);
    xed_message_get (data->message, "filter", &filter, NULL);

    return !filter;
}

static void
message_add_filter_cb (XedMessageBus *bus,
                       XedMessage    *message,
                       XedWindow     *window)
{
    gchar *object_path = NULL;
    gchar *method = NULL;
    gulong id;
    XedMessageType *message_type;
    XedMessage *cbmessage;
    FilterData *filter_data;
    WindowData *data = get_window_data (window);

    xed_message_get (message, "object_path", &object_path, "method", &method, NULL);

    // Check if there exists such a 'callback' message
    if (!object_path || !method)
    {
        g_free (object_path);
        g_free (method);

        return;
    }

    message_type = xed_message_bus_lookup (bus, object_path, method);

    if (!message_type)
    {
        g_free (object_path);
        g_free (method);

        return;
    }

    // Check if the message type has the correct arguments
    if (xed_message_type_lookup (message_type, "id") != G_TYPE_STRING ||
        xed_message_type_lookup (message_type, "location") != G_TYPE_FILE ||
        xed_message_type_lookup (message_type, "is_directory") != G_TYPE_BOOLEAN ||
        xed_message_type_lookup (message_type, "filter") != G_TYPE_BOOLEAN)
    {
        return;
    }

    cbmessage = xed_message_type_instantiate (message_type,
                                             "id", NULL,
                                             "location", NULL,
                                             "is_directory", FALSE,
                                             "filter", FALSE,
                                             NULL);

    // Register the custom filter on the widget
    filter_data = filter_data_new (window, cbmessage);
    id = xed_file_browser_widget_add_filter (data->widget,
                                             (XedFileBrowserWidgetFilterFunc)custom_message_filter_func,
                                             filter_data,
                                             (GDestroyNotify)filter_data_free);

    filter_data->id = id;
}

static void
message_remove_filter_cb (XedMessageBus *bus,
                          XedMessage    *message,
                          WindowData    *data)
{
    gulong id = 0;

    xed_message_get (message, "id", &id, NULL);

    if (!id)
    {
        return;
    }

    xed_file_browser_widget_remove_filter (data->widget, id);
}

static void
message_up_cb (XedMessageBus *bus,
               XedMessage    *message,
               WindowData    *data)
{
    XedFileBrowserStore *store = xed_file_browser_widget_get_browser_store (data->widget);

    xed_file_browser_store_set_virtual_root_up (store);
}

static void
message_history_back_cb (XedMessageBus *bus,
                         XedMessage    *message,
                         WindowData    *data)
{
    xed_file_browser_widget_history_back (data->widget);
}

static void
message_history_forward_cb (XedMessageBus *bus,
                            XedMessage    *message,
                            WindowData    *data)
{
    xed_file_browser_widget_history_forward (data->widget);
}

static void
message_refresh_cb (XedMessageBus *bus,
                    XedMessage    *message,
                    WindowData    *data)
{
    xed_file_browser_widget_refresh (data->widget);
}

static void
message_set_show_hidden_cb (XedMessageBus *bus,
                            XedMessage    *message,
                            WindowData    *data)
{
    gboolean active = FALSE;
    XedFileBrowserStore *store;
    XedFileBrowserStoreFilterMode mode;

    xed_message_get (message, "active", &active, NULL);

    store = xed_file_browser_widget_get_browser_store (data->widget);
    mode = xed_file_browser_store_get_filter_mode (store);

    if (active)
    {
        mode &= ~XED_FILE_BROWSER_STORE_FILTER_MODE_HIDE_HIDDEN;
    }
    else
    {
        mode |= XED_FILE_BROWSER_STORE_FILTER_MODE_HIDE_HIDDEN;
    }

    xed_file_browser_store_set_filter_mode (store, mode);
}

static void
message_set_show_binary_cb (XedMessageBus *bus,
                            XedMessage    *message,
                            WindowData    *data)
{
    gboolean active = FALSE;
    XedFileBrowserStore *store;
    XedFileBrowserStoreFilterMode mode;

    xed_message_get (message, "active", &active, NULL);

    store = xed_file_browser_widget_get_browser_store (data->widget);
    mode = xed_file_browser_store_get_filter_mode (store);

    if (active)
    {
        mode &= ~XED_FILE_BROWSER_STORE_FILTER_MODE_HIDE_BINARY;
    }
    else
    {
        mode |= XED_FILE_BROWSER_STORE_FILTER_MODE_HIDE_BINARY;
    }

    xed_file_browser_store_set_filter_mode (store, mode);
}

static void
message_show_bookmarks_cb (XedMessageBus *bus,
                           XedMessage    *message,
                           WindowData    *data)
{
    xed_file_browser_widget_show_bookmarks (data->widget);
}

static void
message_show_files_cb (XedMessageBus *bus,
                       XedMessage    *message,
                       WindowData    *data)
{
    xed_file_browser_widget_show_files (data->widget);
}

static void
message_add_context_item_cb (XedMessageBus *bus,
                             XedMessage    *message,
                             WindowData    *data)
{
    GtkAction *action = NULL;
    gchar *path = NULL;
    gchar *name;
    GtkUIManager *manager;
    guint merge_id;

    xed_message_get (message, "action", &action, "path", &path, NULL);

    if (!action || !path)
    {
        if (action)
        {
            g_object_unref (action);
        }

        g_free (path);
        return;
    }

    gtk_action_group_add_action (data->merged_actions, action);
    manager = xed_file_browser_widget_get_ui_manager (data->widget);
    name = g_strconcat (gtk_action_get_name (action), "MenuItem", NULL);
    merge_id = gtk_ui_manager_new_merge_id (manager);

    gtk_ui_manager_add_ui (manager,
                           merge_id,
                           path,
                           name,
                           gtk_action_get_name (action),
                           GTK_UI_MANAGER_AUTO,
                           FALSE);

    if (gtk_ui_manager_get_widget (manager, path))
    {
        data->merge_ids = g_list_prepend (data->merge_ids, GINT_TO_POINTER (merge_id));
        xed_message_set (message, "id", merge_id, NULL);
    }
    else
    {
        xed_message_set (message, "id", 0, NULL);
    }

    g_object_unref (action);
    g_free (path);
    g_free (name);
}

static void
message_remove_context_item_cb (XedMessageBus *bus,
                                XedMessage    *message,
                                WindowData    *data)
{
    guint merge_id = 0;
    GtkUIManager *manager;

    xed_message_get (message, "id", &merge_id, NULL);

    if (merge_id == 0)
    {
        return;
    }

    manager = xed_file_browser_widget_get_ui_manager (data->widget);

    data->merge_ids = g_list_remove (data->merge_ids, GINT_TO_POINTER (merge_id));
    gtk_ui_manager_remove_ui (manager, merge_id);
}

static void
message_get_view_cb (XedMessageBus *bus,
                     XedMessage    *message,
                     WindowData    *data)
{
    XedFileBrowserView *view;
    view = xed_file_browser_widget_get_browser_view (data->widget);

    xed_message_set (message, "view", view, NULL);
}

static void
register_methods (XedWindow            *window,
                  XedFileBrowserWidget *widget)
{
    XedMessageBus *bus = xed_window_get_message_bus (window);
    WindowData *data = get_window_data (window);

    /* Register method calls */
    xed_message_bus_register (bus,
                              MESSAGE_OBJECT_PATH, "get_root",
                              1,
                              "location", G_TYPE_FILE,
                              NULL);

    xed_message_bus_register (bus,
                              MESSAGE_OBJECT_PATH, "set_root",
                              1,
                              "location", G_TYPE_FILE,
                              "virtual", G_TYPE_STRING,
                              NULL);

    xed_message_bus_register (bus,
                              MESSAGE_OBJECT_PATH, "set_emblem",
                              0,
                              "id", G_TYPE_STRING,
                              "emblem", G_TYPE_STRING,
                              NULL);

    xed_message_bus_register (bus,
                              MESSAGE_OBJECT_PATH, "add_filter",
                              1,
                              "object_path", G_TYPE_STRING,
                              "method", G_TYPE_STRING,
                              "id", G_TYPE_ULONG,
                              NULL);

    xed_message_bus_register (bus,
                              MESSAGE_OBJECT_PATH, "remove_filter",
                              0,
                              "id", G_TYPE_ULONG,
                              NULL);

    xed_message_bus_register (bus,
                              MESSAGE_OBJECT_PATH, "add_context_item",
                              1,
                              "action", GTK_TYPE_ACTION,
                              "path", G_TYPE_STRING,
                              "id", G_TYPE_UINT,
                              NULL);

    xed_message_bus_register (bus,
                              MESSAGE_OBJECT_PATH, "remove_context_item",
                              0,
                              "id", G_TYPE_UINT,
                              NULL);

    xed_message_bus_register (bus, MESSAGE_OBJECT_PATH, "up", 0, NULL);

    xed_message_bus_register (bus, MESSAGE_OBJECT_PATH, "history_back", 0, NULL);
    xed_message_bus_register (bus, MESSAGE_OBJECT_PATH, "history_forward", 0, NULL);

    xed_message_bus_register (bus, MESSAGE_OBJECT_PATH, "refresh", 0, NULL);

    xed_message_bus_register (bus,
                              MESSAGE_OBJECT_PATH, "set_show_hidden",
                              0,
                              "active", G_TYPE_BOOLEAN,
                              NULL);
    xed_message_bus_register (bus,
                              MESSAGE_OBJECT_PATH, "set_show_binary",
                              0,
                              "active", G_TYPE_BOOLEAN,
                              NULL);

    xed_message_bus_register (bus, MESSAGE_OBJECT_PATH, "show_bookmarks", 0, NULL);
    xed_message_bus_register (bus, MESSAGE_OBJECT_PATH, "show_files", 0, NULL);

    xed_message_bus_register (bus,
                              MESSAGE_OBJECT_PATH, "get_view",
                              1,
                              "view", XED_TYPE_FILE_BROWSER_VIEW,
                              NULL);

    BUS_CONNECT (bus, get_root, data);
    BUS_CONNECT (bus, set_root, data);
    BUS_CONNECT (bus, set_emblem, data);
    BUS_CONNECT (bus, add_filter, window);
    BUS_CONNECT (bus, remove_filter, data);

    BUS_CONNECT (bus, add_context_item, data);
    BUS_CONNECT (bus, remove_context_item, data);

    BUS_CONNECT (bus, up, data);
    BUS_CONNECT (bus, history_back, data);
    BUS_CONNECT (bus, history_forward, data);

    BUS_CONNECT (bus, refresh, data);

    BUS_CONNECT (bus, set_show_hidden, data);
    BUS_CONNECT (bus, set_show_binary, data);

    BUS_CONNECT (bus, show_bookmarks, data);
    BUS_CONNECT (bus, show_files, data);

    BUS_CONNECT (bus, get_view, data);
}

static void
store_row_inserted (XedFileBrowserStore *store,
                    GtkTreePath         *path,
                    GtkTreeIter         *iter,
                    MessageCacheData    *data)
{
    guint flags = 0;

    gtk_tree_model_get (GTK_TREE_MODEL (store), iter,
                        XED_FILE_BROWSER_STORE_COLUMN_FLAGS, &flags,
                        -1);

    if (!FILE_IS_DUMMY (flags) && !FILE_IS_FILTERED (flags))
    {
        WindowData *wdata = get_window_data (data->window);

        set_item_message (wdata, iter, path, data->message);
        xed_message_bus_send_message_sync (wdata->bus, data->message);
    }
}

static void
store_row_deleted (XedFileBrowserStore *store,
                   GtkTreePath         *path,
                   MessageCacheData    *data)
{
    GtkTreeIter iter;
    guint flags = 0;

    if (!gtk_tree_model_get_iter (GTK_TREE_MODEL (store), &iter, path))
    {
        return;
    }

    gtk_tree_model_get (GTK_TREE_MODEL (store), &iter,
                        XED_FILE_BROWSER_STORE_COLUMN_FLAGS, &flags,
                        -1);

    if (!FILE_IS_DUMMY (flags) && !FILE_IS_FILTERED (flags))
    {
        WindowData *wdata = get_window_data (data->window);

        set_item_message (wdata, &iter, path, data->message);
        xed_message_bus_send_message_sync (wdata->bus, data->message);
    }
}

static void
store_virtual_root_changed (XedFileBrowserStore *store,
                            GParamSpec          *spec,
                            MessageCacheData    *data)
{
    WindowData *wdata = get_window_data (data->window);
    GFile *vroot;

    vroot = xed_file_browser_store_get_virtual_root (store);

    if (!vroot)
    {
        return;
    }

    xed_message_set (data->message, "location", vroot, NULL);

    xed_message_bus_send_message_sync (wdata->bus, data->message);

    g_object_unref (vroot);
}

static void
store_begin_loading (XedFileBrowserStore *store,
                     GtkTreeIter         *iter,
                     MessageCacheData    *data)
{
    GtkTreePath *path;
    WindowData *wdata = get_window_data (data->window);

    path = gtk_tree_model_get_path (GTK_TREE_MODEL (store), iter);

    set_item_message (wdata, iter, path, data->message);

    xed_message_bus_send_message_sync (wdata->bus, data->message);
    gtk_tree_path_free (path);
}

static void
store_end_loading (XedFileBrowserStore *store,
                   GtkTreeIter         *iter,
                   MessageCacheData    *data)
{
    GtkTreePath *path;
    WindowData *wdata = get_window_data (data->window);

    path = gtk_tree_model_get_path (GTK_TREE_MODEL (store), iter);

    set_item_message (wdata, iter, path, data->message);

    xed_message_bus_send_message_sync (wdata->bus, data->message);
    gtk_tree_path_free (path);
}

static void
register_signals (XedWindow            *window,
                  XedFileBrowserWidget *widget)
{
    XedMessageBus *bus = xed_window_get_message_bus (window);
    XedFileBrowserStore *store;
    XedMessageType *inserted_type;
    XedMessageType *deleted_type;
    XedMessageType *begin_loading_type;
    XedMessageType *end_loading_type;
    XedMessageType *root_changed_type;

    XedMessage *message;
    WindowData *data;

    /* Register signals */
    root_changed_type = xed_message_bus_register (bus,
                                                  MESSAGE_OBJECT_PATH, "root_changed",
                                                  0,
                                                  "id", G_TYPE_STRING,
                                                  "location", G_TYPE_FILE,
                                                  NULL);

    begin_loading_type = xed_message_bus_register (bus,
                                                   MESSAGE_OBJECT_PATH, "begin_loading",
                                                   0,
                                                   "id", G_TYPE_STRING,
                                                   "location", G_TYPE_FILE,
                                                   NULL);

    end_loading_type = xed_message_bus_register (bus,
                                                 MESSAGE_OBJECT_PATH, "end_loading",
                                                 0,
                                                 "id", G_TYPE_STRING,
                                                 "location", G_TYPE_FILE,
                                                 NULL);

    inserted_type = xed_message_bus_register (bus,
                                              MESSAGE_OBJECT_PATH, "inserted",
                                              0,
                                              "id", G_TYPE_STRING,
                                              "location", G_TYPE_FILE,
                                              "is_directory", G_TYPE_BOOLEAN,
                                              NULL);

    deleted_type = xed_message_bus_register (bus,
                                             MESSAGE_OBJECT_PATH, "deleted",
                                             0,
                                             "id", G_TYPE_STRING,
                                             "location", G_TYPE_FILE,
                                             "is_directory", G_TYPE_BOOLEAN,
                                             NULL);

    store = xed_file_browser_widget_get_browser_store (widget);

    message = xed_message_type_instantiate (inserted_type,
                                            "id", NULL,
                                            "location", NULL,
                                            "is_directory", FALSE,
                                            NULL);

    data = get_window_data (window);

    data->row_inserted_id =
        g_signal_connect_data (store,
                               "row-inserted",
                               G_CALLBACK (store_row_inserted),
                               message_cache_data_new (window, message),
                               (GClosureNotify)message_cache_data_free,
                               0);

    message = xed_message_type_instantiate (deleted_type,
                                            "id", NULL,
                                            "location", NULL,
                                            "is_directory", FALSE,
                                            NULL);
    data->row_deleted_id =
        g_signal_connect_data (store,
                               "row-deleted",
                               G_CALLBACK (store_row_deleted),
                               message_cache_data_new (window, message),
                               (GClosureNotify)message_cache_data_free,
                               0);

    message = xed_message_type_instantiate (root_changed_type,
                                            "id", NULL,
                                            "location", NULL,
                                            NULL);
    data->root_changed_id =
        g_signal_connect_data (store,
                               "notify::virtual-root",
                               G_CALLBACK (store_virtual_root_changed),
                               message_cache_data_new (window, message),
                               (GClosureNotify)message_cache_data_free,
                               0);

    message = xed_message_type_instantiate (begin_loading_type,
                                            "id", NULL,
                                            "location", NULL,
                                            NULL);
    data->begin_loading_id =
        g_signal_connect_data (store,
                               "begin_loading",
                               G_CALLBACK (store_begin_loading),
                               message_cache_data_new (window, message),
                               (GClosureNotify)message_cache_data_free,
                               0);

    message = xed_message_type_instantiate (end_loading_type,
                                            "id", NULL,
                                            "location", NULL,
                                            NULL);
    data->end_loading_id =
        g_signal_connect_data (store,
                               "end_loading",
                               G_CALLBACK (store_end_loading),
                               message_cache_data_new (window, message),
                               (GClosureNotify)message_cache_data_free,
                               0);
}

static void
message_unregistered (XedMessageBus  *bus,
                      XedMessageType *message_type,
                      XedWindow      *window)
{
    gchar *identifier = xed_message_type_identifier (xed_message_type_get_object_path (message_type),
                                                     xed_message_type_get_method (message_type));
    FilterData *data;
    WindowData *wdata = get_window_data (window);

    data = g_hash_table_lookup (wdata->filters, identifier);

    if (data)
    {
        xed_file_browser_widget_remove_filter (wdata->widget, data->id);
    }

    g_free (identifier);
}

void
xed_file_browser_messages_register (XedWindow            *window,
                                    XedFileBrowserWidget *widget)
{
    window_data_new (window, widget);

    register_methods (window, widget);
    register_signals (window, widget);

    g_signal_connect (xed_window_get_message_bus (window), "unregistered",
                      G_CALLBACK (message_unregistered), window);
}

static void
cleanup_signals (XedWindow *window)
{
    WindowData *data = get_window_data (window);
    XedFileBrowserStore *store;

    store = xed_file_browser_widget_get_browser_store (data->widget);

    g_signal_handler_disconnect (store, data->row_inserted_id);
    g_signal_handler_disconnect (store, data->row_deleted_id);
    g_signal_handler_disconnect (store, data->root_changed_id);
    g_signal_handler_disconnect (store, data->begin_loading_id);
    g_signal_handler_disconnect (store, data->end_loading_id);

    g_signal_handlers_disconnect_by_func (data->bus, message_unregistered, window);
}

void
xed_file_browser_messages_unregister (XedWindow *window)
{
    XedMessageBus *bus = xed_window_get_message_bus (window);

    cleanup_signals (window);
    xed_message_bus_unregister_all (bus, MESSAGE_OBJECT_PATH);

    window_data_free (window);
}