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