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