path: root/mail/e-mail-printer.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) 2011 Dan Vratil <dvratil@redhat.com>
 *
 */

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

#include <string.h>
#include <glib/gi18n.h>
#include <gtk/gtk.h>

#include <webkit/webkitdom.h>

#include "e-util/e-util.h"

#include "em-format/e-mail-formatter-print.h"
#include "em-format/e-mail-part-utils.h"

#include "e-mail-printer.h"
#include "e-mail-display.h"

enum {
        BUTTON_SELECT_ALL,
        BUTTON_SELECT_NONE,
        BUTTON_TOP,
        BUTTON_UP,
        BUTTON_DOWN,
        BUTTON_BOTTOM,
        BUTTONS_COUNT
};

#define w(x)

struct _EMailPrinterPrivate {
    EMailFormatterPrint *formatter;
    EMailPartList *parts_list;

    gchar *export_filename;

    GtkListStore *headers;

    WebKitWebView *webview; /* WebView to print from */
    gchar *uri;
    GtkWidget *buttons[BUTTONS_COUNT];
    GtkWidget *treeview;

    GtkPrintOperation *operation;
    GtkPrintOperationAction print_action;
};

G_DEFINE_TYPE (
    EMailPrinter,
    e_mail_printer,
    G_TYPE_OBJECT);

enum {
    PROP_0,
    PROP_PART_LIST
};

enum {
    SIGNAL_DONE,
    LAST_SIGNAL
};

enum {
    COLUMN_ACTIVE,
    COLUMN_HEADER_NAME,
    COLUMN_HEADER_VALUE,
    COLUMN_HEADER_STRUCT,
    LAST_COLUMN
};

static guint signals[LAST_SIGNAL];

static gint
emp_header_name_equal (const EMailFormatterHeader *h1,
                       const EMailFormatterHeader *h2)
{
    if ((h2->value == NULL) || (h1->value == NULL)) {
        return g_strcmp0 (h1->name, h2->name);
    } else {
        if ((g_strcmp0 (h1->name, h2->name) == 0) &&
            (g_strcmp0 (h1->value, h2->value) == 0))
            return 0;
        else
            return 1;
    }
}

static void
emp_draw_footer (GtkPrintOperation *operation,
                 GtkPrintContext *context,
                 gint page_nr)
{
    PangoFontDescription *desc;
    PangoLayout *layout;
    gint n_pages;
    gdouble width, height;
    gchar *text;
    cairo_t *cr;

    cr = gtk_print_context_get_cairo_context (context);
    width = gtk_print_context_get_width (context);
    height = gtk_print_context_get_height (context);

    g_object_get (operation, "n-pages", &n_pages, NULL);
    text = g_strdup_printf (_("Page %d of %d"), page_nr + 1, n_pages);

    cairo_set_source_rgb (cr, 0.1, 0.1, 0.1);
    cairo_fill (cr);

    desc = pango_font_description_from_string ("Sans Regular 10");
    layout = gtk_print_context_create_pango_layout (context);
    pango_layout_set_alignment (layout, PANGO_ALIGN_CENTER);
    pango_layout_set_font_description (layout, desc);
    pango_layout_set_text (layout, text, -1);
    pango_layout_set_width (layout, width * PANGO_SCALE);
    pango_font_description_free (desc);

    cairo_move_to (cr, 0, height + 5);
    pango_cairo_show_layout (cr, layout);

    g_object_unref (layout);
    g_free (text);
}

static void
emp_printing_done (GtkPrintOperation *operation,
                   GtkPrintOperationResult result,
                   gpointer user_data)
{
    EMailPrinter *emp = user_data;

    g_signal_emit (emp, signals[SIGNAL_DONE], 0, operation, result);
}

static gboolean
do_run_print_operation (EMailPrinter *emp)
{
    WebKitWebFrame *frame;

    frame = webkit_web_view_get_main_frame (emp->priv->webview);

    webkit_web_frame_print_full (
        frame, emp->priv->operation, emp->priv->print_action, NULL);

    return FALSE;
}

