/*
 * xed-file-bookmarks-store.c - Xed plugin providing easy file access
 * from the sidepanel
 *
 * Copyright (C) 2006 - Jesse van den Kieboom <jesse@icecrew.nl>
 *
 * 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, 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.
 */

#include <string.h>
#include <glib/gi18n.h>
#include <gio/gio.h>
#include <xed/xed-utils.h>

#include "xed-file-bookmarks-store.h"
#include "xed-file-browser-utils.h"

#define XED_FILE_BOOKMARKS_STORE_GET_PRIVATE(object)( \
        G_TYPE_INSTANCE_GET_PRIVATE((object), XED_TYPE_FILE_BOOKMARKS_STORE, \
        XedFileBookmarksStorePrivate))

struct _XedFileBookmarksStorePrivate
{
    GVolumeMonitor * volume_monitor;
    GFileMonitor * bookmarks_monitor;
};

static void remove_node               (GtkTreeModel * model,
                                       GtkTreeIter * iter);

static void on_fs_changed             (GVolumeMonitor       *monitor,
                                       GObject          *object,
                                       XedFileBookmarksStore    *model);

static void on_bookmarks_file_changed (GFileMonitor * monitor,
                                       GFile * file,
                                       GFile * other_file,
                                       GFileMonitorEvent event_type,
                                       XedFileBookmarksStore * model);
static gboolean find_with_flags       (GtkTreeModel * model,
                                       GtkTreeIter * iter,
                                       gpointer obj,
                                       guint flags,
                                       guint notflags);

G_DEFINE_DYNAMIC_TYPE (XedFileBookmarksStore, xed_file_bookmarks_store, GTK_TYPE_TREE_STORE)

static void
xed_file_bookmarks_store_dispose (GObject * object)
{
    XedFileBookmarksStore *obj = XED_FILE_BOOKMARKS_STORE (object);

    if (obj->priv->volume_monitor != NULL)
    {
        g_signal_handlers_disconnect_by_func (obj->priv->volume_monitor, on_fs_changed, obj);

        g_object_unref (obj->priv->volume_monitor);
        obj->priv->volume_monitor = NULL;
    }

    if (obj->priv->bookmarks_monitor != NULL)
    {
        g_object_unref (obj->priv->bookmarks_monitor);
        obj->priv->bookmarks_monitor = NULL;
    }

    G_OBJECT_CLASS (xed_file_bookmarks_store_parent_class)->dispose (object);
}

static void
xed_file_bookmarks_store_finalize (GObject * object)
{
    G_OBJECT_CLASS (xed_file_bookmarks_store_parent_class)->finalize (object);
}

static void
xed_file_bookmarks_store_class_init (XedFileBookmarksStoreClass *klass)
{
    GObjectClass *object_class = G_OBJECT_CLASS (klass);

    object_class->dispose = xed_file_bookmarks_store_dispose;
    object_class->finalize = xed_file_bookmarks_store_finalize;

    g_type_class_add_private (object_class, sizeof (XedFileBookmarksStorePrivate));
}

static void
xed_file_bookmarks_store_class_finalize  (XedFileBookmarksStoreClass *klass)
{
    /* dummy function - used by G_DEFINE_DYNAMIC_TYPE */
}

static void
xed_file_bookmarks_store_init (XedFileBookmarksStore * obj)
{
    obj->priv = XED_FILE_BOOKMARKS_STORE_GET_PRIVATE (obj);
}

/* Private */
static void
add_node (XedFileBookmarksStore *model,
          GdkPixbuf             *pixbuf,
          const gchar           *name,
          GObject               *obj,
          guint                  flags,
          GtkTreeIter           *iter)
{
    GtkTreeIter newiter;

    gtk_tree_store_append (GTK_TREE_STORE (model), &newiter, NULL);

    gtk_tree_store_set (GTK_TREE_STORE (model), &newiter,
                        XED_FILE_BOOKMARKS_STORE_COLUMN_ICON, pixbuf,
                        XED_FILE_BOOKMARKS_STORE_COLUMN_NAME, name,
                        XED_FILE_BOOKMARKS_STORE_COLUMN_OBJECT, obj,
                        XED_FILE_BOOKMARKS_STORE_COLUMN_FLAGS, flags,
                        -1);

    if (iter != NULL)
    {
        *iter = newiter;
    }
}

