k-9/src/com/android/email/activity/ChooseFolder.java

package com.android.email.activity;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;

import android.app.ListActivity;
import android.content.Intent;
import android.os.Bundle;
import android.os.Handler;
import android.os.Process;
import android.util.Log;
import android.view.View;
import android.view.Window;
import android.widget.AdapterView;
import android.widget.ArrayAdapter;
import android.widget.ListView;
import android.widget.TextView;

import com.android.email.Account;
import com.android.email.Email;
import com.android.email.MessagingController;
import com.android.email.MessagingListener;
import com.android.email.Preferences;
import com.android.email.R;
import com.android.email.mail.Folder;
import com.android.email.mail.MessagingException;

public class ChooseFolder extends ListActivity
{
  String mFolder;
  Account mAccount;
  String	mUID;
  ArrayAdapter<String> adapter;
  private ChooseFolderHandler mHandler = new ChooseFolderHandler();
  String heldInbox = null;
  boolean hideCurrentFolder = true;

  public static final String EXTRA_ACCOUNT = "com.android.email.ChooseFolder_account";
  public static final String EXTRA_CUR_FOLDER = "com.android.email.ChooseFolder_curfolder";
  public static final String EXTRA_NEW_FOLDER = "com.android.email.ChooseFolder_newfolder";
  public static final String EXTRA_MESSAGE_UID = "com.android.email.ChooseFolder_messageuid";
  public static final String EXTRA_SHOW_CURRENT = "com.android.email.ChooseFolder_showcurrent";

  @Override
  public void onCreate(Bundle savedInstanceState)
  {
    super.onCreate(savedInstanceState);

    requestWindowFeature(Window.FEATURE_INDETERMINATE_PROGRESS);
    
    getListView().setTextFilterEnabled(true);
    getListView().setItemsCanFocus(false);
    getListView().setChoiceMode(ListView.CHOICE_MODE_NONE);
    Intent intent = getIntent();
    mAccount = (Account) intent.getSerializableExtra(EXTRA_ACCOUNT);
    mUID = intent.getStringExtra(EXTRA_MESSAGE_UID);
    mFolder = intent.getStringExtra(EXTRA_CUR_FOLDER);
    if (intent.getStringExtra(EXTRA_SHOW_CURRENT) != null) {
    	hideCurrentFolder = false;
    }
    if(mFolder == null)
      mFolder = "";
    
    adapter = new ArrayAdapter<String>(this, android.R.layout.simple_list_item_1);

    setListAdapter(adapter);

    new Thread()
    {
      public void run()
      {
        Process.setThreadPriority(Process.THREAD_PRIORITY_BACKGROUND);
        MessagingController.getInstance(getApplication()).listFolders(mAccount,
            false, mListener);

      }
    }.start();

    this.getListView().setOnItemClickListener(new AdapterView.OnItemClickListener() {  
      public void onItemClick(AdapterView adapterview, View view, int i, long l)
      {
        Intent intent = new Intent();
        intent.putExtra(EXTRA_CUR_FOLDER, mFolder);
        String destFolderName = (String)((TextView)view).getText();
        if (heldInbox != null && getString(R.string.special_mailbox_name_inbox).equals(destFolderName))
        {
          destFolderName = heldInbox;
        }
        
        intent.putExtra(EXTRA_NEW_FOLDER, destFolderName);
        
        intent.putExtra(EXTRA_MESSAGE_UID, mUID);
        setResult(RESULT_OK, intent);
        finish();
      }  
    });  

  }
  
  class ChooseFolderHandler extends Handler
  {

    private static final int MSG_PROGRESS = 2;

    private static final int MSG_DATA_CHANGED = 3;

    public void handleMessage(android.os.Message msg)
    {
      switch (msg.what)
      {
        case MSG_PROGRESS:
          setProgressBarIndeterminateVisibility(msg.arg1 != 0);
          break;
        case MSG_DATA_CHANGED:
          adapter.notifyDataSetChanged();
      }
    }

    public void progress(boolean progress)
    {
      android.os.Message msg = new android.os.Message();
      msg.what = MSG_PROGRESS;
      msg.arg1 = progress ? 1 : 0;
      sendMessage(msg);
    }

    public void dataChanged()
    {
      sendEmptyMessage(MSG_DATA_CHANGED);
    }
  }
  
  private MessagingListener mListener = new MessagingListener()
  {
    public void listFoldersStarted(Account account)
    {
      if (!account.equals(mAccount))
      {
        return;
      }
      mHandler.progress(true);
    }

    @Override
    public void listFoldersFailed(Account account, String message)
    {
      if (!account.equals(mAccount))
      {
        return;
      }
    mHandler.progress(false);
    }

    @Override
    public void listFoldersFinished(Account account)
    {
      if (!account.equals(mAccount))
      {
        return;
      }
      mHandler.progress(false);
    }
    @Override
    public void listFolders(Account account, Folder[] folders)
    {
      if (!account.equals(mAccount))
      {
        return;
      }
      Account.FolderMode aMode = account.getFolderTargetMode();
      Preferences prefs = Preferences.getPreferences(getApplication().getApplicationContext());
      ArrayList<String> localFolders = new ArrayList<String>();

      for (Folder folder : folders)
      {
        String name = folder.getName();

        // Inbox needs to be compared case-insensitively
        if(hideCurrentFolder && (name.equals(mFolder) || (Email.INBOX.equalsIgnoreCase(mFolder) && Email.INBOX.equalsIgnoreCase(name)))) {
          continue;
        }
        try
        {
          folder.refresh(prefs);
          Folder.FolderClass fMode = folder.getDisplayClass();

          if ((aMode == Account.FolderMode.FIRST_CLASS && fMode != Folder.FolderClass.FIRST_CLASS)
              || (aMode == Account.FolderMode.FIRST_AND_SECOND_CLASS &&
                  fMode != Folder.FolderClass.FIRST_CLASS &&
                  fMode != Folder.FolderClass.SECOND_CLASS)
                  || (aMode == Account.FolderMode.NOT_SECOND_CLASS && fMode == Folder.FolderClass.SECOND_CLASS))
          {
            continue;
          }
        }
        catch (MessagingException me)
        {
          Log.e(Email.LOG_TAG, "Couldn't get prefs to check for displayability of folder " + folder.getName(), me);
        }
        
        localFolders.add(folder.getName());
        
      }
      
      Collections.sort(localFolders, new Comparator<String>() {
        public int compare(String aName, String bName)
        {
          if (Email.INBOX.equalsIgnoreCase(aName))
          {
            return -1;
          }
          if (Email.INBOX.equalsIgnoreCase(bName))
          {
            return 1;
          }
          return aName.compareToIgnoreCase(bName); 
        }
      });
      adapter.setNotifyOnChange(false);
      adapter.clear();
      for (String name : localFolders) {
        if (Email.INBOX.equalsIgnoreCase(name))
        {
           adapter.add(getString(R.string.special_mailbox_name_inbox));
           heldInbox = name;
        }
        else {
          adapter.add(name);
        }
      }
      mHandler.dataChanged();
    }
  };
}