static void
emp_start_printing (GObject *object,
                    GParamSpec *pspec,
                    gpointer user_data)
{
    WebKitWebView *web_view;
    WebKitLoadStatus load_status;
    EMailPrinter *emp = user_data;

    web_view = WEBKIT_WEB_VIEW (object);
    load_status = webkit_web_view_get_load_status (web_view);

    if (load_status != WEBKIT_LOAD_FINISHED)
        return;

    /* WebKit reloads the page once more right before starting to print, so
     * disconnect this handler after the first time, so that we don't start
     * another printing operation */
    g_signal_handlers_disconnect_by_func (
        object, emp_start_printing, user_data);

    if (emp->priv->print_action == GTK_PRINT_OPERATION_ACTION_EXPORT) {
        gtk_print_operation_set_export_filename (
            emp->priv->operation, emp->priv->export_filename);
    }

    /* Give WebKit some time to perform layouting and rendering before
     * we start printing. 500ms should be enough in most cases... */
    g_timeout_add_full (
        G_PRIORITY_DEFAULT, 500,
        (GSourceFunc) do_run_print_operation,
        g_object_ref (emp), g_object_unref);
}

static void
emp_run_print_operation (EMailPrinter *emp,
             EMailFormatter *formatter)
{
    EMailPartList *part_list;
    CamelFolder *folder;
    const gchar *message_uid;
    const gchar *default_charset, *charset;
    gchar *mail_uri;

    part_list = emp->priv->parts_list;
    folder = e_mail_part_list_get_folder (part_list);
    message_uid = e_mail_part_list_get_message_uid (part_list);
    default_charset = formatter ? e_mail_formatter_get_default_charset (formatter) : NULL;
    charset = formatter ? e_mail_formatter_get_charset (formatter) : NULL;

    if (!default_charset)
        default_charset = "";
    if (!charset)
        charset = "";

    mail_uri = e_mail_part_build_uri (
        folder, message_uid,
        "__evo-load-image", G_TYPE_BOOLEAN, TRUE,
        "mode", G_TYPE_INT, E_MAIL_FORMATTER_MODE_PRINTING,
        "formatter_default_charset", G_TYPE_STRING, default_charset,
        "formatter_charset", G_TYPE_STRING, charset,
        NULL);

    /* Print_layout is a special EMPart created by EMFormatHTMLPrint */
    if (emp->priv->webview == NULL) {
        EMailFormatter *emp_formatter;

        emp->priv->webview = g_object_new (
            E_TYPE_MAIL_DISPLAY,
            "mode", E_MAIL_FORMATTER_MODE_PRINTING, NULL);
        e_web_view_set_enable_frame_flattening (E_WEB_VIEW (emp->priv->webview), FALSE);
        e_mail_display_set_force_load_images (
            E_MAIL_DISPLAY (emp->priv->webview), TRUE);

        emp_formatter = e_mail_display_get_formatter (E_MAIL_DISPLAY (emp->priv->webview));
        if (default_charset && *default_charset)
            e_mail_formatter_set_default_charset (emp_formatter, default_charset);
        if (charset && *charset)
            e_mail_formatter_set_charset (emp_formatter, charset);

        g_object_ref_sink (emp->priv->webview);
        g_signal_connect (
            emp->priv->webview, "notify::load-status",
            G_CALLBACK (emp_start_printing), emp);

        w ({
            GtkWidget *window = gtk_window_new (GTK_WINDOW_TOPLEVEL);
            GtkWidget *sw = gtk_scrolled_window_new (NULL, NULL);
            gtk_container_add (GTK_CONTAINER (window), sw);
            gtk_container_add (
                GTK_CONTAINER (sw),
                GTK_WIDGET (emp->priv->webview));
            gtk_widget_show_all (window);
        });
    }
    e_mail_display_set_parts_list (
        E_MAIL_DISPLAY (emp->priv->webview), emp->priv->parts_list);
    webkit_web_view_load_uri (emp->priv->webview, mail_uri);

    g_free (mail_uri);
}

