mail/store/ExchangeStore.java

/*
 * Copyright (C) 2009 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.email.mail.store;

import com.android.email.Email;
import com.android.email.mail.AuthenticationFailedException;
import com.android.email.mail.FetchProfile;
import com.android.email.mail.Flag;
import com.android.email.mail.Folder;
import com.android.email.mail.Message;
import com.android.email.mail.MessageRetrievalListener;
import com.android.email.mail.MessagingException;
import com.android.email.mail.Store;
import com.android.email.mail.StoreSynchronizer;
import com.android.email.service.EmailServiceProxy;
import com.android.exchange.Eas;
import com.android.exchange.IEmailService;
import com.android.exchange.IEmailServiceCallback;
import com.android.exchange.SyncManager;
import com.android.exchange.EmailContent.Attachment;

import android.accounts.AccountManager;
import android.accounts.AuthenticatorException;
import android.accounts.Future2;
import android.accounts.Future2Callback;
import android.accounts.OperationCanceledException;
import android.content.Context;
import android.os.Bundle;
import android.os.IBinder;
import android.os.RemoteException;
import android.text.TextUtils;
import android.util.Config;
import android.util.Log;

import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.HashMap;

/**
 * This is a placeholder for use in Exchange implementations.  It is based on the notion of
 * lightweight adapter classes for Store, Folder, and Sender, and a common facade for the common
 * Transport code.
 */
public class ExchangeStore extends Store {
    public static final String LOG_TAG = "ExchangeStore";

    private final Context mContext;
    private URI mUri;
    private PersistentDataCallbacks mCallbacks;

    private final ExchangeTransport mTransport;
    private final HashMap<String, Folder> mFolders = new HashMap<String, Folder>();

    private boolean mPushModeRunning = false;

    /**
     * Factory method.
     */
    public static Store newInstance(String uri, Context context, PersistentDataCallbacks callbacks)
    throws MessagingException {
        return new ExchangeStore(uri, context, callbacks);
    }

    /**
     * eas://user:password@server/domain
     *
     * @param _uri
     * @param application
     * @throws MessagingException
     */
    private ExchangeStore(String _uri, Context context, PersistentDataCallbacks callbacks)
            throws MessagingException {
        mContext = context;
        try {
            mUri = new URI(_uri);
        } catch (URISyntaxException e) {
            throw new MessagingException("Invalid uri for ExchangeStore");
        }
        mCallbacks = callbacks;

        String scheme = mUri.getScheme();
        int connectionSecurity;
        if (scheme.equals("eas")) {
            connectionSecurity = ExchangeTransport.CONNECTION_SECURITY_NONE;
        } else if (scheme.equals("eas+ssl+")) {
            connectionSecurity = ExchangeTransport.CONNECTION_SECURITY_SSL_REQUIRED;
        } else {
            throw new MessagingException("Unsupported protocol");
        }

        mTransport = ExchangeTransport.getInstance(mUri, context);
    }

    /**
     * Retrieve the underlying transport.  Used primarily for testing.
     * @return
     */
    /* package */ ExchangeTransport getTransport() {
        return mTransport;
    }

    @Override
    public void checkSettings() throws MessagingException {
        mTransport.checkSettings(mUri);
    }

    @Override
    public Folder getFolder(String name) throws MessagingException {
        synchronized (mFolders) {
            Folder folder = mFolders.get(name);
            if (folder == null) {
                folder = new ExchangeFolder(this, name);
                mFolders.put(folder.getName(), folder);
            }
            return folder;
        }
    }

    @Override
    public Folder[] getPersonalNamespaces() throws MessagingException {
        return new Folder[] {
                getFolder(ExchangeTransport.FOLDER_INBOX),
        };
    }

    /**
     * For a store that supports push mode, this is the API that enables it or disables it.
     * The store should use this API to start or stop its persistent connection service or thread.
     *
     * <p>Note, may be called multiple times, even after push mode has been started or stopped.
     *
     * @param enablePushMode start or stop push mode delivery
     */
    @Override
    public void enablePushModeDelivery(boolean enablePushMode) {
        if (Config.LOGD && Email.DEBUG) {
            if (enablePushMode && !mPushModeRunning) {
                Log.d(Email.LOG_TAG, "start push mode");
            } else if (!enablePushMode && mPushModeRunning) {
                Log.d(Email.LOG_TAG, "stop push mode");
            } else {
                Log.d(Email.LOG_TAG, enablePushMode ?
                        "push mode already started" : "push mode already stopped");
            }
        }
        mPushModeRunning = enablePushMode;
    }

