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