static gboolean
add_file (XedFileBookmarksStore *model,
          GFile                 *file,
          const gchar           *name,
          guint                  flags,
          GtkTreeIter           *iter)
{
    GdkPixbuf *pixbuf = NULL;
    gboolean native;
    gchar *newname;

    native = g_file_is_native (file);

    if (native && !g_file_query_exists (file, NULL))
    {
        return FALSE;
    }

    if (flags & XED_FILE_BOOKMARKS_STORE_IS_HOME)
    {
        pixbuf = xed_file_browser_utils_pixbuf_from_theme ("user-home", GTK_ICON_SIZE_MENU);
    }
    else if (flags & XED_FILE_BOOKMARKS_STORE_IS_DESKTOP)
    {
        pixbuf = xed_file_browser_utils_pixbuf_from_theme ("user-desktop", GTK_ICON_SIZE_MENU);
    }
    else if (flags & XED_FILE_BOOKMARKS_STORE_IS_ROOT)
    {
        pixbuf = xed_file_browser_utils_pixbuf_from_theme ("drive-harddisk", GTK_ICON_SIZE_MENU);
    }

    if (pixbuf == NULL)
    {
        /* getting the icon is a sync get_info call, so we just do it for local files */
        if (native)
        {
            pixbuf = xed_file_browser_utils_pixbuf_from_file (file, GTK_ICON_SIZE_MENU);
        }
        else
        {
            pixbuf = xed_file_browser_utils_pixbuf_from_theme ("folder", GTK_ICON_SIZE_MENU);
        }
    }

    if (name == NULL)
    {
        newname = xed_file_browser_utils_file_basename (file);
    }
    else
    {
        newname = g_strdup (name);
    }

    add_node (model, pixbuf, newname, G_OBJECT (file), flags, iter);

    if (pixbuf)
    {
        g_object_unref (pixbuf);
    }

    g_free (newname);

    return TRUE;
}

static void
check_mount_separator (XedFileBookmarksStore * model, guint flags,
                       gboolean                added)
{
    GtkTreeIter iter;
    gboolean found;

    found = find_with_flags (GTK_TREE_MODEL (model), &iter, NULL,
                             flags | XED_FILE_BOOKMARKS_STORE_IS_SEPARATOR, 0);

    if (added && !found)
    {
        /* Add the separator */
        add_node (model, NULL, NULL, NULL, flags | XED_FILE_BOOKMARKS_STORE_IS_SEPARATOR, NULL);
    }
    else if (!added && found)
    {
        remove_node (GTK_TREE_MODEL (model), &iter);
    }
}

static void
init_special_directories (XedFileBookmarksStore * model)
{
    gchar const *path;
    GFile * file;

    path = g_get_home_dir ();
    if (path != NULL)
    {
        file = g_file_new_for_path (path);
        add_file (model, file, NULL, XED_FILE_BOOKMARKS_STORE_IS_HOME | XED_FILE_BOOKMARKS_STORE_IS_SPECIAL_DIR, NULL);
        g_object_unref (file);
    }

    // path = g_get_user_special_dir (G_USER_DIRECTORY_DESKTOP);
    // if (path != NULL)
    // {
    //     file = g_file_new_for_path (path);
    //     add_file (model, file, NULL, XED_FILE_BOOKMARKS_STORE_IS_DESKTOP |
    //                                  XED_FILE_BOOKMARKS_STORE_IS_SPECIAL_DIR, NULL);
    //     g_object_unref (file);
    // }

    // path = g_get_user_special_dir (G_USER_DIRECTORY_DOCUMENTS);
    // if (path != NULL)
    // {
    //     file = g_file_new_for_path (path);
    //     add_file (model, file, NULL, XED_FILE_BOOKMARKS_STORE_IS_DOCUMENTS |
    //                                  XED_FILE_BOOKMARKS_STORE_IS_SPECIAL_DIR, NULL);
    //     g_object_unref (file);
    // }

    file = g_file_new_for_uri ("file:///");
    add_file (model, file, _("File System"), XED_FILE_BOOKMARKS_STORE_IS_ROOT, NULL);
    g_object_unref (file);

    check_mount_separator (model, XED_FILE_BOOKMARKS_STORE_IS_ROOT, TRUE);
}