    /**
     * Get class of SettingActivity for this Store class.
     * @return Activity class that has class method actionEditIncomingSettings()
     */
    @Override
    public Class<? extends android.app.Activity> getSettingActivityClass() {
        return com.android.email.activity.setup.AccountSetupExchange.class;
    }

    /**
     * Get class of sync'er for this Store class.  Because exchange Sync rules are so different
     * than IMAP or POP3, it's likely that an Exchange implementation will need its own sync
     * controller.  If so, this function must return a non-null value.
     *
     * @return Message Sync controller, or null to use default
     */
    @Override
    public StoreSynchronizer getMessageSynchronizer() {
        return null;
    }

    /**
     * Inform MessagingController that this store requires message structures to be prefetched
     * before it can fetch message bodies (this is due to EAS protocol restrictions.)
     * @return always true for EAS
     */
    @Override
    public boolean requireStructurePrefetch() {
        return true;
    }

    /**
     * Inform MessagingController that messages sent via EAS will be placed in the Sent folder
     * automatically (server-side) and don't need to be uploaded.
     * @return always false for EAS (assuming server-side copy is supported)
     */
    @Override
    public boolean requireCopyMessageToSentFolder() {
        return false;
    }

    public static class ExchangeTransport {
        public static final int CONNECTION_SECURITY_NONE = 0;
        public static final int CONNECTION_SECURITY_SSL_REQUIRED = 1;

        public static final String FOLDER_INBOX = Email.INBOX;

        private static final String TAG = "ExchangeTransport";
        private final Context mContext;

        private String mHost;
        private String mDomain;
        private String mUsername;
        private String mPassword;

        private static HashMap<String, ExchangeTransport> sUriToInstanceMap =
            new HashMap<String, ExchangeTransport>();
        private static final HashMap<String, Integer> sFolderMap = new HashMap<String, Integer>();

        /**
         * Public factory.  The transport should be a singleton (per Uri)
         */
        public synchronized static ExchangeTransport getInstance(URI uri, Context context)
        throws MessagingException {
            if (!uri.getScheme().equals("eas") && !uri.getScheme().equals("eas+ssl+")) {
                throw new MessagingException("Invalid scheme");
            }

            final String key = uri.toString();
            ExchangeTransport transport = sUriToInstanceMap.get(key);
            if (transport == null) {
                transport = new ExchangeTransport(uri, context);
                sUriToInstanceMap.put(key, transport);
            }
            return transport;
        }

        /**
         * Private constructor - use public factory.
         */
        private ExchangeTransport(URI uri, Context context) throws MessagingException {
            mContext = context;
            setUri(uri);
        }

        /**
         * Use the Uri to set up a newly-constructed transport
         * @param uri
         * @throws MessagingException
         */
        private void setUri(final URI uri) throws MessagingException {
            mHost = uri.getHost();
            if (mHost == null) {
                throw new MessagingException("host not specified");
            }

            mDomain = uri.getPath();
            if (!TextUtils.isEmpty(mDomain)) {
                mDomain = mDomain.substring(1);
            }

            final String userInfo = uri.getUserInfo();
            if (userInfo == null) {
                throw new MessagingException("user information not specifed");
            }
            final String[] uinfo = userInfo.split(":", 2);
            if (uinfo.length != 2) {
                throw new MessagingException("user name and password not specified");
            }
            mUsername = uinfo[0];
            mPassword = uinfo[1];
        }

        /**
         * Blocking call that checks for a useable server connection, credentials, etc.
         * @param uri the server/account to try and connect to
         * @throws MessagingException thrown if the connection, server, account are not useable
         */

        IEmailServiceCallback mCallback = new IEmailServiceCallback () {

            public void status(int statusCode, int progress) throws RemoteException {
                Log.d("Status: ", "Code = " + statusCode + ", progress = " + progress);
            }

            public IBinder asBinder() { return null; }
        };

