/*
* text-highlight.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/>
*
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include "e-mail-formatter-text-highlight.h"
#include "languages.h"
#include <em-format/e-mail-formatter-extension.h>
#include <em-format/e-mail-formatter.h>
#include <em-format/e-mail-part-utils.h>
#include <e-util/e-util.h>
#include <shell/e-shell-settings.h>
#include <shell/e-shell.h>
#include <libebackend/libebackend.h>
#include <libedataserver/libedataserver.h>
#include <glib/gi18n-lib.h>
#include <X11/Xlib.h>
#include <camel/camel.h>
typedef EMailFormatterExtension EMailFormatterTextHighlight;
typedef EMailFormatterExtensionClass EMailFormatterTextHighlightClass;
typedef EExtension EMailFormatterTextHighlightLoader;
typedef EExtensionClass EMailFormatterTextHighlightLoaderClass;
GType e_mail_formatter_text_highlight_get_type (void);
GType e_mail_formatter_text_highlight_loader_get_type (void);
G_DEFINE_DYNAMIC_TYPE (
EMailFormatterTextHighlight,
e_mail_formatter_text_highlight,
E_TYPE_MAIL_FORMATTER_EXTENSION)
G_DEFINE_DYNAMIC_TYPE (
EMailFormatterTextHighlightLoader,
e_mail_formatter_text_highlight_loader,
E_TYPE_EXTENSION)
static gchar *
get_default_font (void)
{
gchar *font;
GSettings *settings;
settings = g_settings_new ("org.gnome.desktop.interface");
font = g_settings_get_string (settings, "monospace-font-name");
g_object_unref (settings);
return font ? font : g_strdup ("monospace 10");
}
static gchar *
get_syntax (EMailPart *part,
const gchar *uri)
{
gchar *syntax = NULL;
CamelContentType *ct = NULL;
if (uri) {
SoupURI *soup_uri = soup_uri_new (uri);
GHashTable *query = soup_form_decode (soup_uri->query);
syntax = g_hash_table_lookup (query, "__formatas");
if (syntax) {
syntax = g_strdup (syntax);
}
g_hash_table_destroy (query);
soup_uri_free (soup_uri);
}
/* Try to detect syntax by content-type first */
if (syntax == NULL) {
ct = camel_mime_part_get_content_type (part->part);
if (ct) {
gchar *mime_type = camel_content_type_simple (ct);
syntax = (gchar *) get_syntax_for_mime_type (mime_type);
syntax = syntax ? g_strdup (syntax) : NULL;
g_free (mime_type);
}
}
/* If it fails or the content type too generic, try to detect it by
* filename extension */
if (syntax == NULL ||
(ct != NULL &&
(camel_content_type_is (ct, "application", "octet-stream") ||
(camel_content_type_is (ct, "text", "plain"))))) {
const gchar *filename = camel_mime_part_get_filename (part->part);
if (filename) {
gchar *ext = g_strrstr (filename, ".");
if (ext) {
syntax = (gchar *) get_syntax_for_ext (ext + 1);
syntax = syntax ? g_strdup (syntax) : NULL;
}
}
}
/* Out of ideas - use plain text */
if (syntax == NULL) {
syntax = g_strdup ("txt");
}
return syntax;
}
static gboolean
emfe_text_highlight_format (EMailFormatterExtension *extension,
EMailFormatter *formatter,
EMailFormatterContext *context,
EMailPart *part,
CamelStream *stream,
GCancellable *cancellable)
{
/* Don't format text/html unless it's an attachment */
CamelContentType *ct = camel_mime_part_get_content_type (part->part);
if (ct && camel_content_type_is (ct, "text", "html")) {
const CamelContentDisposition *disp;
disp = camel_mime_part_get_content_disposition (part->part);
if (!disp || g_strcmp0 (disp->disposition, "attachment") != 0)
return FALSE;
}
if (context->mode == E_MAIL_FORMATTER_MODE_PRINTING) {
CamelDataWrapper *dw;
CamelStream *filter_stream;
CamelMimeFilter *mime_filter;
dw = camel_medium_get_content (CAMEL_MEDIUM (part->part));
if (!dw) {
return FALSE;
}
camel_stream_write_string (
stream, "<pre><div class=\"pre\">", cancellable, NULL);
filter_stream = camel_stream_filter_new (stream);
mime_filter = camel_mime_filter_tohtml_new (
CAMEL_MIME_FILTER_TOHTML_PRE |
CAMEL_MIME_FILTER_TOHTML_CONVERT_SPACES, 0);
camel_stream_filter_add (
CAMEL_STREAM_FILTER (filter_stream), mime_filter);
g_object_unref (mime_filter);
e_mail_formatter_format_text (
formatter, part, filter_stream, cancellable);
camel_stream_flush (filter_stream, cancellable, NULL);
g_object_unref (filter_stream);
camel_stream_write_string (
stream, "</div></pre>", cancellable, NULL);
return TRUE;
} else if (context->mode == E_MAIL_FORMATTER_MODE_RAW) {
gint pipe_stdin, pipe_stdout;
GPid pid;
CamelStream *read, *write, *utf8;
CamelDataWrapper *dw;
gchar *font_family, *font_size, *syntax, *tmp;
GByteArray *ba;
gboolean use_custom_font;
EShell *shell;
EShellSettings *settings;
PangoFontDescription *fd;
const gchar *argv[] = { HIGHLIGHT_COMMAND,
NULL, /* --font= */
NULL, /* --font-size= */
NULL, /* --syntax= */
"--out-format=html",
"--include-style",
"--inline-css",
"--style=bclear",
"--failsafe",
NULL };
dw = camel_medium_get_content (CAMEL_MEDIUM (part->part));
if (!dw) {
return FALSE;
}
syntax = get_syntax (part, context->uri);
/* Use the traditional text/plain formatter for plain-text */
if (g_strcmp0 (syntax, "txt") == 0) {
g_free (syntax);
return FALSE;
}
shell = e_shell_get_default ();
settings = e_shell_get_shell_settings (shell);
fd = NULL;
use_custom_font = e_shell_settings_get_boolean (
settings, "mail-use-custom-fonts");
if (!use_custom_font) {
gchar *font;
font = get_default_font ();
fd = pango_font_description_from_string (font);
g_free (font);
} else {
gchar *font;
font = e_shell_settings_get_string (
settings, "mail-font-monospace");
if (!font)
font = get_default_font ();
fd = pango_font_description_from_string (font);
g_free (font);
}
font_family = g_strdup_printf (
"--font='%s'",
pango_font_description_get_family (fd));
font_size = g_strdup_printf (
"--font-size=%d",
pango_font_description_get_size (fd) / PANGO_SCALE);
argv[1] = font_family;
argv[2] = font_size;
argv[3] = g_strdup_printf ("--syntax=%s", syntax);
g_free (syntax);
if (g_spawn_async_with_pipes (
NULL, (gchar **) argv, NULL,
G_SPAWN_SEARCH_PATH |
G_SPAWN_DO_NOT_REAP_CHILD,
NULL, NULL, &pid, &pipe_stdin, &pipe_stdout, NULL, NULL)) {
write = camel_stream_fs_new_with_fd (pipe_stdin);
read = camel_stream_fs_new_with_fd (pipe_stdout);
/* Decode the content of mime part to the 'utf8' stream */
utf8 = camel_stream_mem_new ();
camel_data_wrapper_decode_to_stream_sync (
dw, utf8, cancellable, NULL);
/* Convert the binary data do someting displayable */
ba = camel_stream_mem_get_byte_array (CAMEL_STREAM_MEM (utf8));
tmp = e_util_utf8_data_make_valid ((gchar *) ba->data, ba->len);
/* Send the sanitized data to the highlighter */
camel_stream_write_string (write, tmp, cancellable, NULL);
g_free (tmp);
g_object_unref (utf8);
g_object_unref (write);
g_spawn_close_pid (pid);
g_seekable_seek (G_SEEKABLE (read), 0, G_SEEK_SET, cancellable, NULL);
camel_stream_write_to_stream (read, stream, cancellable, NULL);
g_object_unref (read);
} else {
/* We can't call e_mail_formatter_format_as on text/plain,
* because text-highlight is registered as an handler for
* text/plain, so we would end up in an endless recursion.
*
* Just return FALSE here and EMailFormatter will automatically
* fall back to the default text/plain formatter */
if (camel_content_type_is (ct, "text", "plain")) {
g_free (font_family);
g_free (font_size);
g_free ((gchar *) argv[3]);
pango_font_description_free (fd);
return FALSE;
} else {
/* In case of any other content, force use of
* text/plain formatter, because returning FALSE
* for text/x-patch or application/php would show
* an error, as there is no other handler registered
* for these */
e_mail_formatter_format_as (
formatter, context, part, stream,
"application/vnd.evolution.plaintext",
cancellable);
}
}
g_free (font_family);
g_free (font_size);
g_free ((gchar *) argv[3]);
pango_font_description_free (fd);
} else {
CamelFolder *folder;
const gchar *message_uid;
gchar *uri, *str;
gchar *syntax;
folder = e_mail_part_list_get_folder (context->part_list);
message_uid = e_mail_part_list_get_message_uid (context->part_list);
syntax = get_syntax (part, NULL);
uri = e_mail_part_build_uri (
folder, message_uid,
"part_id", G_TYPE_STRING, part->id,
"mode", G_TYPE_INT, E_MAIL_FORMATTER_MODE_RAW,
"__formatas", G_TYPE_STRING, syntax,
NULL);
g_free (syntax);
str = g_strdup_printf (
"<div class=\"part-container-nostyle\" >"
"<iframe width=\"100%%\" height=\"10\""
" id=\"%s\" name=\"%s\" "
" frameborder=\"0\" src=\"%s\" "
" style=\"border: 1px solid #%06x; background-color: #%06x;\">"
"</iframe>"
"</div>",
part->id, part->id, uri,
e_color_to_value ((GdkColor *)
e_mail_formatter_get_color (
formatter, E_MAIL_FORMATTER_COLOR_FRAME)),
e_color_to_value ((GdkColor *)
e_mail_formatter_get_color (
formatter, E_MAIL_FORMATTER_COLOR_CONTENT)));
camel_stream_write_string (stream, str, cancellable, NULL);
g_free (str);
g_free (uri);
}
return TRUE;
}
static void
e_mail_formatter_text_highlight_class_init (EMailFormatterExtensionClass *class)
{
class->display_name = _("Text Highlight");
class->description = _("Syntax highlighting of mail parts");
class->mime_types = get_mime_types ();
class->format = emfe_text_highlight_format;
}
static void
e_mail_formatter_text_highlight_class_finalize (EMailFormatterExtensionClass *class)
{
}
static void
e_mail_formatter_text_highlight_init (EMailFormatterExtension *extension)
{
}
static void
mail_formatter_text_highlight_loader_constructed (GObject *object)
{
EExtensible *extensible;
extensible = e_extension_get_extensible (E_EXTENSION (object));
e_mail_extension_registry_add_extension (
E_MAIL_EXTENSION_REGISTRY (extensible),
get_mime_types (),
e_mail_formatter_text_highlight_get_type ());
}
static void
e_mail_formatter_text_highlight_loader_class_init (EExtensionClass *class)
{
GObjectClass *object_class;
object_class = G_OBJECT_CLASS (class);
object_class->constructed = mail_formatter_text_highlight_loader_constructed;
class->extensible_type = E_TYPE_MAIL_FORMATTER_EXTENSION_REGISTRY;
}
static void
e_mail_formatter_text_highlight_loader_class_finalize (EExtensionClass *class)
{
}
static void
e_mail_formatter_text_highlight_loader_init (EExtension *extension)
{
}
void
e_mail_formatter_text_highlight_type_register (GTypeModule *type_module)
{
e_mail_formatter_text_highlight_register_type (type_module);
e_mail_formatter_text_highlight_loader_register_type (type_module);
}
