path: root/modules/mail/e-mail-attachment-handler.c

/*
 * e-mail-attachment-handler.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 <http://www.gnu.org/licenses/>
 *
 *
 * Copyright (C) 1999-2008 Novell, Inc. (www.novell.com)
 *
 */

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

#include "e-mail-attachment-handler.h"

#include <glib/gi18n.h>

#include "mail/e-mail-backend.h"
#include "mail/em-composer-utils.h"

#define E_MAIL_ATTACHMENT_HANDLER_GET_PRIVATE(obj) \
    (G_TYPE_INSTANCE_GET_PRIVATE \
    ((obj), E_TYPE_MAIL_ATTACHMENT_HANDLER, EMailAttachmentHandlerPrivate))

struct _EMailAttachmentHandlerPrivate {
    EMailBackend *backend;
};

static gpointer parent_class;
static GType mail_attachment_handler_type;

static const gchar *ui =
"<ui>"
"  <popup name='context'>"
"    <placeholder name='custom-actions'>"
"      <menuitem action='mail-reply-sender'/>"
"      <menuitem action='mail-reply-all'/>"
"      <menuitem action='mail-forward'/>"
"    </placeholder>"
"  </popup>"
"</ui>";

/* Note: Do not use the info field. */
static GtkTargetEntry target_table[] = {
    { (gchar *) "message/rfc822",   0, 0 },
    { (gchar *) "x-uid-list",   0, 0 }
};

static CamelMimeMessage *
mail_attachment_handler_get_selected_message (EAttachmentHandler *handler)
{
    EAttachment *attachment;
    EAttachmentView *view;
    CamelMimePart *mime_part;
    CamelMimeMessage *message = NULL;
    CamelDataWrapper *outer_wrapper;
    CamelContentType *outer_content_type;
    CamelDataWrapper *inner_wrapper;
    CamelContentType *inner_content_type;
    GList *selected;
    gboolean inner_and_outer_content_types_match;

    view = e_attachment_handler_get_view (handler);

    selected = e_attachment_view_get_selected_attachments (view);
    g_return_val_if_fail (g_list_length (selected) == 1, NULL);

    attachment = E_ATTACHMENT (selected->data);
    mime_part = e_attachment_ref_mime_part (attachment);

    outer_wrapper =
        camel_medium_get_content (CAMEL_MEDIUM (mime_part));
    outer_content_type =
        camel_data_wrapper_get_mime_type_field (outer_wrapper);

    if (!camel_content_type_is (outer_content_type, "message", "rfc822"))
        goto exit;

    inner_wrapper =
        camel_medium_get_content (CAMEL_MEDIUM (outer_wrapper));
    inner_content_type =
        camel_data_wrapper_get_mime_type_field (inner_wrapper);

    inner_and_outer_content_types_match =
        camel_content_type_is (
            inner_content_type,
            outer_content_type->type,
            outer_content_type->subtype);

    if (!inner_and_outer_content_types_match) {
        CamelStream *mem;
        gboolean success;

        /* Create a message copy in case the inner content
         * type doesn't match the mime_part's content type,
         * which can happen for multipart/digest, where it
         * confuses the formatter on reply, which skips all
         * rfc822 subparts. */
        mem = camel_stream_mem_new ();
        camel_data_wrapper_write_to_stream_sync (
            CAMEL_DATA_WRAPPER (outer_wrapper), mem, NULL, NULL);

        g_seekable_seek (
            G_SEEKABLE (mem), 0, G_SEEK_SET, NULL, NULL);
        message = camel_mime_message_new ();
        success = camel_data_wrapper_construct_from_stream_sync (
            CAMEL_DATA_WRAPPER (message), mem, NULL, NULL);
        if (!success)
            g_clear_object (&message);

        g_object_unref (mem);
    }

exit:
    if (message == NULL)
        message = g_object_ref (outer_wrapper);

    g_clear_object (&mime_part);

    g_list_free_full (selected, (GDestroyNotify) g_object_unref);

    return message;
}

