Examples with LocalStore com.fsck.k9.mailstore.LocalStore used on opensource projects

Example 46 with LocalStore

use of com.fsck.k9.mailstore.LocalStore in project k-9 by k9mail.

the class StoreSchemaDefinitionTest method createStoreSchemaDefinition.

private StoreSchemaDefinition createStoreSchemaDefinition() throws MessagingException {
    Context context = createContext();
    Account account = createAccount();
    LockableDatabase lockableDatabase = createLockableDatabase();
    LocalStore localStore = mock(LocalStore.class);
    localStore.database = lockableDatabase;
    when(localStore.getContext()).thenReturn(context);
    when(localStore.getAccount()).thenReturn(account);
    return new StoreSchemaDefinition(localStore);
}

Example 47 with LocalStore

use of com.fsck.k9.mailstore.LocalStore in project k-9 by k9mail.

the class MessagingController method loadSearchResults.

public void loadSearchResults(final Account account, final String folderName, final List<Message> messages, final MessagingListener listener) {
    threadPool.execute(new Runnable() {

        @Override
        public void run() {
            if (listener != null) {
                listener.enableProgressIndicator(true);
            }
            try {
                Store remoteStore = account.getRemoteStore();
                LocalStore localStore = account.getLocalStore();
                if (remoteStore == null || localStore == null) {
                    throw new MessagingException("Could not get store");
                }
                Folder remoteFolder = remoteStore.getFolder(folderName);
                LocalFolder localFolder = localStore.getFolder(folderName);
                if (remoteFolder == null || localFolder == null) {
                    throw new MessagingException("Folder not found");
                }
                loadSearchResultsSynchronous(messages, localFolder, remoteFolder, listener);
            } catch (MessagingException e) {
                Timber.e(e, "Exception in loadSearchResults");
                addErrorMessage(account, null, e);
            } finally {
                if (listener != null) {
                    listener.enableProgressIndicator(false);
                }
            }
        }
    });
}

Example 48 with LocalStore

use of com.fsck.k9.mailstore.LocalStore in project k-9 by k9mail.

the class MessagingController method processPendingMarkAllAsRead.

void processPendingMarkAllAsRead(PendingMarkAllAsRead command, Account account) throws MessagingException {
    String folder = command.folder;
    Folder remoteFolder = null;
    LocalFolder localFolder = null;
    try {
        Store localStore = account.getLocalStore();
        localFolder = (LocalFolder) localStore.getFolder(folder);
        localFolder.open(Folder.OPEN_MODE_RW);
        List<? extends Message> messages = localFolder.getMessages(null, false);
        for (Message message : messages) {
            if (!message.isSet(Flag.SEEN)) {
                message.setFlag(Flag.SEEN, true);
            }
        }
        for (MessagingListener l : getListeners()) {
            l.folderStatusChanged(account, folder, 0);
        }
        if (account.getErrorFolderName().equals(folder)) {
            return;
        }
        Store remoteStore = account.getRemoteStore();
        remoteFolder = remoteStore.getFolder(folder);
        if (!remoteFolder.exists() || !remoteFolder.isFlagSupported(Flag.SEEN)) {
            return;
        }
        remoteFolder.open(Folder.OPEN_MODE_RW);
        if (remoteFolder.getMode() != Folder.OPEN_MODE_RW) {
            return;
        }
        remoteFolder.setFlags(Collections.singleton(Flag.SEEN), true);
        remoteFolder.close();
    } catch (UnsupportedOperationException uoe) {
        Timber.w(uoe, "Could not mark all server-side as read because store doesn't support operation");
    } finally {
        closeFolder(localFolder);
        closeFolder(remoteFolder);
    }
}

Example 49 with LocalStore

use of com.fsck.k9.mailstore.LocalStore in project k-9 by k9mail.

the class MessagingController method synchronizeMailboxSynchronous.

/**
     * Start foreground synchronization of the specified folder. This is generally only called
     * by synchronizeMailbox.
     * <p>
     * TODO Break this method up into smaller chunks.
     */
