/* * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2 of the License, or (at your option) version 3. * * 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 * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with the program; if not, see * * * Authors: * Jeffrey Stedfast * * Copyright (C) 1999-2008 Novell, Inc. (www.novell.com) * */ #ifdef HAVE_CONFIG_H #include #endif #include "em-folder-tree-model.h" #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #define EM_FOLDER_TREE_MODEL_GET_PRIVATE(obj) \ (G_TYPE_INSTANCE_GET_PRIVATE \ ((obj), EM_TYPE_FOLDER_TREE_MODEL, EMFolderTreeModelPrivate)) #define d(x) #define EM_FOLDER_TREE_MODEL_GET_PRIVATE(obj) \ (G_TYPE_INSTANCE_GET_PRIVATE \ ((obj), EM_TYPE_FOLDER_TREE_MODEL, EMFolderTreeModelPrivate)) struct _EMFolderTreeModelPrivate { /* This is set by EMailShellSidebar. It allows new EMFolderTree * instances to initialize their selection and expanded states to * mimic the sidebar. */ GtkTreeSelection *selection; /* weak reference */ EMailSession *session; EMailAccountStore *account_store; /* CamelStore -> EMFolderTreeStoreInfo */ GHashTable *store_index; }; enum { PROP_0, PROP_SELECTION, PROP_SESSION }; enum { LOADING_ROW, LOADED_ROW, LAST_SIGNAL }; static guint signals[LAST_SIGNAL]; G_DEFINE_TYPE (EMFolderTreeModel, em_folder_tree_model, GTK_TYPE_TREE_STORE) static void store_info_free (EMFolderTreeModelStoreInfo *si) { g_signal_handler_disconnect (si->store, si->created_id); g_signal_handler_disconnect (si->store, si->deleted_id); g_signal_handler_disconnect (si->store, si->renamed_id); if (si->subscribed_id > 0) g_signal_handler_disconnect (si->store, si->subscribed_id); if (si->unsubscribed_id > 0) g_signal_handler_disconnect (si->store, si->unsubscribed_id); g_object_unref (si->store); gtk_tree_row_reference_free (si->row); g_hash_table_destroy (si->full_hash); g_free (si); } static gint folder_tree_model_sort (GtkTreeModel *model, GtkTreeIter *a, GtkTreeIter *b, gpointer unused) { EMFolderTreeModel *folder_tree_model; gchar *aname, *bname; CamelService *service_a; CamelService *service_b; gboolean a_is_store; gboolean b_is_store; const gchar *store_uid = NULL; guint32 flags_a, flags_b; gint rv = -2; folder_tree_model = EM_FOLDER_TREE_MODEL (model); gtk_tree_model_get ( model, a, COL_BOOL_IS_STORE, &a_is_store, COL_POINTER_CAMEL_STORE, &service_a, COL_STRING_DISPLAY_NAME, &aname, COL_UINT_FLAGS, &flags_a, -1); gtk_tree_model_get ( model, b, COL_BOOL_IS_STORE, &b_is_store, COL_POINTER_CAMEL_STORE, &service_b, COL_STRING_DISPLAY_NAME, &bname, COL_UINT_FLAGS, &flags_b, -1); if (CAMEL_IS_SERVICE (service_a)) store_uid = camel_service_get_uid (service_a); if (a_is_store && b_is_store) { rv = e_mail_account_store_compare_services ( folder_tree_model->priv->account_store, service_a, service_b); } else if (g_strcmp0 (store_uid, E_MAIL_SESSION_VFOLDER_UID) == 0) { /* UNMATCHED is always last. */ if (g_strcmp0 (aname, _("UNMATCHED")) == 0) rv = 1; else if (g_strcmp0 (bname, _("UNMATCHED")) == 0) rv = -1; } else { /* Inbox is always first. */ if ((flags_a & CAMEL_FOLDER_TYPE_MASK) == CAMEL_FOLDER_TYPE_INBOX) rv = -1; else if ((flags_b & CAMEL_FOLDER_TYPE_MASK) == CAMEL_FOLDER_TYPE_INBOX) rv = 1; } if (rv == -2) { if (aname != NULL && bname != NULL) rv = g_utf8_collate (aname, bname); else if (aname == bname) rv = 0; else if (aname == NULL) rv = -1; else rv = 1; } g_free (aname); g_free (bname); return rv; } static void folder_tree_model_service_removed (EMailAccountStore *account_store, CamelService *service, EMFolderTreeModel *folder_tree_model) { em_folder_tree_model_remove_store ( folder_tree_model, CAMEL_STORE (service)); } static void folder_tree_model_service_enabled (EMailAccountStore *account_store, CamelService *service, EMFolderTreeModel *folder_tree_model) { em_folder_tree_model_add_store ( folder_tree_model, CAMEL_STORE (service)); } static void folder_tree_model_service_disabled (EMailAccountStore *account_store, CamelService *service, EMFolderTreeModel *folder_tree_model) { em_folder_tree_model_remove_store ( folder_tree_model, CAMEL_STORE (service)); } static void folder_tree_model_services_reordered (EMailAccountStore *account_store, gboolean default_restored, EMFolderTreeModel *folder_tree_model) { /* This forces the tree store to re-sort. */ gtk_tree_sortable_set_default_sort_func ( GTK_TREE_SORTABLE (folder_tree_model), folder_tree_model_sort, NULL, NULL); } static void folder_tree_model_selection_finalized_cb (EMFolderTreeModel *model) { model->priv->selection = NULL; g_object_notify (G_OBJECT (model), "selection"); } static void folder_tree_model_set_property (GObject *object, guint property_id, const GValue *value, GParamSpec *pspec) { switch (property_id) { case PROP_SELECTION: em_folder_tree_model_set_selection ( EM_FOLDER_TREE_MODEL (object), g_value_get_object (value)); return; case PROP_SESSION: em_folder_tree_model_set_session ( EM_FOLDER_TREE_MODEL (object), g_value_get_object (value)); return; } G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec); } static void folder_tree_model_get_property (GObject *object, guint property_id, GValue *value, GParamSpec *pspec) { switch (property_id) { case PROP_SELECTION: g_value_set_object ( value, em_folder_tree_model_get_selection ( EM_FOLDER_TREE_MODEL (object))); return; case PROP_SESSION: g_value_set_object ( value, em_folder_tree_model_get_session ( EM_FOLDER_TREE_MODEL (object))); return; } G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec); } static void folder_tree_model_dispose (GObject *object) { EMFolderTreeModelPrivate *priv; priv = EM_FOLDER_TREE_MODEL_GET_PRIVATE (object); if (priv->selection != NULL) { g_object_weak_unref ( G_OBJECT (priv->selection), (GWeakNotify) folder_tree_model_selection_finalized_cb, object); priv->selection = NULL; } if (priv->session != NULL) { g_object_unref (priv->session); priv->session = NULL; } if (priv->account_store != NULL) { g_signal_handlers_disconnect_matched ( priv->account_store, G_SIGNAL_MATCH_DATA, 0, 0, NULL, NULL, object); g_object_unref (priv->account_store); priv->account_store = NULL; } /* Chain up to parent's dispose() method. */ G_OBJECT_CLASS (em_folder_tree_model_parent_class)->dispose (object); } static void folder_tree_model_finalize (GObject *object) { EMFolderTreeModelPrivate *priv; priv = EM_FOLDER_TREE_MODEL_GET_PRIVATE (object); g_hash_table_destroy (priv->store_index); /* Chain up to parent's finalize() method. */ G_OBJECT_CLASS (em_folder_tree_model_parent_class)->finalize (object); } static void folder_tree_model_constructed (GObject *object) { GType col_types[] = { G_TYPE_STRING, /* display name */ G_TYPE_POINTER, /* store object */ G_TYPE_STRING, /* full name */ G_TYPE_STRING, /* icon name */ G_TYPE_UINT, /* unread count */ G_TYPE_UINT, /* flags */ G_TYPE_BOOLEAN, /* is a store node */ G_TYPE_BOOLEAN, /* is a folder node */ G_TYPE_BOOLEAN, /* has not-yet-loaded subfolders */ G_TYPE_UINT, /* last known unread count */ G_TYPE_BOOLEAN, /* folder is a draft folder */ G_TYPE_UINT /* user's sortorder */ }; gtk_tree_store_set_column_types ( GTK_TREE_STORE (object), NUM_COLUMNS, col_types); gtk_tree_sortable_set_default_sort_func ( GTK_TREE_SORTABLE (object), folder_tree_model_sort, NULL, NULL); gtk_tree_sortable_set_sort_column_id ( GTK_TREE_SORTABLE (object), GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID, GTK_SORT_ASCENDING); /* Chain up to parent's constructed() method. */ G_OBJECT_CLASS (em_folder_tree_model_parent_class)-> constructed (object); } static void em_folder_tree_model_class_init (EMFolderTreeModelClass *class) { GObjectClass *object_class; g_type_class_add_private (class, sizeof (EMFolderTreeModelPrivate)); object_class = G_OBJECT_CLASS (class); object_class->set_property = folder_tree_model_set_property; object_class->get_property = folder_tree_model_get_property; object_class->dispose = folder_tree_model_dispose; object_class->finalize = folder_tree_model_finalize; object_class->constructed = folder_tree_model_constructed; g_object_class_install_property ( object_class, PROP_SESSION, g_param_spec_object ( "session", NULL, NULL, E_TYPE_MAIL_SESSION, G_PARAM_READWRITE)); g_object_class_install_property ( object_class, PROP_SELECTION, g_param_spec_object ( "selection", "Selection", NULL, GTK_TYPE_TREE_SELECTION, G_PARAM_READWRITE)); signals[LOADING_ROW] = g_signal_new ( "loading-row", G_OBJECT_CLASS_TYPE (object_class), G_SIGNAL_RUN_FIRST, G_STRUCT_OFFSET (EMFolderTreeModelClass, loading_row), NULL, NULL, e_marshal_VOID__POINTER_POINTER, G_TYPE_NONE, 2, G_TYPE_POINTER, G_TYPE_POINTER); signals[LOADED_ROW] = g_signal_new ( "loaded-row", G_OBJECT_CLASS_TYPE (object_class), G_SIGNAL_RUN_FIRST, G_STRUCT_OFFSET (EMFolderTreeModelClass, loaded_row), NULL, NULL, e_marshal_VOID__POINTER_POINTER, G_TYPE_NONE, 2, G_TYPE_POINTER, G_TYPE_POINTER); } static void folder_tree_model_set_unread_count (EMFolderTreeModel *model, CamelStore *store, const gchar *full, gint unread) { EMFolderTreeModelStoreInfo *si; GtkTreeRowReference *reference; GtkTreeModel *tree_model; GtkTreePath *path; GtkTreeIter parent; GtkTreeIter iter; guint old_unread = 0; g_return_if_fail (EM_IS_FOLDER_TREE_MODEL (model)); g_return_if_fail (CAMEL_IS_STORE (store)); g_return_if_fail (full != NULL); if (unread < 0) return; si = em_folder_tree_model_lookup_store_info (model, store); if (si == NULL) return; reference = g_hash_table_lookup (si->full_hash, full); if (!gtk_tree_row_reference_valid (reference)) return; tree_model = GTK_TREE_MODEL (model); path = gtk_tree_row_reference_get_path (reference); gtk_tree_model_get_iter (tree_model, &iter, path); gtk_tree_path_free (path); gtk_tree_model_get ( tree_model, &iter, COL_UINT_UNREAD_LAST_SEL, &old_unread, -1); gtk_tree_store_set ( GTK_TREE_STORE (model), &iter, COL_UINT_UNREAD, unread, COL_UINT_UNREAD_LAST_SEL, MIN (old_unread, unread), -1); /* Folders are displayed with a bold weight to indicate that * they contain unread messages. We signal that parent rows * have changed here to update them. */ while (gtk_tree_model_iter_parent (tree_model, &parent, &iter)) { path = gtk_tree_model_get_path (tree_model, &parent); gtk_tree_model_row_changed (tree_model, path, &parent); gtk_tree_path_free (path); iter = parent; } } static void em_folder_tree_model_init (EMFolderTreeModel *model) { GHashTable *store_index; store_index = g_hash_table_new_full ( g_direct_hash, g_direct_equal, (GDestroyNotify) NULL, (GDestroyNotify) store_info_free); model->priv = EM_FOLDER_TREE_MODEL_GET_PRIVATE (model); model->priv->store_index = store_index; } EMFolderTreeModel * em_folder_tree_model_new (void) { return g_object_new (EM_TYPE_FOLDER_TREE_MODEL, NULL); } EMFolderTreeModel * em_folder_tree_model_get_default (void) { static EMFolderTreeModel *default_folder_tree_model; if (G_UNLIKELY (default_folder_tree_model == NULL)) default_folder_tree_model = em_folder_tree_model_new (); return default_folder_tree_model; } GtkTreeSelection * em_folder_tree_model_get_selection (EMFolderTreeModel *model) { g_return_val_if_fail (EM_IS_FOLDER_TREE_MODEL (model), NULL); return GTK_TREE_SELECTION (model->priv->selection); } void em_folder_tree_model_set_selection (EMFolderTreeModel *model, GtkTreeSelection *selection) { g_return_if_fail (EM_IS_FOLDER_TREE_MODEL (model)); if (selection != NULL) g_return_if_fail (GTK_IS_TREE_SELECTION (selection)); if (model->priv->selection == selection) return; if (model->priv->selection != NULL) { g_object_weak_unref ( G_OBJECT (model->priv->selection), (GWeakNotify) folder_tree_model_selection_finalized_cb, model); model->priv->selection = NULL; } model->priv->selection = selection; if (model->priv->selection != NULL) g_object_weak_ref ( G_OBJECT (model->priv->selection), (GWeakNotify) folder_tree_model_selection_finalized_cb, model); g_object_notify (G_OBJECT (model), "selection"); } EMailSession * em_folder_tree_model_get_session (EMFolderTreeModel *model) { g_return_val_if_fail (EM_IS_FOLDER_TREE_MODEL (model), NULL); return model->priv->session; } void em_folder_tree_model_set_session (EMFolderTreeModel *model, EMailSession *session) { g_return_if_fail (EM_IS_FOLDER_TREE_MODEL (model)); if (model->priv->session == session) return; if (session != NULL) { g_return_if_fail (E_IS_MAIL_SESSION (session)); g_object_ref (session); } if (model->priv->session != NULL) g_object_unref (model->priv->session); model->priv->session = session; /* FIXME Technically we should be disconnecting this signal * when replacing an old session with a new session, * but at present this function is only called once. */ if (session != NULL) { EMailAccountStore *account_store; MailFolderCache *folder_cache; folder_cache = e_mail_session_get_folder_cache (session); account_store = e_mail_ui_session_get_account_store ( E_MAIL_UI_SESSION (session)); /* Keep our own reference since we connect to its signals. */ g_warn_if_fail (model->priv->account_store == NULL); model->priv->account_store = g_object_ref (account_store); /* No need to connect to "service-added" emissions since it's * always immediately followed by either "service-enabled" or * "service-disabled". */ g_signal_connect ( account_store, "service-removed", G_CALLBACK (folder_tree_model_service_removed), model); g_signal_connect ( account_store, "service-enabled", G_CALLBACK (folder_tree_model_service_enabled), model); g_signal_connect ( account_store, "service-disabled", G_CALLBACK (folder_tree_model_service_disabled), model); g_signal_connect ( account_store, "services-reordered", G_CALLBACK (folder_tree_model_services_reordered), model); g_signal_connect_swapped ( folder_cache, "folder-unread-updated", G_CALLBACK (folder_tree_model_set_unread_count), model); } g_object_notify (G_OBJECT (model), "session"); } /* Helper for em_folder_tree_model_set_folder_info() */ static void folder_tree_model_get_drafts_folder_uri (ESourceRegistry *registry, CamelStore *store, gchar **drafts_folder_uri) { ESource *source; const gchar *extension_name; /* In case we fail... */ *drafts_folder_uri = NULL; source = em_utils_ref_mail_identity_for_store (registry, store); if (source == NULL) return; extension_name = E_SOURCE_EXTENSION_MAIL_COMPOSITION; if (e_source_has_extension (source, extension_name)) { ESourceMailComposition *extension; extension = e_source_get_extension (source, extension_name); *drafts_folder_uri = e_source_mail_composition_dup_drafts_folder (extension); } g_object_unref (source); } /* Helper for em_folder_tree_model_set_folder_info() */ static void folder_tree_model_get_sent_folder_uri (ESourceRegistry *registry, CamelStore *store, gchar **sent_folder_uri) { ESource *source; const gchar *extension_name; /* In case we fail... */ *sent_folder_uri = NULL; source = em_utils_ref_mail_identity_for_store (registry, store); if (source == NULL) return; extension_name = E_SOURCE_EXTENSION_MAIL_SUBMISSION; if (e_source_has_extension (source, extension_name)) { ESourceMailSubmission *extension; extension = e_source_get_extension (source, extension_name); *sent_folder_uri = e_source_mail_submission_dup_sent_folder (extension); } g_object_unref (source); } void em_folder_tree_model_set_folder_info (EMFolderTreeModel *model, GtkTreeIter *iter, EMFolderTreeModelStoreInfo *si, CamelFolderInfo *fi, gint fully_loaded) { GtkTreeRowReference *path_row; GtkTreeStore *tree_store; MailFolderCache *folder_cache; ESourceRegistry *registry; EMailSession *session; guint unread; GtkTreePath *path; GtkTreeIter sub; CamelFolder *folder; gboolean emitted = FALSE; const gchar *uid; const gchar *icon_name; const gchar *display_name; guint32 flags, add_flags = 0; EMEventTargetCustomIcon *target; gboolean load = FALSE; gboolean folder_is_drafts = FALSE; gboolean folder_is_outbox = FALSE; gboolean folder_is_templates = FALSE; gboolean store_is_local; gchar *uri; /* Make sure we don't already know about it. */ if (g_hash_table_lookup (si->full_hash, fi->full_name)) return; tree_store = GTK_TREE_STORE (model); session = em_folder_tree_model_get_session (model); folder_cache = e_mail_session_get_folder_cache (session); registry = e_mail_session_get_registry (session); uid = camel_service_get_uid (CAMEL_SERVICE (si->store)); store_is_local = (g_strcmp0 (uid, E_MAIL_SESSION_LOCAL_UID) == 0); if (!fully_loaded) load = (fi->child == NULL) && !(fi->flags & (CAMEL_FOLDER_NOCHILDREN | CAMEL_FOLDER_NOINFERIORS)); path = gtk_tree_model_get_path (GTK_TREE_MODEL (model), iter); path_row = gtk_tree_row_reference_new (GTK_TREE_MODEL (model), path); gtk_tree_path_free (path); uri = e_mail_folder_uri_build (si->store, fi->full_name); g_hash_table_insert ( si->full_hash, g_strdup (fi->full_name), path_row); /* XXX If we have the folder, and its the Outbox folder, we need * the total count, not unread. We do the same for Drafts. */ /* XXX This is duplicated in mail-folder-cache too, should perhaps * be functionised. */ unread = fi->unread; if (mail_folder_cache_get_folder_from_uri ( folder_cache, uri, &folder) && folder) { folder_is_drafts = em_utils_folder_is_drafts (registry, folder); folder_is_outbox = em_utils_folder_is_outbox (registry, folder); if (folder_is_drafts || folder_is_outbox) { gint total; if ((total = camel_folder_get_message_count (folder)) > 0) { gint deleted = camel_folder_get_deleted_message_count (folder); if (deleted != -1) total -= deleted; } unread = total > 0 ? total : 0; } g_object_unref (folder); } flags = fi->flags; display_name = fi->display_name; if (store_is_local) { if (strcmp (fi->full_name, "Drafts") == 0) { folder_is_drafts = TRUE; display_name = _("Drafts"); } else if (strcmp (fi->full_name, "Templates") == 0) { folder_is_templates = TRUE; display_name = _("Templates"); } else if (strcmp (fi->full_name, "Inbox") == 0) { flags = (flags & ~CAMEL_FOLDER_TYPE_MASK) | CAMEL_FOLDER_TYPE_INBOX; display_name = _("Inbox"); } else if (strcmp (fi->full_name, "Outbox") == 0) { flags = (flags & ~CAMEL_FOLDER_TYPE_MASK) | CAMEL_FOLDER_TYPE_OUTBOX; display_name = _("Outbox"); } else if (strcmp (fi->full_name, "Sent") == 0) { flags = (flags & ~CAMEL_FOLDER_TYPE_MASK) | CAMEL_FOLDER_TYPE_SENT; display_name = _("Sent"); } } if ((flags & CAMEL_FOLDER_TYPE_MASK) == 0) { gchar *drafts_folder_uri; gchar *sent_folder_uri; folder_tree_model_get_drafts_folder_uri ( registry, si->store, &drafts_folder_uri); folder_tree_model_get_sent_folder_uri ( registry, si->store, &sent_folder_uri); if (!folder_is_drafts && drafts_folder_uri != NULL) { folder_is_drafts = e_mail_folder_uri_equal ( CAMEL_SESSION (session), uri, drafts_folder_uri); } if (sent_folder_uri != NULL) { if (e_mail_folder_uri_equal ( CAMEL_SESSION (session), uri, sent_folder_uri)) { add_flags = CAMEL_FOLDER_TYPE_SENT; } } g_free (drafts_folder_uri); g_free (sent_folder_uri); } /* Choose an icon name for the folder. */ icon_name = em_folder_utils_get_icon_name (flags | add_flags); if (g_str_equal (icon_name, "folder")) { if (folder_is_drafts) icon_name = "accessories-text-editor"; else if (folder_is_templates) icon_name = "text-x-generic-template"; } gtk_tree_store_set ( tree_store, iter, COL_STRING_DISPLAY_NAME, display_name, COL_POINTER_CAMEL_STORE, si->store, COL_STRING_FULL_NAME, fi->full_name, COL_STRING_ICON_NAME, icon_name, COL_UINT_FLAGS, flags, COL_BOOL_IS_STORE, FALSE, COL_BOOL_IS_FOLDER, TRUE, COL_BOOL_LOAD_SUBDIRS, load, COL_UINT_UNREAD_LAST_SEL, 0, COL_BOOL_IS_DRAFT, folder_is_drafts, -1); g_free (uri); uri = NULL; target = em_event_target_new_custom_icon ( em_event_peek (), tree_store, iter, fi->full_name, EM_EVENT_CUSTOM_ICON); e_event_emit ( (EEvent *) em_event_peek (), "folder.customicon", (EEventTarget *) target); if (unread != ~0) gtk_tree_store_set ( tree_store, iter, COL_UINT_UNREAD, unread, COL_UINT_UNREAD_LAST_SEL, unread, -1); if (load) { /* create a placeholder node for our subfolders... */ gtk_tree_store_append (tree_store, &sub, iter); gtk_tree_store_set ( tree_store, &sub, COL_STRING_DISPLAY_NAME, _("Loading..."), COL_POINTER_CAMEL_STORE, si->store, COL_STRING_FULL_NAME, NULL, COL_STRING_ICON_NAME, NULL, COL_BOOL_LOAD_SUBDIRS, FALSE, COL_BOOL_IS_STORE, FALSE, COL_BOOL_IS_FOLDER, FALSE, COL_UINT_UNREAD, 0, COL_UINT_UNREAD_LAST_SEL, 0, COL_BOOL_IS_DRAFT, FALSE, -1); path = gtk_tree_model_get_path (GTK_TREE_MODEL (model), iter); g_signal_emit (model, signals[LOADING_ROW], 0, path, iter); gtk_tree_path_free (path); return; } if (fi->child) { fi = fi->child; do { gtk_tree_store_append (tree_store, &sub, iter); if (!emitted) { path = gtk_tree_model_get_path ( GTK_TREE_MODEL (model), iter); g_signal_emit ( model, signals[LOADED_ROW], 0, path, iter); gtk_tree_path_free (path); emitted = TRUE; } em_folder_tree_model_set_folder_info ( model, &sub, si, fi, fully_loaded); fi = fi->next; } while (fi); } if (!emitted) { path = gtk_tree_model_get_path (GTK_TREE_MODEL (model), iter); g_signal_emit (model, signals[LOADED_ROW], 0, path, iter); gtk_tree_path_free (path); } } static void folder_subscribed_cb (CamelStore *store, CamelFolderInfo *fi, EMFolderTreeModel *model) { EMFolderTreeModelStoreInfo *si; GtkTreeRowReference *reference; GtkTreeIter parent, iter; GtkTreePath *path; gboolean load; gchar *dirname, *p; si = em_folder_tree_model_lookup_store_info (model, store); if (si == NULL) return; /* Make sure we don't already know about it? */ if (g_hash_table_lookup (si->full_hash, fi->full_name)) return; /* Get our parent folder's path. */ dirname = g_alloca (strlen (fi->full_name) + 1); strcpy (dirname, fi->full_name); p = strrchr (dirname, '/'); if (p == NULL) { /* User subscribed to a toplevel folder. */ reference = si->row; } else { *p = 0; reference = g_hash_table_lookup (si->full_hash, dirname); } if (!gtk_tree_row_reference_valid (reference)) return; path = gtk_tree_row_reference_get_path (reference); gtk_tree_model_get_iter (GTK_TREE_MODEL (model), &parent, path); gtk_tree_path_free (path); /* Make sure parent's subfolders have already been loaded. */ gtk_tree_model_get ( GTK_TREE_MODEL (model), &parent, COL_BOOL_LOAD_SUBDIRS, &load, -1); if (load) return; gtk_tree_store_append (GTK_TREE_STORE (model), &iter, &parent); em_folder_tree_model_set_folder_info (model, &iter, si, fi, TRUE); } static void folder_unsubscribed_cb (CamelStore *store, CamelFolderInfo *fi, EMFolderTreeModel *model) { EMFolderTreeModelStoreInfo *si; GtkTreeRowReference *reference; GtkTreePath *path; GtkTreeIter iter; si = em_folder_tree_model_lookup_store_info (model, store); if (si == NULL) return; reference = g_hash_table_lookup (si->full_hash, fi->full_name); if (!gtk_tree_row_reference_valid (reference)) return; path = gtk_tree_row_reference_get_path (reference); gtk_tree_model_get_iter (GTK_TREE_MODEL (model), &iter, path); gtk_tree_path_free (path); em_folder_tree_model_remove_folders (model, si, &iter); } static void folder_created_cb (CamelStore *store, CamelFolderInfo *fi, EMFolderTreeModel *model) { EMFolderTreeModelStoreInfo *si; /* We only want created events to do more * work if we don't support subscriptions. */ if (CAMEL_IS_SUBSCRIBABLE (store)) return; /* process "folder-created" event only when store already loaded */ si = em_folder_tree_model_lookup_store_info (model, store); if (si == NULL || g_hash_table_size (si->full_hash) == 0) return; folder_subscribed_cb (store, fi, model); } static void folder_deleted_cb (CamelStore *store, CamelFolderInfo *fi, EMFolderTreeModel *model) { /* We only want deleted events to do more * work if we don't support subscriptions. */ if (CAMEL_IS_SUBSCRIBABLE (store)) return; folder_unsubscribed_cb (store, fi, model); } typedef struct { gchar *old_base; CamelFolderInfo *new; } RenameInfo; static void folder_renamed_cb (CamelStore *store, const gchar *old_name, CamelFolderInfo *info, EMFolderTreeModel *model) { EMFolderTreeModelStoreInfo *si; GtkTreeRowReference *reference; GtkTreeIter root, iter; GtkTreePath *path; gchar *parent, *p; si = em_folder_tree_model_lookup_store_info (model, store); if (si == NULL) return; reference = g_hash_table_lookup (si->full_hash, old_name); if (!gtk_tree_row_reference_valid (reference)) return; path = gtk_tree_row_reference_get_path (reference); gtk_tree_model_get_iter (GTK_TREE_MODEL (model), &iter, path); gtk_tree_path_free (path); em_folder_tree_model_remove_folders (model, si, &iter); parent = g_strdup (info->full_name); p = strrchr (parent, '/'); if (p) *p = 0; if (p == NULL || parent == p) /* renamed to a toplevel folder on the store */ reference = si->row; else reference = g_hash_table_lookup (si->full_hash, parent); g_free (parent); if (!gtk_tree_row_reference_valid (reference)) return; path = gtk_tree_row_reference_get_path (reference); gtk_tree_model_get_iter (GTK_TREE_MODEL (model), &root, path); gtk_tree_path_free (path); gtk_tree_store_append (GTK_TREE_STORE (model), &iter, &root); em_folder_tree_model_set_folder_info (model, &iter, si, info, TRUE); } void em_folder_tree_model_add_store (EMFolderTreeModel *model, CamelStore *store) { EMFolderTreeModelStoreInfo *si; GtkTreeRowReference *reference; GtkTreeStore *tree_store; GtkTreeIter root, iter; GtkTreePath *path; CamelService *service; CamelProvider *provider; CamelURL *service_url; const gchar *display_name; gchar *uri; g_return_if_fail (EM_IS_FOLDER_TREE_MODEL (model)); g_return_if_fail (CAMEL_IS_STORE (store)); tree_store = GTK_TREE_STORE (model); service = CAMEL_SERVICE (store); provider = camel_service_get_provider (service); display_name = camel_service_get_display_name (service); /* Ignore stores that should not be added to the tree model. */ if (provider == NULL) return; if ((provider->flags & CAMEL_PROVIDER_IS_STORAGE) == 0) return; service_url = camel_service_new_camel_url (service); if (em_utils_is_local_delivery_mbox_file (service_url)) { camel_url_free (service_url); return; } uri = camel_url_to_string (service_url, CAMEL_URL_HIDE_ALL); camel_url_free (service_url); si = em_folder_tree_model_lookup_store_info (model, store); if (si != NULL) em_folder_tree_model_remove_store (model, store); /* Add the store to the tree. */ gtk_tree_store_append (tree_store, &iter, NULL); gtk_tree_store_set ( tree_store, &iter, COL_STRING_DISPLAY_NAME, display_name, COL_POINTER_CAMEL_STORE, store, COL_STRING_FULL_NAME, NULL, COL_BOOL_LOAD_SUBDIRS, TRUE, COL_BOOL_IS_STORE, TRUE, -1); g_free (uri); path = gtk_tree_model_get_path (GTK_TREE_MODEL (model), &iter); reference = gtk_tree_row_reference_new (GTK_TREE_MODEL (model), path); si = g_new0 (EMFolderTreeModelStoreInfo, 1); si->store = g_object_ref (store); si->row = reference; /* takes ownership */ si->full_hash = g_hash_table_new_full ( g_str_hash, g_str_equal, (GDestroyNotify) g_free, (GDestroyNotify) gtk_tree_row_reference_free); g_hash_table_insert (model->priv->store_index, store, si); /* Each store has folders, but we don't load them until * the user demands them. */ root = iter; gtk_tree_store_append (tree_store, &iter, &root); gtk_tree_store_set ( tree_store, &iter, COL_STRING_DISPLAY_NAME, _("Loading..."), COL_POINTER_CAMEL_STORE, store, COL_STRING_FULL_NAME, NULL, COL_BOOL_LOAD_SUBDIRS, FALSE, COL_BOOL_IS_STORE, FALSE, COL_BOOL_IS_FOLDER, FALSE, COL_UINT_UNREAD, 0, COL_UINT_UNREAD_LAST_SEL, 0, COL_BOOL_IS_DRAFT, FALSE, -1); /* Listen to store events. */ si->created_id = g_signal_connect ( store, "folder-created", G_CALLBACK (folder_created_cb), model); si->deleted_id = g_signal_connect ( store, "folder-deleted", G_CALLBACK (folder_deleted_cb), model); si->renamed_id = g_signal_connect ( store, "folder_renamed", G_CALLBACK (folder_renamed_cb), model); if (CAMEL_IS_SUBSCRIBABLE (store)) { si->subscribed_id = g_signal_connect ( store, "folder_subscribed", G_CALLBACK (folder_subscribed_cb), model); si->unsubscribed_id = g_signal_connect ( store, "folder_unsubscribed", G_CALLBACK (folder_unsubscribed_cb), model); } g_signal_emit (model, signals[LOADED_ROW], 0, path, &root); gtk_tree_path_free (path); } void em_folder_tree_model_remove_store (EMFolderTreeModel *model, CamelStore *store) { EMFolderTreeModelStoreInfo *si; GtkTreePath *path; GtkTreeIter iter; g_return_if_fail (EM_IS_FOLDER_TREE_MODEL (model)); g_return_if_fail (CAMEL_IS_STORE (store)); si = em_folder_tree_model_lookup_store_info (model, store); if (si == NULL) return; path = gtk_tree_row_reference_get_path (si->row); gtk_tree_model_get_iter (GTK_TREE_MODEL (model), &iter, path); gtk_tree_path_free (path); /* recursively remove subfolders and finally the toplevel store */ em_folder_tree_model_remove_folders (model, si, &iter); } GList * em_folder_tree_model_list_stores (EMFolderTreeModel *model) { g_return_val_if_fail (EM_IS_FOLDER_TREE_MODEL (model), NULL); return g_hash_table_get_keys (model->priv->store_index); } void em_folder_tree_model_remove_folders (EMFolderTreeModel *model, EMFolderTreeModelStoreInfo *si, GtkTreeIter *toplevel) { gchar *full_name; gboolean is_store, go; GtkTreeIter iter; if (gtk_tree_model_iter_children (GTK_TREE_MODEL (model), &iter, toplevel)) { do { GtkTreeIter next = iter; go = gtk_tree_model_iter_next (GTK_TREE_MODEL (model), &next); em_folder_tree_model_remove_folders (model, si, &iter); iter = next; } while (go); } gtk_tree_model_get ( GTK_TREE_MODEL (model), toplevel, COL_STRING_FULL_NAME, &full_name, COL_BOOL_IS_STORE, &is_store, -1); if (full_name != NULL) g_hash_table_remove (si->full_hash, full_name); gtk_tree_store_remove ((GtkTreeStore *) model, toplevel); /* Freeing the GtkTreeRowReference in the store info may finalize * the model. Keep the model alive until the store info is fully * removed from the hash table. */ if (is_store) { g_object_ref (model); g_hash_table_remove (model->priv->store_index, si->store); g_object_unref (model); } g_free (full_name); } gboolean em_folder_tree_model_is_type_inbox (EMFolderTreeModel *model, CamelStore *store, const gchar *full) { EMFolderTreeModelStoreInfo *si; GtkTreeRowReference *reference; GtkTreePath *path; GtkTreeIter iter; guint32 flags; g_return_val_if_fail (EM_IS_FOLDER_TREE_MODEL (model), FALSE); g_return_val_if_fail (CAMEL_IS_STORE (store), FALSE); g_return_val_if_fail (full != NULL, FALSE); si = em_folder_tree_model_lookup_store_info (model, store); if (si == NULL) return FALSE; reference = g_hash_table_lookup (si->full_hash, full); if (!gtk_tree_row_reference_valid (reference)) return FALSE; path = gtk_tree_row_reference_get_path (reference); gtk_tree_model_get_iter (GTK_TREE_MODEL (model), &iter, path); gtk_tree_path_free (path); gtk_tree_model_get ( GTK_TREE_MODEL (model), &iter, COL_UINT_FLAGS, &flags, -1); return ((flags & CAMEL_FOLDER_TYPE_MASK) == CAMEL_FOLDER_TYPE_INBOX); } gchar * em_folder_tree_model_get_folder_name (EMFolderTreeModel *model, CamelStore *store, const gchar *full) { EMFolderTreeModelStoreInfo *si; GtkTreeRowReference *reference; GtkTreePath *path; GtkTreeIter iter; gchar *name = NULL; g_return_val_if_fail (EM_IS_FOLDER_TREE_MODEL (model), NULL); g_return_val_if_fail (CAMEL_IS_STORE (store), NULL); g_return_val_if_fail (full != NULL, NULL); si = em_folder_tree_model_lookup_store_info (model, store); if (si == NULL) return NULL; reference = g_hash_table_lookup (si->full_hash, full); if (!gtk_tree_row_reference_valid (reference)) return NULL; path = gtk_tree_row_reference_get_path (reference); gtk_tree_model_get_iter (GTK_TREE_MODEL (model), &iter, path); gtk_tree_path_free (path); gtk_tree_model_get ( GTK_TREE_MODEL (model), &iter, COL_STRING_DISPLAY_NAME, &name, -1); return name; } EMFolderTreeModelStoreInfo * em_folder_tree_model_lookup_store_info (EMFolderTreeModel *model, CamelStore *store) { g_return_val_if_fail (EM_IS_FOLDER_TREE_MODEL (model), NULL); g_return_val_if_fail (CAMEL_IS_STORE (store), NULL); return g_hash_table_lookup (model->priv->store_index, store); } GtkTreeRowReference * em_folder_tree_model_lookup_uri (EMFolderTreeModel *model, const gchar *folder_uri) { GtkTreeRowReference *reference = NULL; EMFolderTreeModelStoreInfo *si; EMailSession *session; CamelStore *store = NULL; gchar *folder_name = NULL; GError *error = NULL; g_return_val_if_fail (EM_IS_FOLDER_TREE_MODEL (model), NULL); g_return_val_if_fail (folder_uri != NULL, NULL); session = em_folder_tree_model_get_session (model); g_return_val_if_fail (E_IS_MAIL_SESSION (session), NULL); e_mail_folder_uri_parse ( CAMEL_SESSION (session), folder_uri, &store, &folder_name, &error); if (error != NULL) { g_warn_if_fail (store == NULL); g_warn_if_fail (folder_name == NULL); g_warning ("%s: %s", G_STRFUNC, error->message); g_error_free (error); return NULL; } g_return_val_if_fail (CAMEL_IS_STORE (store), NULL); g_return_val_if_fail (folder_name != NULL, NULL); si = g_hash_table_lookup (model->priv->store_index, store); if (si != NULL) reference = g_hash_table_lookup (si->full_hash, folder_name); if (!gtk_tree_row_reference_valid (reference)) reference = NULL; g_object_unref (store); g_free (folder_name); return reference; } void em_folder_tree_model_user_marked_unread (EMFolderTreeModel *model, CamelFolder *folder, guint n_marked) { GtkTreeRowReference *reference; GtkTreePath *path; GtkTreeIter iter; gchar *folder_uri; guint unread; /* The user marked messages in the given folder as unread. * Update our unread counts so we don't misinterpret this * event as new mail arriving. */ g_return_if_fail (EM_IS_FOLDER_TREE_MODEL (model)); g_return_if_fail (CAMEL_IS_FOLDER (folder)); folder_uri = e_mail_folder_uri_from_folder (folder); reference = em_folder_tree_model_lookup_uri (model, folder_uri); g_free (folder_uri); g_return_if_fail (gtk_tree_row_reference_valid (reference)); path = gtk_tree_row_reference_get_path (reference); gtk_tree_model_get_iter (GTK_TREE_MODEL (model), &iter, path); gtk_tree_path_free (path); gtk_tree_model_get ( GTK_TREE_MODEL (model), &iter, COL_UINT_UNREAD, &unread, -1); unread += n_marked; gtk_tree_store_set ( GTK_TREE_STORE (model), &iter, COL_UINT_UNREAD_LAST_SEL, unread, COL_UINT_UNREAD, unread, -1); }