11d124eebbbcb8ac5e55882bb821530fd6d0ef69
[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 *ct;
95                         gchar *content_type;
96                         part = TNY_MIME_PART(tny_iterator_get_current (iter));
97
98                         /* we need to strdown the content type, because
99                          * tny_mime_part_has_content_type does not do it...
100                          */
101                         ct = tny_mime_part_get_content_type (part);
102                         content_type = g_ascii_strdown (ct, strlen(ct));
103                                                 
104                         if (g_str_has_prefix (content_type, mime_type) &&
105                             !tny_mime_part_is_attachment (part)) {
106                                 g_free (content_type);
107                                 break;
108                         }
109                         
110                         if (g_str_has_prefix(content_type, "multipart")) {
111                                 part = modest_tny_msg_actions_find_body_part_from_mime_part (part,
112                                                                                              want_html);
113                                 g_free (content_type);
114                                 if (part)
115                                         break;
116                         }
117
118                         g_free (content_type);
119                         part = NULL;
120                         tny_iterator_next (iter);
121
122                 } while (!tny_iterator_is_done(iter));
123         }
124         
125         /* did we find a matching part? */
126         if (part)
127                 g_object_ref (G_OBJECT(part));
128
129         g_object_unref (G_OBJECT(iter));
130         g_object_unref (G_OBJECT(parts));
131
132         /* if were trying to find an HTML part and couldn't find it,
133          * try to find a text/plain part instead
134          */
135         if (!part && want_html) 
136                 return modest_tny_msg_actions_find_body_part_from_mime_part (msg, FALSE);
137
138         if (!part)
139                 g_printerr ("modest: cannot find body part\n");
140         
141         return part ? part : NULL;
142 }
143
144
145 TnyMimePart*
146 modest_tny_msg_actions_find_body_part (TnyMsg *msg, gboolean want_html)
147 {
148         return modest_tny_msg_actions_find_body_part_from_mime_part (TNY_MIME_PART(msg),
149                                                                      want_html);
150 }
151
152
153 TnyMimePart *
154 modest_tny_msg_actions_find_nth_part (TnyMsg *msg, gint index)
155 {
156         TnyMimePart *part;
157         TnyList *parts;
158         TnyIterator *iter;
159
160         g_return_val_if_fail (msg, NULL);
161         g_return_val_if_fail (index > 0, NULL);
162                 
163         parts = TNY_LIST(tny_simple_list_new());
164         tny_mime_part_get_parts (TNY_MIME_PART(msg), parts);
165         iter  = tny_list_create_iterator (parts);
166
167         part = NULL;
168         
169         if (!tny_iterator_is_done(iter)) {
170                 tny_iterator_nth (iter, index);
171                 part = TNY_MIME_PART(tny_iterator_get_current (iter));
172         }
173
174         g_object_unref (G_OBJECT(iter));
175         g_object_unref (G_OBJECT(parts));
176
177         return part;
178 }
179
180 gchar * 
181 modest_tny_msg_actions_find_body (TnyMsg *msg, gboolean want_html)
182 {
183         const gchar *body;
184
185         body = get_body_text (msg, want_html);
186
187         if (body)
188                 return g_strdup (body);
189         else 
190                 return NULL;
191 }