@VisibleForTesting
void synchronizeMailboxSynchronous(final Account account, final String folder, final MessagingListener listener, Folder providedRemoteFolder) {
    Folder remoteFolder = null;
    LocalFolder tLocalFolder = null;
    Timber.i("Synchronizing folder %s:%s", account.getDescription(), folder);
    for (MessagingListener l : getListeners(listener)) {
        l.synchronizeMailboxStarted(account, folder);
    }
    /*
         * We don't ever sync the Outbox or errors folder
         */
    if (folder.equals(account.getOutboxFolderName()) || folder.equals(account.getErrorFolderName())) {
        for (MessagingListener l : getListeners(listener)) {
            l.synchronizeMailboxFinished(account, folder, 0, 0);
        }
        return;
    }
    Exception commandException = null;
    try {
        Timber.d("SYNC: About to process pending commands for account %s", account.getDescription());
        try {
            processPendingCommandsSynchronous(account);
        } catch (Exception e) {
            addErrorMessage(account, null, e);
            Timber.e(e, "Failure processing command, but allow message sync attempt");
            commandException = e;
        }
        /*
             * Get the message list from the local store and create an index of
             * the uids within the list.
             */
        Timber.v("SYNC: About to get local folder %s", folder);
        final LocalStore localStore = account.getLocalStore();
        tLocalFolder = localStore.getFolder(folder);
        final LocalFolder localFolder = tLocalFolder;
        localFolder.open(Folder.OPEN_MODE_RW);
        localFolder.updateLastUid();
        Map<String, Long> localUidMap = localFolder.getAllMessagesAndEffectiveDates();
        if (providedRemoteFolder != null) {
            Timber.v("SYNC: using providedRemoteFolder %s", folder);
            remoteFolder = providedRemoteFolder;
        } else {
            Store remoteStore = account.getRemoteStore();
            Timber.v("SYNC: About to get remote folder %s", folder);
            remoteFolder = remoteStore.getFolder(folder);
            if (!verifyOrCreateRemoteSpecialFolder(account, folder, remoteFolder, listener)) {
                return;
            }
            /*
                 * Synchronization process:
                 *
                Open the folder
                Upload any local messages that are marked as PENDING_UPLOAD (Drafts, Sent, Trash)
                Get the message count
                Get the list of the newest K9.DEFAULT_VISIBLE_LIMIT messages
                getMessages(messageCount - K9.DEFAULT_VISIBLE_LIMIT, messageCount)
                See if we have each message locally, if not fetch it's flags and envelope
                Get and update the unread count for the folder
                Update the remote flags of any messages we have locally with an internal date newer than the remote message.
                Get the current flags for any messages we have locally but did not just download
                Update local flags
                For any message we have locally but not remotely, delete the local message to keep cache clean.
                Download larger parts of any new messages.
                (Optional) Download small attachments in the background.
                 */
            /*
                 * Open the remote folder. This pre-loads certain metadata like message count.
                 */
            Timber.v("SYNC: About to open remote folder %s", folder);
            remoteFolder.open(Folder.OPEN_MODE_RW);
            if (Expunge.EXPUNGE_ON_POLL == account.getExpungePolicy()) {
                Timber.d("SYNC: Expunging folder %s:%s", account.getDescription(), folder);
                remoteFolder.expunge();
            }
        }
        notificationController.clearAuthenticationErrorNotification(account, true);
        /*
             * Get the remote message count.
             */
        int remoteMessageCount = remoteFolder.getMessageCount();
        int visibleLimit = localFolder.getVisibleLimit();
        if (visibleLimit < 0) {
            visibleLimit = K9.DEFAULT_VISIBLE_LIMIT;
        }
        final List<Message> remoteMessages = new ArrayList<>();
        Map<String, Message> remoteUidMap = new HashMap<>();
        Timber.v("SYNC: Remote message count for folder %s is %d", folder, remoteMessageCount);
        final Date earliestDate = account.getEarliestPollDate();
        long earliestTimestamp = earliestDate != null ? earliestDate.getTime() : 0L;
        int remoteStart = 1;
        if (remoteMessageCount > 0) {
            /* Message numbers start at 1.  */
            if (visibleLimit > 0) {
                remoteStart = Math.max(0, remoteMessageCount - visibleLimit) + 1;
            } else {
                remoteStart = 1;
            }
            Timber.v("SYNC: About to get messages %d through %d for folder %s", remoteStart, remoteMessageCount, folder);
            final AtomicInteger headerProgress = new AtomicInteger(0);
            for (MessagingListener l : getListeners(listener)) {
                l.synchronizeMailboxHeadersStarted(account, folder);
            }
            List<? extends Message> remoteMessageArray = remoteFolder.getMessages(remoteStart, remoteMessageCount, earliestDate, null);
            int messageCount = remoteMessageArray.size();
            for (Message thisMess : remoteMessageArray) {
                headerProgress.incrementAndGet();
                for (MessagingListener l : getListeners(listener)) {
                    l.synchronizeMailboxHeadersProgress(account, folder, headerProgress.get(), messageCount);
                }
                Long localMessageTimestamp = localUidMap.get(thisMess.getUid());
                if (localMessageTimestamp == null || localMessageTimestamp >= earliestTimestamp) {
                    remoteMessages.add(thisMess);
                    remoteUidMap.put(thisMess.getUid(), thisMess);
                }
            }
            Timber.v("SYNC: Got %d messages for folder %s", remoteUidMap.size(), folder);
            for (MessagingListener l : getListeners(listener)) {
                l.synchronizeMailboxHeadersFinished(account, folder, headerProgress.get(), remoteUidMap.size());
            }
        } else if (remoteMessageCount < 0) {
            throw new Exception("Message count " + remoteMessageCount + " for folder " + folder);
        }
        /*
             * Remove any messages that are in the local store but no longer on the remote store or are too old
             */
        MoreMessages moreMessages = localFolder.getMoreMessages();
        if (account.syncRemoteDeletions()) {
            List<String> destroyMessageUids = new ArrayList<>();
            for (String localMessageUid : localUidMap.keySet()) {
                if (remoteUidMap.get(localMessageUid) == null) {
                    destroyMessageUids.add(localMessageUid);
                }
            }
            List<LocalMessage> destroyMessages = localFolder.getMessagesByUids(destroyMessageUids);
            if (!destroyMessageUids.isEmpty()) {
                moreMessages = MoreMessages.UNKNOWN;
                localFolder.destroyMessages(destroyMessages);
                for (Message destroyMessage : destroyMessages) {
                    for (MessagingListener l : getListeners(listener)) {
                        l.synchronizeMailboxRemovedMessage(account, folder, destroyMessage);
                    }
                }
            }
        }
        // noinspection UnusedAssignment, free memory early? (better break up the method!)
        localUidMap = null;
        if (moreMessages == MoreMessages.UNKNOWN) {
            updateMoreMessages(remoteFolder, localFolder, earliestDate, remoteStart);
        }
        /*
             * Now we download the actual content of messages.
             */
        int newMessages = downloadMessages(account, remoteFolder, localFolder, remoteMessages, false, true);
        int unreadMessageCount = localFolder.getUnreadMessageCount();
        for (MessagingListener l : getListeners()) {
            l.folderStatusChanged(account, folder, unreadMessageCount);
        }
        /* Notify listeners that we're finally done. */
        localFolder.setLastChecked(System.currentTimeMillis());
        localFolder.setStatus(null);
        Timber.d("Done synchronizing folder %s:%s @ %tc with %d new messages", account.getDescription(), folder, System.currentTimeMillis(), newMessages);
        for (MessagingListener l : getListeners(listener)) {
            l.synchronizeMailboxFinished(account, folder, remoteMessageCount, newMessages);
        }
        if (commandException != null) {
            String rootMessage = getRootCauseMessage(commandException);
            Timber.e("Root cause failure in %s:%s was '%s'", account.getDescription(), tLocalFolder.getName(), rootMessage);
            localFolder.setStatus(rootMessage);
            for (MessagingListener l : getListeners(listener)) {
                l.synchronizeMailboxFailed(account, folder, rootMessage);
            }
        }
        Timber.i("Done synchronizing folder %s:%s", account.getDescription(), folder);
    } catch (AuthenticationFailedException e) {
        handleAuthenticationFailure(account, true);
        for (MessagingListener l : getListeners(listener)) {
            l.synchronizeMailboxFailed(account, folder, "Authentication failure");
        }
    } catch (Exception e) {
        Timber.e(e, "synchronizeMailbox");
        // If we don't set the last checked, it can try too often during
        // failure conditions
        String rootMessage = getRootCauseMessage(e);
        if (tLocalFolder != null) {
            try {
                tLocalFolder.setStatus(rootMessage);
                tLocalFolder.setLastChecked(System.currentTimeMillis());
            } catch (MessagingException me) {
                Timber.e(e, "Could not set last checked on folder %s:%s", account.getDescription(), tLocalFolder.getName());
            }
        }
        for (MessagingListener l : getListeners(listener)) {
            l.synchronizeMailboxFailed(account, folder, rootMessage);
        }
        notifyUserIfCertificateProblem(account, e, true);
        addErrorMessage(account, null, e);
        Timber.e("Failed synchronizing folder %s:%s @ %tc", account.getDescription(), folder, System.currentTimeMillis());
    } finally {
        if (providedRemoteFolder == null) {
            closeFolder(remoteFolder);
        }
        closeFolder(tLocalFolder);
    }
}

Example 50 with LocalStore

use of com.fsck.k9.mailstore.LocalStore in project k-9 by k9mail.

the class MessagingController method deleteDraft.

public void deleteDraft(final Account account, long id) {
    LocalFolder localFolder = null;
    try {
        LocalStore localStore = account.getLocalStore();
        localFolder = localStore.getFolder(account.getDraftsFolderName());
        localFolder.open(Folder.OPEN_MODE_RW);
        String uid = localFolder.getMessageUidById(id);
        if (uid != null) {
            MessageReference messageReference = new MessageReference(account.getUuid(), account.getDraftsFolderName(), uid, null);
            deleteMessage(messageReference, null);
        }
    } catch (MessagingException me) {
        addErrorMessage(account, null, me);
    } finally {
        closeFolder(localFolder);
    }
}