static void
mail_attachment_handler_forward (GtkAction *action,
                                 EAttachmentHandler *handler)
{
    EMailAttachmentHandlerPrivate *priv;
    GSettings *settings;
    EMailForwardStyle style;
    CamelMimeMessage *message;

    priv = E_MAIL_ATTACHMENT_HANDLER_GET_PRIVATE (handler);

    message = mail_attachment_handler_get_selected_message (handler);
    g_return_if_fail (message != NULL);

    settings = g_settings_new ("org.gnome.evolution.mail");
    style = g_settings_get_enum (settings, "forward-style-name");
    g_object_unref (settings);

    em_utils_forward_message (priv->backend, message, style, NULL, NULL);

    g_object_unref (message);
}

static void
mail_attachment_handler_reply (EAttachmentHandler *handler,
                               EMailReplyType reply_type)
{
    EMailAttachmentHandlerPrivate *priv;
    GSettings *settings;
    EMailReplyStyle style;
    CamelMimeMessage *message;
    EShellBackend *shell_backend;
    EShell *shell;

    priv = E_MAIL_ATTACHMENT_HANDLER_GET_PRIVATE (handler);

    message = mail_attachment_handler_get_selected_message (handler);
    g_return_if_fail (message != NULL);

    settings = g_settings_new ("org.gnome.evolution.mail");
    style = g_settings_get_enum (settings, "reply-style-name");
    g_object_unref (settings);

    shell_backend = E_SHELL_BACKEND (priv->backend);
    shell = e_shell_backend_get_shell (shell_backend);

    em_utils_reply_to_message (
        shell, message, NULL, NULL, reply_type, style, NULL, NULL);

    g_object_unref (message);
}

static void
mail_attachment_handler_reply_all (GtkAction *action,
                                   EAttachmentHandler *handler)
{
    mail_attachment_handler_reply (handler, E_MAIL_REPLY_TO_ALL);
}

static void
mail_attachment_handler_reply_sender (GtkAction *action,
                                      EAttachmentHandler *handler)
{
    mail_attachment_handler_reply (handler, E_MAIL_REPLY_TO_SENDER);
}

static GtkActionEntry standard_entries[] = {

    { "mail-forward",
      "mail-forward",
      N_("_Forward"),
      NULL,
      NULL,  /* XXX Add a tooltip! */
      G_CALLBACK (mail_attachment_handler_forward) },

    { "mail-reply-all",
      "mail-reply-all",
      N_("Reply to _All"),
      NULL,
      NULL,  /* XXX Add a tooltip! */
      G_CALLBACK (mail_attachment_handler_reply_all) },

    { "mail-reply-sender",
      "mail-reply-sender",
      N_("_Reply to Sender"),
      NULL,
      NULL,  /* XXX Add a tooltip! */
      G_CALLBACK (mail_attachment_handler_reply_sender) }
};

static void
mail_attachment_handler_message_rfc822 (EAttachmentView *view,
                                        GdkDragContext *drag_context,
                                        gint x,
                                        gint y,
                                        GtkSelectionData *selection_data,
                                        guint info,
                                        guint time,
                                        EAttachmentHandler *handler)
{
    static GdkAtom atom = GDK_NONE;
    EAttachmentStore *store;
    EAttachment *attachment;
    CamelMimeMessage *message;
    CamelDataWrapper *wrapper;
    CamelStream *stream;
    const gchar *data;
    gboolean success = FALSE;
    gpointer parent;
    gint length;

    if (G_UNLIKELY (atom == GDK_NONE))
        atom = gdk_atom_intern_static_string ("message/rfc822");

    if (gtk_selection_data_get_target (selection_data) != atom)
        return;

    g_signal_stop_emission_by_name (view, "drag-data-received");

    data = (const gchar *) gtk_selection_data_get_data (selection_data);
    length = gtk_selection_data_get_length (selection_data);

    stream = camel_stream_mem_new ();
    camel_stream_write (stream, data, length, NULL, NULL);
    g_seekable_seek (G_SEEKABLE (stream), 0, G_SEEK_SET, NULL, NULL);

    message = camel_mime_message_new ();
    wrapper = CAMEL_DATA_WRAPPER (message);

    if (!camel_data_wrapper_construct_from_stream_sync (
        wrapper, stream, NULL, NULL))
        goto exit;

    store = e_attachment_view_get_store (view);

    parent = gtk_widget_get_toplevel (GTK_WIDGET (view));
    parent = gtk_widget_is_toplevel (parent) ? parent : NULL;

    attachment = e_attachment_new_for_message (message);
    e_attachment_store_add_attachment (store, attachment);
    e_attachment_load_async (
        attachment, (GAsyncReadyCallback)
        e_attachment_load_handle_error, parent);
    g_object_unref (attachment);

    success = TRUE;

exit:
    g_object_unref (message);
    g_object_unref (stream);

    gtk_drag_finish (drag_context, success, FALSE, time);
}

