1 /* 2 * Copyright (C) 2012 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.email.provider; 18 19 import android.annotation.TargetApi; 20 import android.content.ContentUris; 21 import android.content.ContentValues; 22 import android.content.Context; 23 import android.database.Cursor; 24 import android.net.Uri; 25 import android.os.ParcelFileDescriptor; 26 27 import com.android.email.LegacyConversions; 28 import com.android.emailcommon.Logging; 29 import com.android.emailcommon.internet.MimeUtility; 30 import com.android.emailcommon.mail.Message; 31 import com.android.emailcommon.mail.MessagingException; 32 import com.android.emailcommon.mail.Part; 33 import com.android.emailcommon.provider.Account; 34 import com.android.emailcommon.provider.EmailContent; 35 import com.android.emailcommon.provider.EmailContent.Attachment; 36 import com.android.emailcommon.provider.EmailContent.MessageColumns; 37 import com.android.emailcommon.provider.EmailContent.SyncColumns; 38 import com.android.emailcommon.provider.Mailbox; 39 import com.android.emailcommon.utility.ConversionUtilities; 40 import com.android.mail.utils.LogUtils; 41 import com.android.mail.utils.Utils; 42 43 import java.io.IOException; 44 import java.util.ArrayList; 45 46 public class Utilities { 47 /** 48 * Copy one downloaded message (which may have partially-loaded sections) 49 * into a newly created EmailProvider Message, given the account and mailbox 50 * 51 * @param message the remote message we've just downloaded 52 * @param account the account it will be stored into 53 * @param folder the mailbox it will be stored into 54 * @param loadStatus when complete, the message will be marked with this status (e.g. 55 * EmailContent.Message.LOADED) 56 */ copyOneMessageToProvider(Context context, Message message, Account account, Mailbox folder, int loadStatus)57 public static void copyOneMessageToProvider(Context context, Message message, Account account, 58 Mailbox folder, int loadStatus) { 59 EmailContent.Message localMessage = null; 60 Cursor c = null; 61 try { 62 c = context.getContentResolver().query( 63 EmailContent.Message.CONTENT_URI, 64 EmailContent.Message.CONTENT_PROJECTION, 65 EmailContent.MessageColumns.ACCOUNT_KEY + "=?" + 66 " AND " + MessageColumns.MAILBOX_KEY + "=?" + 67 " AND " + SyncColumns.SERVER_ID + "=?", 68 new String[] { 69 String.valueOf(account.mId), 70 String.valueOf(folder.mId), 71 String.valueOf(message.getUid()) 72 }, 73 null); 74 if (c == null) { 75 return; 76 } else if (c.moveToNext()) { 77 localMessage = EmailContent.getContent(context, c, EmailContent.Message.class); 78 } else { 79 localMessage = new EmailContent.Message(); 80 } 81 localMessage.mMailboxKey = folder.mId; 82 localMessage.mAccountKey = account.mId; 83 copyOneMessageToProvider(context, message, localMessage, loadStatus); 84 } finally { 85 if (c != null) { 86 c.close(); 87 } 88 } 89 } 90 91 /** 92 * Copy one downloaded message (which may have partially-loaded sections) 93 * into an already-created EmailProvider Message 94 * 95 * @param message the remote message we've just downloaded 96 * @param localMessage the EmailProvider Message, already created 97 * @param loadStatus when complete, the message will be marked with this status (e.g. 98 * EmailContent.Message.LOADED) 99 * @param context the context to be used for EmailProvider 100 */ copyOneMessageToProvider(Context context, Message message, EmailContent.Message localMessage, int loadStatus)101 public static void copyOneMessageToProvider(Context context, Message message, 102 EmailContent.Message localMessage, int loadStatus) { 103 try { 104 EmailContent.Body body = null; 105 if (localMessage.mId != EmailContent.Message.NO_MESSAGE) { 106 body = EmailContent.Body.restoreBodyWithMessageId(context, localMessage.mId); 107 } 108 if (body == null) { 109 body = new EmailContent.Body(); 110 } 111 try { 112 // Copy the fields that are available into the message object 113 LegacyConversions.updateMessageFields(localMessage, message, 114 localMessage.mAccountKey, localMessage.mMailboxKey); 115 116 // Now process body parts & attachments 117 ArrayList<Part> viewables = new ArrayList<Part>(); 118 ArrayList<Part> attachments = new ArrayList<Part>(); 119 MimeUtility.collectParts(message, viewables, attachments); 120 121 final ConversionUtilities.BodyFieldData data = 122 ConversionUtilities.parseBodyFields(viewables); 123 124 // set body and local message values 125 localMessage.setFlags(data.isQuotedReply, data.isQuotedForward); 126 localMessage.mSnippet = data.snippet; 127 body.mTextContent = data.textContent; 128 body.mHtmlContent = data.htmlContent; 129 130 // Commit the message & body to the local store immediately 131 saveOrUpdate(localMessage, context); 132 body.mMessageKey = localMessage.mId; 133 saveOrUpdate(body, context); 134 135 // process (and save) attachments 136 if (loadStatus != EmailContent.Message.FLAG_LOADED_PARTIAL 137 && loadStatus != EmailContent.Message.FLAG_LOADED_UNKNOWN) { 138 // TODO(pwestbro): What should happen with unknown status? 139 LegacyConversions.updateAttachments(context, localMessage, attachments); 140 LegacyConversions.updateInlineAttachments(context, localMessage, viewables); 141 } else { 142 EmailContent.Attachment att = new EmailContent.Attachment(); 143 // Since we haven't actually loaded the attachment, we're just putting 144 // a dummy placeholder here. When the user taps on it, we'll load the attachment 145 // for real. 146 // TODO: This is not a great way to model this. What we're saying is, we don't 147 // have the complete message, without paying any attention to what we do have. 148 // Did the main body exceed the maximum initial size? If so, we really might 149 // not have any attachments at all, and we just need a button somewhere that 150 // says "load the rest of the message". 151 // Or, what if we were able to load some, but not all of the attachments? 152 // Then we should ideally not be dropping the data we have on the floor. 153 // Also, what behavior we have here really should be based on what protocol 154 // we're dealing with. If it's POP, then we don't actually know how many 155 // attachments we have until we've loaded the complete message. 156 // If it's IMAP, we should know that, and we should load all attachment 157 // metadata we can get, regardless of whether or not we have the complete 158 // message body. 159 att.mFileName = ""; 160 att.mSize = message.getSize(); 161 att.mMimeType = "text/plain"; 162 att.mMessageKey = localMessage.mId; 163 att.mAccountKey = localMessage.mAccountKey; 164 att.mFlags = Attachment.FLAG_DUMMY_ATTACHMENT; 165 att.save(context); 166 localMessage.mFlagAttachment = true; 167 } 168 169 // One last update of message with two updated flags 170 localMessage.mFlagLoaded = loadStatus; 171 172 ContentValues cv = new ContentValues(); 173 cv.put(EmailContent.MessageColumns.FLAG_ATTACHMENT, localMessage.mFlagAttachment); 174 cv.put(EmailContent.MessageColumns.FLAG_LOADED, localMessage.mFlagLoaded); 175 Uri uri = ContentUris.withAppendedId(EmailContent.Message.CONTENT_URI, 176 localMessage.mId); 177 context.getContentResolver().update(uri, cv, null, null); 178 179 } catch (MessagingException me) { 180 LogUtils.e(Logging.LOG_TAG, "Error while copying downloaded message." + me); 181 } 182 183 } catch (RuntimeException rte) { 184 LogUtils.e(Logging.LOG_TAG, "Error while storing downloaded message." + rte.toString()); 185 } catch (IOException ioe) { 186 LogUtils.e(Logging.LOG_TAG, "Error while storing attachment." + ioe.toString()); 187 } 188 } 189 saveOrUpdate(EmailContent content, Context context)190 public static void saveOrUpdate(EmailContent content, Context context) { 191 if (content.isSaved()) { 192 content.update(context, content.toContentValues()); 193 } else { 194 content.save(context); 195 } 196 } 197 198 /** 199 * Converts a string representing a file mode, such as "rw", into a bitmask suitable for use 200 * with {@link android.os.ParcelFileDescriptor#open}. 201 * <p> 202 * @param mode The string representation of the file mode. 203 * @return A bitmask representing the given file mode. 204 * @throws IllegalArgumentException if the given string does not match a known file mode. 205 */ 206 @TargetApi(19) parseMode(String mode)207 public static int parseMode(String mode) { 208 if (Utils.isRunningKitkatOrLater()) { 209 return ParcelFileDescriptor.parseMode(mode); 210 } 211 final int modeBits; 212 if ("r".equals(mode)) { 213 modeBits = ParcelFileDescriptor.MODE_READ_ONLY; 214 } else if ("w".equals(mode) || "wt".equals(mode)) { 215 modeBits = ParcelFileDescriptor.MODE_WRITE_ONLY 216 | ParcelFileDescriptor.MODE_CREATE 217 | ParcelFileDescriptor.MODE_TRUNCATE; 218 } else if ("wa".equals(mode)) { 219 modeBits = ParcelFileDescriptor.MODE_WRITE_ONLY 220 | ParcelFileDescriptor.MODE_CREATE 221 | ParcelFileDescriptor.MODE_APPEND; 222 } else if ("rw".equals(mode)) { 223 modeBits = ParcelFileDescriptor.MODE_READ_WRITE 224 | ParcelFileDescriptor.MODE_CREATE; 225 } else if ("rwt".equals(mode)) { 226 modeBits = ParcelFileDescriptor.MODE_READ_WRITE 227 | ParcelFileDescriptor.MODE_CREATE 228 | ParcelFileDescriptor.MODE_TRUNCATE; 229 } else { 230 throw new IllegalArgumentException("Bad mode '" + mode + "'"); 231 } 232 return modeBits; 233 } 234 } 235