X-Git-Url: http://git.maemo.org/git/?p=modest;a=blobdiff_plain;f=src%2Fmodest-formatter.c;h=a503ed7544077a62fafd07640ae16e1f6a150a27;hp=5392b062486d9c3890f2cfdb5b458e2fd8a5120f;hb=3013a1c7e1739de8758a9ffa8c58c927262c2d0a;hpb=0a600fce312da2fafdbe63d2c60b428fac0df208

diff --git a/src/modest-formatter.c b/src/modest-formatter.c
index 5392b06..a503ed7 100644
--- a/src/modest-formatter.c
+++ b/src/modest-formatter.c
@@ -33,12 +33,15 @@
 #include <tny-simple-list.h>
 #include <tny-gtk-text-buffer-stream.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 MAX_BODY_LENGTH 4096
+#define LINE_WRAP 78
+#define MAX_BODY_LINES 1024
+#define MAX_BODY_LENGTH 1024*128
 
 typedef struct _ModestFormatterPrivate ModestFormatterPrivate;
 struct _ModestFormatterPrivate {
@@ -70,40 +73,76 @@ 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;
+	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);
 	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 (mp_stream)) {
+	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 (mp_stream, buffer, next_read);
-		if (n_read > 0) {
+		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, n_read);
+			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);
@@ -112,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;
 }