static void
set_header_visible (EMailPrinter *emp,
                    EMailFormatterHeader *header,
                    gint index,
                    gboolean visible)
{
    WebKitDOMDocument *document;
    WebKitDOMNodeList *headers;
    WebKitDOMElement *element;
    WebKitDOMCSSStyleDeclaration *style;

    document = webkit_web_view_get_dom_document (emp->priv->webview);
    headers = webkit_dom_document_get_elements_by_class_name (document, "header-item");

    g_return_if_fail (index < webkit_dom_node_list_get_length (headers));

    element = WEBKIT_DOM_ELEMENT (webkit_dom_node_list_item (headers, index));
    style = webkit_dom_element_get_style (element);
    webkit_dom_css_style_declaration_set_property (
        style,
        "display", (visible ? "table-row" : "none"), "", NULL);
}

static void
header_active_renderer_toggled_cb (GtkCellRendererToggle *renderer,
                                   gchar *path,
                                   EMailPrinter *emp)
{
    GtkTreeIter iter;
    GtkTreePath *p;
    gboolean active;
    EMailFormatterHeader *header;
    gint *indices;

    gtk_tree_model_get_iter_from_string (
        GTK_TREE_MODEL (emp->priv->headers),
        &iter, path);

    gtk_tree_model_get (
        GTK_TREE_MODEL (emp->priv->headers), &iter,
        COLUMN_ACTIVE, &active, -1);
    gtk_tree_model_get (
        GTK_TREE_MODEL (emp->priv->headers), &iter,
        COLUMN_HEADER_STRUCT, &header, -1);
    gtk_list_store_set (
        GTK_LIST_STORE (emp->priv->headers), &iter,
        COLUMN_ACTIVE, !active, -1);

    p = gtk_tree_path_new_from_string (path);
    indices = gtk_tree_path_get_indices (p);
    set_header_visible (emp, header, indices[0], !active);
    gtk_tree_path_free (p);
}

static void
emp_headers_tab_toggle_selection (GtkWidget *button,
                                  gpointer user_data)
{
    EMailPrinter *emp = user_data;
    GtkTreeIter iter;
    gboolean select;

    if (button == emp->priv->buttons[BUTTON_SELECT_ALL])
        select = TRUE;
    else if (button == emp->priv->buttons[BUTTON_SELECT_NONE])
        select = FALSE;
    else
        return;

    if (!gtk_tree_model_get_iter_first (GTK_TREE_MODEL (emp->priv->headers), &iter))
        return;

    do {
        EMailFormatterHeader *header;
        GtkTreePath *path;
        gint *indices;

        gtk_tree_model_get (
            GTK_TREE_MODEL (emp->priv->headers), &iter,
            COLUMN_HEADER_STRUCT, &header, -1);
        gtk_list_store_set (
            GTK_LIST_STORE (emp->priv->headers), &iter,
            COLUMN_ACTIVE, select, -1);

        path = gtk_tree_model_get_path (GTK_TREE_MODEL (emp->priv->headers), &iter);
        indices = gtk_tree_path_get_indices (path);
        set_header_visible (emp, header, indices[0], select);
        gtk_tree_path_free (path);

    } while (gtk_tree_model_iter_next (GTK_TREE_MODEL (emp->priv->headers), &iter));
}

