* all:
[modest] / src / modest-tny-msg-actions.c
index 60b3efc..ccce454 100644 (file)
  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
-
-#include <gtk/gtk.h>
+#include <string.h>
 #include <gtkhtml/gtkhtml.h>
-#include <tny-text-buffer-stream.h>
-#include <tny-mime-part-iface.h>
-#include <tny-msg-iface.h>
-#include <tny-list-iface.h>
-#include <tny-list.h>
+#include <tny-gtk-text-buffer-stream.h>
+#include <tny-simple-list.h>
+#include <tny-folder.h>
 
 #ifdef HAVE_CONFIG_H
 #include <config.h>
 #include "modest-tny-msg-actions.h"
 #include "modest-text-utils.h"
 
-static gchar *
-quote_msg (TnyMsgIface* src, const gchar * from, time_t sent_date, gint limit)
+static const gchar *
+get_body_text (TnyMsg *msg, gboolean want_html)
 {
-       TnyStreamIface *stream;
-       TnyMimePartIface *body;
+       TnyStream *stream;
+       TnyMimePart *body;
        GtkTextBuffer *buf;
        GtkTextIter start, end;
        const gchar *to_quote;
-       gchar *quoted;
 
-       body = modest_tny_msg_actions_find_body_part(src, FALSE);
+       body = modest_tny_msg_actions_find_body_part(msg, want_html);
        if (!body)
                return NULL;
 
        buf = gtk_text_buffer_new (NULL);
-       stream = TNY_STREAM_IFACE (tny_text_buffer_stream_new (buf));
-       tny_stream_iface_reset (stream);
-       tny_mime_part_iface_decode_to_stream (body, stream);
-       tny_stream_iface_reset (stream);
+       stream = TNY_STREAM (tny_gtk_text_buffer_stream_new (buf));
+       tny_stream_reset (stream);
+       tny_mime_part_decode_to_stream (body, stream);
+       tny_stream_reset (stream);
 
        g_object_unref (G_OBJECT(stream));
        g_object_unref (G_OBJECT(body));
        
        gtk_text_buffer_get_bounds (buf, &start, &end);
        to_quote = gtk_text_buffer_get_text (buf, &start, &end, FALSE);
-       quoted = modest_text_utils_quote (to_quote, from, sent_date, limit);
        g_object_unref (buf);
 
-       return quoted;
-}
-
-
-gchar*
-modest_tny_msg_actions_quote (TnyMsgIface * self, const gchar * from,
-                             time_t sent_date, gint limit,
-                             const gchar * to_quote)
-{
-       /* 2 cases: */
-
-       /* a) quote text from selection */
-       if (to_quote != NULL) 
-               return modest_text_utils_quote (to_quote, from, sent_date,
-                                               limit);
-       
-       /* b) try to find a text/plain part in the msg and quote it */
-       return quote_msg (self, from, sent_date, limit);
+       return to_quote;
 }
 
