X-Git-Url: http://git.maemo.org/git/?p=modest;a=blobdiff_plain;f=src%2Fmodest-formatter.c;h=aade0e3bf32ab7619cf97888105aa14bd2fe97b3;hp=73fdfbb3e0ccc0a59951c3e92eca9ece645e2483;hb=12a672c559d983c4e49a7e4054ee14c0177ecb1c;hpb=b3189f988bc953765fd3cdb325c4dd883ac9f608 diff --git a/src/modest-formatter.c b/src/modest-formatter.c index 73fdfbb..aade0e3 100644 --- a/src/modest-formatter.c +++ b/src/modest-formatter.c @@ -32,12 +32,16 @@ #include #include #include -#include -#include +#include #include "modest-formatter.h" #include "modest-text-utils.h" #include "modest-tny-platform-factory.h" -#include +#include "modest-runtime.h" +#include "modest-stream-html-to-text.h" + +#define LINE_WRAP 78 +#define MAX_BODY_LINES 1024 +#define MAX_BODY_LENGTH 1024*128 typedef struct _ModestFormatterPrivate ModestFormatterPrivate; struct _ModestFormatterPrivate { @@ -55,27 +59,98 @@ typedef gchar* FormatterFunc (ModestFormatter *self, const gchar *text, TnyHeade static TnyMsg *modest_formatter_do (ModestFormatter *self, TnyMimePart *body, TnyHeader *header, FormatterFunc func, GList *attachments); -static gchar* modest_formatter_wrapper_cite (ModestFormatter *self, const gchar *text, TnyHeader *header, GList *attachments); -static gchar* modest_formatter_wrapper_quote (ModestFormatter *self, const gchar *text, TnyHeader *header, GList *attachments); -static gchar* modest_formatter_wrapper_inline (ModestFormatter *self, const gchar *text, TnyHeader *header, GList *attachments); +static gchar* modest_formatter_wrapper_cite (ModestFormatter *self, const gchar *text, + TnyHeader *header, GList *attachments); +static gchar* modest_formatter_wrapper_quote (ModestFormatter *self, const gchar *text, + TnyHeader *header, GList *attachments); +static gchar* modest_formatter_wrapper_inline (ModestFormatter *self, const gchar *text, + TnyHeader *header, GList *attachments); + +static TnyMimePart *find_body_parent (TnyMimePart *part); static gchar * extract_text (ModestFormatter *self, TnyMimePart *body) { + TnyStream *mp_stream; TnyStream *stream; + TnyStream *input_stream = NULL; GtkTextBuffer *buf; GtkTextIter start, end; - gchar *text, *converted_text; + gchar *text; ModestFormatterPrivate *priv; + gint total, lines, total_lines, line_chars; + gboolean is_html, first_time; buf = gtk_text_buffer_new (NULL); stream = TNY_STREAM (tny_gtk_text_buffer_stream_new (buf)); tny_stream_reset (stream); - tny_mime_part_decode_to_stream (body, stream); + mp_stream = tny_mime_part_get_decoded_stream (body); + + is_html = (g_strcmp0 (tny_mime_part_get_content_type (body), "text/html") == 0); + if (is_html) { + input_stream = modest_stream_html_to_text_new (mp_stream); + } else { + input_stream = g_object_ref (mp_stream); + } + + total = 0; + total_lines = 0; + line_chars = 0; + lines = 0; + + first_time = TRUE; + while (!tny_stream_is_eos (input_stream)) { + gchar buffer [128]; + gchar *offset; + gint n_read; + gint next_read; + + next_read = MIN (128, MAX_BODY_LENGTH - total); + if (next_read == 0) + break; + n_read = tny_stream_read (input_stream, buffer, next_read); + + if (G_UNLIKELY (n_read < 0)) + break; + + offset = buffer; + while (offset < buffer + n_read) { + + if (*offset == '\n') { + total_lines ++; + line_chars = 0; + } else { + line_chars ++; + if (line_chars >= LINE_WRAP) { + total_lines ++; + line_chars = 0; + } + } + if (total_lines >= MAX_BODY_LINES) + break; + offset++; + } + + if (offset - buffer > 0) { + gint n_write = 0, to_write = 0; + + to_write = offset - buffer; + n_write = tny_stream_write (stream, buffer, to_write); + total += n_write; + } else if (n_read == -1) { + break; + } + + if (total_lines >= MAX_BODY_LINES) + break; + } + tny_stream_reset (stream); g_object_unref (G_OBJECT(stream)); - + g_object_unref (G_OBJECT (mp_stream)); + g_object_unref (G_OBJECT (input_stream)); + gtk_text_buffer_get_bounds (buf, &start, &end); text = gtk_text_buffer_get_text (buf, &start, &end, FALSE); g_object_unref (G_OBJECT(buf)); @@ -83,15 +158,6 @@ extract_text (ModestFormatter *self, TnyMimePart *body) /* Convert to desired content type if needed */ priv = MODEST_FORMATTER_GET_PRIVATE (self); - if (strcmp (tny_mime_part_get_content_type (body), priv->content_type) == 0) { - if (!strcmp (priv->content_type, "text/html")) - converted_text = modest_text_utils_convert_to_html (text); - else - converted_text = g_strdup (text); - - g_free (text); - text = converted_text; - } return text; } @@ -103,13 +169,12 @@ construct_from_text (TnyMimePart *part, TnyStream *text_body_stream; /* Create the stream */ - text_body_stream = TNY_STREAM (tny_camel_stream_new - (camel_stream_mem_new_with_buffer - (text, strlen(text)))); + text_body_stream = TNY_STREAM (tny_camel_mem_stream_new_with_buffer + (text, strlen(text))); /* Construct MIME part */ tny_stream_reset (text_body_stream); - tny_mime_part_construct_from_stream (part, text_body_stream, content_type); + tny_mime_part_construct (part, text_body_stream, content_type, "7bit"); tny_stream_reset (text_body_stream); /* Clean */ @@ -117,7 +182,8 @@ construct_from_text (TnyMimePart *part, } static TnyMsg * -modest_formatter_do (ModestFormatter *self, TnyMimePart *body, TnyHeader *header, FormatterFunc func, GList *attachments) +modest_formatter_do (ModestFormatter *self, TnyMimePart *body, TnyHeader *header, FormatterFunc func, + GList *attachments) { TnyMsg *new_msg = NULL; gchar *body_text = NULL, *txt = NULL; @@ -129,7 +195,7 @@ modest_formatter_do (ModestFormatter *self, TnyMimePart *body, TnyHeader *header g_return_val_if_fail (func, NULL); /* Build new part */ - new_msg = modest_formatter_create_message (self, TRUE, attachments != NULL); + new_msg = modest_formatter_create_message (self, TRUE, attachments != NULL, FALSE); body_part = modest_formatter_create_body_part (self, new_msg); if (body) @@ -173,20 +239,24 @@ modest_formatter_attach (ModestFormatter *self, TnyMsg *msg, TnyHeader *header) TnyMsg *new_msg = NULL; TnyMimePart *body_part = NULL; ModestFormatterPrivate *priv; - TnyPlatformFactory *fact; + gchar *txt; - fact = modest_runtime_get_platform_factory (); /* Build new part */ - new_msg = modest_formatter_create_message (self, TRUE, TRUE); + new_msg = modest_formatter_create_message (self, TRUE, TRUE, FALSE); body_part = modest_formatter_create_body_part (self, new_msg); /* Create the two parts */ priv = MODEST_FORMATTER_GET_PRIVATE (self); - construct_from_text (body_part, "", priv->content_type); + txt = modest_text_utils_cite ("", priv->content_type, priv->signature, + NULL, tny_header_get_date_sent (header)); + construct_from_text (body_part, txt, priv->content_type); + g_free (txt); g_object_unref (body_part); - /* Add parts */ - tny_mime_part_add_part (TNY_MIME_PART (new_msg), TNY_MIME_PART (msg)); + if (msg) { + /* Add parts */ + tny_mime_part_add_part (TNY_MIME_PART (new_msg), TNY_MIME_PART (msg)); + } return new_msg; } @@ -212,6 +282,7 @@ modest_formatter_instance_init (GTypeInstance *instance, gpointer g_class) ModestFormatterPrivate *priv = MODEST_FORMATTER_GET_PRIVATE (self); priv->content_type = NULL; + priv->signature = NULL; } static void @@ -271,46 +342,62 @@ modest_formatter_get_type (void) /****************/ static gchar * -modest_formatter_wrapper_cite (ModestFormatter *self, const gchar *text, TnyHeader *header, GList *attachments) +modest_formatter_wrapper_cite (ModestFormatter *self, const gchar *text, TnyHeader *header, + GList *attachments) { + gchar *result, *from; ModestFormatterPrivate *priv = MODEST_FORMATTER_GET_PRIVATE (self); - - return modest_text_utils_cite (text, - priv->content_type, - priv->signature, - tny_header_get_from (header), - tny_header_get_date_sent (header)); + + from = tny_header_dup_from (header); + result = modest_text_utils_cite (text, + priv->content_type, + priv->signature, + from, + tny_header_get_date_sent (header)); + g_free (from); + return result; } static gchar * -modest_formatter_wrapper_inline (ModestFormatter *self, const gchar *text, TnyHeader *header, GList *attachments) +modest_formatter_wrapper_inline (ModestFormatter *self, const gchar *text, TnyHeader *header, + GList *attachments) { + gchar *result, *from, *to, *subject; ModestFormatterPrivate *priv = MODEST_FORMATTER_GET_PRIVATE (self); - return modest_text_utils_inline (text, - priv->content_type, - priv->signature, - tny_header_get_from (header), - tny_header_get_date_sent (header), - tny_header_get_to (header), - tny_header_get_subject (header)); + from = tny_header_dup_from (header); + to = tny_header_dup_to (header); + subject = tny_header_dup_subject (header); + result = modest_text_utils_inline (text, + priv->content_type, + priv->signature, + from, + tny_header_get_date_sent (header), + to, + subject); + g_free (subject); + g_free (to); + g_free (from); + return result; } static gchar * -modest_formatter_wrapper_quote (ModestFormatter *self, const gchar *text, TnyHeader *header, GList *attachments) +modest_formatter_wrapper_quote (ModestFormatter *self, const gchar *text, TnyHeader *header, + GList *attachments) { ModestFormatterPrivate *priv = MODEST_FORMATTER_GET_PRIVATE (self); GList *filenames = NULL; GList *node = NULL; gchar *result = NULL; - + gchar *from; + /* First we need a GList of attachments filenames */ for (node = attachments; node != NULL; node = g_list_next (node)) { TnyMimePart *part = (TnyMimePart *) node->data; gchar *filename = NULL; if (TNY_IS_MSG (part)) { TnyHeader *header = tny_msg_get_header (TNY_MSG (part)); - filename = g_strdup (tny_header_get_subject (header)); + filename = tny_header_dup_subject (header); if ((filename == NULL)||(filename[0] == '\0')) { g_free (filename); filename = g_strdup (_("mail_va_no_subject")); @@ -318,21 +405,24 @@ modest_formatter_wrapper_quote (ModestFormatter *self, const gchar *text, TnyHea g_object_unref (header); } else { filename = g_strdup (tny_mime_part_get_filename (part)); - if ((filename == NULL)||(filename[0] == '\0')) + if ((filename == NULL)||(filename[0] == '\0')) { + g_free (filename); filename = g_strdup (""); + } } - filenames = g_list_append (filenames, filename); + filenames = g_list_prepend (filenames, filename); } - filenames = g_list_reverse (filenames); /* TODO: get 80 from the configuration */ + from = tny_header_dup_from (header); result = modest_text_utils_quote (text, priv->content_type, priv->signature, - tny_header_get_from (header), + from, tny_header_get_date_sent (header), filenames, - 80); + 72); + g_free (from); g_list_foreach (filenames, (GFunc) g_free, NULL); g_list_free (filenames); @@ -340,21 +430,44 @@ modest_formatter_wrapper_quote (ModestFormatter *self, const gchar *text, TnyHea } TnyMsg * -modest_formatter_create_message (ModestFormatter *self, gboolean single_body, gboolean has_attachments) +modest_formatter_create_message (ModestFormatter *self, gboolean single_body, + gboolean has_attachments, gboolean has_images) { TnyMsg *result = NULL; TnyPlatformFactory *fact = NULL; TnyMimePart *body_mime_part = NULL; + TnyMimePart *related_mime_part = NULL; + fact = modest_runtime_get_platform_factory (); result = tny_platform_factory_new_msg (fact); if (has_attachments) { tny_mime_part_set_content_type (TNY_MIME_PART (result), "multipart/mixed"); + if (has_images) { + related_mime_part = tny_platform_factory_new_mime_part (fact); + tny_mime_part_set_content_type (related_mime_part, "multipart/related"); + tny_mime_part_add_part (TNY_MIME_PART (result), related_mime_part); + } else { + related_mime_part = g_object_ref (result); + } + + if (!single_body) { + body_mime_part = tny_platform_factory_new_mime_part (fact); + tny_mime_part_set_content_type (body_mime_part, "multipart/alternative"); + tny_mime_part_add_part (TNY_MIME_PART (related_mime_part), body_mime_part); + g_object_unref (body_mime_part); + } + + g_object_unref (related_mime_part); + } else if (has_images) { + tny_mime_part_set_content_type (TNY_MIME_PART (result), "multipart/related"); + if (!single_body) { body_mime_part = tny_platform_factory_new_mime_part (fact); tny_mime_part_set_content_type (body_mime_part, "multipart/alternative"); tny_mime_part_add_part (TNY_MIME_PART (result), body_mime_part); g_object_unref (body_mime_part); } + } else if (!single_body) { tny_mime_part_set_content_type (TNY_MIME_PART (result), "multipart/alternative"); } @@ -362,35 +475,32 @@ modest_formatter_create_message (ModestFormatter *self, gboolean single_body, gb return result; } -TnyMimePart * -modest_formatter_create_body_part (ModestFormatter *self, TnyMsg *msg) +TnyMimePart * +find_body_parent (TnyMimePart *part) { - TnyMimePart *result = NULL; const gchar *msg_content_type = NULL; - TnyPlatformFactory *fact = NULL; + msg_content_type = tny_mime_part_get_content_type (part); - fact = modest_runtime_get_platform_factory (); - msg_content_type = tny_mime_part_get_content_type (TNY_MIME_PART (msg)); - /* First it checks if the main part is alternative */ - if ((msg_content_type != NULL) && - (!g_strcasecmp (msg_content_type, "multipart/alternative"))) { - result = tny_platform_factory_new_mime_part (fact); - tny_mime_part_add_part (TNY_MIME_PART (msg), result); - return result; - } else if ((msg_content_type != NULL) && - (!g_strcasecmp (msg_content_type, "multipart/mixed"))) { - TnyList *parts = NULL; + if ((msg_content_type != NULL) && + (!g_ascii_strcasecmp (msg_content_type, "multipart/alternative"))) + return g_object_ref (part); + else if ((msg_content_type != NULL) && + (g_str_has_prefix (msg_content_type, "multipart/"))) { TnyIterator *iter = NULL; TnyMimePart *alternative_part = NULL; - - parts = TNY_LIST (tny_simple_list_new ()); - tny_mime_part_get_parts (TNY_MIME_PART (msg), parts); + TnyMimePart *related_part = NULL; + TnyList *parts = TNY_LIST (tny_simple_list_new ()); + tny_mime_part_get_parts (TNY_MIME_PART (part), parts); iter = tny_list_create_iterator (parts); + while (!tny_iterator_is_done (iter)) { TnyMimePart *part = TNY_MIME_PART (tny_iterator_get_current (iter)); - if (part && !g_strcasecmp(tny_mime_part_get_content_type (part), "multipart/alternative")) { + if (part && !g_ascii_strcasecmp(tny_mime_part_get_content_type (part), "multipart/alternative")) { alternative_part = part; break; + } else if (part && !g_ascii_strcasecmp (tny_mime_part_get_content_type (part), "multipart/related")) { + related_part = part; + break; } if (part) @@ -398,18 +508,38 @@ modest_formatter_create_body_part (ModestFormatter *self, TnyMsg *msg) tny_iterator_next (iter); } + g_object_unref (iter); + g_object_unref (parts); + if (related_part) { + TnyMimePart *result; + result = find_body_parent (related_part); + g_object_unref (related_part); + return result; + } else if (alternative_part) + return alternative_part; + else + return g_object_ref (part); + } else + return NULL; +} + +TnyMimePart * +modest_formatter_create_body_part (ModestFormatter *self, TnyMsg *msg) +{ + TnyMimePart *result = NULL; + TnyPlatformFactory *fact = NULL; + TnyMimePart *parent = NULL; + + parent = find_body_parent (TNY_MIME_PART (msg)); + fact = modest_runtime_get_platform_factory (); + if (parent != NULL) { result = tny_platform_factory_new_mime_part (fact); - if (alternative_part != NULL) { - tny_mime_part_add_part (alternative_part, result); - } else { - tny_mime_part_add_part (TNY_MIME_PART (msg), result); - } - g_object_unref (G_OBJECT (parts)); - return result; + tny_mime_part_add_part (TNY_MIME_PART (parent), result); + g_object_unref (parent); } else { - /* We add a reference as this method is intended to obtain - a ref'ed part */ - g_object_ref (msg); - return TNY_MIME_PART (msg); + result = g_object_ref (msg); } + + return result; + }