From 18d72d727c74a3f970e6c1b7c00b17ba234842fb Mon Sep 17 00:00:00 2001
From: Christian Basler <chrigu.meyer@gmail.com>
Date: Thu, 23 Feb 2017 17:37:15 +0100
Subject: [PATCH 1/3] Fixed notification title for when there are more than 5
 messages

---
 .../dissem/apps/abit/notification/NewMessageNotification.java | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/app/src/main/java/ch/dissem/apps/abit/notification/NewMessageNotification.java b/app/src/main/java/ch/dissem/apps/abit/notification/NewMessageNotification.java
index 06e4c02..3bddc2e 100644
--- a/app/src/main/java/ch/dissem/apps/abit/notification/NewMessageNotification.java
+++ b/app/src/main/java/ch/dissem/apps/abit/notification/NewMessageNotification.java
@@ -92,14 +92,12 @@ public class NewMessageNotification extends AbstractNotification {
         numberOfUnacknowledgedMessages) {
         NotificationCompat.Builder builder = new NotificationCompat.Builder(ctx);
         builder.setSmallIcon(R.drawable.ic_notification_new_message)
-            .setContentTitle(ctx.getString(R.string.n_new_messages, unacknowledged.size()))
+            .setContentTitle(ctx.getString(R.string.n_new_messages, numberOfUnacknowledgedMessages))
             .setContentText(ctx.getString(R.string.app_name));
 
         NotificationCompat.InboxStyle inboxStyle = new NotificationCompat.InboxStyle();
         //noinspection SynchronizationOnLocalVariableOrMethodParameter
         synchronized (unacknowledged) {
-            inboxStyle.setBigContentTitle(ctx.getString(R.string.n_new_messages,
-                numberOfUnacknowledgedMessages));
             for (Plaintext msg : unacknowledged) {
                 Spannable sb = new SpannableString(msg.getFrom() + " " + msg.getSubject());
                 sb.setSpan(SPAN_EMPHASIS, 0, String.valueOf(msg.getFrom()).length(), Spannable

From 65c03bd6380eeb2fb9d8e5c5446d09e0be300803 Mon Sep 17 00:00:00 2001
From: Christian Basler <chrigu.meyer@gmail.com>
Date: Thu, 23 Feb 2017 17:38:00 +0100
Subject: [PATCH 2/3] Load lists asynchronously

---
 .../ch/dissem/apps/abit/MainActivity.java     | 156 +++++++++++-------
 .../dissem/apps/abit/MessageListFragment.java |  63 +++++--
 .../ch/dissem/apps/abit/SettingsFragment.java |   8 +-
 3 files changed, 148 insertions(+), 79 deletions(-)

diff --git a/app/src/main/java/ch/dissem/apps/abit/MainActivity.java b/app/src/main/java/ch/dissem/apps/abit/MainActivity.java
index 812f6cf..7b98637 100644
--- a/app/src/main/java/ch/dissem/apps/abit/MainActivity.java
+++ b/app/src/main/java/ch/dissem/apps/abit/MainActivity.java
@@ -18,6 +18,7 @@ package ch.dissem.apps.abit;
 
 import android.content.Intent;
 import android.graphics.Point;
+import android.os.AsyncTask;
 import android.os.Bundle;
 import android.support.v4.app.Fragment;
 import android.support.v7.app.AppCompatActivity;
@@ -53,8 +54,9 @@ import org.slf4j.LoggerFactory;
 import java.io.Serializable;
 import java.lang.ref.WeakReference;
 import java.util.ArrayList;
-import java.util.Collection;
+import java.util.Collections;
 import java.util.List;
+import java.util.Objects;
 
 import ch.dissem.apps.abit.dialog.AddIdentityDialogFragment;
 import ch.dissem.apps.abit.dialog.FullNodeDialogActivity;
@@ -125,12 +127,6 @@ public class MainActivity extends AppCompatActivity
         super.onCreate(savedInstanceState);
         instance = new WeakReference<>(this);
         bmc = Singleton.getBitmessageContext(this);
-        List<Label> labels = bmc.messages().getLabels();
-        if (getIntent().hasExtra(EXTRA_SHOW_LABEL)) {
-            selectedLabel = (Label) getIntent().getSerializableExtra(EXTRA_SHOW_LABEL);
-        } else if (selectedLabel == null) {
-            selectedLabel = labels.get(0);
-        }
 
         setContentView(R.layout.activity_message_list);
 
@@ -155,7 +151,7 @@ public class MainActivity extends AppCompatActivity
             listFragment.setActivateOnItemClick(true);
         }
 
-        createDrawer(toolbar, labels);
+        createDrawer(toolbar);
 
         // handle intents
         Intent intent = getIntent();
@@ -217,22 +213,8 @@ public class MainActivity extends AppCompatActivity
         }
     }
 
-    private void createDrawer(Toolbar toolbar, Collection<Label> labels) {
+    private void createDrawer(Toolbar toolbar) {
         final ArrayList<IProfile> profiles = new ArrayList<>();
-        for (BitmessageAddress identity : bmc.addresses().getIdentities()) {
-            LOG.info("Adding identity " + identity.getAddress());
-            profiles.add(new ProfileDrawerItem()
-                .withIcon(new Identicon(identity))
-                .withName(identity.toString())
-                .withNameShown(true)
-                .withEmail(identity.getAddress())
-                .withTag(identity)
-            );
-        }
-        if (profiles.isEmpty()) {
-            // Create an initial identity
-            Singleton.getIdentity(this);
-        }
         profiles.add(new ProfileSettingDrawerItem()
             .withName(getString(R.string.add_identity))
             .withDescription(getString(R.string.add_identity_summary))
@@ -288,43 +270,7 @@ public class MainActivity extends AppCompatActivity
             accountHeader.setActiveProfile(profiles.get(0), true);
         }
 
-        ArrayList<IDrawerItem> drawerItems = new ArrayList<>();
-        for (Label label : labels) {
-            PrimaryDrawerItem item = new PrimaryDrawerItem()
-                .withName(label.toString())
-                .withTag(label);
-            if (label.getType() == null) {
-                item.withIcon(CommunityMaterial.Icon.cmd_label)
-                    .withIconColor(label.getColor());
-            } else {
-                switch (label.getType()) {
-                    case INBOX:
-                        item.withIcon(GoogleMaterial.Icon.gmd_inbox);
-                        break;
-                    case DRAFT:
-                        item.withIcon(CommunityMaterial.Icon.cmd_file);
-                        break;
-                    case OUTBOX:
-                        item.withIcon(CommunityMaterial.Icon.cmd_outbox);
-                        break;
-                    case SENT:
-                        item.withIcon(CommunityMaterial.Icon.cmd_send);
-                        break;
-                    case BROADCAST:
-                        item.withIcon(CommunityMaterial.Icon.cmd_rss);
-                        break;
-                    case UNREAD:
-                        item.withIcon(GoogleMaterial.Icon.gmd_markunread_mailbox);
-                        break;
-                    case TRASH:
-                        item.withIcon(GoogleMaterial.Icon.gmd_delete);
-                        break;
-                    default:
-                        item.withIcon(CommunityMaterial.Icon.cmd_label);
-                }
-            }
-            drawerItems.add(item);
-        }
+        final ArrayList<IDrawerItem> drawerItems = new ArrayList<>();
         drawerItems.add(new PrimaryDrawerItem()
             .withName(R.string.archive)
             .withTag(null)
@@ -397,6 +343,59 @@ public class MainActivity extends AppCompatActivity
             })
             .withShowDrawerOnFirstLaunch(true)
             .build();
+
+        new AsyncTask<Void, Void, List<BitmessageAddress>>() {
+            @Override
+            protected List<BitmessageAddress> doInBackground(Void... params) {
+                List<BitmessageAddress> identities = bmc.addresses().getIdentities();
+                if (identities.isEmpty()) {
+                    // Create an initial identity
+                    Singleton.getIdentity(MainActivity.this);
+                }
+                return identities;
+            }
+
+            @Override
+            protected void onPostExecute(List<BitmessageAddress> identities) {
+                for (BitmessageAddress identity : identities) {
+                    addIdentityEntry(identity);
+                }
+            }
+        }.execute();
+
+        new AsyncTask<Void, Void, List<Label>>() {
+            @Override
+            protected List<Label> doInBackground(Void... params) {
+                return bmc.messages().getLabels();
+            }
+
+            @Override
+            protected void onPostExecute(List<Label> labels) {
+                if (getIntent().hasExtra(EXTRA_SHOW_LABEL)) {
+                    selectedLabel = (Label) getIntent().getSerializableExtra(EXTRA_SHOW_LABEL);
+                } else if (selectedLabel == null) {
+                    selectedLabel = labels.get(0);
+                }
+                for (Label label : labels) {
+                    addLabelEntry(label);
+                }
+                showSelectedLabel();
+            }
+        }.execute();
+    }
+
+    @Override
+    protected void onSaveInstanceState(Bundle savedInstanceState) {
+        super.onSaveInstanceState(savedInstanceState);
+        savedInstanceState.putSerializable("selectedLabel", selectedLabel);
+    }
+
+    @Override
+    @SuppressWarnings("unchecked")
+    protected void onRestoreInstanceState(Bundle savedInstanceState) {
+        selectedLabel = (Label) savedInstanceState.getSerializable("selectedLabel");
+        showSelectedLabel();
+        super.onRestoreInstanceState(savedInstanceState);
     }
 
     private void addIdentityDialog() {
@@ -429,6 +428,43 @@ public class MainActivity extends AppCompatActivity
         }
     }
 