-
-
-TnyMimePartIface *
-modest_tny_msg_actions_find_body_part (TnyMsgIface *msg, gboolean want_html)
+static TnyMimePart*
+modest_tny_msg_actions_find_body_part_from_mime_part (TnyMimePart *msg, gboolean want_html)
 {
        const gchar *mime_type = want_html ? "text/html" : "text/plain";
-       TnyMimePartIface *part;
-       TnyListIface *parts;
-       TnyIteratorIface *iter;
+       TnyMimePart *part = NULL;
+       TnyList *parts;
+       TnyIterator *iter;
 
        if (!msg)
                return NULL;
 
-       parts = TNY_LIST_IFACE(tny_list_new());
-       tny_msg_iface_get_parts ((TnyMsgIface*)msg, parts);
-
-       iter  = tny_list_iface_create_iterator(parts);
-
-       while (!tny_iterator_iface_is_done(iter)) {
-
-               part = TNY_MIME_PART_IFACE(tny_iterator_iface_current (iter));
-               
-               if (tny_mime_part_iface_content_type_is (part, mime_type) &&
-                   !tny_mime_part_iface_is_attachment (part)) {
-                       break;
-               }
-               part = NULL;
-               tny_iterator_iface_next (iter);
+       parts = TNY_LIST (tny_simple_list_new());
+       tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
+
+       iter  = tny_list_create_iterator(parts);
+
+       /* no parts? assume it's single-part message */
+       if (tny_iterator_is_done(iter)) 
+               return TNY_MIME_PART (g_object_ref(G_OBJECT(msg)));
+       else {
+               do {
+                       const gchar *ct;
+                       gchar *content_type;
+                       part = TNY_MIME_PART(tny_iterator_get_current (iter));
+
+                       /* we need to strdown the content type, because
+                        * tny_mime_part_has_content_type does not do it...
+                        */
+                       ct = tny_mime_part_get_content_type (part);
+                       content_type = g_ascii_strdown (ct, strlen(ct));
+                                               
+                       if (g_str_has_prefix (content_type, mime_type) &&
+                           !tny_mime_part_is_attachment (part)) {
+                               g_free (content_type);
+                               break;
+                       }
+                       
+                       if (g_str_has_prefix(content_type, "multipart")) {
+                               part = modest_tny_msg_actions_find_body_part_from_mime_part (part,
+                                                                                            want_html);
+                               g_free (content_type);
+                               if (part)
+                                       break;
+                       }
+
+                       g_free (content_type);
+                       part = NULL;
+                       tny_iterator_next (iter);
+
+               } while (!tny_iterator_is_done(iter));
        }
-
+       
        /* did we find a matching part? */
        if (part)
                g_object_ref (G_OBJECT(part));
@@ -132,33 +132,59 @@ modest_tny_msg_actions_find_body_part (TnyMsgIface *msg, gboolean want_html)
         * try to find a text/plain part instead
         */
        if (!part && want_html) 
-               return modest_tny_msg_actions_find_body_part (msg, FALSE);
+               return modest_tny_msg_actions_find_body_part_from_mime_part (msg, FALSE);
 
+       if (!part)
+               g_printerr ("modest: cannot find body part\n");
+       
        return part ? part : NULL;
 }
 
 
+TnyMimePart*
+modest_tny_msg_actions_find_body_part (TnyMsg *msg, gboolean want_html)
+{
+       return modest_tny_msg_actions_find_body_part_from_mime_part (TNY_MIME_PART(msg),
+                                                                    want_html);
+}
 
-TnyMimePartIface *
-modest_tny_msg_actions_find_nth_part (TnyMsgIface *msg, gint index)
+
+TnyMimePart *
+modest_tny_msg_actions_find_nth_part (TnyMsg *msg, gint index)
 {
-       TnyMimePartIface *part;
-       TnyListIface *parts;
-       TnyIteratorIface *iter;
+       TnyMimePart *part;
+       TnyList *parts;
+       TnyIterator *iter;
 
        g_return_val_if_fail (msg, NULL);
        g_return_val_if_fail (index > 0, NULL);
                
-       parts = TNY_LIST_IFACE(tny_list_new());
-       tny_msg_iface_get_parts ((TnyMsgIface*)msg, parts);
-       iter  = tny_list_iface_create_iterator ((TnyListIface*)parts);
-       if (!tny_iterator_iface_has_first(iter))
-               return NULL;
+       parts = TNY_LIST(tny_simple_list_new());
+       tny_mime_part_get_parts (TNY_MIME_PART(msg), parts);
+       iter  = tny_list_create_iterator (parts);
 
-       part = TNY_MIME_PART_IFACE(tny_iterator_iface_nth (iter, index));
+       part = NULL;
+       
+       if (!tny_iterator_is_done(iter)) {
+               tny_iterator_nth (iter, index);
+               part = TNY_MIME_PART(tny_iterator_get_current (iter));
+       }
 
        g_object_unref (G_OBJECT(iter));
        g_object_unref (G_OBJECT(parts));
 
        return part;
 }
+
+gchar * 
+modest_tny_msg_actions_find_body (TnyMsg *msg, gboolean want_html)
+{
+       const gchar *body;
+
+       body = get_body_text (msg, want_html);
+
+       if (body)
+               return g_strdup (body);
+       else 
+               return NULL;
+}