X-Git-Url: http://git.maemo.org/git/?a=blobdiff_plain;f=src%2Fwidgets%2Fmodest-tny-stream-gtkhtml.c;h=3bdd6b3339c534888b40818b8773258a461f46d7;hb=ca885ed046b02fd495ff07fbe4283c9692cfa442;hp=943a9e184fe8a3813ae28f2f433e9335c6c2781d;hpb=85dbfb5b2586e248ed9cef94d0e43f8e24865caf;p=modest

diff --git a/src/widgets/modest-tny-stream-gtkhtml.c b/src/widgets/modest-tny-stream-gtkhtml.c
index 943a9e1..3bdd6b3 100644
--- a/src/widgets/modest-tny-stream-gtkhtml.c
+++ b/src/widgets/modest-tny-stream-gtkhtml.c
@@ -31,6 +31,7 @@
 /* modest-tny-stream-gtkhtml.c */
 
 #include "modest-tny-stream-gtkhtml.h"
+#include "modest-gtkhtml-mime-part-view.h"
 #include <tny-stream.h>
 #include <gtkhtml/gtkhtml-stream.h>
 #include <gtkhtml/gtkhtml-search.h>
@@ -42,6 +43,8 @@ static void  modest_tny_stream_gtkhtml_finalize     (GObject *obj);
 
 static void  modest_tny_stream_gtkhml_iface_init (gpointer g_iface, gpointer iface_data);
 
+static void  stop_streams (ModestGtkhtmlMimePartView *view, gpointer userdata);
+
 /* list my signals */
 enum {
 	/* MY_SIGNAL_1, */
@@ -53,6 +56,10 @@ typedef struct _ModestTnyStreamGtkhtmlPrivate ModestTnyStreamGtkhtmlPrivate;
 struct _ModestTnyStreamGtkhtmlPrivate {
 	GtkHTMLStream *stream;
 	GtkHTML *html;
+	guint stop_streams_id;
+
+	gssize max_size;
+	gssize current_size;
 };
 #define MODEST_TNY_STREAM_GTKHTML_GET_PRIVATE(o)      (G_TYPE_INSTANCE_GET_PRIVATE((o), \
                                                        MODEST_TYPE_TNY_STREAM_GTKHTML, \
@@ -118,6 +125,10 @@ modest_tny_stream_gtkhtml_init (ModestTnyStreamGtkhtml *obj)
 
 	priv->stream  = NULL;
 	priv->html = NULL;
+	priv->stop_streams_id = 0;
+
+	priv->max_size = 0;
+	priv->current_size = 0;
 }
 
 static void
@@ -127,6 +138,12 @@ modest_tny_stream_gtkhtml_finalize (GObject *obj)
 
 	priv = MODEST_TNY_STREAM_GTKHTML_GET_PRIVATE(obj);
 	priv->stream = NULL;
+
+	if (priv->stop_streams_id > 0) {
+		g_signal_handler_disconnect (G_OBJECT (priv->html), priv->stop_streams_id);
+		priv->stop_streams_id = 0;
+	}
+
 	if (priv->html) {
 		g_object_unref (priv->html);
 		priv->html = NULL;
@@ -147,6 +164,10 @@ modest_tny_stream_gtkhtml_new (GtkHTMLStream *stream, GtkHTML *html)
 	priv->stream = stream;
 	priv->html = g_object_ref (html);
 
+	priv->stop_streams_id = g_signal_connect (G_OBJECT (html), "stop-streams",
+						  G_CALLBACK (stop_streams), obj);
+	priv->current_size = 0;
+
 	return obj;
 }
 
@@ -176,10 +197,29 @@ gtkhtml_write (TnyStream *self, const char *buffer, size_t n)
 	if (n == 0 || !buffer)
 		return 0;
 
-	if (!GTK_WIDGET_VISIBLE (priv->html))
+	if (!priv->html || !GTK_WIDGET_VISIBLE (priv->html))
 		return -1;
 
