* src/gnome/modest-account-view-window.c:
[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;
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         header_content_type = g_strstrip (header_content_type);
410         header_content_type_lower = header_content_type?g_ascii_strdown (header_content_type, -1):NULL;
411         if (header_content_type_lower && 
412             g_str_has_prefix (header_content_type_lower, "multipart/") &&
413             strstr (header_content_type_lower, "application/")) {
414                 g_free (header_content_type_lower);
415                 g_free (header_content_type);
416                 return NULL;
417         }
418         g_free (header_content_type_lower);
419         g_free (header_content_type);
420
421         parts = TNY_LIST (tny_simple_list_new());
422         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
423
424         iter  = tny_list_create_iterator(parts);
425
426         /* no parts? assume it's single-part message */
427         if (tny_iterator_is_done(iter)) {
428                 const gchar *content_type;
429                 g_object_unref (G_OBJECT(iter));
430                 content_type = tny_mime_part_get_content_type (msg);
431                 /* if this part cannot be a supported body return NULL */
432                 if (!g_str_has_prefix (content_type, "text/")) {
433                         return NULL;
434                 } else {
435                         return TNY_MIME_PART (g_object_ref(G_OBJECT(msg)));
436                 }
437         } else {
438                 do {
439                         gchar *tmp, *content_type = NULL;
440                         gboolean has_content_disp_name = FALSE;
441
442                         part = TNY_MIME_PART(tny_iterator_get_current (iter));
443
444                         if (!part) {
445                                 g_warning ("%s: not a valid mime part", __FUNCTION__);
446                                 continue;
447                         }
448
449                         /* it's a message --> ignore */
450                         if (part && TNY_IS_MSG (part)) {
451                                 g_object_unref (part);
452                                 tny_iterator_next (iter);
453                                 continue;
454                         }                       
455
456                         /* we need to strdown the content type, because
457                          * tny_mime_part_has_content_type does not do it...
458                          */
459                         content_type = g_ascii_strdown (tny_mime_part_get_content_type (part), -1);
460                         
461                         /* mime-parts with a content-disposition header (either 'inline' or 'attachment')
462                          * and a 'name=' thingy cannot be body parts
463                          */
464                                 
465                         tmp = modest_tny_mime_part_get_header_value (part, "Content-Disposition");
466                         if (tmp) {
467                                 gchar *content_disp = g_ascii_strdown(tmp, -1);
468                                 g_free (tmp);
469                                 has_content_disp_name = g_strstr_len (content_disp, strlen(content_disp), "name=") != NULL;
470                                 g_free (content_disp);
471                         }
472                         
473                         if (g_str_has_prefix (content_type, desired_mime_type) && !has_content_disp_name) {
474                                 /* we found the desired mime-type! */
475                                 g_free (content_type);
476                                 break;
477
478                         } else  if (g_str_has_prefix(content_type, "multipart")) {
479
480                                 /* multipart? recurse! */
481                                 g_object_unref (part);
482                                 g_free (content_type);
483                                 part = modest_tny_msg_find_body_part_from_mime_part (part, want_html);
484                                 if (part)
485                                         break;
486                         } else
487                                 g_free (content_type);
488                         
489                         if (part) {
490                                 g_object_unref (G_OBJECT(part));
491                                 part = NULL;
492                         }
493                         
494                         tny_iterator_next (iter);
495                         
496                 } while (!tny_iterator_is_done(iter));
497         }
498         
499         g_object_unref (G_OBJECT(iter));
500         g_object_unref (G_OBJECT(parts));
501
502         /* if were trying to find an HTML part and couldn't find it,
503          * try to find a text/plain part instead
504          */
505         if (!part && want_html) 
506                 return modest_tny_msg_find_body_part_from_mime_part (msg, FALSE);
507         else
508                 return part; /* this maybe NULL, this is not an error; some message just don't have a body
509                               * part */
510 }
511
512
513 TnyMimePart*
514 modest_tny_msg_find_body_part (TnyMsg *msg, gboolean want_html)
515 {
516         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
517         
518         return modest_tny_msg_find_body_part_from_mime_part (TNY_MIME_PART(msg),
519                                                              want_html);
520 }
521
522
523 #define MODEST_TNY_MSG_PARENT_UID "parent-uid"
524
525 static TnyMsg *
526 create_reply_forward_mail (TnyMsg *msg, TnyHeader *header, const gchar *from,
527                            const gchar *signature, gboolean is_reply,
528                            guint type /*ignored*/, GList *attachments)
529 {
530         TnyMsg *new_msg;
531         TnyHeader *new_header;
532         gchar *new_subject;
533         TnyMimePart *body = NULL;
534         ModestFormatter *formatter;
535         gchar *subject_prefix;
536         gboolean no_text_part;
537         
538         if (header)
539                 g_object_ref (header);
540         else
541                 header = tny_msg_get_header (msg);
542
543         /* Get body from original msg. Always look for the text/plain
544            part of the message to create the reply/forwarded mail */
545         if (msg != NULL)
546                 body   = modest_tny_msg_find_body_part (msg, FALSE);
547         
548         if (modest_conf_get_bool (modest_runtime_get_conf (), MODEST_CONF_PREFER_FORMATTED_TEXT,
549                                   NULL))
550                 formatter = modest_formatter_new ("text/html", signature);
551         else
552                 formatter = modest_formatter_new ("text/plain", signature);
553         
554
555         /* if we don't have a text-part */
556         no_text_part = (!body) || (strcmp (tny_mime_part_get_content_type (body), "text/html")==0);
557         
558         /* when we're reply, include the text part if we have it, or nothing otherwise. */
559         if (is_reply)
560                 new_msg = modest_formatter_quote  (formatter, no_text_part ? NULL: body, header,
561                                                     attachments);
562         else {
563                 /* for attachements; inline if there is a text part, and include the
564                  * full old mail if there was none */
565                 if (no_text_part) {
566                         new_msg = modest_formatter_attach (formatter, msg, header);
567                 } else { 
568                         new_msg = modest_formatter_inline  (formatter, body, header,
569                                                             attachments);
570                 }
571         }
572         
573         g_object_unref (G_OBJECT(formatter));
574         if (body)
575                 g_object_unref (G_OBJECT(body));
576         
577         /* Fill the header */
578         new_header = tny_msg_get_header (new_msg);      
579         tny_header_set_from (new_header, from);
580         tny_header_set_replyto (new_header, from);
581
582         /* Change the subject */
583         if (is_reply)
584                 subject_prefix = g_strconcat (_("mail_va_re"), ":", NULL);
585         else
586                 subject_prefix = g_strconcat (_("mail_va_fw"), ":", NULL);
587         new_subject =
588                 (gchar *) modest_text_utils_derived_subject (tny_header_get_subject(header),
589                                                              subject_prefix);
590         g_free (subject_prefix);
591         tny_header_set_subject (new_header, (const gchar *) new_subject);
592         g_free (new_subject);
593         
594         /* get the parent uid, and set it as a gobject property on the new msg */
595         if (new_msg) {
596                 gchar* parent_uid = modest_tny_folder_get_header_unique_id (header);
597                 g_object_set_data_full (G_OBJECT(new_msg), MODEST_TNY_MSG_PARENT_UID,
598                                         parent_uid, g_free);
599         }
600
601         /* set modest as the X-Mailer
602          * we could this in the platform factory, but then the header
603          * would show up before all the others.
604          */
605         tny_mime_part_set_header_pair (TNY_MIME_PART (msg), "X-Mailer", "Modest "
606                                        VERSION);
607
608         /* Clean */
609         g_object_unref (G_OBJECT (new_header));
610         g_object_unref (G_OBJECT (header));
611         /* ugly to unref it here instead of in the calling func */
612
613         if (!is_reply & !no_text_part) {
614                 add_attachments (TNY_MIME_PART (new_msg), attachments, FALSE);
615         }
616
617         return new_msg;
618 }
619
620 const gchar*
621 modest_tny_msg_get_parent_uid (TnyMsg *msg)
622 {
623         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
624         
625         return g_object_get_data (G_OBJECT(msg), MODEST_TNY_MSG_PARENT_UID);
626 }
627
628
629
630 static void
631 add_if_attachment (gpointer data, gpointer user_data)
632 {
633         TnyMimePart *part;
634         GList **attachments_list;
635
636         part = TNY_MIME_PART (data);
637         attachments_list = ((GList **) user_data);
638
639         if ((tny_mime_part_is_attachment (part))||(TNY_IS_MSG (part))) {
640                 *attachments_list = g_list_prepend (*attachments_list, part);
641                 g_object_ref (part);
642         }
643 }
644
645 TnyMsg* 
646 modest_tny_msg_create_forward_msg (TnyMsg *msg, 
647                                    const gchar *from,
648                                    const gchar *signature,
649                                    ModestTnyMsgForwardType forward_type)
650 {
651         TnyMsg *new_msg;
652         TnyList *parts = NULL;
653         GList *attachments_list = NULL;
654
655         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
656         
657         /* Add attachments */
658         parts = TNY_LIST (tny_simple_list_new());
659         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
660         tny_list_foreach (parts, add_if_attachment, &attachments_list);
661
662         new_msg = create_reply_forward_mail (msg, NULL, from, signature, FALSE, forward_type,
663                                              attachments_list);
664
665         /* Clean */
666         if (attachments_list) {
667                 g_list_foreach (attachments_list, (GFunc) g_object_unref, NULL);
668                 g_list_free (attachments_list);
669         }
670         g_object_unref (G_OBJECT (parts));
671
672         return new_msg;
673 }
674
675
676
677 static gint
678 count_addresses (const gchar* addresses)
679 {
680         gint count = 1;
681
682         if (!addresses)
683                 return 0;
684         
685         while (*addresses) {
686                 if (*addresses == ',' || *addresses == ';')
687                         ++count;
688                 ++addresses;
689         }
690         
691         return count;
692 }
693
694
695 /* get the new To:, based on the old header,
696  * result is newly allocated or NULL in case of error
697  * */
698 static gchar*
699 get_new_to (TnyMsg *msg, TnyHeader *header, const gchar* from,
700             ModestTnyMsgReplyMode reply_mode)
701 {
702         const gchar* old_reply_to;
703         const gchar* old_from;
704         gchar* new_to;
705         
706         /* according to RFC2369 (http://www.faqs.org/rfcs/rfc2369.html), we
707          * can identify Mailing-List posts by the List-Help header.
708          * for mailing lists, both the Reply-To: and From: should be included
709          * in the new To:; for now, we're ignoring List-Post
710          */
711         gchar* list_help = modest_tny_mime_part_get_header_value (TNY_MIME_PART(msg), 
712                                                                   "List-Help");
713         gboolean is_mailing_list = (list_help != NULL);
714         g_free (list_help);
715
716
717         /* reply to sender, use ReplyTo or From */
718         //old_reply_to = tny_header_get_replyto (header);
719         old_reply_to = modest_tny_mime_part_get_header_value (TNY_MIME_PART(msg), 
720                                                               "Reply-To"); 
721         old_from     = tny_header_get_from (header);
722         
723         if (!old_from &&  !old_reply_to) {
724                 g_warning ("%s: failed to get either Reply-To: or From: from header",
725                            __FUNCTION__);
726                 return NULL;
727         }
728         
729         /* for mailing lists, use both Reply-To and From if we did a
730          * 'Reply All:'
731          * */
732         if (is_mailing_list && reply_mode == MODEST_TNY_MSG_REPLY_MODE_ALL &&
733             old_reply_to && old_from && strcmp (old_from, old_reply_to) != 0)
734                 new_to = g_strjoin (",", old_reply_to, old_from, NULL);
735         else
736                 /* otherwise use either Reply-To: (preferred) or From: */
737                 new_to = g_strdup (old_reply_to ? old_reply_to : old_from);
738
739         /* in case of ReplyAll, we need to add the Recipients in the old To: */
740         if (reply_mode == MODEST_TNY_MSG_REPLY_MODE_ALL) {
741                 const gchar *old_to = tny_header_get_to (header);
742                 if (!old_to) 
743                         g_warning ("%s: no To: address found in source mail",
744                                    __FUNCTION__);
745                 else {
746                         /* append the old To: */
747                         gchar *tmp = g_strjoin (",", new_to, old_to, NULL);
748                         g_free (new_to);
749                         new_to = tmp;
750                 }
751
752                 /* remove duplicate entries */
753                 gchar *tmp = modest_text_utils_remove_duplicate_addresses (new_to);
754                 g_free (new_to);
755                 new_to = tmp;
756                 
757                 /* now, strip me (the new From:) from the new_to, but only if
758                  * there are >1 addresses there */
759                 if (count_addresses (new_to) > 1) {
760                         gchar *tmp = modest_text_utils_remove_address (new_to, from);
761                         g_free (new_to);
762                         new_to = tmp;
763                 }
764         }
765
766         return new_to;
767 }
768
769
770 /* get the new Cc:, based on the old header,
771  * result is newly allocated or NULL in case of error */
772 static gchar*
773 get_new_cc (TnyHeader *header, const gchar* from)
774 {
775         const gchar *old_cc;
776
777         old_cc = tny_header_get_cc (header);
778         if (!old_cc)
779                 return NULL;
780
781         /* remove me (the new From:) from the Cc: list */
782         return modest_text_utils_remove_address (old_cc, from);
783 }
784
785
786 TnyMsg* 
787 modest_tny_msg_create_reply_msg (TnyMsg *msg,
788                                  TnyHeader *header,
789                                  const gchar *from,
790                                  const gchar *signature,
791                                  ModestTnyMsgReplyType reply_type,
792                                  ModestTnyMsgReplyMode reply_mode)
793 {
794         TnyMsg *new_msg = NULL;
795         TnyHeader *new_header;
796         gchar *new_to = NULL;
797         TnyList *parts = NULL;
798         GList *attachments_list = NULL;
799
800         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
801         
802         parts = TNY_LIST (tny_simple_list_new());
803         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
804         tny_list_foreach (parts, add_if_attachment, &attachments_list);
805
806         new_msg = create_reply_forward_mail (msg, header, from, signature, TRUE, reply_type,
807                                              attachments_list);
808         if (attachments_list != NULL) {
809                 g_list_foreach (attachments_list, (GFunc) g_object_unref, NULL);
810                 g_list_free (attachments_list);
811         }
812         if (parts)
813                 g_object_unref (G_OBJECT (parts));
814
815         /* Fill the header */
816         if (header)
817                 g_object_ref (header);
818         else
819                 header = tny_msg_get_header (msg);
820
821         
822         new_header = tny_msg_get_header(new_msg);
823         new_to = get_new_to (msg, header, from, reply_mode);
824         if (!new_to)
825                 g_warning ("%s: failed to get new To:", __FUNCTION__);
826         else {
827                 tny_header_set_to (new_header, new_to);
828                 g_free (new_to);
829         }       
830         
831         if (reply_mode == MODEST_TNY_MSG_REPLY_MODE_ALL) {
832                 gchar *new_cc = get_new_cc (header, from);
833                 if (new_cc) { 
834                         tny_header_set_cc (new_header, new_cc);
835                         g_free (new_cc);
836                 }
837         }
838         
839         /* Clean */
840         g_object_unref (G_OBJECT (new_header));
841         g_object_unref (G_OBJECT (header));
842
843         return new_msg;
844 }
845
846
847 static gboolean
848 is_ascii(const gchar *s)
849 {
850         if (!s)
851                 return TRUE;
852         while (s[0]) {
853                 if (s[0] & 128 || s[0] < 32)
854                         return FALSE;
855                 s++;
856         }
857         return TRUE;
858 }
859
860 static char *
861 get_content_type(const gchar *s)
862 {
863         GString *type;
864         
865         type = g_string_new("text/plain");
866         if (!is_ascii(s)) {
867                 if (g_utf8_validate(s, -1, NULL)) {
868                         g_string_append(type, "; charset=\"utf-8\"");
869                 } else {
870                         /* it should be impossible to reach this, but better safe than sorry */
871                         g_warning("invalid utf8 in message");
872                         g_string_append(type, "; charset=\"latin1\"");
873                 }
874         }
875         return g_string_free(type, FALSE);
876 }
877
878 guint64
879 modest_tny_msg_estimate_size (const gchar *plain_body, const gchar *html_body,
880                               guint64 parts_count,
881                               guint64 parts_size)
882 {
883         guint64 result;
884
885         /* estimation of headers size */
886         result = 1024;
887
888         /* We add a 20% of size due to the increase in 7bit encoding */
889         if (plain_body) {
890                 result += strlen (plain_body) * 120 / 100;
891         }
892         if (html_body) {
893                 result += strlen (html_body) * 120 / 100;
894         }
895
896         /* 256 bytes per additional part because of their headers */
897         result += parts_count * 256;
898
899         /* 150% of increase per encoding */
900         result += parts_size * 3 / 2;
901
902         return result;
903 }