git.maemo.org Git - modest/blob - src/modest-tny-msg.c

   1 /* Copyright (c) 2006, Nokia Corporation
   2  * All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are
   6  * met:
   7  *
   8  * * Redistributions of source code must retain the above copyright
   9  *   notice, this list of conditions and the following disclaimer.
  10  * * Redistributions in binary form must reproduce the above copyright
  11  *   notice, this list of conditions and the following disclaimer in the
  12  *   documentation and/or other materials provided with the distribution.
  13  * * Neither the name of the Nokia Corporation nor the names of its
  14  *   contributors may be used to endorse or promote products derived from
  15  *   this software without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
  18  * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  19  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
  20  * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
  21  * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  22  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  23  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  24  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  25  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  26  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  27  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29
  30 #include <string.h>
  31 #include <gtkhtml/gtkhtml.h>
  32 #include <tny-gtk-text-buffer-stream.h>
  33 #include <tny-simple-list.h>
  34 #include <tny-folder.h>
  35 #include <modest-runtime.h>
  36 #include "modest-formatter.h"
  37 #include <tny-camel-stream.h>
  38 #include <camel/camel-stream-mem.h>
  39
  40 #ifdef HAVE_CONFIG_H
  41 #include <config.h>
  42 #endif /*HAVE_CONFIG_H */
  43
  44 #include <modest-tny-msg.h>
  45 #include "modest-text-utils.h"
  46
  47 static TnyMimePart * add_body_part (TnyMsg *msg, const gchar *body,
  48                                     const gchar *content_type);
  49 static TnyMimePart * add_html_body_part (TnyMsg *msg, const gchar *body);
  50 static void add_attachments (TnyMsg *msg, GList *attachments_list);
  51 static char * get_content_type(const gchar *s);
  52 static gboolean is_ascii(const gchar *s);
  53
  54 TnyMsg*
  55 modest_tny_msg_new (const gchar* mailto, const gchar* from, const gchar *cc,
  56                     const gchar *bcc, const gchar* subject, const gchar *body,
  57                     GSList *attachments)
  58 {
  59         TnyMsg *new_msg;
  60         TnyHeader *header;
  61         gchar *content_type;
  62
  63         /* Create new msg */
  64         new_msg = modest_formatter_create_message (NULL, TRUE, (attachments != NULL));
  65         header  = tny_msg_get_header (new_msg);
  66
  67         if ((from != NULL) && (strlen(from) > 0)) {
  68                 tny_header_set_from (TNY_HEADER (header), from);
  69                 tny_header_set_replyto (TNY_HEADER (header), from);
  70         }
  71         if ((mailto != NULL) && (strlen(mailto) > 0))
  72                 tny_header_set_to (TNY_HEADER (header), mailto);
  73         if ((cc != NULL) && (strlen(cc) > 0))
  74                 tny_header_set_cc (TNY_HEADER (header), cc);
  75         if ((bcc != NULL) && (strlen(bcc) > 0))
  76                 tny_header_set_bcc (TNY_HEADER (header), bcc);
  77
  78         if ((subject != NULL) && (strlen(subject) > 0))
  79                 tny_header_set_subject (TNY_HEADER (header), subject);
  80
  81         content_type = get_content_type(body);
  82
  83         /* Add the body of the new mail */
  84         /* This is needed even if body is NULL or empty. */
  85         add_body_part (new_msg, body, content_type);
  86         g_free (content_type);
  87
  88         /* Add attachments */
  89         if (attachments)
  90                 add_attachments (new_msg, (GList*) attachments);
  91
  92         return new_msg;
  93 }
  94
  95 TnyMsg*
  96 modest_tny_msg_new_html_plain (const gchar* mailto, const gchar* from, const gchar *cc,
  97                                const gchar *bcc, const gchar* subject,
  98                                const gchar *html_body, const gchar *plain_body,
  99                                GSList *attachments)
 100 {
 101         TnyMsg *new_msg;
 102         TnyHeader *header;
 103         gchar *content_type;
 104
 105         /* Create new msg */
 106         new_msg = modest_formatter_create_message (NULL, FALSE, (attachments != NULL));
 107         header  = tny_msg_get_header (new_msg);
 108
 109         if ((from != NULL) && (strlen(from) > 0)) {
 110                 tny_header_set_from (TNY_HEADER (header), from);
 111                 tny_header_set_replyto (TNY_HEADER (header), from);
 112         }
 113         if ((mailto != NULL) && (strlen(mailto) > 0))
 114                 tny_header_set_to (TNY_HEADER (header), mailto);
 115         if ((cc != NULL) && (strlen(cc) > 0))
 116                 tny_header_set_cc (TNY_HEADER (header), cc);
 117         if ((bcc != NULL) && (strlen(bcc) > 0))
 118                 tny_header_set_bcc (TNY_HEADER (header), bcc);
 119
 120         if ((subject != NULL) && (strlen(subject) > 0))
 121                 tny_header_set_subject (TNY_HEADER (header), subject);
 122
 123         content_type = get_content_type(plain_body);
 124
 125         /* Add the body of the new mail */
 126         add_body_part (new_msg, plain_body, content_type);
 127         add_html_body_part (new_msg, html_body);
 128         g_free (content_type);
 129
 130         /* Add attachments */
 131         add_attachments (new_msg, (GList*) attachments);
 132
 133         return new_msg;
 134 }
 135
 136
 137 /* FIXME: this func copy from modest-mail-operation: refactor */
 138 static TnyMimePart *
 139 add_body_part (TnyMsg *msg,
 140                const gchar *body,
 141                const gchar *content_type)
 142 {
 143         TnyMimePart *text_body_part = NULL;
 144         TnyStream *text_body_stream;
 145
 146         /* Create the stream */
 147         text_body_stream = TNY_STREAM (tny_camel_stream_new
 148                                        (camel_stream_mem_new_with_buffer
 149                                         (body, (body ? strlen(body) : 0))));
 150
 151         text_body_part = modest_formatter_create_body_part (NULL, msg);
 152
 153         /* Construct MIME part */
 154         tny_stream_reset (text_body_stream);
 155         tny_mime_part_construct_from_stream (text_body_part,
 156                                              text_body_stream,
 157                                              content_type);
 158         tny_stream_reset (text_body_stream);
 159
 160         g_object_unref (G_OBJECT(text_body_part));
 161
 162         /* Clean */
 163         g_object_unref (text_body_stream);
 164
 165         return text_body_part;
 166 }
 167
 168 static TnyMimePart *
 169 add_html_body_part (TnyMsg *msg,
 170                     const gchar *body)
 171 {
 172         TnyMimePart *html_body_part = NULL;
 173         TnyStream *html_body_stream;
 174
 175         /* Create the stream */
 176         html_body_stream = TNY_STREAM (tny_camel_stream_new
 177                                        (camel_stream_mem_new_with_buffer
 178                                         (body, strlen(body))));
 179
 180         /* Create body part if needed */
 181         html_body_part = modest_formatter_create_body_part (NULL, msg);
 182
 183         /* Construct MIME part */
 184         tny_stream_reset (html_body_stream);
 185         tny_mime_part_construct_from_stream (html_body_part,
 186                                              html_body_stream,
 187                                              "text/html; charset=utf-8");
 188         tny_stream_reset (html_body_stream);
 189
 190         g_object_unref (G_OBJECT(html_body_part));
 191
 192         /* Clean */
 193         g_object_unref (html_body_stream);
 194
 195         return html_body_part;
 196 }
 197
 198 static void
 199 add_attachments (TnyMsg *msg, GList *attachments_list)
 200 {
 201         GList *pos;
 202         TnyMimePart *attachment_part, *old_attachment;
 203         const gchar *attachment_content_type;
 204         const gchar *attachment_filename;
 205         const gchar *attachment_cid;
 206         TnyStream *attachment_stream;
 207
 208         for (pos = (GList *)attachments_list; pos; pos = pos->next) {
 209
 210                 old_attachment = pos->data;
 211                 attachment_filename = tny_mime_part_get_filename (old_attachment);
 212                 attachment_stream = tny_mime_part_get_stream (old_attachment);
 213                 if (TNY_IS_MSG (old_attachment)) {
 214                         attachment_part = TNY_MIME_PART (tny_platform_factory_new_msg (modest_runtime_get_platform_factory ()));
 215                 } else {
 216                         attachment_part = tny_platform_factory_new_mime_part (
 217                                 modest_runtime_get_platform_factory());
 218                 }
 219                 attachment_content_type = tny_mime_part_get_content_type (old_attachment);
 220                 attachment_cid = tny_mime_part_get_content_id (old_attachment);
 221
 222                 tny_stream_reset (attachment_stream);
 223                 tny_mime_part_construct_from_stream (attachment_part,
 224                                                      attachment_stream,
 225                                                      attachment_content_type);
 226                 tny_stream_reset (attachment_stream);
 227
 228                 tny_mime_part_set_filename (attachment_part, attachment_filename);
 229                 tny_mime_part_set_content_id (attachment_part, attachment_cid);
 230
 231                 tny_mime_part_add_part (TNY_MIME_PART (msg), attachment_part);
 232                 g_object_unref (attachment_stream);
 233 /*              g_object_unref (attachment_part); */
 234         }
 235 }
 236
 237
 238 gchar *
 239 modest_tny_msg_get_body (TnyMsg *msg, gboolean want_html)
 240 {
 241         TnyStream *stream;
 242         TnyMimePart *body;
 243         GtkTextBuffer *buf;
 244         GtkTextIter start, end;
 245         gchar *to_quote;
 246
 247         body = modest_tny_msg_find_body_part(msg, want_html);
 248         if (!body)
 249                 return NULL;
 250
 251         buf = gtk_text_buffer_new (NULL);
 252         stream = TNY_STREAM (tny_gtk_text_buffer_stream_new (buf));
 253         tny_stream_reset (stream);
 254         tny_mime_part_decode_to_stream (body, stream);
 255         tny_stream_reset (stream);
 256
 257         gtk_text_buffer_get_bounds (buf, &start, &end);
 258         to_quote = gtk_text_buffer_get_text (buf, &start, &end, FALSE);
 259         if (tny_mime_part_content_type_is (body, "text/plain")) {
 260                 gchar *to_quote_converted = modest_text_utils_convert_to_html (to_quote);
 261                 g_free (to_quote);
 262                 to_quote = to_quote_converted;
 263         }
 264
 265         g_object_unref (buf);
 266         g_object_unref (G_OBJECT(stream));
 267         g_object_unref (G_OBJECT(body));
 268
 269         return to_quote;
 270 }
 271
 272
 273 TnyMimePart*
 274 modest_tny_msg_find_body_part_from_mime_part (TnyMimePart *msg, gboolean want_html)
 275 {
 276         const gchar *mime_type = want_html ? "text/html" : "text/plain";
 277         TnyMimePart *part = NULL;
 278         TnyList *parts;
 279         TnyIterator *iter;
 280
 281         if (!msg)
 282                 return NULL;
 283
 284         parts = TNY_LIST (tny_simple_list_new());
 285         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
 286
 287         iter  = tny_list_create_iterator(parts);
 288
 289         /* no parts? assume it's single-part message */
 290         if (tny_iterator_is_done(iter))
 291                 return TNY_MIME_PART (g_object_ref(G_OBJECT(msg)));
 292         else {
 293                 gchar *content_type = NULL;
 294                 do {
 295                         part = TNY_MIME_PART(tny_iterator_get_current (iter));
 296
 297                         /* we need to strdown the content type, because
 298                          * tny_mime_part_has_content_type does not do it...
 299                          */
 300                         content_type = g_ascii_strdown
 301                                 (tny_mime_part_get_content_type (part), -1);
 302
 303                         if (g_str_has_prefix (content_type, mime_type) &&
 304                             !tny_mime_part_is_attachment (part))
 305                                 break;
 306
 307                         if (g_str_has_prefix(content_type, "multipart")) {
 308                                 part = modest_tny_msg_find_body_part_from_mime_part (part, want_html);
 309                                 if (part)
 310                                         break;
 311                         }
 312                         if (part)
 313                                 g_object_unref (G_OBJECT(part));
 314
 315                         part = NULL;
 316
 317                         g_free (content_type);
 318                         content_type = NULL;
 319
 320                         tny_iterator_next (iter);
 321
 322                 } while (!tny_iterator_is_done(iter));
 323                 g_free (content_type);
 324         }
 325
 326         g_object_unref (G_OBJECT(iter));
 327         g_object_unref (G_OBJECT(parts));
 328
 329         /* if were trying to find an HTML part and couldn't find it,
 330          * try to find a text/plain part instead
 331          */
 332         if (!part && want_html)
 333                 return modest_tny_msg_find_body_part_from_mime_part (msg, FALSE);
 334         else
 335                 return part; /* this maybe NULL, this is not an error; some message just don't have a body
 336                               * part */
 337 }
 338
 339
 340 TnyMimePart*
 341 modest_tny_msg_find_body_part (TnyMsg *msg, gboolean want_html)
 342 {
 343         return modest_tny_msg_find_body_part_from_mime_part (TNY_MIME_PART(msg),
 344                                                              want_html);
 345 }
 346
 347
 348 static TnyMsg *
 349 create_reply_forward_mail (TnyMsg *msg, const gchar *from, const gchar *signature,
 350                            gboolean is_reply, guint type, GList *attachments)
 351 {
 352         TnyMsg *new_msg;
 353         TnyHeader *new_header, *header;
 354         gchar *new_subject;
 355         TnyMimePart *body;
 356         ModestFormatter *formatter;
 357         gchar *subject_prefix;
 358
 359         /* Get body from original msg. Always look for the text/plain
 360            part of the message to create the reply/forwarded mail */
 361         header = tny_msg_get_header (msg);
 362         body   = modest_tny_msg_find_body_part (msg, FALSE);
 363
 364         /* TODO: select the formatter from account prefs */
 365         if (modest_conf_get_bool (modest_runtime_get_conf (), MODEST_CONF_PREFER_FORMATTED_TEXT, NULL))
 366                 formatter = modest_formatter_new ("text/html", signature);
 367         else
 368                 formatter = modest_formatter_new ("text/plain", signature);
 369
 370         /* Format message body */
 371         if (is_reply) {
 372                 switch (type) {
 373                 case MODEST_TNY_MSG_REPLY_TYPE_CITE:
 374                 default:
 375                         new_msg = modest_formatter_cite  (formatter, body, header);
 376                         break;
 377                 case MODEST_TNY_MSG_REPLY_TYPE_QUOTE:
 378                         new_msg = modest_formatter_quote (formatter, body, header, attachments);
 379                         break;
 380                 }
 381         } else {
 382                 switch (type) {
 383                 case MODEST_TNY_MSG_FORWARD_TYPE_INLINE:
 384                 default:
 385                         new_msg = modest_formatter_inline  (formatter, body, header, attachments);
 386                         break;
 387                 case MODEST_TNY_MSG_FORWARD_TYPE_ATTACHMENT:
 388                         new_msg = modest_formatter_attach (formatter, body, header);
 389                         break;
 390                 }
 391         }
 392         g_object_unref (G_OBJECT(formatter));
 393         g_object_unref (G_OBJECT(body));
 394
 395         /* Fill the header */
 396         new_header = tny_msg_get_header (new_msg);
 397         tny_header_set_from (new_header, from);
 398         tny_header_set_replyto (new_header, from);
 399
 400         /* Change the subject */
 401         if (is_reply)
 402                 subject_prefix = g_strconcat (_("mail_va_re"), ":", NULL);
 403         else
 404                 subject_prefix = g_strconcat (_("mail_va_fw"), ":", NULL);
 405         new_subject =
 406                 (gchar *) modest_text_utils_derived_subject (tny_header_get_subject(header),
 407                                                              subject_prefix);
 408         g_free (subject_prefix);
 409         tny_header_set_subject (new_header, (const gchar *) new_subject);
 410         g_free (new_subject);
 411
 412         /* Clean */
 413         g_object_unref (G_OBJECT (new_header));
 414         g_object_unref (G_OBJECT (header));
 415
 416         return new_msg;
 417 }
 418
 419 static void
 420 add_if_attachment (gpointer data, gpointer user_data)
 421 {
 422         TnyMimePart *part;
 423         GList **attachments_list;
 424
 425         part = TNY_MIME_PART (data);
 426         attachments_list = ((GList **) user_data);
 427
 428         if ((tny_mime_part_is_attachment (part))||(TNY_IS_MSG (part)))
 429                 *attachments_list = g_list_prepend (*attachments_list, part);
 430 }
 431
 432 TnyMsg*
 433 modest_tny_msg_create_forward_msg (TnyMsg *msg,
 434                                    const gchar *from,
 435                                    const gchar *signature,
 436                                    ModestTnyMsgForwardType forward_type)
 437 {
 438         TnyMsg *new_msg;
 439         TnyList *parts = NULL;
 440         GList *attachments_list = NULL;
 441
 442         /* Add attachments */
 443         parts = TNY_LIST (tny_simple_list_new());
 444         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
 445         tny_list_foreach (parts, add_if_attachment, &attachments_list);
 446
 447         new_msg = create_reply_forward_mail (msg, from, signature, FALSE, forward_type, attachments_list);
 448         add_attachments (new_msg, attachments_list);
 449
 450         /* Clean */
 451         if (attachments_list)
 452                 g_list_free (attachments_list);
 453         g_object_unref (G_OBJECT (parts));
 454
 455         return new_msg;
 456 }
 457
 458 TnyMsg*
 459 modest_tny_msg_create_reply_msg (TnyMsg *msg,
 460                                  const gchar *from,
 461                                  const gchar *signature,
 462                                  ModestTnyMsgReplyType reply_type,
 463                                  ModestTnyMsgReplyMode reply_mode)
 464 {
 465         TnyMsg *new_msg = NULL;
 466         TnyHeader *new_header, *header;
 467         const gchar* reply_to;
 468         gchar *new_cc = NULL;
 469         const gchar *cc = NULL, *bcc = NULL;
 470         GString *tmp = NULL;
 471         TnyList *parts = NULL;
 472         GList *attachments_list = NULL;
 473
 474         /* Add attachments */
 475         parts = TNY_LIST (tny_simple_list_new());
 476         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
 477         tny_list_foreach (parts, add_if_attachment, &attachments_list);
 478
 479         new_msg = create_reply_forward_mail (msg, from, signature, TRUE, reply_type, attachments_list);
 480         if (attachments_list) {
 481                 g_list_foreach (attachments_list, (GFunc) g_object_unref, NULL);
 482                 g_list_free (attachments_list);
 483         }
 484         g_object_unref (G_OBJECT (parts));
 485
 486         /* Fill the header */
 487         header = tny_msg_get_header (msg);
 488         new_header = tny_msg_get_header (new_msg);
 489         reply_to = tny_header_get_replyto (header);
 490
 491         if (reply_to)
 492                 tny_header_set_to (new_header, reply_to);
 493         else
 494                 tny_header_set_to (new_header, tny_header_get_from (header));
 495
 496         switch (reply_mode) {
 497         case MODEST_TNY_MSG_REPLY_MODE_SENDER:
 498                 /* Do not fill neither cc nor bcc */
 499                 break;
 500         case MODEST_TNY_MSG_REPLY_MODE_LIST:
 501                 /* TODO */
 502                 break;
 503         case MODEST_TNY_MSG_REPLY_MODE_ALL:
 504                 /* Concatenate to, cc and bcc */
 505                 cc = tny_header_get_cc (header);
 506                 bcc = tny_header_get_bcc (header);
 507
 508                 tmp = g_string_new (tny_header_get_to (header));
 509                 if (cc)  g_string_append_printf (tmp, ",%s",cc);
 510                 if (bcc) g_string_append_printf (tmp, ",%s",bcc);
 511
 512                /* Remove my own address from the cc list. TODO:
 513                   remove also the To: of the new message, needed due
 514                   to the new reply_to feature */
 515                 new_cc = (gchar *)
 516                         modest_text_utils_remove_address ((const gchar *) tmp->str,
 517                                                           from);
 518                 /* FIXME: remove also the mails from the new To: */
 519                 tny_header_set_cc (new_header, new_cc);
 520
 521                 /* Clean */
 522                 g_string_free (tmp, TRUE);
 523                 g_free (new_cc);
 524                 break;
 525         }
 526
 527         /* Clean */
 528         g_object_unref (G_OBJECT (new_header));
 529         g_object_unref (G_OBJECT (header));
 530
 531         return new_msg;
 532 }
 533
 534
 535 static gboolean
 536 is_ascii(const gchar *s)
 537 {
 538         if (!s)
 539                 return TRUE;
 540         while (s[0]) {
 541                 if (s[0] & 128 || s[0] < 32)
 542                         return FALSE;
 543                 s++;
 544         }
 545         return TRUE;
 546 }
 547
 548 static char *
 549 get_content_type(const gchar *s)
 550 {
 551         GString *type;
 552
 553         type = g_string_new("text/plain");
 554         if (!is_ascii(s)) {
 555                 if (g_utf8_validate(s, -1, NULL)) {
 556                         g_string_append(type, "; charset=\"utf-8\"");
 557                 } else {
 558                         /* it should be impossible to reach this, but better safe than sorry */
 559                         g_warning("invalid utf8 in message");
 560                         g_string_append(type, "; charset=\"latin1\"");
 561                 }
 562         }
 563         return g_string_free(type, FALSE);
 564 }