static void
mail_attachment_handler_x_uid_list (EAttachmentView *view,
                                    GdkDragContext *drag_context,
                                    gint x,
                                    gint y,
                                    GtkSelectionData *selection_data,
                                    guint info,
                                    guint time,
                                    EAttachmentHandler *handler)
{
    static GdkAtom atom = GDK_NONE;
    EMailAttachmentHandlerPrivate *priv;
    CamelDataWrapper *wrapper;
    CamelMimeMessage *message;
    CamelMultipart *multipart;
    CamelMimePart *mime_part;
    CamelFolder *folder = NULL;
    EAttachment *attachment;
    EAttachmentStore *store;
    EMailSession *session;
    GPtrArray *uids;
    const gchar *data;
    const gchar *cp, *end;
    gchar *description;
    gpointer parent;
    gint length;
    guint ii;
    GError *local_error = NULL;

    if (G_UNLIKELY (atom == GDK_NONE))
        atom = gdk_atom_intern_static_string ("x-uid-list");

    if (gtk_selection_data_get_target (selection_data) != atom)
        return;

    store = e_attachment_view_get_store (view);
    priv = E_MAIL_ATTACHMENT_HANDLER_GET_PRIVATE (handler);

    parent = gtk_widget_get_toplevel (GTK_WIDGET (view));
    parent = gtk_widget_is_toplevel (parent) ? parent : NULL;

    uids = g_ptr_array_new ();

    data = (const gchar *) gtk_selection_data_get_data (selection_data);
    length = gtk_selection_data_get_length (selection_data);

    /* The UID list is delimited by NUL characters.
     * Brilliant.  So we can't use g_strsplit(). */

    cp = data;
    end = data + length;

    while (cp < end) {
        const gchar *start = cp;

        while (cp < end && *cp != '\0')
            cp++;

        /* Skip the first string. */
        if (start > data)
            g_ptr_array_add (uids, g_strndup (start, cp - start));

        cp++;
    }

    if (uids->len == 0)
        goto exit;

    session = e_mail_backend_get_session (priv->backend);

    /* The first string is the folder URI. */
    /* FIXME Not passing a GCancellable here. */
    folder = e_mail_session_uri_to_folder_sync (
        session, data, 0, NULL, &local_error);
    if (folder == NULL)
        goto exit;

    /* Handle one message. */
    if (uids->len == 1) {
        const gchar *message_uid;

        message_uid = g_ptr_array_index (uids, 0);

        /* FIXME Not passing a GCancellable here. */
        message = camel_folder_get_message_sync (
            folder, message_uid, NULL, &local_error);
        if (message == NULL)
            goto exit;

        attachment = e_attachment_new_for_message (message);
        e_attachment_store_add_attachment (store, attachment);
        e_attachment_load_async (
            attachment, (GAsyncReadyCallback)
            e_attachment_load_handle_error, parent);
        g_object_unref (attachment);

        g_object_unref (message);
        goto exit;
    }

    /* Build a multipart/digest message out of the UIDs. */

    multipart = camel_multipart_new ();
    wrapper = CAMEL_DATA_WRAPPER (multipart);
    camel_data_wrapper_set_mime_type (wrapper, "multipart/digest");
    camel_multipart_set_boundary (multipart, NULL);

    for (ii = 0; ii < uids->len; ii++) {
        /* FIXME Not passing a GCancellable here. */
        message = camel_folder_get_message_sync (
            folder, uids->pdata[ii], NULL, &local_error);
        if (message == NULL) {
            g_object_unref (multipart);
            goto exit;
        }

        mime_part = camel_mime_part_new ();
        wrapper = CAMEL_DATA_WRAPPER (message);
        camel_mime_part_set_disposition (mime_part, "inline");
        camel_medium_set_content (
            CAMEL_MEDIUM (mime_part), wrapper);
        camel_mime_part_set_content_type (mime_part, "message/rfc822");
        camel_multipart_add_part (multipart, mime_part);
        g_object_unref (mime_part);

        g_object_unref (message);
    }

    mime_part = camel_mime_part_new ();
    wrapper = CAMEL_DATA_WRAPPER (multipart);
    camel_medium_set_content (CAMEL_MEDIUM (mime_part), wrapper);

    description = g_strdup_printf (
        ngettext (
            "%d attached message",
            "%d attached messages",
            uids->len),
        uids->len);
    camel_mime_part_set_description (mime_part, description);
    g_free (description);

    attachment = e_attachment_new ();
    e_attachment_set_mime_part (attachment, mime_part);
    e_attachment_store_add_attachment (store, attachment);
    e_attachment_load_async (
        attachment, (GAsyncReadyCallback)
        e_attachment_load_handle_error, parent);
    g_object_unref (attachment);

    g_object_unref (mime_part);
    g_object_unref (multipart);

exit:
    if (local_error != NULL) {
        const gchar *folder_name = data;

        if (folder != NULL)
            folder_name = camel_folder_get_display_name (folder);

        e_alert_run_dialog_for_args (
            parent, "mail-composer:attach-nomessages",
            folder_name, local_error->message, NULL);

        g_clear_error (&local_error);
    }

    if (folder != NULL)
        g_object_unref (folder);

    g_ptr_array_free (uids, TRUE);

    g_signal_stop_emission_by_name (view, "drag-data-received");
}

