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-tny-platform-factory.h>
  36 #include <tny-camel-stream.h>
  37 #include <camel/camel-stream-mem.h>
  38
  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, gboolean has_attachments);
  49 static void add_attachments (TnyMsg *msg, GList *attachments_list);
  50 static char * get_content_type(const gchar *s);
  51 static gboolean is_ascii(const gchar *s);
  52
  53 TnyMsg*
  54 modest_tny_msg_new (const gchar* mailto, const gchar* from, const gchar *cc,
  55                     const gchar *bcc, const gchar* subject, const gchar *body,
  56                     GSList *attachments)
  57 {
  58         TnyPlatformFactory *fact;
  59         TnyMsg *new_msg;
  60         TnyHeader *header;
  61         gchar *content_type;
  62
  63         /* Create new msg */
  64         fact    = modest_tny_platform_factory_get_instance ();
  65         new_msg = tny_platform_factory_new_msg (fact);
  66         header  = tny_platform_factory_new_header (fact);
  67
  68         /* WARNING: set the header before assign values to it */
  69         tny_msg_set_header (new_msg, header);
  70         tny_header_set_from (TNY_HEADER (header), from);
  71         tny_header_set_replyto (TNY_HEADER (header), from);
  72         tny_header_set_to (TNY_HEADER (header), mailto);
  73         tny_header_set_cc (TNY_HEADER (header), cc);
  74         tny_header_set_bcc (TNY_HEADER (header), bcc);
  75         tny_header_set_subject (TNY_HEADER (header), subject);
  76
  77         content_type = get_content_type(body);
  78
  79
  80         /* Add the body of the new mail */
  81         add_body_part (new_msg, body, content_type, (attachments ? TRUE: FALSE));
  82
  83         /* Add attachments */
  84         add_attachments (new_msg, (GList*) attachments);
  85
  86         return new_msg;
  87 }
  88
  89
  90 /* FIXME: this func copy from modest-mail-operation: refactor */
  91 static TnyMimePart *
  92 add_body_part (TnyMsg *msg,
  93                const gchar *body,
  94                const gchar *content_type,
  95                gboolean has_attachments)
  96 {
  97         TnyMimePart *text_body_part = NULL;
  98         TnyStream *text_body_stream;
  99         TnyPlatformFactory *fact;
 100
 101         fact = modest_tny_platform_factory_get_instance ();
 102
 103         /* Create the stream */
 104         text_body_stream = TNY_STREAM (tny_camel_stream_new
 105                                        (camel_stream_mem_new_with_buffer
 106                                         (body, strlen(body))));
 107
 108         /* Create body part if needed */
 109         if (has_attachments)
 110                 text_body_part = tny_platform_factory_new_mime_part (fact);
 111         else
 112                 text_body_part = TNY_MIME_PART(msg);
 113
 114         /* Construct MIME part */
 115         tny_stream_reset (text_body_stream);
 116         tny_mime_part_construct_from_stream (text_body_part,
 117                                              text_body_stream,
 118                                              content_type);
 119         tny_stream_reset (text_body_stream);
 120
 121         /* Add part if needed */
 122         if (has_attachments) {
 123                 tny_mime_part_add_part (TNY_MIME_PART (msg), text_body_part);
 124                 g_object_unref (G_OBJECT(text_body_part));
 125         }
 126
 127         /* Clean */
 128         g_object_unref (text_body_stream);
 129
 130         return text_body_part;
 131 }
 132
 133 static void
 134 add_attachments (TnyMsg *msg, GList *attachments_list)
 135 {
 136         GList *pos;
 137         TnyMimePart *attachment_part, *old_attachment;
 138         const gchar *attachment_content_type;
 139         const gchar *attachment_filename;
 140         TnyStream *attachment_stream;
 141         TnyPlatformFactory *fact;
 142
 143         fact = modest_tny_platform_factory_get_instance ();
 144         for (pos = (GList *)attachments_list; pos; pos = pos->next) {
 145
 146                 old_attachment = pos->data;
 147                 attachment_filename = tny_mime_part_get_filename (old_attachment);
 148                 attachment_stream = tny_mime_part_get_stream (old_attachment);
 149                 attachment_part = tny_platform_factory_new_mime_part (fact);
 150
 151                 attachment_content_type = tny_mime_part_get_content_type (old_attachment);
 152
 153                 tny_mime_part_construct_from_stream (attachment_part,
 154                                                      attachment_stream,
 155                                                      attachment_content_type);
 156                 tny_stream_reset (attachment_stream);
 157
 158                 tny_mime_part_set_filename (attachment_part, attachment_filename);
 159
 160                 tny_mime_part_add_part (TNY_MIME_PART (msg), attachment_part);
 161 /*              g_object_unref (attachment_part); */
 162         }
 163 }
 164
 165
 166 gchar *
 167 modest_tny_msg_get_body (TnyMsg *msg, gboolean want_html)
 168 {
 169         TnyStream *stream;
 170         TnyMimePart *body;
 171         GtkTextBuffer *buf;
 172         GtkTextIter start, end;
 173         gchar *to_quote;
 174
 175         body = modest_tny_msg_find_body_part(msg, want_html);
 176         if (!body)
 177                 return NULL;
 178
 179         buf = gtk_text_buffer_new (NULL);
 180         stream = TNY_STREAM (tny_gtk_text_buffer_stream_new (buf));
 181         tny_stream_reset (stream);
 182         tny_mime_part_decode_to_stream (body, stream);
 183         tny_stream_reset (stream);
 184
 185         g_object_unref (G_OBJECT(stream));
 186         g_object_unref (G_OBJECT(body));
 187
 188         gtk_text_buffer_get_bounds (buf, &start, &end);
 189         to_quote = gtk_text_buffer_get_text (buf, &start, &end, FALSE);
 190         g_object_unref (buf);
 191
 192         return to_quote;
 193 }
 194
 195
 196 TnyMimePart*
 197 modest_tny_msg_find_body_part_from_mime_part (TnyMimePart *msg, gboolean want_html)
 198 {
 199         const gchar *mime_type = want_html ? "text/html" : "text/plain";
 200         TnyMimePart *part = NULL;
 201         TnyList *parts;
 202         TnyIterator *iter;
 203
 204         if (!msg)
 205                 return NULL;
 206
 207         parts = TNY_LIST (tny_simple_list_new());
 208         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
 209
 210         iter  = tny_list_create_iterator(parts);
 211
 212         /* no parts? assume it's single-part message */
 213         if (tny_iterator_is_done(iter))
 214                 return TNY_MIME_PART (g_object_ref(G_OBJECT(msg)));
 215         else {
 216                 gchar *content_type = NULL;
 217                 do {
 218                         part = TNY_MIME_PART(tny_iterator_get_current (iter));
 219
 220                         /* we need to strdown the content type, because
 221                          * tny_mime_part_has_content_type does not do it...
 222                          */
 223                         content_type = g_ascii_strdown
 224                                 (tny_mime_part_get_content_type (part), -1);
 225
 226                         if (g_str_has_prefix (content_type, mime_type) &&
 227                             !tny_mime_part_is_attachment (part))
 228                                 break;
 229
 230                         if (g_str_has_prefix(content_type, "multipart")) {
 231                                 part = modest_tny_msg_find_body_part_from_mime_part (part, want_html);
 232                                 if (part)
 233                                         break;
 234                         }
 235                         if (part)
 236                                 g_object_unref (G_OBJECT(part));
 237
 238                         part = NULL;
 239
 240                         g_free (content_type);
 241                         content_type = NULL;
 242
 243                         tny_iterator_next (iter);
 244
 245                 } while (!tny_iterator_is_done(iter));
 246                 g_free (content_type);
 247         }
 248
 249         g_object_unref (G_OBJECT(iter));
 250         g_object_unref (G_OBJECT(parts));
 251
 252         /* if were trying to find an HTML part and couldn't find it,
 253          * try to find a text/plain part instead
 254          */
 255         if (!part && want_html)
 256                 return modest_tny_msg_find_body_part_from_mime_part (msg, FALSE);
 257         else
 258                 return part; /* this maybe NULL, this is not an error; some message just don't have a body
 259                               * part */
 260 }
 261
 262
 263 TnyMimePart*
 264 modest_tny_msg_find_body_part (TnyMsg *msg, gboolean want_html)
 265 {
 266         return modest_tny_msg_find_body_part_from_mime_part (TNY_MIME_PART(msg),
 267                                                              want_html);
 268 }
 269
 270
 271
 272 static gboolean
 273 is_ascii(const gchar *s)
 274 {
 275         while (s[0]) {
 276                 if (s[0] & 128 || s[0] < 32)
 277                         return FALSE;
 278                 s++;
 279         }
 280         return TRUE;
 281 }
 282
 283 static char *
 284 get_content_type(const gchar *s)
 285 {
 286         GString *type;
 287
 288         type = g_string_new("text/plain");
 289         if (!is_ascii(s)) {
 290                 if (g_utf8_validate(s, -1, NULL)) {
 291                         g_string_append(type, "; charset=\"utf-8\"");
 292                 } else {
 293                         /* it should be impossible to reach this, but better safe than sorry */
 294                         g_warning("invalid utf8 in message");
 295                         g_string_append(type, "; charset=\"latin1\"");
 296                 }
 297         }
 298         return g_string_free(type, FALSE);
 299 }