static void
get_fs_properties (gpointer    fs,
                   gchar     **name,
                   GdkPixbuf **pixbuf,
                   guint      *flags)
{
    GIcon *icon = NULL;

    *flags = XED_FILE_BOOKMARKS_STORE_IS_FS;
    *name = NULL;
    *pixbuf = NULL;

    if (G_IS_DRIVE (fs))
    {
        icon = g_drive_get_icon (G_DRIVE (fs));
        *name = g_drive_get_name (G_DRIVE (fs));

        *flags |= XED_FILE_BOOKMARKS_STORE_IS_DRIVE;
    }
    else if (G_IS_VOLUME (fs))
    {
        icon = g_volume_get_icon (G_VOLUME (fs));
        *name = g_volume_get_name (G_VOLUME (fs));

        *flags |= XED_FILE_BOOKMARKS_STORE_IS_VOLUME;
    }
    else if (G_IS_MOUNT (fs))
    {
        icon = g_mount_get_icon (G_MOUNT (fs));
        *name = g_mount_get_name (G_MOUNT (fs));

        *flags |= XED_FILE_BOOKMARKS_STORE_IS_MOUNT;
    }

    if (icon)
    {
        *pixbuf = xed_file_browser_utils_pixbuf_from_icon (icon, GTK_ICON_SIZE_MENU);
        g_object_unref (icon);
    }
}


static void
add_fs (XedFileBookmarksStore *model,
        gpointer               fs,
        guint                  flags,
        GtkTreeIter           *iter)
{
    gchar *name;
    GdkPixbuf *pixbuf;
    guint fsflags;

    get_fs_properties (fs, &name, &pixbuf, &fsflags);
    add_node (model, pixbuf, name, fs, flags | fsflags, iter);

    if (pixbuf)
    {
        g_object_unref (pixbuf);
    }

    g_free (name);
    check_mount_separator (model, XED_FILE_BOOKMARKS_STORE_IS_FS, TRUE);
}

static void
process_volume_cb (GVolume               *volume,
                   XedFileBookmarksStore *model)
{
    GMount *mount;
    guint flags = XED_FILE_BOOKMARKS_STORE_NONE;
    mount = g_volume_get_mount (volume);

    /* CHECK: should we use the LOCAL/REMOTE thing still? */
    if (mount)
    {
        /* Show mounted volume */
        add_fs (model, mount, flags, NULL);
        g_object_unref (mount);
    }
    else if (g_volume_can_mount (volume))
    {
        /* We also show the unmounted volume here so users can
           mount it if they want to access it */
        add_fs (model, volume, flags, NULL);
    }
}

static void
process_drive_novolumes (XedFileBookmarksStore *model,
                         GDrive                *drive)
{
    if (g_drive_is_media_removable (drive) &&
       !g_drive_is_media_check_automatic (drive) &&
        g_drive_can_poll_for_media (drive))
    {
        /* This can be the case for floppy drives or other
           drives where media detection fails. We show the
           drive and poll for media when the user activates
           it */
        add_fs (model, drive, XED_FILE_BOOKMARKS_STORE_NONE, NULL);
    }
}

static void
process_drive_cb (GDrive                *drive,
                  XedFileBookmarksStore *model)
{
    GList *volumes;

    volumes = g_drive_get_volumes (drive);

    if (volumes)
    {
        /* Add all volumes for the drive */
        g_list_foreach (volumes, (GFunc)process_volume_cb, model);
        g_list_free (volumes);
    }
    else
    {
        process_drive_novolumes (model, drive);
    }
}