+    public void addLabelEntry(Label label) {
+        PrimaryDrawerItem item = new PrimaryDrawerItem()
+            .withName(label.toString())
+            .withTag(label);
+        if (label.getType() == null) {
+            item.withIcon(CommunityMaterial.Icon.cmd_label)
+                .withIconColor(label.getColor());
+        } else {
+            switch (label.getType()) {
+                case INBOX:
+                    item.withIcon(GoogleMaterial.Icon.gmd_inbox);
+                    break;
+                case DRAFT:
+                    item.withIcon(CommunityMaterial.Icon.cmd_file);
+                    break;
+                case OUTBOX:
+                    item.withIcon(CommunityMaterial.Icon.cmd_outbox);
+                    break;
+                case SENT:
+                    item.withIcon(CommunityMaterial.Icon.cmd_send);
+                    break;
+                case BROADCAST:
+                    item.withIcon(CommunityMaterial.Icon.cmd_rss);
+                    break;
+                case UNREAD:
+                    item.withIcon(GoogleMaterial.Icon.gmd_markunread_mailbox);
+                    break;
+                case TRASH:
+                    item.withIcon(GoogleMaterial.Icon.gmd_delete);
+                    break;
+                default:
+                    item.withIcon(CommunityMaterial.Icon.cmd_label);
+            }
+        }
+        drawer.addItemAtPosition(item, drawer.getDrawerItems().size() - 3);
+    }
+
     public void updateIdentityEntry(BitmessageAddress identity) {
         for (IProfile profile : accountHeader.getProfiles()) {
             if (profile instanceof ProfileDrawerItem) {
diff --git a/app/src/main/java/ch/dissem/apps/abit/MessageListFragment.java b/app/src/main/java/ch/dissem/apps/abit/MessageListFragment.java
index fbd2f00..55e91e7 100644
--- a/app/src/main/java/ch/dissem/apps/abit/MessageListFragment.java
+++ b/app/src/main/java/ch/dissem/apps/abit/MessageListFragment.java
@@ -17,6 +17,7 @@
 package ch.dissem.apps.abit;
 
 import android.content.Intent;
+import android.os.AsyncTask;
 import android.os.Bundle;
 import android.support.v4.app.Fragment;
 import android.support.v4.content.ContextCompat;
@@ -37,7 +38,9 @@ import com.h6ah4i.android.widget.advrecyclerview.swipeable.RecyclerViewSwipeMana
 import com.h6ah4i.android.widget.advrecyclerview.touchguard.RecyclerViewTouchActionGuardManager;
 import com.h6ah4i.android.widget.advrecyclerview.utils.WrapperAdapterUtils;
 
+import java.util.Collections;
 import java.util.List;
+import java.util.Objects;
 
 import ch.dissem.apps.abit.adapter.SwipeableMessageAdapter;
 import ch.dissem.apps.abit.listener.ActionBarListener;
@@ -97,20 +100,29 @@ public class MessageListFragment extends Fragment implements ListHolder {
         MainActivity activity = (MainActivity) getActivity();
         messageRepo = Singleton.getMessageRepository(activity);
 
-        doUpdateList(activity.getSelectedLabel());
+        currentLabel = activity.getSelectedLabel();
+        doUpdateList(currentLabel);
     }
 
     @Override
     public void updateList(Label label) {
-        currentLabel = label;
-
-        if (!isVisible()) return;
+        if (!isResumed()) {
+            currentLabel = label;
+            return;
+        }
 
+        if (!Objects.equals(currentLabel, label)) {
+            adapter.setData(label, Collections.<Plaintext>emptyList());
+            adapter.notifyDataSetChanged();
+        }
         doUpdateList(label);
     }
 
-    private void doUpdateList(Label label) {
-        List<Plaintext> messages = Singleton.getMessageRepository(getContext()).findMessages(label);
+    private void doUpdateList(final Label label) {
+        currentLabel = label;
+        if (emptyTrashMenuItem != null) {
+            emptyTrashMenuItem.setVisible(label != null && label.getType() == Label.Type.TRASH);
+        }
         if (getActivity() instanceof ActionBarListener) {
             if (label != null) {
                 ((ActionBarListener) getActivity()).updateTitle(label.toString());
@@ -118,11 +130,21 @@ public class MessageListFragment extends Fragment implements ListHolder {
                 ((ActionBarListener) getActivity()).updateTitle(getString(R.string.archive));
             }
         }
-        if (emptyTrashMenuItem != null) {
-            emptyTrashMenuItem.setVisible(label != null && label.getType() == Label.Type.TRASH);
-        }
-        adapter.setData(label, messages);
-        adapter.notifyDataSetChanged();
+        new AsyncTask<Void, Void, List<Plaintext>>() {
+
+            @Override
+            protected List<Plaintext> doInBackground(Void... params) {
+                return messageRepo.findMessages(label);
+            }
+
+            @Override
+            protected void onPostExecute(List<Plaintext> messages) {
+                if (adapter != null) {
+                    adapter.setData(label, messages);
+                    adapter.notifyDataSetChanged();
+                }
+            }
+        }.execute();
     }
 
     @Override
@@ -276,11 +298,20 @@ public class MessageListFragment extends Fragment implements ListHolder {
             case R.id.empty_trash:
                 if (currentLabel.getType() != Label.Type.TRASH) return true;
 
-                MessageRepository repo = Singleton.getMessageRepository(getContext());
-                for (Plaintext message : repo.findMessages(currentLabel)) {
-                    repo.remove(message);
-                }
-                updateList(currentLabel);
+                new AsyncTask<Void, Void, Void>() {
+                    @Override
+                    protected Void doInBackground(Void... params) {
+                        for (Plaintext message : messageRepo.findMessages(currentLabel)) {
+                            messageRepo.remove(message);
+                        }
+                        return null;
+                    }
+
+                    @Override
+                    protected void onPostExecute(Void aVoid) {
+                        updateList(currentLabel);
+                    }
+                }.execute();
                 return true;
             default:
                 return false;
diff --git a/app/src/main/java/ch/dissem/apps/abit/SettingsFragment.java b/app/src/main/java/ch/dissem/apps/abit/SettingsFragment.java
index 15fc6b0..6879025 100644
--- a/app/src/main/java/ch/dissem/apps/abit/SettingsFragment.java
+++ b/app/src/main/java/ch/dissem/apps/abit/SettingsFragment.java
@@ -67,23 +67,25 @@ public class SettingsFragment
             @Override
             public boolean onPreferenceClick(Preference preference) {
                 new AsyncTask<Void, Void, Void>() {
+                    private Context ctx = getActivity().getApplicationContext();
+
                     @Override
                     protected void onPreExecute() {
                         cleanup.setEnabled(false);
-                        Toast.makeText(getActivity(), R.string.cleanup_notification_start, Toast
+                        Toast.makeText(ctx, R.string.cleanup_notification_start, Toast
                             .LENGTH_SHORT).show();
                     }
 
                     @Override
                     protected Void doInBackground(Void... voids) {
-                        Singleton.getBitmessageContext(getActivity()).cleanup();
+                        Singleton.getBitmessageContext(ctx).cleanup();
                         return null;
                     }
 
                     @Override
                     protected void onPostExecute(Void aVoid) {
                         Toast.makeText(
-                            getActivity(),
+                            ctx,
                             R.string.cleanup_notification_end,
                             Toast.LENGTH_LONG
                         ).show();

From 300da1730d17f144a8d189f3c69eaa77866935be Mon Sep 17 00:00:00 2001
From: Christian Basler <chrigu.meyer@gmail.com>
Date: Fri, 24 Feb 2017 17:33:59 +0100
Subject: [PATCH 3/3] Archive is now handled somewhat differently, so we can
 distinguish between 'archive' and 'not initialized'.

---
 .../ch/dissem/apps/abit/MainActivity.java     |  9 ++------
 .../dissem/apps/abit/MessageListFragment.java | 20 ++++++++++++-----
 .../abit/adapter/SwipeableMessageAdapter.java |  5 +++--
 .../repository/AndroidMessageRepository.java  | 22 +++++++++++++++----
 4 files changed, 37 insertions(+), 19 deletions(-)

diff --git a/app/src/main/java/ch/dissem/apps/abit/MainActivity.java b/app/src/main/java/ch/dissem/apps/abit/MainActivity.java
index 7b98637..c994207 100644
--- a/app/src/main/java/ch/dissem/apps/abit/MainActivity.java
+++ b/app/src/main/java/ch/dissem/apps/abit/MainActivity.java
@@ -54,14 +54,13 @@ import org.slf4j.LoggerFactory;
 import java.io.Serializable;
 import java.lang.ref.WeakReference;
 import java.util.ArrayList;
-import java.util.Collections;
 import java.util.List;
-import java.util.Objects;
 
 import ch.dissem.apps.abit.dialog.AddIdentityDialogFragment;
 import ch.dissem.apps.abit.dialog.FullNodeDialogActivity;
 import ch.dissem.apps.abit.listener.ActionBarListener;
 import ch.dissem.apps.abit.listener.ListSelectionListener;
+import ch.dissem.apps.abit.repository.AndroidMessageRepository;
 import ch.dissem.apps.abit.service.BitmessageService;
 import ch.dissem.apps.abit.service.Singleton;
 import ch.dissem.apps.abit.synchronization.SyncAdapter;
@@ -273,7 +272,7 @@ public class MainActivity extends AppCompatActivity
         final ArrayList<IDrawerItem> drawerItems = new ArrayList<>();
         drawerItems.add(new PrimaryDrawerItem()
             .withName(R.string.archive)
-            .withTag(null)
+            .withTag(AndroidMessageRepository.LABEL_ARCHIVE)
             .withIcon(CommunityMaterial.Icon.cmd_archive)
         );
         drawerItems.add(new DividerDrawerItem());
@@ -330,10 +329,6 @@ public class MainActivity extends AppCompatActivity
                                 startActivity(new Intent(MainActivity.this, SettingsActivity
                                     .class));
                                 break;
-                            case R.string.archive:
-                                selectedLabel = null;
-                                showSelectedLabel();
-                                break;
                             case R.string.full_node:
                                 return true;
                         }
diff --git a/app/src/main/java/ch/dissem/apps/abit/MessageListFragment.java b/app/src/main/java/ch/dissem/apps/abit/MessageListFragment.java
index 55e91e7..f2bc0fb 100644
--- a/app/src/main/java/ch/dissem/apps/abit/MessageListFragment.java
+++ b/app/src/main/java/ch/dissem/apps/abit/MessageListFragment.java
@@ -100,8 +100,7 @@ public class MessageListFragment extends Fragment implements ListHolder {
         MainActivity activity = (MainActivity) getActivity();
         messageRepo = Singleton.getMessageRepository(activity);
 
-        currentLabel = activity.getSelectedLabel();
-        doUpdateList(currentLabel);
+        doUpdateList(activity.getSelectedLabel());
     }
 
     @Override
@@ -119,15 +118,24 @@ public class MessageListFragment extends Fragment implements ListHolder {
     }
 
     private void doUpdateList(final Label label) {
+        if (label == null) {
+            if (getActivity() instanceof ActionBarListener) {
+                ((ActionBarListener) getActivity()).updateTitle(getString(R.string.app_name));
+            }
+            adapter.setData(null, Collections.<Plaintext>emptyList());
+            adapter.notifyDataSetChanged();
+            return;
+        }
         currentLabel = label;
         if (emptyTrashMenuItem != null) {
-            emptyTrashMenuItem.setVisible(label != null && label.getType() == Label.Type.TRASH);
+            emptyTrashMenuItem.setVisible(label.getType() == Label.Type.TRASH);
         }
         if (getActivity() instanceof ActionBarListener) {
-            if (label != null) {
-                ((ActionBarListener) getActivity()).updateTitle(label.toString());
+            ActionBarListener actionBarListener = (ActionBarListener) getActivity();
+            if ("archive".equals(label.toString())) {
+                actionBarListener.updateTitle(getString(R.string.archive));
             } else {
-                ((ActionBarListener) getActivity()).updateTitle(getString(R.string.archive));
+                actionBarListener.updateTitle(label.toString());
             }
         }
         new AsyncTask<Void, Void, List<Plaintext>>() {
diff --git a/app/src/main/java/ch/dissem/apps/abit/adapter/SwipeableMessageAdapter.java b/app/src/main/java/ch/dissem/apps/abit/adapter/SwipeableMessageAdapter.java
index f50a3f6..7990c62 100644
--- a/app/src/main/java/ch/dissem/apps/abit/adapter/SwipeableMessageAdapter.java
+++ b/app/src/main/java/ch/dissem/apps/abit/adapter/SwipeableMessageAdapter.java
@@ -45,6 +45,7 @@ import ch.dissem.apps.abit.util.Assets;
 import ch.dissem.bitmessage.entity.Plaintext;
 import ch.dissem.bitmessage.entity.valueobject.Label;
 
+import static ch.dissem.apps.abit.repository.AndroidMessageRepository.LABEL_ARCHIVE;
 import static ch.dissem.apps.abit.util.Strings.normalizeWhitespaces;
 
 /**
@@ -199,7 +200,7 @@ public class SwipeableMessageAdapter
 
     @Override
     public int onGetSwipeReactionType(ViewHolder holder, int position, int x, int y) {
-        if (label == null || label.getType() == Label.Type.TRASH) {
+        if (label == LABEL_ARCHIVE || label.getType() == Label.Type.TRASH) {
             return REACTION_CAN_SWIPE_LEFT | REACTION_CAN_NOT_SWIPE_RIGHT_WITH_RUBBER_BAND_EFFECT;
         }
         return REACTION_CAN_SWIPE_BOTH_H;
@@ -217,7 +218,7 @@ public class SwipeableMessageAdapter
                 bgRes = R.drawable.bg_swipe_item_left;
                 break;
             case DRAWABLE_SWIPE_RIGHT_BACKGROUND:
-                if (label == null || label.getType() == Label.Type.TRASH) {
+                if (label == LABEL_ARCHIVE || label.getType() == Label.Type.TRASH) {
                     bgRes = R.drawable.bg_swipe_item_neutral;
                 } else {
                     bgRes = R.drawable.bg_swipe_item_right;
diff --git a/app/src/main/java/ch/dissem/apps/abit/repository/AndroidMessageRepository.java b/app/src/main/java/ch/dissem/apps/abit/repository/AndroidMessageRepository.java
index a1500f4..40bc9c6 100644
--- a/app/src/main/java/ch/dissem/apps/abit/repository/AndroidMessageRepository.java
+++ b/app/src/main/java/ch/dissem/apps/abit/repository/AndroidMessageRepository.java
@@ -48,6 +48,8 @@ import static java.lang.String.valueOf;
 public class AndroidMessageRepository extends AbstractMessageRepository {
     private static final Logger LOG = LoggerFactory.getLogger(AndroidMessageRepository.class);
 
+    public static final Label LABEL_ARCHIVE = new Label("archive", null, 0);
+
     private static final String TABLE_NAME = "Message";
     private static final String COLUMN_ID = "id";
     private static final String COLUMN_IV = "iv";
@@ -82,6 +84,15 @@ public class AndroidMessageRepository extends AbstractMessageRepository {
         this.context = ctx;
     }
 
+    @Override
+    public List<Plaintext> findMessages(Label label) {
+        if (label == LABEL_ARCHIVE) {
+            return super.findMessages((Label) null);
+        } else {
+            return super.findMessages(label);
+        }
+    }
+
     public List<Label> findLabels(String where) {
         List<Label> result = new LinkedList<>();
 
@@ -153,15 +164,18 @@ public class AndroidMessageRepository extends AbstractMessageRepository {
     public int countUnread(Label label) {
         String[] args;
         String where;
-        if (label != null) {
-            where = "id IN (SELECT message_id FROM Message_Label WHERE label_id=?) AND ";
+        if (label == null){
+            return 0;
+        }
+        if (label == LABEL_ARCHIVE) {
+            where = "";
             args = new String[]{
-                label.getId().toString(),
                 Label.Type.UNREAD.name()
             };
         } else {
-            where = "";
+            where = "id IN (SELECT message_id FROM Message_Label WHERE label_id=?) AND ";
             args = new String[]{
+                label.getId().toString(),
                 Label.Type.UNREAD.name()
             };
         }