static void
mail_attachment_handler_update_actions (EAttachmentView *view,
                                        EAttachmentHandler *handler)
{
    EAttachment *attachment;
    CamelMimePart *mime_part;
    GtkActionGroup *action_group;
    GList *selected;
    gboolean visible = FALSE;

    selected = e_attachment_view_get_selected_attachments (view);

    if (g_list_length (selected) != 1)
        goto exit;

    attachment = E_ATTACHMENT (selected->data);

    if (e_attachment_get_loading (attachment) ||
        e_attachment_get_saving (attachment))
        goto exit;

    mime_part = e_attachment_ref_mime_part (attachment);

    if (mime_part != NULL) {
        CamelMedium *medium;
        CamelDataWrapper *content;

        medium = CAMEL_MEDIUM (mime_part);
        content = camel_medium_get_content (medium);
        visible = CAMEL_IS_MIME_MESSAGE (content);

        g_object_unref (mime_part);
    }

exit:
    action_group = e_attachment_view_get_action_group (view, "mail");
    gtk_action_group_set_visible (action_group, visible);

    g_list_foreach (selected, (GFunc) g_object_unref, NULL);
    g_list_free (selected);
}

static void
mail_attachment_handler_dispose (GObject *object)
{
    EMailAttachmentHandlerPrivate *priv;

    priv = E_MAIL_ATTACHMENT_HANDLER_GET_PRIVATE (object);

    g_clear_object (&priv->backend);

    /* Chain up to parent's dispose() method. */
    G_OBJECT_CLASS (parent_class)->dispose (object);
}