static void
init_drives (XedFileBookmarksStore *model)
{
    GList *drives;

    drives = g_volume_monitor_get_connected_drives (model->priv->volume_monitor);

    g_list_foreach (drives, (GFunc)process_drive_cb, model);
    g_list_foreach (drives, (GFunc)g_object_unref, NULL);
    g_list_free (drives);
}

static void
process_volume_nodrive_cb (GVolume               *volume,
                           XedFileBookmarksStore *model)
{
    GDrive *drive;

    drive = g_volume_get_drive (volume);

    if (drive)
    {
        g_object_unref (drive);
        return;
    }

    process_volume_cb (volume, model);
}

static void
init_volumes (XedFileBookmarksStore *model)
{
    GList *volumes;

    volumes = g_volume_monitor_get_volumes (model->priv->volume_monitor);

    g_list_foreach (volumes, (GFunc)process_volume_nodrive_cb, model);
    g_list_foreach (volumes, (GFunc)g_object_unref, NULL);
    g_list_free (volumes);
}

static void
process_mount_novolume_cb (GMount                *mount,
                           XedFileBookmarksStore *model)
{
    GVolume *volume;

    volume = g_mount_get_volume (mount);

    if (volume)
    {
        g_object_unref (volume);
    }
    else if (!g_mount_is_shadowed (mount))
    {
        /* Add the mount */
        add_fs (model, mount, XED_FILE_BOOKMARKS_STORE_NONE, NULL);
    }
}

static void
init_mounts (XedFileBookmarksStore *model)
{
    GList *mounts;

    mounts = g_volume_monitor_get_mounts (model->priv->volume_monitor);

    g_list_foreach (mounts, (GFunc)process_mount_novolume_cb, model);
    g_list_foreach (mounts, (GFunc)g_object_unref, NULL);
    g_list_free (mounts);
}

static void
init_fs (XedFileBookmarksStore * model)
{
    if (model->priv->volume_monitor == NULL)
    {
        const gchar **ptr;
        const gchar *signals[] = {
            "drive-connected", "drive-changed", "drive-disconnected",
            "volume-added", "volume-removed", "volume-changed",
            "mount-added", "mount-removed", "mount-changed",
            NULL
        };

        model->priv->volume_monitor = g_volume_monitor_get ();

        /* Connect signals */
        for (ptr = signals; *ptr; ptr++)
        {
            g_signal_connect (model->priv->volume_monitor, *ptr, G_CALLBACK (on_fs_changed), model);
        }
    }

    /* First go through all the connected drives */
    init_drives (model);

    /* Then add all volumes, not associated with a drive */
    init_volumes (model);

    /* Then finally add all mounts that have no volume */
    init_mounts (model);
}

static gboolean
add_bookmark (XedFileBookmarksStore * model,
              gchar const           * name,
              gchar const           * uri)
{
    GFile * file;
    gboolean ret;
    guint flags = XED_FILE_BOOKMARKS_STORE_IS_BOOKMARK;
    GtkTreeIter iter;

    file = g_file_new_for_uri (uri);

    if (g_file_is_native (file))
    {
        flags |= XED_FILE_BOOKMARKS_STORE_IS_LOCAL_BOOKMARK;
    }
    else
    {
        flags |= XED_FILE_BOOKMARKS_STORE_IS_REMOTE_BOOKMARK;
    }

    ret = add_file (model, file, name, flags, &iter);

    g_object_unref (file);

    return ret;
}

static gchar *
get_bookmarks_file (void)
{
    return g_build_filename (g_get_user_config_dir (), "gtk-3.0", "bookmarks", NULL);
}

static gchar *
get_legacy_bookmarks_file (void)
{
    return g_build_filename (g_get_home_dir (), ".gtk-bookmarks", NULL);
}

static gboolean
parse_bookmarks_file (XedFileBookmarksStore *model,
                      const gchar           *bookmarks,
                      gboolean              *added)

