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