Decode the mime part when copying it
[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 #include "modest-tny-mime-part.h"
44
45
46 #ifdef HAVE_CONFIG_H
47 #include <config.h>
48 #endif /*HAVE_CONFIG_H */
49
50 #include <modest-tny-msg.h>
51 #include "modest-text-utils.h"
52
53 static TnyMimePart * add_body_part (TnyMsg *msg, const gchar *body,
54                                     const gchar *content_type);
55 static TnyMimePart * add_html_body_part (TnyMsg *msg, const gchar *body);
56 static void add_attachments (TnyMimePart *part, GList *attachments_list, gboolean add_inline);
57 static void add_images (TnyMsg *msg, GList *attachments_list);
58 static char * get_content_type(const gchar *s);
59 static gboolean is_ascii(const gchar *s);
60
61
62 TnyMsg*
63 modest_tny_msg_new (const gchar* mailto, const gchar* from, const gchar *cc,
64                     const gchar *bcc, const gchar* subject, const gchar *body,
65                     GList *attachments)
66 {
67         TnyMsg *new_msg;
68         TnyHeader *header;
69         gchar *content_type;
70         
71         /* Create new msg */
72         new_msg = modest_formatter_create_message (NULL, TRUE, (attachments != NULL), FALSE);
73         header  = tny_msg_get_header (new_msg);
74         
75         if ((from != NULL) && (strlen(from) > 0)) {
76                 tny_header_set_from (TNY_HEADER (header), from);
77                 tny_header_set_replyto (TNY_HEADER (header), from);
78         }
79         if ((mailto != NULL) && (strlen(mailto) > 0)) 
80                 tny_header_set_to (TNY_HEADER (header), mailto);
81         if ((cc != NULL) && (strlen(cc) > 0)) 
82                 tny_header_set_cc (TNY_HEADER (header), cc);
83         if ((bcc != NULL) && (strlen(bcc) > 0)) 
84                 tny_header_set_bcc (TNY_HEADER (header), bcc);
85         
86         if ((subject != NULL) && (strlen(subject) > 0)) 
87                 tny_header_set_subject (TNY_HEADER (header), subject);
88
89         content_type = get_content_type(body);
90
91         /* set modest as the X-Mailer
92          * we could this in the platform factory, but then the header
93          * would show up before all the others.
94          */
95         tny_mime_part_set_header_pair (TNY_MIME_PART (new_msg), "X-Mailer", "Modest "
96                                        VERSION);
97         
98         /* Add the body of the new mail */
99         /* This is needed even if body is NULL or empty. */
100         add_body_part (new_msg, body, content_type);
101         g_free (content_type);
102                        
103         /* Add attachments */
104         if (attachments)
105                 add_attachments (TNY_MIME_PART (new_msg), attachments, FALSE);
106         if (header)
107                 g_object_unref(header);
108
109         return new_msg;
110 }
111
112 TnyMsg*
113 modest_tny_msg_new_html_plain (const gchar* mailto, const gchar* from, const gchar *cc,
114                                const gchar *bcc, const gchar* subject, 
115                                const gchar *html_body, const gchar *plain_body,
116                                GList *attachments, GList *images)
117 {
118         TnyMsg *new_msg;
119         TnyHeader *header;
120         gchar *content_type;
121         
122         /* Create new msg */
123         new_msg = modest_formatter_create_message (NULL, FALSE, (attachments != NULL), (images != NULL));
124         header  = tny_msg_get_header (new_msg);
125         
126         if ((from != NULL) && (strlen(from) > 0)) {
127                 tny_header_set_from (TNY_HEADER (header), from);
128                 tny_header_set_replyto (TNY_HEADER (header), from);
129         }
130         if ((mailto != NULL) && (strlen(mailto) > 0)) 
131                 tny_header_set_to (TNY_HEADER (header), mailto);
132         if ((cc != NULL) && (strlen(cc) > 0)) 
133                 tny_header_set_cc (TNY_HEADER (header), cc);
134         if ((bcc != NULL) && (strlen(bcc) > 0)) 
135                 tny_header_set_bcc (TNY_HEADER (header), bcc);
136         
137         if ((subject != NULL) && (strlen(subject) > 0)) 
138                 tny_header_set_subject (TNY_HEADER (header), subject);
139
140         content_type = get_content_type(plain_body);
141         
142         /* set modest as the X-Mailer
143          * we could this in the platform factory, but then the header
144          * would show up before all the others.
145          */
146         tny_mime_part_set_header_pair (TNY_MIME_PART (new_msg), "X-Mailer", "Modest "
147                                        VERSION);
148
149         /* Add the body of the new mail */      
150         add_body_part (new_msg, plain_body, content_type);
151         add_html_body_part (new_msg, html_body);
152         g_free (content_type);
153                        
154         /* Add attachments */
155         add_attachments (TNY_MIME_PART (new_msg), attachments, FALSE);
156         add_images (new_msg, images);
157         if (header)
158                 g_object_unref(header);
159
160         return new_msg;
161 }
162
163
164 /* FIXME: this func copy from modest-mail-operation: refactor */
165 static TnyMimePart *
166 add_body_part (TnyMsg *msg, 
167                const gchar *body,
168                const gchar *content_type)
169 {
170         TnyMimePart *text_body_part = NULL;
171         TnyStream *text_body_stream;
172
173         /* Create the stream */
174         text_body_stream = TNY_STREAM (tny_camel_stream_new
175                                        (camel_stream_mem_new_with_buffer
176                                         (body, (body ? strlen(body) : 0))));
177
178         text_body_part = modest_formatter_create_body_part (NULL, msg);
179
180         /* Construct MIME part */
181         tny_stream_reset (text_body_stream);
182         tny_mime_part_construct (text_body_part,
183                                  text_body_stream,
184                                  content_type, "7bit");
185         tny_stream_reset (text_body_stream);
186
187         g_object_unref (G_OBJECT(text_body_part));
188
189         /* Clean */
190         g_object_unref (text_body_stream);
191
192         return text_body_part;
193 }
194
195 static TnyMimePart *
196 add_html_body_part (TnyMsg *msg, 
197                     const gchar *body)
198 {
199         TnyMimePart *html_body_part = NULL;
200         TnyStream *html_body_stream;
201
202         /* Create the stream */
203         html_body_stream = TNY_STREAM (tny_camel_stream_new
204                                        (camel_stream_mem_new_with_buffer
205                                         (body, strlen(body))));
206
207         /* Create body part if needed */
208         html_body_part = modest_formatter_create_body_part (NULL, msg);
209
210         /* Construct MIME part */
211         tny_stream_reset (html_body_stream);
212         tny_mime_part_construct (html_body_part,
213                                  html_body_stream,
214                                  "text/html; charset=utf-8", 
215                                  "7bit"); /* Sometimes it might be needed 
216                                              to make this one a 8bit! */
217         tny_stream_reset (html_body_stream);
218
219         g_object_unref (G_OBJECT(html_body_part));
220
221         /* Clean */
222         g_object_unref (html_body_stream);
223
224         return html_body_part;
225 }
226
227 static TnyMimePart *
228 copy_mime_part (TnyMimePart *part)
229 {
230         TnyMimePart *result = NULL;
231         const gchar *attachment_content_type;
232         const gchar *attachment_filename;
233         const gchar *attachment_cid;
234         TnyList *parts;
235         TnyIterator *iterator;
236         TnyStream *attachment_stream;
237         const gchar *enc;
238         
239         if (TNY_IS_MSG (part)) {
240                 g_object_ref (part);
241                 return part;
242         }
243
244         result = tny_platform_factory_new_mime_part (
245                 modest_runtime_get_platform_factory());
246
247         attachment_content_type = tny_mime_part_get_content_type (part);
248
249         /* get mime part headers */
250         attachment_filename = tny_mime_part_get_filename (part);
251         attachment_cid = tny_mime_part_get_content_id (part);
252         
253         /* fill the stream */
254         attachment_stream = tny_mime_part_get_decoded_stream (part);
255         enc = tny_mime_part_get_transfer_encoding (part);
256         tny_stream_reset (attachment_stream);
257         tny_mime_part_construct (result,
258                                  attachment_stream,
259                                  attachment_content_type, 
260                                  enc);
261         tny_stream_reset (attachment_stream);
262         
263         /* set other mime part fields */
264         tny_mime_part_set_filename (result, attachment_filename);
265         tny_mime_part_set_content_id (result, attachment_cid);
266
267         /* copy subparts */
268         parts = tny_simple_list_new ();
269         tny_mime_part_get_parts (part, parts);
270         iterator = tny_list_create_iterator (parts);
271         while (!tny_iterator_is_done (iterator)) {
272                 TnyMimePart *subpart = TNY_MIME_PART (tny_iterator_get_current (iterator));
273                 if (subpart) {
274                         const gchar *subpart_cid;
275                         TnyMimePart *subpart_copy = copy_mime_part (subpart);
276                         subpart_cid = tny_mime_part_get_content_id (subpart);
277                         tny_mime_part_add_part (result, subpart_copy);
278                         if (subpart_cid)
279                                 tny_mime_part_set_content_id (result, subpart_cid);
280                         g_object_unref (subpart);
281                         g_object_unref (subpart_copy);
282                 }
283
284                 tny_iterator_next (iterator);
285         }
286         g_object_unref (iterator);
287         g_object_unref (parts);
288         g_object_unref (attachment_stream);
289
290         return result;
291 }
292
293 static void
294 add_attachments (TnyMimePart *part, GList *attachments_list, gboolean add_inline)
295 {
296         GList *pos;
297         TnyMimePart *attachment_part, *old_attachment;
298
299         for (pos = (GList *)attachments_list; pos; pos = pos->next) {
300
301                 old_attachment = pos->data;
302                 if (!tny_mime_part_is_purged (old_attachment)) {
303                         const gchar *old_cid;
304                         old_cid = tny_mime_part_get_content_id (old_attachment);
305                         attachment_part = copy_mime_part (old_attachment);
306                         tny_mime_part_set_header_pair (attachment_part, "Content-Disposition", 
307                                                        add_inline?"inline":"attachment");
308                         tny_mime_part_set_transfer_encoding (TNY_MIME_PART (attachment_part), "base64");
309                         tny_mime_part_add_part (TNY_MIME_PART (part), attachment_part);
310                         if (old_cid)
311                                 tny_mime_part_set_content_id (attachment_part, old_cid);
312                         g_object_unref (attachment_part);
313                 }
314         }
315 }
316
317 static void
318 add_images (TnyMsg *msg, GList *images_list)
319 {
320         TnyMimePart *related_part = NULL;
321         const gchar *content_type;
322
323         content_type = tny_mime_part_get_content_type (TNY_MIME_PART (msg));
324
325         if ((content_type != NULL) && !strcasecmp (content_type, "multipart/related")) {
326                 related_part = g_object_ref (msg);
327         } else if ((content_type != NULL) && !strcasecmp (content_type, "multipart/mixed")) {
328                 TnyList *parts = TNY_LIST (tny_simple_list_new ());
329                 TnyIterator *iter = NULL;
330                 tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
331                 iter = tny_list_create_iterator (parts);
332
333                 while (!tny_iterator_is_done (iter)) {
334                         TnyMimePart *part = TNY_MIME_PART (tny_iterator_get_current (iter));
335                         if (part && !g_strcasecmp (tny_mime_part_get_content_type (part), "multipart/related")) {
336                                 related_part = part;
337                                 break;
338                         } else {
339                                 g_object_unref (part);
340                         }
341                         tny_iterator_next (iter);
342                 }
343                 g_object_unref (iter);
344                 g_object_unref (parts);
345         }
346
347         if (related_part != NULL) {
348                 /* TODO: attach images in their proper place */
349                 add_attachments (related_part, images_list, TRUE);
350                 g_object_unref (related_part);
351         }
352 }
353
354
355 gchar * 
356 modest_tny_msg_get_body (TnyMsg *msg, gboolean want_html, gboolean *is_html)
357 {
358         TnyStream *stream;
359         TnyMimePart *body;
360         GtkTextBuffer *buf;
361         GtkTextIter start, end;
362         gchar *to_quote;
363         gboolean result_was_html = TRUE;
364
365         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
366         
367         body = modest_tny_msg_find_body_part(msg, want_html);
368         if (!body)
369                 return NULL;
370
371         buf = gtk_text_buffer_new (NULL);
372         stream = TNY_STREAM (tny_gtk_text_buffer_stream_new (buf));
373         tny_stream_reset (stream);
374         tny_mime_part_decode_to_stream (body, stream, NULL);
375         tny_stream_reset (stream);
376         
377         gtk_text_buffer_get_bounds (buf, &start, &end);
378         to_quote = gtk_text_buffer_get_text (buf, &start, &end, FALSE);
379         if (tny_mime_part_content_type_is (body, "text/plain")) {
380                 gchar *to_quote_converted = modest_text_utils_convert_to_html (to_quote);
381                 g_free (to_quote);
382                 to_quote = to_quote_converted;
383                 result_was_html = FALSE;
384         }
385
386         g_object_unref (buf);
387         g_object_unref (G_OBJECT(stream));
388         g_object_unref (G_OBJECT(body));
389
390         if (is_html != NULL)
391                 *is_html = result_was_html;
392
393         return to_quote;
394 }
395
396
397 TnyMimePart*
398 modest_tny_msg_find_body_part_from_mime_part (TnyMimePart *msg, gboolean want_html)
399 {
400         const gchar *desired_mime_type = want_html ? "text/html" : "text/plain";
401         TnyMimePart *part = NULL;
402         TnyList *parts = NULL;
403         TnyIterator *iter = NULL;
404         gchar *header_content_type;
405         gchar *header_content_type_lower;
406         
407         if (!msg)
408                 return NULL;
409
410         /* If it's an application multipart, then we don't get into as we don't
411          * support them (for example application/sml or wap messages */
412         header_content_type = modest_tny_mime_part_get_header_value (msg, "Content-Type");
413         header_content_type = g_strstrip (header_content_type);
414         header_content_type_lower = header_content_type?g_ascii_strdown (header_content_type, -1):NULL;
415         if (header_content_type_lower && 
416             g_str_has_prefix (header_content_type_lower, "multipart/") &&
417             strstr (header_content_type_lower, "application/")) {
418                 g_free (header_content_type_lower);
419                 g_free (header_content_type);
420                 return NULL;
421         }
422         g_free (header_content_type_lower);
423         g_free (header_content_type);
424
425         parts = TNY_LIST (tny_simple_list_new());
426         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
427
428         iter  = tny_list_create_iterator(parts);
429
430         /* no parts? assume it's single-part message */
431         if (tny_iterator_is_done(iter)) {
432                 const gchar *content_type;
433                 g_object_unref (G_OBJECT(iter));
434                 content_type = tny_mime_part_get_content_type (msg);
435                 /* if this part cannot be a supported body return NULL */
436                 if (!g_str_has_prefix (content_type, "text/")) {
437                         return NULL;
438                 } else {
439                         return TNY_MIME_PART (g_object_ref(G_OBJECT(msg)));
440                 }
441         } else {
442                 do {
443                         gchar *tmp, *content_type = NULL;
444                         gboolean has_content_disp_name = FALSE;
445
446                         part = TNY_MIME_PART(tny_iterator_get_current (iter));
447
448                         if (!part) {
449                                 g_warning ("%s: not a valid mime part", __FUNCTION__);
450                                 continue;
451                         }
452
453                         /* it's a message --> ignore */
454                         if (part && TNY_IS_MSG (part)) {
455                                 g_object_unref (part);
456                                 tny_iterator_next (iter);
457                                 continue;
458                         }                       
459
460                         /* we need to strdown the content type, because
461                          * tny_mime_part_has_content_type does not do it...
462                          */
463                         content_type = g_ascii_strdown (tny_mime_part_get_content_type (part), -1);
464                         
465                         /* mime-parts with a content-disposition header (either 'inline' or 'attachment')
466                          * and a 'name=' thingy cannot be body parts
467                          */
468                                 
469                         tmp = modest_tny_mime_part_get_header_value (part, "Content-Disposition");
470                         if (tmp) {
471                                 gchar *content_disp = g_ascii_strdown(tmp, -1);
472                                 g_free (tmp);
473                                 has_content_disp_name = g_strstr_len (content_disp, strlen(content_disp), "name=") != NULL;
474                                 g_free (content_disp);
475                         }
476                         
477                         if (g_str_has_prefix (content_type, desired_mime_type) && !has_content_disp_name) {
478                                 /* we found the desired mime-type! */
479                                 g_free (content_type);
480                                 break;
481
482                         } else  if (g_str_has_prefix(content_type, "multipart")) {
483
484                                 /* multipart? recurse! */
485                                 g_object_unref (part);
486                                 g_free (content_type);
487                                 part = modest_tny_msg_find_body_part_from_mime_part (part, want_html);
488                                 if (part)
489                                         break;
490                         } else
491                                 g_free (content_type);
492                         
493                         if (part) {
494                                 g_object_unref (G_OBJECT(part));
495                                 part = NULL;
496                         }
497                         
498                         tny_iterator_next (iter);
499                         
500                 } while (!tny_iterator_is_done(iter));
501         }
502         
503         g_object_unref (G_OBJECT(iter));
504         g_object_unref (G_OBJECT(parts));
505
506         /* if were trying to find an HTML part and couldn't find it,
507          * try to find a text/plain part instead
508          */
509         if (!part && want_html) 
510                 return modest_tny_msg_find_body_part_from_mime_part (msg, FALSE);
511         else
512                 return part; /* this maybe NULL, this is not an error; some message just don't have a body
513                               * part */
514 }
515
516
517 TnyMimePart*
518 modest_tny_msg_find_body_part (TnyMsg *msg, gboolean want_html)
519 {
520         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
521         
522         return modest_tny_msg_find_body_part_from_mime_part (TNY_MIME_PART(msg),
523                                                              want_html);
524 }
525
526
527 #define MODEST_TNY_MSG_PARENT_UID "parent-uid"
528
529 static TnyMsg *
530 create_reply_forward_mail (TnyMsg *msg, TnyHeader *header, const gchar *from,
531                            const gchar *signature, gboolean is_reply,
532                            guint type /*ignored*/, GList *attachments)
533 {
534         TnyMsg *new_msg;
535         TnyHeader *new_header;
536         gchar *new_subject;
537         TnyMimePart *body = NULL;
538         ModestFormatter *formatter;
539         gchar *subject_prefix;
540         gboolean no_text_part;
541         
542         if (header)
543                 g_object_ref (header);
544         else
545                 header = tny_msg_get_header (msg);
546
547         /* Get body from original msg. Always look for the text/plain
548            part of the message to create the reply/forwarded mail */
549         if (msg != NULL)
550                 body   = modest_tny_msg_find_body_part (msg, FALSE);
551         
552         if (modest_conf_get_bool (modest_runtime_get_conf (), MODEST_CONF_PREFER_FORMATTED_TEXT,
553                                   NULL))
554                 formatter = modest_formatter_new ("text/html", signature);
555         else
556                 formatter = modest_formatter_new ("text/plain", signature);
557         
558
559         /* if we don't have a text-part */
560         no_text_part = (!body) || (strcmp (tny_mime_part_get_content_type (body), "text/html")==0);
561         
562         /* when we're reply, include the text part if we have it, or nothing otherwise. */
563         if (is_reply)
564                 new_msg = modest_formatter_quote  (formatter, no_text_part ? NULL: body, header,
565                                                     attachments);
566         else {
567                 /* for attachements; inline if there is a text part, and include the
568                  * full old mail if there was none */
569                 if (no_text_part) {
570                         new_msg = modest_formatter_attach (formatter, msg, header);
571                 } else { 
572                         new_msg = modest_formatter_inline  (formatter, body, header,
573                                                             attachments);
574                 }
575         }
576         
577         g_object_unref (G_OBJECT(formatter));
578         if (body)
579                 g_object_unref (G_OBJECT(body));
580         
581         /* Fill the header */
582         new_header = tny_msg_get_header (new_msg);      
583         tny_header_set_from (new_header, from);
584         tny_header_set_replyto (new_header, from);
585
586         /* Change the subject */
587         if (is_reply)
588                 subject_prefix = g_strconcat (_("mail_va_re"), ":", NULL);
589         else
590                 subject_prefix = g_strconcat (_("mail_va_fw"), ":", NULL);
591         new_subject =
592                 (gchar *) modest_text_utils_derived_subject (tny_header_get_subject(header),
593                                                              subject_prefix);
594         g_free (subject_prefix);
595         tny_header_set_subject (new_header, (const gchar *) new_subject);
596         g_free (new_subject);
597         
598         /* get the parent uid, and set it as a gobject property on the new msg */
599         if (new_msg) {
600                 gchar* parent_uid = modest_tny_folder_get_header_unique_id (header);
601                 g_object_set_data_full (G_OBJECT(new_msg), MODEST_TNY_MSG_PARENT_UID,
602                                         parent_uid, g_free);
603         }
604
605         /* set modest as the X-Mailer
606          * we could this in the platform factory, but then the header
607          * would show up before all the others.
608          */
609         tny_mime_part_set_header_pair (TNY_MIME_PART (msg), "X-Mailer", "Modest "
610                                        VERSION);
611
612         /* Clean */
613         g_object_unref (G_OBJECT (new_header));
614         g_object_unref (G_OBJECT (header));
615         /* ugly to unref it here instead of in the calling func */
616
617         if (!is_reply & !no_text_part) {
618                 add_attachments (TNY_MIME_PART (new_msg), attachments, FALSE);
619         }
620
621         return new_msg;
622 }
623
624 const gchar*
625 modest_tny_msg_get_parent_uid (TnyMsg *msg)
626 {
627         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
628         
629         return g_object_get_data (G_OBJECT(msg), MODEST_TNY_MSG_PARENT_UID);
630 }
631
632
633
634 static void
635 add_if_attachment (gpointer data, gpointer user_data)
636 {
637         TnyMimePart *part;
638         GList **attachments_list;
639
640         part = TNY_MIME_PART (data);
641         attachments_list = ((GList **) user_data);
642
643         if ((tny_mime_part_is_attachment (part))||(TNY_IS_MSG (part))) {
644                 *attachments_list = g_list_prepend (*attachments_list, part);
645                 g_object_ref (part);
646         }
647 }
648
649 TnyMsg* 
650 modest_tny_msg_create_forward_msg (TnyMsg *msg, 
651                                    const gchar *from,
652                                    const gchar *signature,
653                                    ModestTnyMsgForwardType forward_type)
654 {
655         TnyMsg *new_msg;
656         TnyList *parts = NULL;
657         GList *attachments_list = NULL;
658
659         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
660         
661         /* Add attachments */
662         parts = TNY_LIST (tny_simple_list_new());
663         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
664         tny_list_foreach (parts, add_if_attachment, &attachments_list);
665
666         new_msg = create_reply_forward_mail (msg, NULL, from, signature, FALSE, forward_type,
667                                              attachments_list);
668
669         /* Clean */
670         if (attachments_list) {
671                 g_list_foreach (attachments_list, (GFunc) g_object_unref, NULL);
672                 g_list_free (attachments_list);
673         }
674         g_object_unref (G_OBJECT (parts));
675
676         return new_msg;
677 }
678
679
680
681 static gint
682 count_addresses (const gchar* addresses)
683 {
684         gint count = 1;
685
686         if (!addresses)
687                 return 0;
688         
689         while (*addresses) {
690                 if (*addresses == ',' || *addresses == ';')
691                         ++count;
692                 ++addresses;
693         }
694         
695         return count;
696 }
697
698
699 /* get the new To:, based on the old header,
700  * result is newly allocated or NULL in case of error
701  * */
702 static gchar*
703 get_new_to (TnyMsg *msg, TnyHeader *header, const gchar* from,
704             ModestTnyMsgReplyMode reply_mode)
705 {
706         const gchar* old_reply_to;
707         const gchar* old_from;
708         gchar* new_to;
709         
710         /* according to RFC2369 (http://www.faqs.org/rfcs/rfc2369.html), we
711          * can identify Mailing-List posts by the List-Help header.
712          * for mailing lists, both the Reply-To: and From: should be included
713          * in the new To:; for now, we're ignoring List-Post
714          */
715         gchar* list_help = modest_tny_mime_part_get_header_value (TNY_MIME_PART(msg), 
716                                                                   "List-Help");
717         gboolean is_mailing_list = (list_help != NULL);
718         g_free (list_help);
719
720
721         /* reply to sender, use ReplyTo or From */
722         //old_reply_to = tny_header_get_replyto (header);
723         old_reply_to = modest_tny_mime_part_get_header_value (TNY_MIME_PART(msg), 
724                                                               "Reply-To"); 
725         old_from     = tny_header_get_from (header);
726         
727         if (!old_from &&  !old_reply_to) {
728                 g_warning ("%s: failed to get either Reply-To: or From: from header",
729                            __FUNCTION__);
730                 return NULL;
731         }
732         
733         /* for mailing lists, use both Reply-To and From if we did a
734          * 'Reply All:'
735          * */
736         if (is_mailing_list && reply_mode == MODEST_TNY_MSG_REPLY_MODE_ALL &&
737             old_reply_to && old_from && strcmp (old_from, old_reply_to) != 0)
738                 new_to = g_strjoin (",", old_reply_to, old_from, NULL);
739         else
740                 /* otherwise use either Reply-To: (preferred) or From: */
741                 new_to = g_strdup (old_reply_to ? old_reply_to : old_from);
742
743         /* in case of ReplyAll, we need to add the Recipients in the old To: */
744         if (reply_mode == MODEST_TNY_MSG_REPLY_MODE_ALL) {
745                 const gchar *old_to = tny_header_get_to (header);
746                 if (!old_to) 
747                         g_warning ("%s: no To: address found in source mail",
748                                    __FUNCTION__);
749                 else {
750                         /* append the old To: */
751                         gchar *tmp = g_strjoin (",", new_to, old_to, NULL);
752                         g_free (new_to);
753                         new_to = tmp;
754                 }
755
756                 /* remove duplicate entries */
757                 gchar *tmp = modest_text_utils_remove_duplicate_addresses (new_to);
758                 g_free (new_to);
759                 new_to = tmp;
760                 
761                 /* now, strip me (the new From:) from the new_to, but only if
762                  * there are >1 addresses there */
763                 if (count_addresses (new_to) > 1) {
764                         gchar *tmp = modest_text_utils_remove_address (new_to, from);
765                         g_free (new_to);
766                         new_to = tmp;
767                 }
768         }
769
770         return new_to;
771 }
772
773
774 /* get the new Cc:, based on the old header,
775  * result is newly allocated or NULL in case of error */
776 static gchar*
777 get_new_cc (TnyHeader *header, const gchar* from)
778 {
779         const gchar *old_cc;
780
781         old_cc = tny_header_get_cc (header);
782         if (!old_cc)
783                 return NULL;
784
785         /* remove me (the new From:) from the Cc: list */
786         return modest_text_utils_remove_address (old_cc, from);
787 }
788
789
790 TnyMsg* 
791 modest_tny_msg_create_reply_msg (TnyMsg *msg,
792                                  TnyHeader *header,
793                                  const gchar *from,
794                                  const gchar *signature,
795                                  ModestTnyMsgReplyType reply_type,
796                                  ModestTnyMsgReplyMode reply_mode)
797 {
798         TnyMsg *new_msg = NULL;
799         TnyHeader *new_header;
800         gchar *new_to = NULL;
801         TnyList *parts = NULL;
802         GList *attachments_list = NULL;
803
804         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
805         
806         parts = TNY_LIST (tny_simple_list_new());
807         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
808         tny_list_foreach (parts, add_if_attachment, &attachments_list);
809
810         new_msg = create_reply_forward_mail (msg, header, from, signature, TRUE, reply_type,
811                                              attachments_list);
812         if (attachments_list) {
813                 g_list_foreach (attachments_list, (GFunc) g_object_unref, NULL);
814                 g_list_free (attachments_list);
815         }
816         if (parts)
817                 g_object_unref (G_OBJECT (parts));
818
819         /* Fill the header */
820         if (header)
821                 g_object_ref (header);
822         else
823                 header = tny_msg_get_header (msg);
824
825         
826         new_header = tny_msg_get_header(new_msg);
827         new_to = get_new_to (msg, header, from, reply_mode);
828         if (!new_to)
829                 g_warning ("%s: failed to get new To:", __FUNCTION__);
830         else {
831                 tny_header_set_to (new_header, new_to);
832                 g_free (new_to);
833         }       
834         
835         if (reply_mode == MODEST_TNY_MSG_REPLY_MODE_ALL) {
836                 gchar *new_cc = get_new_cc (header, from);
837                 if (new_cc) { 
838                         tny_header_set_cc (new_header, new_cc);
839                         g_free (new_cc);
840                 }
841         }
842         
843         /* Clean */
844         g_object_unref (G_OBJECT (new_header));
845         g_object_unref (G_OBJECT (header));
846
847         return new_msg;
848 }
849
850
851 static gboolean
852 is_ascii(const gchar *s)
853 {
854         if (!s)
855                 return TRUE;
856         while (s[0]) {
857                 if (s[0] & 128 || s[0] < 32)
858                         return FALSE;
859                 s++;
860         }
861         return TRUE;
862 }
863
864 static char *
865 get_content_type(const gchar *s)
866 {
867         GString *type;
868         
869         type = g_string_new("text/plain");
870         if (!is_ascii(s)) {
871                 if (g_utf8_validate(s, -1, NULL)) {
872                         g_string_append(type, "; charset=\"utf-8\"");
873                 } else {
874                         /* it should be impossible to reach this, but better safe than sorry */
875                         g_warning("invalid utf8 in message");
876                         g_string_append(type, "; charset=\"latin1\"");
877                 }
878         }
879         return g_string_free(type, FALSE);
880 }
881
882 guint64
883 modest_tny_msg_estimate_size (const gchar *plain_body, const gchar *html_body,
884                               guint64 parts_count,
885                               guint64 parts_size)
886 {
887         guint64 result;
888
889         /* estimation of headers size */
890         result = 1024;
891
892         /* We add a 20% of size due to the increase in 7bit encoding */
893         if (plain_body) {
894                 result += strlen (plain_body) * 120 / 100;
895         }
896         if (html_body) {
897                 result += strlen (html_body) * 120 / 100;
898         }
899
900         /* 256 bytes per additional part because of their headers */
901         result += parts_count * 256;
902
903         /* 150% of increase per encoding */
904         result += parts_size * 3 / 2;
905
906         return result;
907 }