{
    GError *error = NULL;
    gchar *contents;
    gchar **lines;
    gchar **line;

    if (!g_file_get_contents (bookmarks, &contents, NULL, &error))
    {
        /* The bookmarks file doesn't exist (which is perfectly fine) */
        g_error_free (error);

        return FALSE;
    }

    lines = g_strsplit (contents, "\n", 0);

    for (line = lines; *line; ++line)
    {
        if (**line)
        {
            GFile *location;

            gchar *pos;
            gchar *name;

            /* CHECK: is this really utf8? */
            pos = g_utf8_strchr (*line, -1, ' ');

            if (pos != NULL)
            {
                *pos = '\0';
                name = pos + 1;
            }
            else
            {
                name = NULL;
            }

            /* the bookmarks file should contain valid
            * URIs, but paranoia is good */
            location = g_file_new_for_uri (*line);
            if (xed_utils_is_valid_location (location))
            {
                *added |= add_bookmark (model, name, *line);
            }
            g_object_unref (location);
        }
    }

    g_strfreev (lines);
    g_free (contents);

    /* Add a watch */
    if (model->priv->bookmarks_monitor == NULL)
    {
        GFile * file;

        file = g_file_new_for_path (bookmarks);
        model->priv->bookmarks_monitor = g_file_monitor_file (file, G_FILE_MONITOR_NONE, NULL, NULL);
        g_object_unref (file);

        g_signal_connect (model->priv->bookmarks_monitor, "changed",
                          G_CALLBACK (on_bookmarks_file_changed), model);
    }

    return TRUE;
}

static void
init_bookmarks (XedFileBookmarksStore *model)
{
    gchar *bookmarks;
    gboolean added = FALSE;

    bookmarks = get_bookmarks_file ();

    if (!parse_bookmarks_file (model, bookmarks, &added))
    {
        g_free (bookmarks);

        /* try the old location (gtk <= 3.4) */
        bookmarks = get_legacy_bookmarks_file ();
        parse_bookmarks_file (model, bookmarks, &added);
    }

    if (added)
    {
        /* Bookmarks separator */
        add_node (model, NULL, NULL, NULL,
                  XED_FILE_BOOKMARKS_STORE_IS_BOOKMARK |
                  XED_FILE_BOOKMARKS_STORE_IS_SEPARATOR, NULL);
    }

    g_free (bookmarks);
}

static gint flags_order[] = {
    XED_FILE_BOOKMARKS_STORE_IS_HOME,
    XED_FILE_BOOKMARKS_STORE_IS_DESKTOP,
    XED_FILE_BOOKMARKS_STORE_IS_SPECIAL_DIR,
    XED_FILE_BOOKMARKS_STORE_IS_ROOT,
    XED_FILE_BOOKMARKS_STORE_IS_FS,
    XED_FILE_BOOKMARKS_STORE_IS_BOOKMARK,
    -1
};

static gint
utf8_casecmp (gchar const *s1, const gchar * s2)
{
    gchar *n1;
    gchar *n2;
    gint result;

    n1 = g_utf8_casefold (s1, -1);
    n2 = g_utf8_casefold (s2, -1);

    result = g_utf8_collate (n1, n2);

    g_free (n1);
    g_free (n2);

    return result;
}

static gint
bookmarks_compare_names (GtkTreeModel * model,
                         GtkTreeIter  * a,
                         GtkTreeIter  * b)
{
    gchar *n1;
    gchar *n2;
    gint result;
    guint f1;
    guint f2;

    gtk_tree_model_get (model, a,
                        XED_FILE_BOOKMARKS_STORE_COLUMN_NAME, &n1,
                        XED_FILE_BOOKMARKS_STORE_COLUMN_FLAGS, &f1,
                        -1);
    gtk_tree_model_get (model, b,
                        XED_FILE_BOOKMARKS_STORE_COLUMN_NAME, &n2,
                        XED_FILE_BOOKMARKS_STORE_COLUMN_FLAGS, &f2,
                        -1);

    /* do not sort actual bookmarks to keep same order as in caja */
    if ((f1 & XED_FILE_BOOKMARKS_STORE_IS_BOOKMARK) && (f2 & XED_FILE_BOOKMARKS_STORE_IS_BOOKMARK))
    {
        result = 0;
    }
    else if (n1 == NULL && n2 == NULL)
    {
        result = 0;
    }
    else if (n1 == NULL)
    {
        result = -1;
    }
    else if (n2 == NULL)
    {
        result = 1;
    }
    else
    {
        result = utf8_casecmp (n1, n2);
    }

    g_free (n1);
    g_free (n2);

    return result;
}