+	if (priv->max_size > 0) {
+
+		/* We only use the maximum size for write method, and even we
+		 * ignore and fake as we would do a successfull read */
+		if (priv->current_size >= priv->max_size)
+			return n;
+
+		if (priv->current_size + n > priv->max_size)
+			n = priv->max_size - priv->current_size;
+	}
+
+	if (!g_main_context_is_owner (NULL))
+		gdk_threads_enter ();
+
 	gtk_html_stream_write (priv->stream, buffer, n);
+
+	if (!g_main_context_is_owner (NULL))
+		gdk_threads_leave ();
+	priv->current_size += n;
+
 	return n; /* hmmm */
 }
 
@@ -198,10 +238,21 @@ gtkhtml_close (TnyStream *self)
 	g_return_val_if_fail (self, 0);
 	priv = MODEST_TNY_STREAM_GTKHTML_GET_PRIVATE(self);
 	
-	if (GTK_WIDGET_VISIBLE (priv->html)) {
+	if (priv->html && GTK_WIDGET_VISIBLE (priv->html)) {
+		if (!g_main_context_is_owner (NULL))
+			gdk_threads_enter ();
+
 		gtk_html_stream_close   (priv->stream, GTK_HTML_STREAM_OK);
+
+		if (!g_main_context_is_owner (NULL))
+			gdk_threads_leave ();
+
 	}
 	priv->stream = NULL;
+	if (priv->html && priv->stop_streams_id > 0) {
+		g_signal_handler_disconnect (G_OBJECT (priv->html), priv->stop_streams_id);
+		priv->stop_streams_id = 0;
+	}
 	if (priv->html) {
 		g_object_unref (priv->html);
 		priv->html = NULL;
@@ -232,6 +283,59 @@ gtkhtml_write_to_stream (TnyStream *self, TnyStream *output)
 	return 0;
 }
 
+static void
+stop_streams (ModestGtkhtmlMimePartView *view, gpointer userdata)
+{
+	ModestTnyStreamGtkhtml *self = (ModestTnyStreamGtkhtml *) userdata;
+	ModestTnyStreamGtkhtmlPrivate *priv;
+	
+	g_return_if_fail (self);
+	priv = MODEST_TNY_STREAM_GTKHTML_GET_PRIVATE(self);
+
+	if (priv->html && priv->stop_streams_id > 0) {
+		g_signal_handler_disconnect (G_OBJECT (priv->html), priv->stop_streams_id);
+		priv->stop_streams_id = 0;
+	}
+	
+	if (priv->html) {
+		g_object_unref (priv->html);
+		priv->html = NULL;
+	}
+}
+
+void 
+modest_tny_stream_gtkhtml_set_max_size (ModestTnyStreamGtkhtml *stream, 
+					gssize max_size)
+{
+	ModestTnyStreamGtkhtmlPrivate *priv;
+
+	g_return_if_fail (MODEST_IS_TNY_STREAM_GTKHTML (stream));
+	priv = MODEST_TNY_STREAM_GTKHTML_GET_PRIVATE (stream);
+
+	priv->max_size = max_size;
+}
+
+gssize 
+modest_tny_stream_gtkhtml_get_max_size (ModestTnyStreamGtkhtml *stream)
+{
+	ModestTnyStreamGtkhtmlPrivate *priv;
+
+	g_return_val_if_fail (MODEST_IS_TNY_STREAM_GTKHTML (stream), 0);
+	priv = MODEST_TNY_STREAM_GTKHTML_GET_PRIVATE (stream);
+
+	return priv->max_size;
+}
+
+gboolean
+modest_tny_stream_gtkhtml_limit_reached (ModestTnyStreamGtkhtml *self)
+{
+	ModestTnyStreamGtkhtmlPrivate *priv;
+
+	g_return_val_if_fail (MODEST_IS_TNY_STREAM_GTKHTML (self), 0);
+	priv = MODEST_TNY_STREAM_GTKHTML_GET_PRIVATE (self);
+
+	return (priv->max_size > 0) && (priv->current_size >= priv->max_size);
+}
 
 static void
 modest_tny_stream_gtkhml_iface_init (gpointer g_iface, gpointer iface_data)