* src/modest-ui-actions.[ch]:
[modest] / src / modest-tny-msg.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 #include <string.h>
31 #include <gtkhtml/gtkhtml.h>
32 #include <tny-gtk-text-buffer-stream.h>
33 #include <tny-simple-list.h>
34 #include <tny-folder.h>
35 #include <modest-runtime.h>
36 #include "modest-formatter.h"
37 #include <tny-camel-stream.h>
38 #include <camel/camel-stream-mem.h>
39
40 #ifdef HAVE_CONFIG_H
41 #include <config.h>
42 #endif /*HAVE_CONFIG_H */
43
44 #include <modest-tny-msg.h>
45 #include "modest-text-utils.h"
46
47 static TnyMimePart * add_body_part (TnyMsg *msg, const gchar *body,
48                                     const gchar *content_type, gboolean has_attachments);
49 static TnyMimePart * add_html_body_part (TnyMsg *msg, const gchar *body);
50 static void add_attachments (TnyMsg *msg, GList *attachments_list);
51 static char * get_content_type(const gchar *s);
52 static gboolean is_ascii(const gchar *s);
53
54 TnyMsg*
55 modest_tny_msg_new (const gchar* mailto, const gchar* from, const gchar *cc,
56                     const gchar *bcc, const gchar* subject, const gchar *body,
57                     GSList *attachments)
58 {
59         TnyPlatformFactory *fact;
60         TnyMsg *new_msg;
61         TnyHeader *header;
62         gchar *content_type;
63         
64         /* Create new msg */
65         fact    = modest_runtime_get_platform_factory ();
66         new_msg = tny_platform_factory_new_msg (fact);
67         header  = tny_msg_get_header (new_msg);
68         
69         tny_header_set_from (TNY_HEADER (header), from);
70         tny_header_set_replyto (TNY_HEADER (header), from);
71         tny_header_set_to (TNY_HEADER (header), mailto);
72         tny_header_set_cc (TNY_HEADER (header), cc);
73         tny_header_set_bcc (TNY_HEADER (header), bcc);
74
75         if (subject)
76                 tny_header_set_subject (TNY_HEADER (header), subject);
77
78         content_type = get_content_type(body);
79                 
80         /* Add the body of the new mail */
81         /* This is needed even if body is NULL or empty. */
82         add_body_part (new_msg, body, content_type, (attachments ? TRUE: FALSE));
83         g_free (content_type);
84                        
85         /* Add attachments */
86         if (attachments)
87                 add_attachments (new_msg, (GList*) attachments);
88
89         return new_msg;
90 }
91
92 TnyMsg*
93 modest_tny_msg_new_html_plain (const gchar* mailto, const gchar* from, const gchar *cc,
94                                const gchar *bcc, const gchar* subject, 
95                                const gchar *html_body, const gchar *plain_body,
96                                GSList *attachments)
97 {
98         TnyPlatformFactory *fact;
99         TnyMsg *new_msg;
100         TnyHeader *header;
101         gchar *content_type;
102         
103         /* Create new msg */
104         fact    = modest_runtime_get_platform_factory ();
105         new_msg = tny_platform_factory_new_msg (fact);
106         header  = tny_msg_get_header (new_msg);
107         
108         tny_header_set_from (TNY_HEADER (header), from);
109         tny_header_set_replyto (TNY_HEADER (header), from);
110         tny_header_set_to (TNY_HEADER (header), mailto);
111         tny_header_set_cc (TNY_HEADER (header), cc);
112         tny_header_set_bcc (TNY_HEADER (header), bcc);
113         tny_header_set_subject (TNY_HEADER (header), subject);
114
115         content_type = get_content_type(plain_body);
116                 
117         /* Add the body of the new mail */      
118         add_body_part (new_msg, plain_body, content_type, TRUE);
119         add_html_body_part (new_msg, html_body);
120         g_free (content_type);
121                        
122         /* Add attachments */
123         add_attachments (new_msg, (GList*) attachments);
124
125         return new_msg;
126 }
127
128
129 /* FIXME: this func copy from modest-mail-operation: refactor */
130 static TnyMimePart *
131 add_body_part (TnyMsg *msg, 
132                const gchar *body,
133                const gchar *content_type,
134                gboolean has_attachments)
135 {
136         TnyMimePart *text_body_part = NULL;
137         TnyStream *text_body_stream;
138
139         /* Create the stream */
140         text_body_stream = TNY_STREAM (tny_camel_stream_new
141                                        (camel_stream_mem_new_with_buffer
142                                         (body, (body ? strlen(body) : 0))));
143
144         /* Create body part if needed */
145         if (has_attachments)
146                 text_body_part = tny_platform_factory_new_mime_part
147                         (modest_runtime_get_platform_factory ());
148         else
149                 text_body_part = TNY_MIME_PART(msg);
150
151         /* Construct MIME part */
152         tny_stream_reset (text_body_stream);
153         tny_mime_part_construct_from_stream (text_body_part,
154                                              text_body_stream,
155                                              content_type);
156         tny_stream_reset (text_body_stream);
157
158         /* Add part if needed */
159         if (has_attachments) {
160                 tny_mime_part_add_part (TNY_MIME_PART (msg), text_body_part);
161                 g_object_unref (G_OBJECT(text_body_part));
162         }
163
164         /* Clean */
165         g_object_unref (text_body_stream);
166
167         return text_body_part;
168 }
169
170 static TnyMimePart *
171 add_html_body_part (TnyMsg *msg, 
172                     const gchar *body)
173 {
174         TnyMimePart *html_body_part = NULL;
175         TnyStream *html_body_stream;
176
177         /* Create the stream */
178         html_body_stream = TNY_STREAM (tny_camel_stream_new
179                                        (camel_stream_mem_new_with_buffer
180                                         (body, strlen(body))));
181
182         /* Create body part if needed */
183         html_body_part = tny_platform_factory_new_mime_part
184                 (modest_runtime_get_platform_factory ());
185
186         /* Construct MIME part */
187         tny_stream_reset (html_body_stream);
188         tny_mime_part_construct_from_stream (html_body_part,
189                                              html_body_stream,
190                                              "text/html; charset=utf-8");
191         tny_stream_reset (html_body_stream);
192
193         /* Add part if needed */
194         tny_mime_part_add_part (TNY_MIME_PART (msg), html_body_part);
195         g_object_unref (G_OBJECT(html_body_part));
196
197         /* Clean */
198         g_object_unref (html_body_stream);
199
200         return html_body_part;
201 }
202
203 static void
204 add_attachments (TnyMsg *msg, GList *attachments_list)
205 {
206         GList *pos;
207         TnyMimePart *attachment_part, *old_attachment;
208         const gchar *attachment_content_type;
209         const gchar *attachment_filename;
210         const gchar *attachment_cid;
211         TnyStream *attachment_stream;
212
213         for (pos = (GList *)attachments_list; pos; pos = pos->next) {
214
215                 old_attachment = pos->data;
216                 attachment_filename = tny_mime_part_get_filename (old_attachment);
217                 attachment_stream = tny_mime_part_get_stream (old_attachment);
218                 attachment_part = tny_platform_factory_new_mime_part (
219                         modest_runtime_get_platform_factory());
220                 
221                 attachment_content_type = tny_mime_part_get_content_type (old_attachment);
222                 attachment_cid = tny_mime_part_get_content_id (old_attachment);
223                                  
224                 tny_mime_part_construct_from_stream (attachment_part,
225                                                      attachment_stream,
226                                                      attachment_content_type);
227                 tny_stream_reset (attachment_stream);
228                 
229                 tny_mime_part_set_filename (attachment_part, attachment_filename);
230                 tny_mime_part_set_content_id (attachment_part, attachment_cid);
231                 
232                 tny_mime_part_add_part (TNY_MIME_PART (msg), attachment_part);
233 /*              g_object_unref (attachment_part); */
234         }
235 }
236
237
238 gchar * 
239 modest_tny_msg_get_body (TnyMsg *msg, gboolean want_html)
240 {
241         TnyStream *stream;
242         TnyMimePart *body;
243         GtkTextBuffer *buf;
244         GtkTextIter start, end;
245         gchar *to_quote;
246
247         body = modest_tny_msg_find_body_part(msg, want_html);
248         if (!body)
249                 return NULL;
250
251         buf = gtk_text_buffer_new (NULL);
252         stream = TNY_STREAM (tny_gtk_text_buffer_stream_new (buf));
253         tny_stream_reset (stream);
254         tny_mime_part_decode_to_stream (body, stream);
255         tny_stream_reset (stream);
256
257         g_object_unref (G_OBJECT(stream));
258         g_object_unref (G_OBJECT(body));
259         
260         gtk_text_buffer_get_bounds (buf, &start, &end);
261         to_quote = gtk_text_buffer_get_text (buf, &start, &end, FALSE);
262         if (tny_mime_part_content_type_is (body, "text/plain")) {
263                 gchar *to_quote_converted = modest_text_utils_convert_to_html (to_quote);
264                 g_free (to_quote);
265                 to_quote = to_quote_converted;
266         }
267         g_object_unref (buf);
268
269         return to_quote;
270 }
271
272
273 TnyMimePart*
274 modest_tny_msg_find_body_part_from_mime_part (TnyMimePart *msg, gboolean want_html)
275 {
276         const gchar *mime_type = want_html ? "text/html" : "text/plain";
277         TnyMimePart *part = NULL;
278         TnyList *parts;
279         TnyIterator *iter;
280
281         if (!msg)
282                 return NULL;
283
284         parts = TNY_LIST (tny_simple_list_new());
285         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
286
287         iter  = tny_list_create_iterator(parts);
288
289         /* no parts? assume it's single-part message */
290         if (tny_iterator_is_done(iter)) 
291                 return TNY_MIME_PART (g_object_ref(G_OBJECT(msg)));
292         else {
293                 gchar *content_type = NULL;
294                 do {
295                         part = TNY_MIME_PART(tny_iterator_get_current (iter));
296
297                         /* we need to strdown the content type, because
298                          * tny_mime_part_has_content_type does not do it...
299                          */
300                         content_type = g_ascii_strdown
301                                 (tny_mime_part_get_content_type (part), -1);
302                                                         
303                         if (g_str_has_prefix (content_type, mime_type) &&
304                             !tny_mime_part_is_attachment (part))
305                                 break;
306                         
307                         if (g_str_has_prefix(content_type, "multipart")) {
308                                 part = modest_tny_msg_find_body_part_from_mime_part (part, want_html);
309                                 if (part)
310                                         break;
311                         }
312                         if (part)
313                                 g_object_unref (G_OBJECT(part));
314
315                         part = NULL;
316                         
317                         g_free (content_type);
318                         content_type = NULL;
319
320                         tny_iterator_next (iter);
321                         
322                 } while (!tny_iterator_is_done(iter));
323                 g_free (content_type);
324         }
325         
326         g_object_unref (G_OBJECT(iter));
327         g_object_unref (G_OBJECT(parts));
328
329         /* if were trying to find an HTML part and couldn't find it,
330          * try to find a text/plain part instead
331          */
332         if (!part && want_html) 
333                 return modest_tny_msg_find_body_part_from_mime_part (msg, FALSE);
334         else
335                 return part; /* this maybe NULL, this is not an error; some message just don't have a body
336                               * part */
337 }
338
339
340 TnyMimePart*
341 modest_tny_msg_find_body_part (TnyMsg *msg, gboolean want_html)
342 {
343         return modest_tny_msg_find_body_part_from_mime_part (TNY_MIME_PART(msg),
344                                                              want_html);
345 }
346
347
348 static TnyMsg *
349 create_reply_forward_mail (TnyMsg *msg, const gchar *from, const gchar *signature, gboolean is_reply, guint type)
350 {
351         TnyMsg *new_msg;
352         TnyHeader *new_header, *header;
353         gchar *new_subject;
354         TnyMimePart *body;
355         ModestFormatter *formatter;
356         gchar *subject_prefix;
357
358         /* Get body from original msg. Always look for the text/plain
359            part of the message to create the reply/forwarded mail */
360         header = tny_msg_get_header (msg);
361         body   = modest_tny_msg_find_body_part (msg, FALSE);
362
363         /* TODO: select the formatter from account prefs */
364         if (modest_conf_get_bool (modest_runtime_get_conf (), MODEST_CONF_PREFER_FORMATTED_TEXT, NULL))
365                 formatter = modest_formatter_new ("text/html", signature);
366         else
367                 formatter = modest_formatter_new ("text/plain", signature);
368
369         /* Format message body */
370         if (is_reply) {
371                 switch (type) {
372                 case MODEST_TNY_MSG_REPLY_TYPE_CITE:
373                 default:
374                         new_msg = modest_formatter_cite  (formatter, body, header);
375                         break;
376                 case MODEST_TNY_MSG_REPLY_TYPE_QUOTE:
377                         new_msg = modest_formatter_quote (formatter, body, header);
378                         break;
379                 }
380         } else {
381                 switch (type) {
382                 case MODEST_TNY_MSG_FORWARD_TYPE_INLINE:
383                 default:
384                         new_msg = modest_formatter_inline  (formatter, body, header);
385                         break;
386                 case MODEST_TNY_MSG_FORWARD_TYPE_ATTACHMENT:
387                         new_msg = modest_formatter_attach (formatter, body, header);
388                         break;
389                 }
390         }
391         g_object_unref (G_OBJECT(formatter));
392         g_object_unref (G_OBJECT(body));
393         
394         /* Fill the header */
395         new_header = tny_msg_get_header (new_msg);      
396         tny_header_set_from (new_header, from);
397         tny_header_set_replyto (new_header, from);
398
399         /* Change the subject */
400         if (is_reply)
401                 subject_prefix = g_strconcat (_("mail_va_re"), ":", NULL);
402         else
403                 subject_prefix = g_strconcat (_("mail_va_fw"), ":", NULL);
404         new_subject =
405                 (gchar *) modest_text_utils_derived_subject (tny_header_get_subject(header),
406                                                              subject_prefix);
407         g_free (subject_prefix);
408         tny_header_set_subject (new_header, (const gchar *) new_subject);
409         g_free (new_subject);
410
411         /* Clean */
412         g_object_unref (G_OBJECT (new_header));
413         g_object_unref (G_OBJECT (header));
414
415         return new_msg;
416 }
417
418 static void
419 add_if_attachment (gpointer data, gpointer user_data)
420 {
421         TnyMimePart *part;
422         GList *attachments_list;
423
424         part = TNY_MIME_PART (data);
425         attachments_list = (GList *) user_data;
426
427         if (tny_mime_part_is_attachment (part))
428                 attachments_list = g_list_prepend (attachments_list, part);
429 }
430
431 TnyMsg* 
432 modest_tny_msg_create_forward_msg (TnyMsg *msg, 
433                                    const gchar *from,
434                                    const gchar *signature,
435                                    ModestTnyMsgForwardType forward_type)
436 {
437         TnyMsg *new_msg;
438         TnyList *parts = NULL;
439         GList *attachments_list = NULL;
440
441         new_msg = create_reply_forward_mail (msg, from, signature, FALSE, forward_type);
442
443         /* Add attachments */
444         parts = TNY_LIST (tny_simple_list_new());
445         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
446         tny_list_foreach (parts, add_if_attachment, attachments_list);
447         add_attachments (new_msg, attachments_list);
448
449         /* Clean */
450         if (attachments_list)
451                 g_list_free (attachments_list);
452         g_object_unref (G_OBJECT (parts));
453
454         return new_msg;
455 }
456
457 TnyMsg* 
458 modest_tny_msg_create_reply_msg (TnyMsg *msg, 
459                                  const gchar *from,
460                                  const gchar *signature,
461                                  ModestTnyMsgReplyType reply_type,
462                                  ModestTnyMsgReplyMode reply_mode)
463 {
464         TnyMsg *new_msg = NULL;
465         TnyHeader *new_header, *header;
466         const gchar* reply_to;
467         gchar *new_cc = NULL;
468         const gchar *cc = NULL, *bcc = NULL;
469         GString *tmp = NULL;
470
471         new_msg = create_reply_forward_mail (msg, from, signature, TRUE, reply_type);
472
473         /* Fill the header */
474         header = tny_msg_get_header (msg);
475         new_header = tny_msg_get_header (new_msg);
476         reply_to = tny_header_get_replyto (header);
477
478         if (reply_to)
479                 tny_header_set_to (new_header, reply_to);
480         else
481                 tny_header_set_to (new_header, tny_header_get_from (header));
482
483         switch (reply_mode) {
484         case MODEST_TNY_MSG_REPLY_MODE_SENDER:
485                 /* Do not fill neither cc nor bcc */
486                 break;
487         case MODEST_TNY_MSG_REPLY_MODE_LIST:
488                 /* TODO */
489                 break;
490         case MODEST_TNY_MSG_REPLY_MODE_ALL:
491                 /* Concatenate to, cc and bcc */
492                 cc = tny_header_get_cc (header);
493                 bcc = tny_header_get_bcc (header);
494
495                 tmp = g_string_new (tny_header_get_to (header));
496                 if (cc)  g_string_append_printf (tmp, ",%s",cc);
497                 if (bcc) g_string_append_printf (tmp, ",%s",bcc);
498
499                /* Remove my own address from the cc list. TODO:
500                   remove also the To: of the new message, needed due
501                   to the new reply_to feature */
502                 new_cc = (gchar *)
503                         modest_text_utils_remove_address ((const gchar *) tmp->str,
504                                                           from);
505                 /* FIXME: remove also the mails from the new To: */
506                 tny_header_set_cc (new_header, new_cc);
507
508                 /* Clean */
509                 g_string_free (tmp, TRUE);
510                 g_free (new_cc);
511                 break;
512         }
513
514         /* Clean */
515         g_object_unref (G_OBJECT (new_header));
516         g_object_unref (G_OBJECT (header));
517
518         return new_msg;
519 }
520
521
522 static gboolean
523 is_ascii(const gchar *s)
524 {
525         if (!s)
526                 return TRUE;
527         while (s[0]) {
528                 if (s[0] & 128 || s[0] < 32)
529                         return FALSE;
530                 s++;
531         }
532         return TRUE;
533 }
534
535 static char *
536 get_content_type(const gchar *s)
537 {
538         GString *type;
539         
540         type = g_string_new("text/plain");
541         if (!is_ascii(s)) {
542                 if (g_utf8_validate(s, -1, NULL)) {
543                         g_string_append(type, "; charset=\"utf-8\"");
544                 } else {
545                         /* it should be impossible to reach this, but better safe than sorry */
546                         g_warning("invalid utf8 in message");
547                         g_string_append(type, "; charset=\"latin1\"");
548                 }
549         }
550         return g_string_free(type, FALSE);
551 }