static gint
bookmarks_compare_flags (GtkTreeModel * model,
                         GtkTreeIter  * a,
                         GtkTreeIter * b)
{
    guint f1;
    guint f2;
    gint *flags;
    guint sep;

    sep = XED_FILE_BOOKMARKS_STORE_IS_SEPARATOR;

    gtk_tree_model_get (model, a, XED_FILE_BOOKMARKS_STORE_COLUMN_FLAGS, &f1, -1);
    gtk_tree_model_get (model, b, XED_FILE_BOOKMARKS_STORE_COLUMN_FLAGS, &f2, -1);

    for (flags = flags_order; *flags != -1; ++flags)
    {
        if ((f1 & *flags) != (f2 & *flags))
        {
            if (f1 & *flags)
            {
                return -1;
            }
            else
            {
                return 1;
            }
        }
        else if ((f1 & *flags) && (f1 & sep) != (f2 & sep))
        {
            if (f1 & sep)
            {
                return -1;
            }
            else
            {
                return 1;
            }
        }
    }

    return 0;
}

static gint
bookmarks_compare_func (GtkTreeModel *model,
                        GtkTreeIter  *a,
                        GtkTreeIter  *b,
                        gpointer      user_data)
{
    gint result;

    result = bookmarks_compare_flags (model, a, b);

    if (result == 0)
    {
        result = bookmarks_compare_names (model, a, b);
    }

    return result;
}

static gboolean
find_with_flags (GtkTreeModel *model,
                 GtkTreeIter  *iter,
                 gpointer      obj,
                 guint         flags,
                 guint         notflags)
{
    GtkTreeIter child;
    guint childflags = 0;
    GObject * childobj;
    gboolean fequal;

    if (!gtk_tree_model_get_iter_first (model, &child))
    {
        return FALSE;
    }

    do {
        gtk_tree_model_get (model, &child,
                            XED_FILE_BOOKMARKS_STORE_COLUMN_OBJECT,
                            &childobj,
                            XED_FILE_BOOKMARKS_STORE_COLUMN_FLAGS,
                            &childflags, -1);

        fequal = (obj == childobj);

        if (childobj)
        {
            g_object_unref (childobj);
        }

        if ((obj == NULL || fequal) &&
            (childflags & flags) == flags
            && !(childflags & notflags))
        {
            *iter = child;
            return TRUE;
        }
    } while (gtk_tree_model_iter_next (model, &child));

    return FALSE;
}

static void
remove_node (GtkTreeModel *model,
             GtkTreeIter  *iter)
{
    guint flags;

    gtk_tree_model_get (model, iter, XED_FILE_BOOKMARKS_STORE_COLUMN_FLAGS, &flags, -1);

    if (!(flags & XED_FILE_BOOKMARKS_STORE_IS_SEPARATOR))
    {
        if (flags & XED_FILE_BOOKMARKS_STORE_IS_FS)
        {
            check_mount_separator (XED_FILE_BOOKMARKS_STORE (model),
                                   flags & XED_FILE_BOOKMARKS_STORE_IS_FS,
                                   FALSE);
        }
    }

    gtk_tree_store_remove (GTK_TREE_STORE (model), iter);
}

static void
remove_bookmarks (XedFileBookmarksStore *model)
{
    GtkTreeIter iter;

    while (find_with_flags (GTK_TREE_MODEL (model), &iter, NULL,
                            XED_FILE_BOOKMARKS_STORE_IS_BOOKMARK, 0))
    {
        remove_node (GTK_TREE_MODEL (model), &iter);
    }
}

static void
initialize_fill (XedFileBookmarksStore *model)
{
    init_special_directories (model);
    init_fs (model);
    init_bookmarks (model);
}