        /**
         * Here's where we check the settings for EAS.
         * @param uri the URI of the account to create
         * @throws MessagingException if we can't authenticate the account
         */
        public void checkSettings(URI uri) throws MessagingException {
            setUri(uri);
            boolean ssl = uri.getScheme().contains("ssl+");
            try {
                IEmailService svc = EmailServiceProxy.getService(mContext, SyncManager.class);
                int result = svc.validate("eas", mHost, mUsername, mPassword, ssl ? 443 : 80, ssl);
                if (result != MessagingException.NO_ERROR) {
                    if (result == MessagingException.AUTHENTICATION_FAILED) {
                        throw new AuthenticationFailedException("Authentication failed.");
                    } else {
                        throw new MessagingException(result);
                    }
                } else {
                    // This code was taken from sample code in AccountsTester
                    Bundle options = new Bundle();
                    options.putString("username", mUsername);
                    options.putString("password", mPassword);
                    Future2Callback callback = new Future2Callback() {
                        public void run(Future2 future) {
                            try {
                                Bundle bundle = future.getResult();
                                bundle.keySet();
                                Log.d(TAG, "account added: " + bundle);
                            } catch (OperationCanceledException e) {
                                Log.d(TAG, "addAccount was canceled");
                            } catch (IOException e) {
                                Log.d(TAG, "addAccount failed: " + e);
                            } catch (AuthenticatorException e) {
                                Log.d(TAG, "addAccount failed: " + e);
                            }

                        }
                    };
                    // Here's where we tell AccountManager about the new account.  The addAccount
                    // method in AccountManager calls the addAccount method in our authenticator
                    // service (EasAuthenticatorService)
                    AccountManager.get(mContext).addAccount(Eas.ACCOUNT_MANAGER_TYPE, null, null,
                            options, null, callback, null);
                 }
                svc.loadAttachment(0, new Attachment(), mCallback);
             } catch (RemoteException e) {
                throw new MessagingException("Call to validate generated an exception", e);
            }
        }

        /**
         * Typical helper function:  Return existence of a given folder
         */
        public boolean isFolderAvailable(final String folder) {
            return sFolderMap.containsKey(folder);
        }
    }

    public static class ExchangeFolder extends Folder {

        private final ExchangeTransport mTransport;
        @SuppressWarnings("unused")
        private final ExchangeStore mStore;
        @SuppressWarnings("unused")
        private final String mName;

        @SuppressWarnings("unused")
        private PersistentDataCallbacks mPersistenceCallbacks;

        public ExchangeFolder(ExchangeStore store, String name)
                throws MessagingException {
            mStore = store;
            mTransport = store.getTransport();
            mName = name;
            if (!mTransport.isFolderAvailable(name)) {
                throw new MessagingException("folder not supported: " + name);
            }
        }

        @Override
        public void appendMessages(Message[] messages) throws MessagingException {
            // TODO Implement this function
        }

        @Override
        public void close(boolean expunge) throws MessagingException {
            mPersistenceCallbacks = null;
            // TODO Implement this function
        }

        @Override
        public void copyMessages(Message[] msgs, Folder folder, MessageUpdateCallbacks callbacks)
                throws MessagingException {
            // TODO Implement this function
        }

        @Override
        public boolean create(FolderType type) throws MessagingException {
            // TODO Implement this function
            return false;
        }

        @Override
        public void delete(boolean recurse) throws MessagingException {
            // TODO Implement this function
        }

        @Override
        public boolean exists() throws MessagingException {
            // TODO Implement this function
            return false;
        }

        @Override
        public Message[] expunge() throws MessagingException {
            // TODO Implement this function
            return null;
        }

        @Override
        public void fetch(Message[] messages, FetchProfile fp, MessageRetrievalListener listener)
                throws MessagingException {
            // TODO Implement this function
        }

        @Override
        public Message getMessage(String uid) throws MessagingException {
            // TODO Implement this function
            return null;
        }

        @Override
        public int getMessageCount() throws MessagingException {
            // TODO Implement this function
            return 0;
        }

        @Override
        public Message[] getMessages(int start, int end, MessageRetrievalListener listener)
                throws MessagingException {
            // TODO Implement this function
            return null;
        }

        @Override
        public Message[] getMessages(MessageRetrievalListener listener) throws MessagingException {
            // TODO Implement this function
            return null;
        }

        @Override
        public Message[] getMessages(String[] uids, MessageRetrievalListener listener)
                throws MessagingException {
            // TODO Implement this function
            return null;
        }

        @Override
        public OpenMode getMode() throws MessagingException {
            // TODO Implement this function
            return null;
        }

        @Override
        public String getName() {
            // TODO Implement this function
            return null;
        }

        @Override
        public Flag[] getPermanentFlags() throws MessagingException {
            // TODO Implement this function
            return null;
        }

        @Override
        public int getUnreadMessageCount() throws MessagingException {
            // TODO Implement this function
            return 0;
        }

        @Override
        public boolean isOpen() {
            // TODO Implement this function
            return false;
        }

        @Override
        public void open(OpenMode mode, PersistentDataCallbacks callbacks)
                throws MessagingException {
            mPersistenceCallbacks = callbacks;
            // TODO Implement this function
        }

        @Override
        public void setFlags(Message[] messages, Flag[] flags, boolean value)
                throws MessagingException {
            // TODO Implement this function
        }


    }


}