On creating forward, don't include purged parts (fixes NB#124831).
[modest] / src / modest-formatter.c
index 4e1e367..8cdd091 100644 (file)
 #include <tny-header.h>
 #include <tny-simple-list.h>
 #include <tny-gtk-text-buffer-stream.h>
-#include <tny-camel-stream.h>
-#include <camel/camel-stream-mem.h>
+#include <tny-camel-mem-stream.h>
+#include <tny-camel-html-to-text-stream.h>
 #include "modest-formatter.h"
 #include "modest-text-utils.h"
 #include "modest-tny-platform-factory.h"
 #include <modest-runtime.h>
 
+#define LINE_WRAP 78
+#define MAX_BODY_LINES 1024
+#define MAX_BODY_LENGTH 1024*128
+
 typedef struct _ModestFormatterPrivate ModestFormatterPrivate;
 struct _ModestFormatterPrivate {
        gchar *content_type;
@@ -67,19 +71,78 @@ static TnyMimePart *find_body_parent (TnyMimePart *part);
 static gchar *
 extract_text (ModestFormatter *self, TnyMimePart *body)
 {
+       TnyStream *mp_stream;
        TnyStream *stream;
+       TnyStream *input_stream;
        GtkTextBuffer *buf;
        GtkTextIter start, end;
-       gchar *text, *converted_text;
+       gchar *text;
        ModestFormatterPrivate *priv;
+       gint total, total_lines, line_chars;
 
        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, NULL);
+       mp_stream = tny_mime_part_get_decoded_stream (body);
+
+       if (g_strcmp0 (tny_mime_part_get_content_type (body), "text/html") == 0) {
+               input_stream = tny_camel_html_to_text_stream_new (mp_stream);
+       } else {
+               input_stream = g_object_ref (mp_stream);
+       }
+
+       total = 0;
+       total_lines = 0;
+       line_chars = 0;
+
+       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);
+
+               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;
+                       n_write = tny_stream_write (stream, buffer, offset - buffer);
+                       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);
@@ -88,15 +151,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;
 }
 
@@ -108,13 +162,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 */
@@ -173,12 +226,39 @@ modest_formatter_inline (ModestFormatter *self, TnyMimePart *body, TnyHeader *he
        return modest_formatter_do (self, body, header, modest_formatter_wrapper_inline, attachments);
 }
 
+static void
+remove_purgeds (TnyMimePart *part)
+{
+       TnyList *parts;
+       TnyIterator *iterator;
+
+       parts = TNY_LIST (tny_simple_list_new ());
+       tny_mime_part_get_parts (part, parts);
+
+       for (iterator = tny_list_create_iterator (parts);
+            !tny_iterator_is_done (iterator);
+            tny_iterator_next (iterator)) {
+               TnyMimePart *current;
+
+               current = (TnyMimePart *) tny_iterator_get_current (iterator);
+               if (tny_mime_part_is_purged (current)) {
+                       tny_mime_part_del_part (part, current);
+               }
+
+               g_object_unref (current);
+       }
+
+       g_object_unref (iterator);
+       g_object_unref (parts);
+}
+
 TnyMsg *
 modest_formatter_attach (ModestFormatter *self, TnyMsg *msg, TnyHeader *header)
 {
        TnyMsg *new_msg = NULL;
        TnyMimePart *body_part = NULL;
        ModestFormatterPrivate *priv;
+       gchar *txt;
 
        /* Build new part */
        new_msg     = modest_formatter_create_message (self, TRUE, TRUE, FALSE);
@@ -186,11 +266,20 @@ modest_formatter_attach (ModestFormatter *self, TnyMsg *msg, TnyHeader *header)
 
        /* 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) {
+
+               /* Remove purgeds */
+               remove_purgeds (TNY_MIME_PART (msg));
+
+               /* Add parts */
+               tny_mime_part_add_part (TNY_MIME_PART (new_msg), TNY_MIME_PART (msg));
+       }
 
        return new_msg;
 }
@@ -216,6 +305,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
@@ -278,28 +368,40 @@ static gchar *
 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) 
 {
+       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 *
@@ -310,14 +412,15 @@ modest_formatter_wrapper_quote (ModestFormatter *self, const gchar *text, TnyHea
        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"));
@@ -325,21 +428,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);
+       g_free (from);
 
        g_list_foreach (filenames, (GFunc) g_free, NULL);
        g_list_free (filenames);