static void
emp_headers_tab_selection_changed (GtkTreeSelection *selection,
                                   gpointer user_data)
{
    EMailPrinter *emp = user_data;
    gboolean enabled;
    GList *selected_rows;
    GtkTreeIter iter;
    GtkTreeModel *model;
    GtkTreePath *path;

    if (gtk_tree_selection_count_selected_rows (selection) == 0) {
        gtk_widget_set_sensitive (emp->priv->buttons[BUTTON_TOP], FALSE);
        gtk_widget_set_sensitive (emp->priv->buttons[BUTTON_UP], FALSE);
        gtk_widget_set_sensitive (emp->priv->buttons[BUTTON_DOWN], FALSE);
        gtk_widget_set_sensitive (emp->priv->buttons[BUTTON_BOTTOM], FALSE);

        return;
    }

    model = GTK_TREE_MODEL (emp->priv->headers);
    selected_rows = gtk_tree_selection_get_selected_rows (selection, &model);

    path = gtk_tree_path_copy (selected_rows->data);
    enabled = gtk_tree_path_prev (path);
    gtk_widget_set_sensitive (emp->priv->buttons[BUTTON_TOP], enabled);
    gtk_widget_set_sensitive (emp->priv->buttons[BUTTON_UP], enabled);

    gtk_tree_model_get_iter (model, &iter, g_list_last (selected_rows)->data);
    enabled = gtk_tree_model_iter_next (model, &iter);
    gtk_widget_set_sensitive (emp->priv->buttons[BUTTON_DOWN], enabled);
    gtk_widget_set_sensitive (emp->priv->buttons[BUTTON_BOTTOM], enabled);

    g_list_foreach (selected_rows, (GFunc) gtk_tree_path_free, NULL);
    g_list_free (selected_rows);
    gtk_tree_path_free (path);
}

static void
emp_headers_tab_move (GtkWidget *button,
                      gpointer user_data)
{
    EMailPrinter *emp = user_data;
    GtkTreeSelection *selection;
    GList *selected_rows, *references, *l;
    GtkTreePath *path;
    GtkTreeModel *model;
    GtkTreeIter iter;
    GtkTreeRowReference *selection_middle;
    gint *indices;

    WebKitDOMDocument *document;
    WebKitDOMNodeList *headers;
    WebKitDOMNode *header, *parent;

    model = GTK_TREE_MODEL (emp->priv->headers);
    selection = gtk_tree_view_get_selection  (GTK_TREE_VIEW (emp->priv->treeview));
    selected_rows = gtk_tree_selection_get_selected_rows (selection, &model);

        /* The order of header rows in the HMTL document should be in sync with
       order of headers in the listview and in efhp->headers_list */
    document = webkit_web_view_get_dom_document (emp->priv->webview);
    headers = webkit_dom_document_get_elements_by_class_name (document, "header-item");

    l = g_list_nth (selected_rows, g_list_length (selected_rows) / 2);
    selection_middle = gtk_tree_row_reference_new (model, l->data);

    references = NULL;
    for (l = selected_rows; l; l = l->next) {
        references = g_list_prepend (
            references,
            gtk_tree_row_reference_new (model, l->data));
    }

    if (button == emp->priv->buttons[BUTTON_TOP]) {

        for (l = references; l; l = l->next) {
                        /* Move the rows in the view  */
            path = gtk_tree_row_reference_get_path (l->data);
            gtk_tree_model_get_iter (model, &iter, path);
            gtk_list_store_move_after (emp->priv->headers, &iter, NULL);

                        /* Move the header row in HTML document */
            indices = gtk_tree_path_get_indices (path);
            header = webkit_dom_node_list_item (headers, indices[0]);
            parent = webkit_dom_node_get_parent_node (header);
            webkit_dom_node_remove_child (parent, header, NULL);
            webkit_dom_node_insert_before (parent, header,
                webkit_dom_node_get_first_child (parent), NULL);

            gtk_tree_path_free (path);
        }

       } else if (button == emp->priv->buttons[BUTTON_UP]) {

        GtkTreeIter *iter_prev;
        WebKitDOMNode *node2;

        references = g_list_reverse (references);

        for (l = references; l; l = l->next) {

            path = gtk_tree_row_reference_get_path (l->data);
            gtk_tree_model_get_iter (model, &iter, path);
            iter_prev = gtk_tree_iter_copy (&iter);
            gtk_tree_model_iter_previous (model, iter_prev);

            gtk_list_store_move_before (emp->priv->headers, &iter, iter_prev);

            indices = gtk_tree_path_get_indices (path);
            header = webkit_dom_node_list_item (headers, indices[0]);
            node2 = webkit_dom_node_get_previous_sibling (header);
            parent = webkit_dom_node_get_parent_node (header);

            webkit_dom_node_remove_child (parent, header, NULL);
            webkit_dom_node_insert_before (parent, header, node2, NULL);

            gtk_tree_path_free (path);
            gtk_tree_iter_free (iter_prev);
        }

    } else if (button == emp->priv->buttons[BUTTON_DOWN]) {

        GtkTreeIter *iter_next;
        WebKitDOMNode *node2;

        for (l = references; l; l = l->next) {

            path = gtk_tree_row_reference_get_path (l->data);
            gtk_tree_model_get_iter (model, &iter, path);
            iter_next = gtk_tree_iter_copy (&iter);
            gtk_tree_model_iter_next (model, iter_next);

            gtk_list_store_move_after (emp->priv->headers, &iter, iter_next);

            indices = gtk_tree_path_get_indices (path);
            header = webkit_dom_node_list_item (headers, indices[0]);
            node2 = webkit_dom_node_get_next_sibling (header);
            parent = webkit_dom_node_get_parent_node (header);

            webkit_dom_node_remove_child (parent, header, NULL);
            webkit_dom_node_insert_before (parent, header,
                webkit_dom_node_get_next_sibling (node2), NULL);

            gtk_tree_path_free (path);
            gtk_tree_iter_free (iter_next);
        }

    } else if (button == emp->priv->buttons[BUTTON_BOTTOM]) {

        references = g_list_reverse (references);

        for (l = references; l; l = l->next) {
            path = gtk_tree_row_reference_get_path (l->data);
            gtk_tree_model_get_iter (model, &iter, path);
            gtk_list_store_move_before (emp->priv->headers, &iter, NULL);

                        /* Move the header row in HTML document */
            indices = gtk_tree_path_get_indices (path);
            header = webkit_dom_node_list_item (headers, indices[0]);
            parent = webkit_dom_node_get_parent_node (header);
            webkit_dom_node_remove_child (parent, header, NULL);
            webkit_dom_node_append_child (parent, header, NULL);

            gtk_tree_path_free (path);
        }
    };

    g_list_foreach (references, (GFunc) gtk_tree_row_reference_free, NULL);
    g_list_free (references);

        /* Keep the selection in middle of the screen */
    path = gtk_tree_row_reference_get_path (selection_middle);
    gtk_tree_view_scroll_to_cell (
        GTK_TREE_VIEW (emp->priv->treeview),
        path, COLUMN_ACTIVE, TRUE, 0.5, 0.5);
    gtk_tree_path_free (path);
    gtk_tree_row_reference_free (selection_middle);

    g_list_foreach (selected_rows, (GFunc) gtk_tree_path_free, NULL);
    g_list_free (selected_rows);

    emp_headers_tab_selection_changed (selection, user_data);
}

static GtkWidget *
emp_create_headers_tab (GtkPrintOperation *operation,
                        EMailPrinter *emp)
{
    GtkWidget *vbox, *hbox, *scw, *button;
    GtkTreeView *view;
    GtkTreeSelection *selection;
    GtkTreeViewColumn *column;
    GtkCellRenderer *renderer;

    hbox = gtk_box_new (GTK_ORIENTATION_HORIZONTAL, 5);
    vbox = gtk_box_new (GTK_ORIENTATION_VERTICAL, 5);
    gtk_box_pack_end (GTK_BOX (hbox), vbox, FALSE, FALSE, 5);

    emp->priv->treeview = gtk_tree_view_new_with_model (
        GTK_TREE_MODEL (emp->priv->headers));
    view = GTK_TREE_VIEW (emp->priv->treeview);
    selection = gtk_tree_view_get_selection (view);
    gtk_tree_selection_set_mode (selection, GTK_SELECTION_MULTIPLE);
    g_signal_connect (
        selection, "changed",
        G_CALLBACK (emp_headers_tab_selection_changed), emp);

    renderer = gtk_cell_renderer_toggle_new ();
    g_signal_connect (
        renderer, "toggled",
        G_CALLBACK (header_active_renderer_toggled_cb), emp);
    column = gtk_tree_view_column_new_with_attributes (
        _("Print"), renderer,
        "active", COLUMN_ACTIVE, NULL);
    gtk_tree_view_append_column (view, column);

    renderer = gtk_cell_renderer_text_new ();
    column = gtk_tree_view_column_new_with_attributes (
        _("Header Name"), renderer,
        "text", COLUMN_HEADER_NAME, NULL);
    gtk_tree_view_append_column (view, column);

    renderer = gtk_cell_renderer_text_new ();
    column = gtk_tree_view_column_new_with_attributes (
        _("Header Value"), renderer,
        "text", COLUMN_HEADER_VALUE, NULL);
    gtk_tree_view_append_column (view, column);

    scw = gtk_scrolled_window_new (NULL, NULL);
    gtk_container_add (GTK_CONTAINER (scw), GTK_WIDGET (view));
    gtk_box_pack_start (GTK_BOX (hbox), scw, TRUE, TRUE, 0);

    button = gtk_button_new_from_stock (GTK_STOCK_SELECT_ALL);
    emp->priv->buttons[BUTTON_SELECT_ALL] = button;
    g_signal_connect (
        button, "clicked",
        G_CALLBACK (emp_headers_tab_toggle_selection), emp);
    gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, TRUE, 5);

    button = gtk_button_new_from_stock (GTK_STOCK_CLEAR);
    emp->priv->buttons[BUTTON_SELECT_NONE] = button;
    g_signal_connect (
        button, "clicked",
        G_CALLBACK (emp_headers_tab_toggle_selection), emp);
    gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, TRUE, 5);

    button = gtk_button_new_from_stock (GTK_STOCK_GOTO_TOP);
    emp->priv->buttons[BUTTON_TOP] = button;
    gtk_widget_set_sensitive (button, FALSE);
    g_signal_connect (
        button, "clicked",
        G_CALLBACK (emp_headers_tab_move), emp);
    gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, TRUE, 5);

    button = gtk_button_new_from_stock (GTK_STOCK_GO_UP);
    emp->priv->buttons[BUTTON_UP] = button;
    gtk_widget_set_sensitive (button, FALSE);
    g_signal_connect (
        button, "clicked",
        G_CALLBACK (emp_headers_tab_move), emp);
    gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, TRUE, 5);

    button = gtk_button_new_from_stock (GTK_STOCK_GO_DOWN);
    emp->priv->buttons[BUTTON_DOWN] = button;
    gtk_widget_set_sensitive (button, FALSE);
    g_signal_connect (
        button, "clicked",
        G_CALLBACK (emp_headers_tab_move), emp);
    gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, TRUE, 5);

    button = gtk_button_new_from_stock (GTK_STOCK_GOTO_BOTTOM);
    emp->priv->buttons[BUTTON_BOTTOM] = button;
    gtk_widget_set_sensitive (button, FALSE);
    g_signal_connect (
        button, "clicked",
        G_CALLBACK (emp_headers_tab_move), emp);
    gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, TRUE, 5);

    gtk_print_operation_set_custom_tab_label (operation, _("Headers"));
    gtk_widget_show_all (hbox);

    return hbox;
}

static void
emp_set_parts_list (EMailPrinter *emp,
                    EMailPartList *parts_list)
{
    CamelMediumHeader *header;
    CamelMimeMessage *message;
    GArray *headers;
    gint i;
    GtkTreeIter last_known = { 0 };

    g_return_if_fail (parts_list);

    emp->priv->parts_list = g_object_ref (parts_list);

    if (emp->priv->headers)
        g_object_unref (emp->priv->headers);
    emp->priv->headers = gtk_list_store_new (
        5,
        G_TYPE_BOOLEAN, G_TYPE_STRING, G_TYPE_STRING, G_TYPE_POINTER, G_TYPE_INT);

    message = e_mail_part_list_get_message (parts_list);
    headers = camel_medium_get_headers (CAMEL_MEDIUM (message));
    if (!headers)
        return;

    for (i = 0; i < headers->len; i++) {
        GtkTreeIter iter;
        GList *found_header;
        EMailFormatterHeader *emfh;

        header = &g_array_index (headers, CamelMediumHeader, i);
        emfh = e_mail_formatter_header_new (header->name, header->value);

        found_header = g_queue_find_custom (
                (GQueue *) e_mail_formatter_get_headers (
                    E_MAIL_FORMATTER (emp->priv->formatter)),
                emfh, (GCompareFunc) emp_header_name_equal);

        if (!found_header) {
            emfh->flags |= E_MAIL_FORMATTER_HEADER_FLAG_HIDDEN;
            e_mail_formatter_add_header_struct (
                E_MAIL_FORMATTER (emp->priv->formatter), emfh);
            gtk_list_store_append (emp->priv->headers, &iter);
        } else {
            if (gtk_list_store_iter_is_valid (emp->priv->headers, &last_known))
                gtk_list_store_insert_after (emp->priv->headers, &iter, &last_known);
            else
                gtk_list_store_insert_after (emp->priv->headers, &iter, NULL);

            last_known = iter;
        }

        gtk_list_store_set (
            emp->priv->headers, &iter,
            COLUMN_ACTIVE, (found_header != NULL),
            COLUMN_HEADER_NAME, emfh->name,
            COLUMN_HEADER_VALUE, emfh->value,
            COLUMN_HEADER_STRUCT, emfh, -1);
    }

    camel_medium_free_headers (CAMEL_MEDIUM (message), headers);
}

static void
emp_set_property (GObject *object,
                  guint property_id,
                  const GValue *value,
                  GParamSpec *pspec)
{
    EMailPrinter *emp = E_MAIL_PRINTER (object);

    switch (property_id) {

        case PROP_PART_LIST:
            emp_set_parts_list (emp, g_value_get_pointer (value));
            return;
    }

    G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
}

static void
emp_get_property (GObject *object,
                  guint property_id,
                  GValue *value,
                  GParamSpec *pspec)
{
    EMailPrinter *emp = E_MAIL_PRINTER (object);

    switch (property_id) {

        case PROP_PART_LIST:
            g_value_set_pointer (value, emp->priv->parts_list);
            return;
    }

    G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
}

static void
emp_finalize (GObject *object)
{
    EMailPrinterPrivate *priv = E_MAIL_PRINTER (object)->priv;

    if (priv->formatter) {
        g_object_unref (priv->formatter);
        priv->formatter = NULL;
    }

    if (priv->headers) {
        GtkTreeIter iter;

        if (gtk_tree_model_get_iter_first (GTK_TREE_MODEL (priv->headers), &iter)) {
            do {
                EMailFormatterHeader *header = NULL;
                gtk_tree_model_get (
                    GTK_TREE_MODEL (priv->headers), &iter,
                    COLUMN_HEADER_STRUCT, &header, -1);
                e_mail_formatter_header_free (header);
            } while (gtk_tree_model_iter_next (GTK_TREE_MODEL (priv->headers), &iter));
        }
        g_object_unref (priv->headers);
        priv->headers = NULL;
    }

    if (priv->webview) {
        g_object_unref (priv->webview);
        priv->webview = NULL;
    }

    if (priv->uri) {
        g_free (priv->uri);
        priv->uri = NULL;
    }

    if (priv->operation) {
        g_object_unref (priv->operation);
        priv->operation = NULL;
    }

    if (priv->parts_list) {
        g_object_unref (priv->parts_list);
        priv->parts_list = NULL;
    }

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

static void
e_mail_printer_class_init (EMailPrinterClass *class)
{
    GObjectClass *object_class;

    g_type_class_add_private (class, sizeof (EMailPrinterPrivate));

    object_class = G_OBJECT_CLASS (class);
    object_class->set_property = emp_set_property;
    object_class->get_property = emp_get_property;
    object_class->finalize = emp_finalize;

    g_object_class_install_property (
        object_class,
        PROP_PART_LIST,
        g_param_spec_pointer (
            "parts-list",
            "Parts List",
            NULL,
            G_PARAM_READWRITE |
            G_PARAM_CONSTRUCT_ONLY));

    signals[SIGNAL_DONE] = g_signal_new (
        "done",
        G_TYPE_FROM_CLASS (class),
        G_SIGNAL_RUN_FIRST,
        G_STRUCT_OFFSET (EMailPrinterClass, done),
        NULL, NULL,
        e_marshal_VOID__OBJECT_INT,
        G_TYPE_NONE, 2,
        GTK_TYPE_PRINT_OPERATION, G_TYPE_INT);
}

static void
e_mail_printer_init (EMailPrinter *emp)
{
    emp->priv = G_TYPE_INSTANCE_GET_PRIVATE (
        emp, E_TYPE_MAIL_PRINTER, EMailPrinterPrivate);

    emp->priv->formatter = (EMailFormatterPrint *) e_mail_formatter_print_new ();
    emp->priv->headers = NULL;
    emp->priv->webview = NULL;
}

EMailPrinter *
e_mail_printer_new (EMailPartList *source)
{
    EMailPrinter *emp;

    emp = g_object_new (
        E_TYPE_MAIL_PRINTER,
        "parts-list", source, NULL);

    return emp;
}

void
e_mail_printer_print (EMailPrinter *emp,
                      GtkPrintOperationAction action,
              EMailFormatter *formatter,
                      GCancellable *cancellable)
{
    g_return_if_fail (E_IS_MAIL_PRINTER (emp));

    if (emp->priv->operation)
        g_object_unref (emp->priv->operation);
    emp->priv->operation = e_print_operation_new ();
    emp->priv->print_action = action;
    gtk_print_operation_set_unit (emp->priv->operation, GTK_UNIT_PIXEL);

    gtk_print_operation_set_show_progress (emp->priv->operation, TRUE);
    g_signal_connect (
        emp->priv->operation, "create-custom-widget",
        G_CALLBACK (emp_create_headers_tab), emp);
    g_signal_connect (
        emp->priv->operation, "done",
        G_CALLBACK (emp_printing_done), emp);
    g_signal_connect (
        emp->priv->operation, "draw-page",
        G_CALLBACK (emp_draw_footer), NULL);

    if (cancellable)
        g_signal_connect_swapped (
            cancellable, "cancelled",
            G_CALLBACK (gtk_print_operation_cancel), emp->priv->operation);

    emp_run_print_operation (emp, formatter);
}

const gchar *
e_mail_printer_get_export_filename (EMailPrinter *printer)
{
    g_return_val_if_fail (E_IS_MAIL_PRINTER (printer), NULL);

    return printer->priv->export_filename;
}

void
e_mail_printer_set_export_filename (EMailPrinter *printer,
                                    const gchar *filename)
{
    g_return_if_fail (E_IS_MAIL_PRINTER (printer));

    if (printer->priv->export_filename)
      g_free (printer->priv->export_filename);

    printer->priv->export_filename = g_strdup (filename);
}