/* * e-attachment-store.c * * 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 * * * Copyright (C) 1999-2008 Novell, Inc. (www.novell.com) * */ #ifdef HAVE_CONFIG_H #include #endif #include "e-attachment-store.h" #include "e-icon-factory.h" #include #include #include #include "e-mktemp.h" #define E_ATTACHMENT_STORE_GET_PRIVATE(obj) \ (G_TYPE_INSTANCE_GET_PRIVATE \ ((obj), E_TYPE_ATTACHMENT_STORE, EAttachmentStorePrivate)) struct _EAttachmentStorePrivate { GHashTable *attachment_index; guint ignore_row_changed : 1; }; enum { PROP_0, PROP_NUM_ATTACHMENTS, PROP_NUM_LOADING, PROP_TOTAL_SIZE }; G_DEFINE_TYPE ( EAttachmentStore, e_attachment_store, GTK_TYPE_LIST_STORE) static void attachment_store_get_property (GObject *object, guint property_id, GValue *value, GParamSpec *pspec) { switch (property_id) { case PROP_NUM_ATTACHMENTS: g_value_set_uint ( value, e_attachment_store_get_num_attachments ( E_ATTACHMENT_STORE (object))); return; case PROP_NUM_LOADING: g_value_set_uint ( value, e_attachment_store_get_num_loading ( E_ATTACHMENT_STORE (object))); return; case PROP_TOTAL_SIZE: g_value_set_uint64 ( value, e_attachment_store_get_total_size ( E_ATTACHMENT_STORE (object))); return; } G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec); } static void attachment_store_dispose (GObject *object) { e_attachment_store_remove_all (E_ATTACHMENT_STORE (object)); /* Chain up to parent's dispose() method. */ G_OBJECT_CLASS (e_attachment_store_parent_class)->dispose (object); } static void attachment_store_finalize (GObject *object) { EAttachmentStorePrivate *priv; priv = E_ATTACHMENT_STORE_GET_PRIVATE (object); g_hash_table_destroy (priv->attachment_index); /* Chain up to parent's finalize() method. */ G_OBJECT_CLASS (e_attachment_store_parent_class)->finalize (object); } static void e_attachment_store_class_init (EAttachmentStoreClass *class) { GObjectClass *object_class; g_type_class_add_private (class, sizeof (EAttachmentStorePrivate)); object_class = G_OBJECT_CLASS (class); object_class->get_property = attachment_store_get_property; object_class->dispose = attachment_store_dispose; object_class->finalize = attachment_store_finalize; g_object_class_install_property ( object_class, PROP_NUM_ATTACHMENTS, g_param_spec_uint ( "num-attachments", "Num Attachments", NULL, 0, G_MAXUINT, 0, G_PARAM_READABLE)); g_object_class_install_property ( object_class, PROP_NUM_LOADING, g_param_spec_uint ( "num-loading", "Num Loading", NULL, 0, G_MAXUINT, 0, G_PARAM_READABLE)); g_object_class_install_property ( object_class, PROP_TOTAL_SIZE, g_param_spec_uint64 ( "total-size", "Total Size", NULL, 0, G_MAXUINT64, 0, G_PARAM_READABLE)); } static void e_attachment_store_init (EAttachmentStore *store) { GType types[E_ATTACHMENT_STORE_NUM_COLUMNS]; GHashTable *attachment_index; gint column = 0; attachment_index = g_hash_table_new_full ( g_direct_hash, g_direct_equal, (GDestroyNotify) g_object_unref, (GDestroyNotify) gtk_tree_row_reference_free); store->priv = E_ATTACHMENT_STORE_GET_PRIVATE (store); store->priv->attachment_index = attachment_index; types[column++] = E_TYPE_ATTACHMENT; /* COLUMN_ATTACHMENT */ types[column++] = G_TYPE_STRING; /* COLUMN_CAPTION */ types[column++] = G_TYPE_STRING; /* COLUMN_CONTENT_TYPE */ types[column++] = G_TYPE_STRING; /* COLUMN_DESCRIPTION */ types[column++] = G_TYPE_ICON; /* COLUMN_ICON */ types[column++] = G_TYPE_BOOLEAN; /* COLUMN_LOADING */ types[column++] = G_TYPE_INT; /* COLUMN_PERCENT */ types[column++] = G_TYPE_BOOLEAN; /* COLUMN_SAVING */ types[column++] = G_TYPE_UINT64; /* COLUMN_SIZE */ g_assert (column == E_ATTACHMENT_STORE_NUM_COLUMNS); gtk_list_store_set_column_types ( GTK_LIST_STORE (store), G_N_ELEMENTS (types), types); } GtkTreeModel * e_attachment_store_new (void) { return g_object_new (E_TYPE_ATTACHMENT_STORE, NULL); } void e_attachment_store_add_attachment (EAttachmentStore *store, EAttachment *attachment) { GtkTreeRowReference *reference; GtkTreeModel *model; GtkTreePath *path; GtkTreeIter iter; g_return_if_fail (E_IS_ATTACHMENT_STORE (store)); g_return_if_fail (E_IS_ATTACHMENT (attachment)); gtk_list_store_append (GTK_LIST_STORE (store), &iter); gtk_list_store_set ( GTK_LIST_STORE (store), &iter, E_ATTACHMENT_STORE_COLUMN_ATTACHMENT, attachment, -1); model = GTK_TREE_MODEL (store); path = gtk_tree_model_get_path (model, &iter); reference = gtk_tree_row_reference_new (model, path); gtk_tree_path_free (path); g_hash_table_insert ( store->priv->attachment_index, g_object_ref (attachment), reference); /* This lets the attachment tell us when to update. */ e_attachment_set_reference (attachment, reference); g_object_freeze_notify (G_OBJECT (store)); g_object_notify (G_OBJECT (store), "num-attachments"); g_object_notify (G_OBJECT (store), "total-size"); g_object_thaw_notify (G_OBJECT (store)); } gboolean e_attachment_store_remove_attachment (EAttachmentStore *store, EAttachment *attachment) { GtkTreeRowReference *reference; GHashTable *hash_table; GtkTreeModel *model; GtkTreePath *path; GtkTreeIter iter; g_return_val_if_fail (E_IS_ATTACHMENT_STORE (store), FALSE); g_return_val_if_fail (E_IS_ATTACHMENT (attachment), FALSE); hash_table = store->priv->attachment_index; reference = g_hash_table_lookup (hash_table, attachment); if (reference == NULL) return FALSE; if (!gtk_tree_row_reference_valid (reference)) { g_hash_table_remove (hash_table, attachment); return FALSE; } e_attachment_cancel (attachment); e_attachment_set_reference (attachment, NULL); model = gtk_tree_row_reference_get_model (reference); path = gtk_tree_row_reference_get_path (reference); gtk_tree_model_get_iter (model, &iter, path); gtk_tree_path_free (path); gtk_list_store_remove (GTK_LIST_STORE (store), &iter); g_hash_table_remove (hash_table, attachment); g_object_freeze_notify (G_OBJECT (store)); g_object_notify (G_OBJECT (store), "num-attachments"); g_object_notify (G_OBJECT (store), "total-size"); g_object_thaw_notify (G_OBJECT (store)); return TRUE; } void e_attachment_store_remove_all (EAttachmentStore *store) { GList *list, *iter; g_return_if_fail (E_IS_ATTACHMENT_STORE (store)); if (!g_hash_table_size (store->priv->attachment_index)) return; /* Clear the list store before cancelling EAttachment load/save * operations. This will invalidate the EAttachment's tree row * reference so it won't try to update the row's icon column in * response to the cancellation. That can create problems when * the list store is being disposed. */ gtk_list_store_clear (GTK_LIST_STORE (store)); g_object_freeze_notify (G_OBJECT (store)); list = e_attachment_store_get_attachments (store); for (iter = list; iter; iter = iter->next) { EAttachment *attachment = iter->data; e_attachment_cancel (attachment); g_hash_table_remove (store->priv->attachment_index, iter->data); } g_list_foreach (list, (GFunc) g_object_unref, NULL); g_list_free (list); g_object_notify (G_OBJECT (store), "num-attachments"); g_object_notify (G_OBJECT (store), "total-size"); g_object_thaw_notify (G_OBJECT (store)); } void e_attachment_store_add_to_multipart (EAttachmentStore *store, CamelMultipart *multipart, const gchar *default_charset) { GList *list, *iter; g_return_if_fail (E_IS_ATTACHMENT_STORE (store)); g_return_if_fail (CAMEL_MULTIPART (multipart)); list = e_attachment_store_get_attachments (store); for (iter = list; iter != NULL; iter = iter->next) { EAttachment *attachment = iter->data; /* Skip the attachment if it's still loading. */ if (!e_attachment_get_loading (attachment)) e_attachment_add_to_multipart ( attachment, multipart, default_charset); } g_list_foreach (list, (GFunc) g_object_unref, NULL); g_list_free (list); } GList * e_attachment_store_get_attachments (EAttachmentStore *store) { GList *list = NULL; GtkTreeModel *model; GtkTreeIter iter; gboolean valid; g_return_val_if_fail (E_IS_ATTACHMENT_STORE (store), NULL); model = GTK_TREE_MODEL (store); valid = gtk_tree_model_get_iter_first (model, &iter); while (valid) { EAttachment *attachment; gint column_id; column_id = E_ATTACHMENT_STORE_COLUMN_ATTACHMENT; gtk_tree_model_get (model, &iter, column_id, &attachment, -1); list = g_list_prepend (list, attachment); valid = gtk_tree_model_iter_next (model, &iter); } return g_list_reverse (list); } guint e_attachment_store_get_num_attachments (EAttachmentStore *store) { g_return_val_if_fail (E_IS_ATTACHMENT_STORE (store), 0); return g_hash_table_size (store->priv->attachment_index); } guint e_attachment_store_get_num_loading (EAttachmentStore *store) { GList *list, *iter; guint num_loading = 0; g_return_val_if_fail (E_IS_ATTACHMENT_STORE (store), 0); list = e_attachment_store_get_attachments (store); for (iter = list; iter != NULL; iter = iter->next) { EAttachment *attachment = iter->data; if (e_attachment_get_loading (attachment)) num_loading++; } g_list_foreach (list, (GFunc) g_object_unref, NULL); g_list_free (list); return num_loading; } goffset e_attachment_store_get_total_size (EAttachmentStore *store) { GList *list, *iter; goffset total_size = 0; g_return_val_if_fail (E_IS_ATTACHMENT_STORE (store), 0); list = e_attachment_store_get_attachments (store); for (iter = list; iter != NULL; iter = iter->next) { EAttachment *attachment = iter->data; GFileInfo *file_info; file_info = e_attachment_ref_file_info (attachment); if (file_info != NULL) { total_size += g_file_info_get_size (file_info); g_object_unref (file_info); } } g_list_foreach (list, (GFunc) g_object_unref, NULL); g_list_free (list); return total_size; } static void update_preview_cb (GtkFileChooser *file_chooser, gpointer data) { GtkWidget *preview; gchar *filename = NULL; GdkPixbuf *pixbuf; gtk_file_chooser_set_preview_widget_active (file_chooser, FALSE); gtk_image_clear (GTK_IMAGE (data)); preview = GTK_WIDGET (data); filename = gtk_file_chooser_get_preview_filename (file_chooser); if (filename == NULL) return; pixbuf = gdk_pixbuf_new_from_file_at_size (filename, 128, 128, NULL); g_free (filename); if (!pixbuf) return; gtk_file_chooser_set_preview_widget_active (file_chooser, TRUE); gtk_image_set_from_pixbuf (GTK_IMAGE (preview), pixbuf); g_object_unref (pixbuf); } void e_attachment_store_run_load_dialog (EAttachmentStore *store, GtkWindow *parent) { GtkFileChooser *file_chooser; GtkWidget *dialog; GtkWidget *option; GtkImage *preview; GSList *files, *iter; const gchar *disposition; gboolean active; gint response; g_return_if_fail (E_IS_ATTACHMENT_STORE (store)); g_return_if_fail (GTK_IS_WINDOW (parent)); dialog = gtk_file_chooser_dialog_new ( _("Add Attachment"), parent, GTK_FILE_CHOOSER_ACTION_OPEN, GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL, _("A_ttach"), GTK_RESPONSE_OK, NULL); file_chooser = GTK_FILE_CHOOSER (dialog); gtk_file_chooser_set_local_only (file_chooser, FALSE); gtk_file_chooser_set_select_multiple (file_chooser, TRUE); gtk_dialog_set_default_response (GTK_DIALOG (dialog), GTK_RESPONSE_OK); gtk_window_set_icon_name (GTK_WINDOW (dialog), "mail-attachment"); preview = GTK_IMAGE (gtk_image_new ()); gtk_file_chooser_set_preview_widget ( GTK_FILE_CHOOSER (file_chooser), GTK_WIDGET (preview)); g_signal_connect ( file_chooser, "update-preview", G_CALLBACK (update_preview_cb), preview); option = gtk_check_button_new_with_mnemonic ( _("_Suggest automatic display of attachment")); gtk_file_chooser_set_extra_widget (file_chooser, option); gtk_widget_show (option); response = gtk_dialog_run (GTK_DIALOG (dialog)); if (response != GTK_RESPONSE_OK) goto exit; files = gtk_file_chooser_get_files (file_chooser); active = gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (option)); disposition = active ? "inline" : "attachment"; for (iter = files; iter != NULL; iter = g_slist_next (iter)) { EAttachment *attachment; GFile *file = iter->data; attachment = e_attachment_new (); e_attachment_set_file (attachment, file); e_attachment_set_disposition (attachment, disposition); e_attachment_store_add_attachment (store, attachment); e_attachment_load_async ( attachment, (GAsyncReadyCallback) e_attachment_load_handle_error, parent); g_object_unref (attachment); } g_slist_foreach (files, (GFunc) g_object_unref, NULL); g_slist_free (files); exit: gtk_widget_destroy (dialog); } GFile * e_attachment_store_run_save_dialog (EAttachmentStore *store, GList *attachment_list, GtkWindow *parent) { GtkFileChooser *file_chooser; GtkFileChooserAction action; GtkWidget *dialog; GtkBox *extra_box; GtkWidget *extra_box_widget; GtkBox *extract_box; GtkWidget *extract_box_widget; GSList *extract_group; GtkWidget *extract_only, *extract_org; GtkToggleButton *extract_only_toggle; GtkWidget *extract_yes; GtkToggleButton *extract_yes_toggle; GFile *destination; const gchar *title; gint response; guint length; g_return_val_if_fail (E_IS_ATTACHMENT_STORE (store), NULL); length = g_list_length (attachment_list); if (length == 0) return NULL; title = ngettext ("Save Attachment", "Save Attachments", length); if (length == 1) action = GTK_FILE_CHOOSER_ACTION_SAVE; else action = GTK_FILE_CHOOSER_ACTION_SELECT_FOLDER; dialog = gtk_file_chooser_dialog_new ( title, parent, action, GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL, GTK_STOCK_SAVE, GTK_RESPONSE_OK, NULL); file_chooser = GTK_FILE_CHOOSER (dialog); gtk_file_chooser_set_local_only (file_chooser, FALSE); gtk_file_chooser_set_do_overwrite_confirmation (file_chooser, TRUE); gtk_dialog_set_default_response (GTK_DIALOG (dialog), GTK_RESPONSE_OK); gtk_window_set_icon_name (GTK_WINDOW (dialog), "mail-attachment"); extra_box_widget = gtk_box_new (GTK_ORIENTATION_VERTICAL, 0); extra_box = GTK_BOX (extra_box_widget); extract_yes = gtk_check_button_new_with_mnemonic ( _("E_xtract files from the attachment if it is an archive")); extract_yes_toggle = GTK_TOGGLE_BUTTON (extract_yes); gtk_box_pack_start (extra_box, extract_yes, FALSE, FALSE, 0); extract_box_widget = gtk_box_new (GTK_ORIENTATION_VERTICAL, 0); extract_box = GTK_BOX (extract_box_widget); gtk_box_pack_start (extra_box, extract_box_widget, FALSE, FALSE, 5); extract_only = gtk_radio_button_new_with_mnemonic (NULL, _("Save extracted files only")); extract_only_toggle = GTK_TOGGLE_BUTTON (extract_only); extract_group = gtk_radio_button_get_group (GTK_RADIO_BUTTON (extract_only)); gtk_box_pack_start (extract_box, extract_only, FALSE, FALSE, 0); extract_org = gtk_radio_button_new_with_mnemonic (extract_group, _("Save extracted files and the original archive")); gtk_box_pack_start (extract_box, extract_org, FALSE, FALSE, 0); g_object_bind_property (extract_yes, "active", extract_box, "visible", 0); gtk_toggle_button_set_active (extract_yes_toggle, TRUE); gtk_toggle_button_set_active (extract_only_toggle, TRUE); gtk_widget_show_all (extra_box_widget); gtk_file_chooser_set_extra_widget (file_chooser, extra_box_widget); if (action == GTK_FILE_CHOOSER_ACTION_SAVE) { EAttachment *attachment; AutoarPref *arpref; GSettings *settings; GFileInfo *file_info; const gchar *name = NULL; gchar *mime_type; attachment = attachment_list->data; file_info = e_attachment_ref_file_info (attachment); if (file_info != NULL) name = g_file_info_get_display_name (file_info); if (name == NULL) /* Translators: Default attachment filename. */ name = _("attachment.dat"); gtk_file_chooser_set_current_name (file_chooser, name); mime_type = e_attachment_dup_mime_type (attachment); settings = g_settings_new (AUTOAR_PREF_DEFAULT_GSCHEMA_ID); arpref = autoar_pref_new_with_gsettings (settings); if (!autoar_pref_check_file_name (arpref, name) && !autoar_pref_check_mime_type_d (arpref, mime_type)) { gtk_toggle_button_set_active (extract_yes_toggle, FALSE); } g_clear_object (&file_info); g_clear_object (&settings); g_clear_object (&arpref); g_free (mime_type); } response = gtk_dialog_run (GTK_DIALOG (dialog)); if (response == GTK_RESPONSE_OK) { gboolean save_self, save_extracted; destination = gtk_file_chooser_get_file (file_chooser); save_self = !gtk_toggle_button_get_active (extract_yes_toggle) || !gtk_toggle_button_get_active (extract_only_toggle); save_extracted = gtk_toggle_button_get_active (extract_yes_toggle); if (action == GTK_FILE_CHOOSER_ACTION_SAVE) { e_attachment_set_save_self (attachment_list->data, save_self); e_attachment_set_save_extracted (attachment_list->data, save_extracted); } else { AutoarPref *arpref; GSettings *settings; GList *iter; settings = g_settings_new (AUTOAR_PREF_DEFAULT_GSCHEMA_ID); arpref = autoar_pref_new_with_gsettings (settings); for (iter = attachment_list; iter != NULL; iter = iter->next) { EAttachment *attachment; GFileInfo *file_info; const gchar *name; gchar *mime_type; attachment = iter->data; file_info = e_attachment_ref_file_info (attachment); name = g_file_info_get_display_name (file_info); mime_type = e_attachment_dup_mime_type (attachment); if ((name != NULL && autoar_pref_check_file_name (arpref, name)) || autoar_pref_check_mime_type_d (arpref, mime_type)) { e_attachment_set_save_self (attachment, save_self); e_attachment_set_save_extracted (attachment, save_extracted); } else { e_attachment_set_save_self (attachment, TRUE); e_attachment_set_save_extracted (attachment, FALSE); } g_object_unref (file_info); g_free (mime_type); } g_object_unref (settings); g_object_unref (arpref); } } else { destination = NULL; } gtk_widget_destroy (dialog); return destination; } /******************** e_attachment_store_get_uris_async() ********************/ typedef struct _UriContext UriContext; struct _UriContext { GSimpleAsyncResult *simple; GList *attachment_list; GError *error; gchar **uris; gint index; }; static UriContext * attachment_store_uri_context_new (EAttachmentStore *store, GList *attachment_list, GAsyncReadyCallback callback, gpointer user_data) { UriContext *uri_context; GSimpleAsyncResult *simple; guint length; gchar **uris; simple = g_simple_async_result_new ( G_OBJECT (store), callback, user_data, e_attachment_store_get_uris_async); /* Add one for NULL terminator. */ length = g_list_length (attachment_list) + 1; uris = g_malloc0 (sizeof (gchar *) * length); uri_context = g_slice_new0 (UriContext); uri_context->simple = simple; uri_context->attachment_list = g_list_copy (attachment_list); uri_context->uris = uris; g_list_foreach ( uri_context->attachment_list, (GFunc) g_object_ref, NULL); return uri_context; } static void attachment_store_uri_context_free (UriContext *uri_context) { g_object_unref (uri_context->simple); /* The attachment list should be empty now. */ g_warn_if_fail (uri_context->attachment_list == NULL); /* So should the error. */ g_warn_if_fail (uri_context->error == NULL); g_strfreev (uri_context->uris); g_slice_free (UriContext, uri_context); } static void attachment_store_get_uris_save_cb (EAttachment *attachment, GAsyncResult *result, UriContext *uri_context) { GSimpleAsyncResult *simple; GFile *file; gchar **uris; gchar *uri; GError *error = NULL; file = e_attachment_save_finish (attachment, result, &error); /* Remove the attachment from the list. */ uri_context->attachment_list = g_list_remove ( uri_context->attachment_list, attachment); g_object_unref (attachment); if (file != NULL) { uri = g_file_get_uri (file); uri_context->uris[uri_context->index++] = uri; g_object_unref (file); } else if (error != NULL) { /* If this is the first error, cancel the other jobs. */ if (uri_context->error == NULL) { g_propagate_error (&uri_context->error, error); g_list_foreach ( uri_context->attachment_list, (GFunc) e_attachment_cancel, NULL); error = NULL; /* Otherwise, we can only report back one error. So if * this is something other than cancellation, dump it to * the terminal. */ } else if (!g_error_matches ( error, G_IO_ERROR, G_IO_ERROR_CANCELLED)) g_warning ("%s", error->message); } if (error != NULL) g_error_free (error); /* If there's still jobs running, let them finish. */ if (uri_context->attachment_list != NULL) return; /* Steal the URI list. */ uris = uri_context->uris; uri_context->uris = NULL; /* And the error. */ error = uri_context->error; uri_context->error = NULL; simple = uri_context->simple; if (error == NULL) g_simple_async_result_set_op_res_gpointer (simple, uris, NULL); else g_simple_async_result_take_error (simple, error); g_simple_async_result_complete (simple); attachment_store_uri_context_free (uri_context); } void e_attachment_store_get_uris_async (EAttachmentStore *store, GList *attachment_list, GAsyncReadyCallback callback, gpointer user_data) { GFile *temp_directory; UriContext *uri_context; GList *iter, *trash = NULL; gchar *template; gchar *path; g_return_if_fail (E_IS_ATTACHMENT_STORE (store)); uri_context = attachment_store_uri_context_new ( store, attachment_list, callback, user_data); /* Grab the copied attachment list. */ attachment_list = uri_context->attachment_list; /* First scan the list for attachments with a GFile. */ for (iter = attachment_list; iter != NULL; iter = iter->next) { EAttachment *attachment = iter->data; GFile *file; file = e_attachment_ref_file (attachment); if (file != NULL) { gchar *uri; uri = g_file_get_uri (file); uri_context->uris[uri_context->index++] = uri; /* Mark the list node for deletion. */ trash = g_list_prepend (trash, iter); g_object_unref (attachment); g_object_unref (file); } } /* Expunge the list. */ for (iter = trash; iter != NULL; iter = iter->next) { GList *link = iter->data; attachment_list = g_list_delete_link (attachment_list, link); } g_list_free (trash); uri_context->attachment_list = attachment_list; /* If we got them all then we're done. */ if (attachment_list == NULL) { GSimpleAsyncResult *simple; gchar **uris; /* Steal the URI list. */ uris = uri_context->uris; uri_context->uris = NULL; simple = uri_context->simple; g_simple_async_result_set_op_res_gpointer (simple, uris, NULL); g_simple_async_result_complete (simple); attachment_store_uri_context_free (uri_context); return; } /* Any remaining attachments in the list should have MIME parts * only, so we need to save them all to a temporary directory. * We use a directory so the files can retain their basenames. * XXX This could trigger a blocking temp directory cleanup. */ template = g_strdup_printf (PACKAGE "-%s-XXXXXX", g_get_user_name ()); path = e_mkdtemp (template); g_free (template); /* XXX Let's hope errno got set properly. */ if (path == NULL) { GSimpleAsyncResult *simple; simple = uri_context->simple; g_simple_async_result_set_error ( simple, G_FILE_ERROR, g_file_error_from_errno (errno), "%s", g_strerror (errno)); g_simple_async_result_complete (simple); attachment_store_uri_context_free (uri_context); return; } temp_directory = g_file_new_for_path (path); for (iter = attachment_list; iter != NULL; iter = iter->next) e_attachment_save_async ( E_ATTACHMENT (iter->data), temp_directory, (GAsyncReadyCallback) attachment_store_get_uris_save_cb, uri_context); g_object_unref (temp_directory); g_free (path); } gchar ** e_attachment_store_get_uris_finish (EAttachmentStore *store, GAsyncResult *result, GError **error) { GSimpleAsyncResult *simple; gchar **uris; g_return_val_if_fail (E_IS_ATTACHMENT_STORE (store), NULL); g_return_val_if_fail (G_IS_SIMPLE_ASYNC_RESULT (result), NULL); simple = G_SIMPLE_ASYNC_RESULT (result); uris = g_simple_async_result_get_op_res_gpointer (simple); g_simple_async_result_propagate_error (simple, error); return uris; } /********************** e_attachment_store_load_async() **********************/ typedef struct _LoadContext LoadContext; struct _LoadContext { GSimpleAsyncResult *simple; GList *attachment_list; GError *error; }; static LoadContext * attachment_store_load_context_new (EAttachmentStore *store, GList *attachment_list, GAsyncReadyCallback callback, gpointer user_data) { LoadContext *load_context; GSimpleAsyncResult *simple; simple = g_simple_async_result_new ( G_OBJECT (store), callback, user_data, e_attachment_store_load_async); load_context = g_slice_new0 (LoadContext); load_context->simple = simple; load_context->attachment_list = g_list_copy (attachment_list); g_list_foreach ( load_context->attachment_list, (GFunc) g_object_ref, NULL); return load_context; } static void attachment_store_load_context_free (LoadContext *load_context) { g_object_unref (load_context->simple); /* The attachment list should be empty now. */ g_warn_if_fail (load_context->attachment_list == NULL); /* So should the error. */ g_warn_if_fail (load_context->error == NULL); g_slice_free (LoadContext, load_context); } static void attachment_store_load_ready_cb (EAttachment *attachment, GAsyncResult *result, LoadContext *load_context) { GSimpleAsyncResult *simple; GError *error = NULL; e_attachment_load_finish (attachment, result, &error); /* Remove the attachment from the list. */ load_context->attachment_list = g_list_remove ( load_context->attachment_list, attachment); g_object_unref (attachment); if (error != NULL) { /* If this is the first error, cancel the other jobs. */ if (load_context->error == NULL) { g_propagate_error (&load_context->error, error); g_list_foreach ( load_context->attachment_list, (GFunc) e_attachment_cancel, NULL); error = NULL; /* Otherwise, we can only report back one error. So if * this is something other than cancellation, dump it to * the terminal. */ } else if (!g_error_matches ( error, G_IO_ERROR, G_IO_ERROR_CANCELLED)) g_warning ("%s", error->message); } if (error != NULL) g_error_free (error); /* If there's still jobs running, let them finish. */ if (load_context->attachment_list != NULL) return; /* Steal the error. */ error = load_context->error; load_context->error = NULL; simple = load_context->simple; if (error == NULL) g_simple_async_result_set_op_res_gboolean (simple, TRUE); else g_simple_async_result_take_error (simple, error); g_simple_async_result_complete (simple); attachment_store_load_context_free (load_context); } void e_attachment_store_load_async (EAttachmentStore *store, GList *attachment_list, GAsyncReadyCallback callback, gpointer user_data) { LoadContext *load_context; GList *iter; g_return_if_fail (E_IS_ATTACHMENT_STORE (store)); load_context = attachment_store_load_context_new ( store, attachment_list, callback, user_data); if (attachment_list == NULL) { GSimpleAsyncResult *simple; simple = load_context->simple; g_simple_async_result_set_op_res_gboolean (simple, TRUE); g_simple_async_result_complete (simple); attachment_store_load_context_free (load_context); return; } for (iter = attachment_list; iter != NULL; iter = iter->next) { EAttachment *attachment = E_ATTACHMENT (iter->data); e_attachment_store_add_attachment (store, attachment); e_attachment_load_async ( attachment, (GAsyncReadyCallback) attachment_store_load_ready_cb, load_context); } } gboolean e_attachment_store_load_finish (EAttachmentStore *store, GAsyncResult *result, GError **error) { GSimpleAsyncResult *simple; gboolean success; g_return_val_if_fail (E_IS_ATTACHMENT_STORE (store), FALSE); g_return_val_if_fail (G_IS_SIMPLE_ASYNC_RESULT (result), FALSE); simple = G_SIMPLE_ASYNC_RESULT (result); success = g_simple_async_result_get_op_res_gboolean (simple); g_simple_async_result_propagate_error (simple, error); return success; } /********************** e_attachment_store_save_async() **********************/ typedef struct _SaveContext SaveContext; struct _SaveContext { GSimpleAsyncResult *simple; GFile *destination; gchar *filename_prefix; GFile *fresh_directory; GFile *trash_directory; GList *attachment_list; GError *error; gchar **uris; gint index; }; static SaveContext * attachment_store_save_context_new (EAttachmentStore *store, GFile *destination, const gchar *filename_prefix, GAsyncReadyCallback callback, gpointer user_data) { SaveContext *save_context; GSimpleAsyncResult *simple; GList *attachment_list; guint length; gchar **uris; simple = g_simple_async_result_new ( G_OBJECT (store), callback, user_data, e_attachment_store_save_async); attachment_list = e_attachment_store_get_attachments (store); /* Add one for NULL terminator. */ length = g_list_length (attachment_list) + 1; uris = g_malloc0 (sizeof (gchar *) * length); save_context = g_slice_new0 (SaveContext); save_context->simple = simple; save_context->destination = g_object_ref (destination); save_context->filename_prefix = g_strdup (filename_prefix); save_context->attachment_list = attachment_list; save_context->uris = uris; return save_context; } static void attachment_store_save_context_free (SaveContext *save_context) { g_object_unref (save_context->simple); /* The attachment list should be empty now. */ g_warn_if_fail (save_context->attachment_list == NULL); /* So should the error. */ g_warn_if_fail (save_context->error == NULL); if (save_context->destination) { g_object_unref (save_context->destination); save_context->destination = NULL; } g_free (save_context->filename_prefix); save_context->filename_prefix = NULL; if (save_context->fresh_directory) { g_object_unref (save_context->fresh_directory); save_context->fresh_directory = NULL; } if (save_context->trash_directory) { g_object_unref (save_context->trash_directory); save_context->trash_directory = NULL; } g_strfreev (save_context->uris); g_slice_free (SaveContext, save_context); } static void attachment_store_move_file (SaveContext *save_context, GFile *source, GFile *destination, GError **error) { gchar *tmpl; gchar *path; g_return_if_fail (save_context != NULL); g_return_if_fail (source != NULL); g_return_if_fail (destination != NULL); g_return_if_fail (error != NULL); /* Attachments are all saved to a temporary directory. * Now we need to move the existing destination directory * out of the way (if it exists). Instead of testing for * existence we'll just attempt the move and ignore any * G_IO_ERROR_NOT_FOUND errors. */ /* First, however, we need another temporary directory to * move the existing destination directory to. Note we're * not actually creating the directory yet, just picking a * name for it. The usual raciness with this approach * applies here (read up on mktemp(3)), but worst case is * we get a spurious G_IO_ERROR_WOULD_MERGE error and the * user has to try saving attachments again. */ tmpl = g_strdup_printf (PACKAGE "-%s-XXXXXX", g_get_user_name ()); path = e_mktemp (tmpl); g_free (tmpl); save_context->trash_directory = g_file_new_for_path (path); g_free (path); /* XXX No asynchronous move operation in GIO? */ g_file_move ( destination, save_context->trash_directory, G_FILE_COPY_NONE, NULL, NULL, NULL, error); if (*error != NULL && !g_error_matches (*error, G_IO_ERROR, G_IO_ERROR_NOT_FOUND)) return; g_clear_error (error); /* Now we can move the file from the temporary directory * to the user-specified destination. */ g_file_move ( source, destination, G_FILE_COPY_NONE, NULL, NULL, NULL, error); } static void attachment_store_save_cb (EAttachment *attachment, GAsyncResult *result, SaveContext *save_context) { GSimpleAsyncResult *simple; GFile *file; gchar **uris; GError *error = NULL; file = e_attachment_save_finish (attachment, result, &error); /* Remove the attachment from the list. */ save_context->attachment_list = g_list_remove ( save_context->attachment_list, attachment); g_object_unref (attachment); if (file != NULL) { /* Assemble the file's final URI from its basename. */ gchar *basename; gchar *uri; GFile *source = NULL, *destination = NULL; basename = g_file_get_basename (file); g_object_unref (file); source = g_file_get_child (save_context->fresh_directory, basename); if (save_context->filename_prefix && *save_context->filename_prefix) { gchar *tmp = basename; basename = g_strconcat (save_context->filename_prefix, basename, NULL); g_free (tmp); } file = save_context->destination; destination = g_file_get_child (file, basename); uri = g_file_get_uri (destination); /* move them file-by-file */ attachment_store_move_file (save_context, source, destination, &error); if (!error) save_context->uris[save_context->index++] = uri; g_object_unref (source); g_object_unref (destination); } if (error != NULL) { /* If this is the first error, cancel the other jobs. */ if (save_context->error == NULL) { g_propagate_error (&save_context->error, error); g_list_foreach ( save_context->attachment_list, (GFunc) e_attachment_cancel, NULL); error = NULL; /* Otherwise, we can only report back one error. So if * this is something other than cancellation, dump it to * the terminal. */ } else if (!g_error_matches ( error, G_IO_ERROR, G_IO_ERROR_CANCELLED)) g_warning ("%s", error->message); } g_clear_error (&error); /* If there's still jobs running, let them finish. */ if (save_context->attachment_list != NULL) return; /* If an error occurred while saving, we're done. */ if (save_context->error != NULL) { /* Steal the error. */ error = save_context->error; save_context->error = NULL; simple = save_context->simple; g_simple_async_result_take_error (simple, error); g_simple_async_result_complete (simple); attachment_store_save_context_free (save_context); return; } if (error != NULL) { simple = save_context->simple; g_simple_async_result_take_error (simple, error); g_simple_async_result_complete (simple); attachment_store_save_context_free (save_context); return; } /* clean-up left directory */ g_file_delete (save_context->fresh_directory, NULL, NULL); /* And the URI list. */ uris = save_context->uris; save_context->uris = NULL; simple = save_context->simple; g_simple_async_result_set_op_res_gpointer (simple, uris, NULL); g_simple_async_result_complete (simple); attachment_store_save_context_free (save_context); } /* * @filename_prefix: prefix to use for a file name; can be %NULL for none **/ void e_attachment_store_save_async (EAttachmentStore *store, GFile *destination, const gchar *filename_prefix, GAsyncReadyCallback callback, gpointer user_data) { SaveContext *save_context; GList *attachment_list, *iter; GFile *temp_directory; gchar *template; gchar *path; g_return_if_fail (E_IS_ATTACHMENT_STORE (store)); g_return_if_fail (G_IS_FILE (destination)); save_context = attachment_store_save_context_new ( store, destination, filename_prefix, callback, user_data); attachment_list = save_context->attachment_list; /* Deal with an empty attachment store. The caller will get * an empty NULL-terminated list as opposed to NULL, to help * distinguish it from an error. */ if (attachment_list == NULL) { GSimpleAsyncResult *simple; gchar **uris; /* Steal the URI list. */ uris = save_context->uris; save_context->uris = NULL; simple = save_context->simple; g_simple_async_result_set_op_res_gpointer (simple, uris, NULL); g_simple_async_result_complete (simple); attachment_store_save_context_free (save_context); return; } /* Save all attachments to a temporary directory, which we'll * then move to its proper location. We use a directory so * files can retain their basenames. * XXX This could trigger a blocking temp directory cleanup. */ template = g_strdup_printf (PACKAGE "-%s-XXXXXX", g_get_user_name ()); path = e_mkdtemp (template); g_free (template); /* XXX Let's hope errno got set properly. */ if (path == NULL) { GSimpleAsyncResult *simple; simple = save_context->simple; g_simple_async_result_set_error ( simple, G_FILE_ERROR, g_file_error_from_errno (errno), "%s", g_strerror (errno)); g_simple_async_result_complete (simple); attachment_store_save_context_free (save_context); return; } temp_directory = g_file_new_for_path (path); save_context->fresh_directory = temp_directory; g_free (path); for (iter = attachment_list; iter != NULL; iter = iter->next) e_attachment_save_async ( E_ATTACHMENT (iter->data), temp_directory, (GAsyncReadyCallback) attachment_store_save_cb, save_context); } gchar ** e_attachment_store_save_finish (EAttachmentStore *store, GAsyncResult *result, GError **error) { GSimpleAsyncResult *simple; gchar **uris; g_return_val_if_fail (E_IS_ATTACHMENT_STORE (store), NULL); g_return_val_if_fail (G_IS_SIMPLE_ASYNC_RESULT (result), NULL); simple = G_SIMPLE_ASYNC_RESULT (result); uris = g_simple_async_result_get_op_res_gpointer (simple); g_simple_async_result_propagate_error (simple, error); return uris; }