* Added ModestMailOperationQueue
[modest] / src / modest-tny-msg-actions.c
1 /* Copyright (c) 2006, Nokia Corporation
2  * All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
8  * * Redistributions of source code must retain the above copyright
9  *   notice, this list of conditions and the following disclaimer.
10  * * Redistributions in binary form must reproduce the above copyright
11  *   notice, this list of conditions and the following disclaimer in the
12  *   documentation and/or other materials provided with the distribution.
13  * * Neither the name of the Nokia Corporation nor the names of its
14  *   contributors may be used to endorse or promote products derived from
15  *   this software without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
18  * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
19  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
20  * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
21  * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
22  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
23  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
24  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
25  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
26  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
27  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29
30
31 #include <gtk/gtk.h>
32 #include <gtkhtml/gtkhtml.h>
33 #include <tny-gtk-text-buffer-stream.h>
34 #include <tny-simple-list.h>
35 #include <tny-folder.h>
36
37 #ifdef HAVE_CONFIG_H
38 #include <config.h>
39 #endif /*HAVE_CONFIG_H */
40
41 #include "modest-tny-msg-actions.h"
42 #include "modest-text-utils.h"
43
44 static const gchar *
45 get_body_text (TnyMsg *msg, gboolean want_html)
46 {
47         TnyStream *stream;
48         TnyMimePart *body;
49         GtkTextBuffer *buf;
50         GtkTextIter start, end;
51         const gchar *to_quote;
52
53         body = modest_tny_msg_actions_find_body_part(msg, want_html);
54         if (!body)
55                 return NULL;
56
57         buf = gtk_text_buffer_new (NULL);
58         stream = TNY_STREAM (tny_gtk_text_buffer_stream_new (buf));
59         tny_stream_reset (stream);
60         tny_mime_part_decode_to_stream (body, stream);
61         tny_stream_reset (stream);
62
63         g_object_unref (G_OBJECT(stream));
64         g_object_unref (G_OBJECT(body));
65         
66         gtk_text_buffer_get_bounds (buf, &start, &end);
67         to_quote = gtk_text_buffer_get_text (buf, &start, &end, FALSE);
68         g_object_unref (buf);
69
70         return to_quote;
71 }
72
73 static TnyMimePart*
74 modest_tny_msg_actions_find_body_part_from_mime_part (TnyMimePart *msg, gboolean want_html)
75 {
76         const gchar *mime_type = want_html ? "text/html" : "text/plain";
77         TnyMimePart *part = NULL;
78         TnyList *parts;
79         TnyIterator *iter;
80
81         if (!msg)
82                 return NULL;
83
84         parts = TNY_LIST (tny_simple_list_new());
85         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
86
87         iter  = tny_list_create_iterator(parts);
88
89         /* no parts? assume it's single-part message */
90         if (tny_iterator_is_done(iter)) 
91                 return TNY_MIME_PART (g_object_ref(G_OBJECT(msg)));
92         else {
93                 do {
94                         const gchar* content_type;
95                         part = TNY_MIME_PART(tny_iterator_get_current (iter));
96                         
97                         if (tny_mime_part_content_type_is (part, mime_type) &&
98                             !tny_mime_part_is_attachment (part)) 
99                                 break;
100                         
101                         content_type = tny_mime_part_get_content_type (part);
102                         if (g_str_has_prefix(content_type, "multipart")) {
103                                 part = modest_tny_msg_actions_find_body_part_from_mime_part (part,
104                                                                                              want_html);
105                                 if (part)
106                                         break;
107                         }
108
109                         part = NULL;
110                         tny_iterator_next (iter);
111
112                 } while (!tny_iterator_is_done(iter));
113         }
114         
115         /* did we find a matching part? */
116         if (part)
117                 g_object_ref (G_OBJECT(part));
118
119         g_object_unref (G_OBJECT(iter));
120         g_object_unref (G_OBJECT(parts));
121
122         /* if were trying to find an HTML part and couldn't find it,
123          * try to find a text/plain part instead
124          */
125         if (!part && want_html) 
126                 return modest_tny_msg_actions_find_body_part_from_mime_part (msg, FALSE);
127
128         if (!part)
129                 g_warning ("cannot find body part");
130         
131         return part ? part : NULL;
132 }
133
134
135 TnyMimePart*
136 modest_tny_msg_actions_find_body_part (TnyMsg *msg, gboolean want_html)
137 {
138         return modest_tny_msg_actions_find_body_part_from_mime_part (TNY_MIME_PART(msg),
139                                                                      want_html);
140 }
141
142
143 TnyMimePart *
144 modest_tny_msg_actions_find_nth_part (TnyMsg *msg, gint index)
145 {
146         TnyMimePart *part;
147         TnyList *parts;
148         TnyIterator *iter;
149
150         g_return_val_if_fail (msg, NULL);
151         g_return_val_if_fail (index > 0, NULL);
152                 
153         parts = TNY_LIST(tny_simple_list_new());
154         tny_mime_part_get_parts (TNY_MIME_PART(msg), parts);
155         iter  = tny_list_create_iterator (parts);
156
157         part = NULL;
158         
159         if (!tny_iterator_is_done(iter)) {
160                 tny_iterator_nth (iter, index);
161                 part = TNY_MIME_PART(tny_iterator_get_current (iter));
162         }
163
164         g_object_unref (G_OBJECT(iter));
165         g_object_unref (G_OBJECT(parts));
166
167         return part;
168 }
169
170 gchar * 
171 modest_tny_msg_actions_find_body (TnyMsg *msg, gboolean want_html)
172 {
173         const gchar *body;
174
175         body = get_body_text (msg, want_html);
176
177         if (body)
178                 return g_strdup (body);
179         else 
180                 return NULL;
181 }