static void
mail_attachment_handler_constructed (GObject *object)
{
    EMailAttachmentHandlerPrivate *priv;
    EShell *shell;
    EShellBackend *shell_backend;
    EAttachmentHandler *handler;
    EAttachmentView *view;
    GtkActionGroup *action_group;
    GtkUIManager *ui_manager;
    GError *error = NULL;

    handler = E_ATTACHMENT_HANDLER (object);
    priv = E_MAIL_ATTACHMENT_HANDLER_GET_PRIVATE (object);

    /* Chain up to parent's constructed() method. */
    G_OBJECT_CLASS (parent_class)->constructed (object);

    shell = e_shell_get_default ();
    shell_backend = e_shell_get_backend_by_name (shell, "mail");
    priv->backend = g_object_ref (shell_backend);

    view = e_attachment_handler_get_view (handler);

    action_group = e_attachment_view_add_action_group (view, "mail");
    gtk_action_group_add_actions (
        action_group, standard_entries,
        G_N_ELEMENTS (standard_entries), handler);

    ui_manager = e_attachment_view_get_ui_manager (view);
    gtk_ui_manager_add_ui_from_string (ui_manager, ui, -1, &error);

    if (error != NULL) {
        g_warning ("%s", error->message);
        g_error_free (error);
    }

    g_signal_connect (
        view, "update-actions",
        G_CALLBACK (mail_attachment_handler_update_actions),
        handler);

    g_signal_connect (
        view, "drag-data-received",
        G_CALLBACK (mail_attachment_handler_message_rfc822),
        handler);

    g_signal_connect (
        view, "drag-data-received",
        G_CALLBACK (mail_attachment_handler_x_uid_list),
        handler);
}

static GdkDragAction
mail_attachment_handler_get_drag_actions (EAttachmentHandler *handler)
{
    return GDK_ACTION_COPY;
}

static const GtkTargetEntry *
mail_attachment_handler_get_target_table (EAttachmentHandler *handler,
                                          guint *n_targets)
{
    if (n_targets != NULL)
        *n_targets = G_N_ELEMENTS (target_table);

    return target_table;
}

static void
mail_attachment_handler_class_init (EMailAttachmentHandlerClass *class)
{
    GObjectClass *object_class;
    EAttachmentHandlerClass *handler_class;

    parent_class = g_type_class_peek_parent (class);
    g_type_class_add_private (class, sizeof (EMailAttachmentHandlerPrivate));

    object_class = G_OBJECT_CLASS (class);
    object_class->dispose = mail_attachment_handler_dispose;
    object_class->constructed = mail_attachment_handler_constructed;

    handler_class = E_ATTACHMENT_HANDLER_CLASS (class);
    handler_class->get_drag_actions = mail_attachment_handler_get_drag_actions;
    handler_class->get_target_table = mail_attachment_handler_get_target_table;
}

static void
mail_attachment_handler_init (EMailAttachmentHandler *handler)
{
    handler->priv = E_MAIL_ATTACHMENT_HANDLER_GET_PRIVATE (handler);
}

GType
e_mail_attachment_handler_get_type (void)
{
    return mail_attachment_handler_type;
}

void
e_mail_attachment_handler_register_type (GTypeModule *type_module)
{
    static const GTypeInfo type_info = {
        sizeof (EMailAttachmentHandlerClass),
        (GBaseInitFunc) NULL,
        (GBaseFinalizeFunc) NULL,
        (GClassInitFunc) mail_attachment_handler_class_init,
        (GClassFinalizeFunc) NULL,
        NULL,  /* class_data */
        sizeof (EMailAttachmentHandler),
        0,     /* n_preallocs */
        (GInstanceInitFunc) mail_attachment_handler_init,
        NULL   /* value_table */
    };

    mail_attachment_handler_type = g_type_module_register_type (
        type_module, E_TYPE_ATTACHMENT_HANDLER,
        "EMailAttachmentHandler", &type_info, 0);
}