/* Public */
XedFileBookmarksStore *
xed_file_bookmarks_store_new (void)
{
    XedFileBookmarksStore *model;
    GType column_types[] = {
        GDK_TYPE_PIXBUF,
        G_TYPE_STRING,
        G_TYPE_OBJECT,
        G_TYPE_UINT
    };

    model = g_object_new (XED_TYPE_FILE_BOOKMARKS_STORE, NULL);
    gtk_tree_store_set_column_types (GTK_TREE_STORE (model),
                                     XED_FILE_BOOKMARKS_STORE_N_COLUMNS,
                                     column_types);

    gtk_tree_sortable_set_default_sort_func (GTK_TREE_SORTABLE (model),
                                             bookmarks_compare_func,
                                             NULL, NULL);
    gtk_tree_sortable_set_sort_column_id (GTK_TREE_SORTABLE (model),
                                          GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID,
                                          GTK_SORT_ASCENDING);

    initialize_fill (model);

    return model;
}

GFile *
xed_file_bookmarks_store_get_location (XedFileBookmarksStore *model,
                                       GtkTreeIter           *iter)
{
    GObject *obj;
    GFile *file = NULL;
    guint flags;
    GFile *ret = NULL;
    gboolean isfs;

    g_return_val_if_fail (XED_IS_FILE_BOOKMARKS_STORE (model), NULL);
    g_return_val_if_fail (iter != NULL, NULL);

    gtk_tree_model_get (GTK_TREE_MODEL (model), iter,
                        XED_FILE_BOOKMARKS_STORE_COLUMN_FLAGS,
                        &flags,
                        XED_FILE_BOOKMARKS_STORE_COLUMN_OBJECT,
                        &obj,
                        -1);

    if (obj == NULL)
    {
        return NULL;
    }

    isfs = (flags & XED_FILE_BOOKMARKS_STORE_IS_FS);

    if (isfs && (flags & XED_FILE_BOOKMARKS_STORE_IS_MOUNT))
    {
        file = g_mount_get_root (G_MOUNT (obj));
    }
    else if (!isfs)
    {
        file = G_FILE (g_object_ref (obj));
    }

    g_object_unref (obj);

    if (file)
    {
        ret = g_file_dup (file);
        g_object_unref (file);
    }

    return ret;
}

void
xed_file_bookmarks_store_refresh (XedFileBookmarksStore *model)
{
    gtk_tree_store_clear (GTK_TREE_STORE (model));
    initialize_fill (model);
}

static void
on_fs_changed (GVolumeMonitor        *monitor,
               GObject               *object,
               XedFileBookmarksStore *model)
{
    GtkTreeModel *tree_model = GTK_TREE_MODEL (model);
    guint flags = XED_FILE_BOOKMARKS_STORE_IS_FS;
    guint noflags = XED_FILE_BOOKMARKS_STORE_IS_SEPARATOR;
    GtkTreeIter iter;

    /* clear all fs items */
    while (find_with_flags (tree_model, &iter, NULL, flags, noflags))
        remove_node (tree_model, &iter);

    /* then reinitialize */
    init_fs (model);
}

static void
on_bookmarks_file_changed (GFileMonitor * monitor,
               GFile * file,
               GFile * other_file,
               GFileMonitorEvent event_type,
               XedFileBookmarksStore * model)
{
    switch (event_type) {
    case G_FILE_MONITOR_EVENT_CHANGED:
    case G_FILE_MONITOR_EVENT_CREATED:
        /* Re-initialize bookmarks */
        remove_bookmarks (model);
        init_bookmarks (model);
        break;
    case G_FILE_MONITOR_EVENT_DELETED: // FIXME: shouldn't we also monitor the directory?
        /* Remove bookmarks */
        remove_bookmarks (model);
        g_object_unref (monitor);
        model->priv->bookmarks_monitor = NULL;
        break;
    default:
        break;
    }
}

void
_xed_file_bookmarks_store_register_type (GTypeModule *type_module)
{
    xed_file_bookmarks_store_register_type (type_module);
}

// ex:ts=8:noet: