path: root/e-util/e-mail-signature-preview.c

/*
 * e-mail-signature-preview.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.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, see <http://www.gnu.org/licenses/>.
 *
 */

#include "e-mail-signature-preview.h"

#include <fcntl.h>
#include <string.h>
#include <unistd.h>
#include <glib/gstdio.h>

#include "e-alert-sink.h"

#define E_MAIL_SIGNATURE_PREVIEW_GET_PRIVATE(obj) \
    (G_TYPE_INSTANCE_GET_PRIVATE \
    ((obj), E_TYPE_MAIL_SIGNATURE_PREVIEW, EMailSignaturePreviewPrivate))

#define SOURCE_IS_MAIL_SIGNATURE(source) \
    (e_source_has_extension ((source), E_SOURCE_EXTENSION_MAIL_SIGNATURE))

struct _EMailSignaturePreviewPrivate {
    ESourceRegistry *registry;
    GCancellable *cancellable;
    gchar *source_uid;
};

enum {
    PROP_0,
    PROP_REGISTRY,
    PROP_SOURCE_UID
};

enum {
    REFRESH,
    LAST_SIGNAL
};

static guint signals[LAST_SIGNAL];

G_DEFINE_TYPE (
    EMailSignaturePreview,
    e_mail_signature_preview,
    E_TYPE_WEB_VIEW)

static void
replace_local_image_links (WebKitDOMDocument *document)
{
    gint ii, length;
    WebKitDOMNodeList *list;

    list = webkit_dom_document_query_selector_all (
        document, "img[src^=\"file://\"]", NULL);
    length = webkit_dom_node_list_get_length (list);

    for (ii = 0; ii < length; ii++) {
        gchar *src, *new_src;
        WebKitDOMHTMLImageElement *img;

        img = WEBKIT_DOM_HTML_IMAGE_ELEMENT (
            webkit_dom_node_list_item (list, ii));
        src = webkit_dom_html_image_element_get_src (img);

        /* this forms "evo-file://", which can be loaded,
         * while "file://" cannot be, due to WebKit policy */
        new_src = g_strconcat ("evo-", src, NULL);
        webkit_dom_html_image_element_set_src (img, new_src);
        g_free (new_src);
        g_free (src);
    }

    list = webkit_dom_document_get_elements_by_tag_name ( document, "iframe");
    length = webkit_dom_node_list_get_length (list);
    for (ii = 0; ii < length; ii++) {
        WebKitDOMDocument *content_document;
        WebKitDOMHTMLIFrameElement *iframe;

        iframe = WEBKIT_DOM_HTML_IFRAME_ELEMENT (
            webkit_dom_node_list_item (list, ii));

        content_document =
            webkit_dom_html_iframe_element_get_content_document (iframe);

        if (content_document && WEBKIT_DOM_IS_DOCUMENT (content_document))
            replace_local_image_links (content_document);
    }
}

static void
signature_preview_document_loaded_cb (WebKitWebView *web_view,
                                      WebKitWebFrame *web_frame,
                                      gpointer user_data)
{
    replace_local_image_links (webkit_web_view_get_dom_document (web_view));
}

static void
mail_signature_preview_load_cb (ESource *source,
                                GAsyncResult *result,
                                EMailSignaturePreview *preview)
{
    ESourceMailSignature *extension;
    const gchar *extension_name;
    const gchar *mime_type;
    gchar *contents = NULL;
    GError *error = NULL;

    e_source_mail_signature_load_finish (
        source, result, &contents, NULL, &error);

    /* Ignore cancellations. */
    if (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_CANCELLED)) {
        g_warn_if_fail (contents == NULL);
        g_object_unref (preview);
        g_error_free (error);
        return;

    } else if (error != NULL) {
        g_warn_if_fail (contents == NULL);
        e_alert_submit (
            E_ALERT_SINK (preview),
            "widgets:no-load-signature",
            error->message, NULL);
        g_object_unref (preview);
        g_error_free (error);
        return;
    }

    g_return_if_fail (contents != NULL);

    extension_name = E_SOURCE_EXTENSION_MAIL_SIGNATURE;
    extension = e_source_get_extension (source, extension_name);
    mime_type = e_source_mail_signature_get_mime_type (extension);

    if (g_strcmp0 (mime_type, "text/html") == 0) {
        webkit_web_view_load_string (
            WEBKIT_WEB_VIEW (preview), contents,
            "text/html", "UTF-8", "file:///");
    } else {
        gchar *string;

        string = g_markup_printf_escaped ("<pre>%s</pre>", contents);
        webkit_web_view_load_string (
            WEBKIT_WEB_VIEW (preview), string,
            "text/html", "UTF-8", "file:///");
        g_free (string);
    }

    g_free (contents);

    g_object_unref (preview);
}

static void
mail_signature_preview_set_registry (EMailSignaturePreview *preview,
                                     ESourceRegistry *registry)
{
    g_return_if_fail (E_IS_SOURCE_REGISTRY (registry));
    g_return_if_fail (preview->priv->registry == NULL);

    preview->priv->registry = g_object_ref (registry);
}

static void
mail_signature_preview_set_property (GObject *object,
                                guint property_id,
                                const GValue *value,
                                GParamSpec *pspec)
{
    switch (property_id) {
        case PROP_REGISTRY:
            mail_signature_preview_set_registry (
                E_MAIL_SIGNATURE_PREVIEW (object),
                g_value_get_object (value));
            return;

        case PROP_SOURCE_UID:
            e_mail_signature_preview_set_source_uid (
                E_MAIL_SIGNATURE_PREVIEW (object),
                g_value_get_string (value));
            return;
    }

    G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
}

static void
mail_signature_preview_get_property (GObject *object,
                                guint property_id,
                                GValue *value,
                                GParamSpec *pspec)
{
    switch (property_id) {
        case PROP_REGISTRY:
            g_value_set_object (
                value,
                e_mail_signature_preview_get_registry (
                E_MAIL_SIGNATURE_PREVIEW (object)));
            return;

        case PROP_SOURCE_UID:
            g_value_set_string (
                value,
                e_mail_signature_preview_get_source_uid (
                E_MAIL_SIGNATURE_PREVIEW (object)));
            return;
    }

    G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
}

static void
mail_signature_preview_dispose (GObject *object)
{
    EMailSignaturePreviewPrivate *priv;

    priv = E_MAIL_SIGNATURE_PREVIEW_GET_PRIVATE (object);

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

    if (priv->cancellable != NULL) {
        g_cancellable_cancel (priv->cancellable);
        g_object_unref (priv->cancellable);
        priv->cancellable = NULL;
    }

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

static void
mail_signature_preview_finalize (GObject *object)
{
    EMailSignaturePreviewPrivate *priv;

    priv = E_MAIL_SIGNATURE_PREVIEW_GET_PRIVATE (object);

    g_free (priv->source_uid);

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

static void
mail_signature_preview_refresh (EMailSignaturePreview *preview)
{
    ESourceRegistry *registry;
    ESource *source;
    const gchar *extension_name;
    const gchar *source_uid;

    /* Cancel any unfinished refreshes. */
    if (preview->priv->cancellable != NULL) {
        g_cancellable_cancel (preview->priv->cancellable);
        g_object_unref (preview->priv->cancellable);
        preview->priv->cancellable = NULL;
    }

    source_uid = e_mail_signature_preview_get_source_uid (preview);

    if (source_uid == NULL)
        goto fail;

    registry = e_mail_signature_preview_get_registry (preview);
    source = e_source_registry_ref_source (registry, source_uid);

    if (source == NULL)
        goto fail;

    extension_name = E_SOURCE_EXTENSION_MAIL_SIGNATURE;
    if (!e_source_has_extension (source, extension_name)) {
        g_object_unref (source);
        goto fail;
    }

    preview->priv->cancellable = g_cancellable_new ();

    e_source_mail_signature_load (
        source, G_PRIORITY_DEFAULT,
        preview->priv->cancellable, (GAsyncReadyCallback)
        mail_signature_preview_load_cb, g_object_ref (preview));

    g_object_unref (source);

    return;

fail:
    e_web_view_clear (E_WEB_VIEW (preview));
}

static void
e_mail_signature_preview_class_init (EMailSignaturePreviewClass *class)
{
    GObjectClass *object_class;

    g_type_class_add_private (class, sizeof (EMailSignaturePreviewPrivate));

    object_class = G_OBJECT_CLASS (class);
    object_class->set_property = mail_signature_preview_set_property;
    object_class->get_property = mail_signature_preview_get_property;
    object_class->dispose = mail_signature_preview_dispose;
    object_class->finalize = mail_signature_preview_finalize;

    class->refresh = mail_signature_preview_refresh;

    g_object_class_install_property (
        object_class,
        PROP_REGISTRY,
        g_param_spec_object (
            "registry",
            "Registry",
            NULL,
            E_TYPE_SOURCE_REGISTRY,
            G_PARAM_READWRITE |
            G_PARAM_CONSTRUCT_ONLY |
            G_PARAM_STATIC_STRINGS));

    g_object_class_install_property (
        object_class,
        PROP_SOURCE_UID,
        g_param_spec_string (
            "source-uid",
            "Source UID",
            NULL,
            NULL,
            G_PARAM_READWRITE |
            G_PARAM_STATIC_STRINGS));

    signals[REFRESH] = g_signal_new (
        "refresh",
        G_TYPE_FROM_CLASS (class),
        G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
        G_STRUCT_OFFSET (EMailSignaturePreviewClass, refresh),
        NULL, NULL,
        g_cclosure_marshal_VOID__VOID,
        G_TYPE_NONE, 0);
}

static void
e_mail_signature_preview_init (EMailSignaturePreview *preview)
{
    preview->priv = E_MAIL_SIGNATURE_PREVIEW_GET_PRIVATE (preview);

    g_signal_connect (
        preview, "document-load-finished",
        G_CALLBACK (signature_preview_document_loaded_cb), NULL);
}

GtkWidget *
e_mail_signature_preview_new (ESourceRegistry *registry)
{
    g_return_val_if_fail (E_IS_SOURCE_REGISTRY (registry), NULL);

    return g_object_new (
        E_TYPE_MAIL_SIGNATURE_PREVIEW,
        "registry", registry, NULL);
}

void
e_mail_signature_preview_refresh (EMailSignaturePreview *preview)
{
    g_return_if_fail (E_IS_MAIL_SIGNATURE_PREVIEW (preview));

    g_signal_emit (preview, signals[REFRESH], 0);
}

ESourceRegistry *
e_mail_signature_preview_get_registry (EMailSignaturePreview *preview)
{
    g_return_val_if_fail (E_IS_MAIL_SIGNATURE_PREVIEW (preview), NULL);

    return preview->priv->registry;
}

const gchar *
e_mail_signature_preview_get_source_uid (EMailSignaturePreview *preview)
{
    g_return_val_if_fail (E_IS_MAIL_SIGNATURE_PREVIEW (preview), NULL);

    return preview->priv->source_uid;
}

void
e_mail_signature_preview_set_source_uid (EMailSignaturePreview *preview,
                                         const gchar *source_uid)
{
    g_return_if_fail (E_IS_MAIL_SIGNATURE_PREVIEW (preview));

    /* Avoid repeatedly loading the same signature file. */
    if (g_strcmp0 (source_uid, preview->priv->source_uid) == 0)
        return;

    g_free (preview->priv->source_uid);
    preview->priv->source_uid = g_strdup (source_uid);

    g_object_notify (G_OBJECT (preview), "source-uid");

    e_mail_signature_preview_refresh (preview);
}