From c7b39eca114ac168e760586b113891f048c1906f Mon Sep 17 00:00:00 2001
From: Ruben Pollan <meskio@sindominio.net>
Date: Sun, 20 Oct 2013 18:25:01 +0200
Subject: [PATCH] Add deselect to batch selection mode

---
 .../thoughtcrime/securesms/ConversationListAdapter.java   | 8 ++++++--
 .../thoughtcrime/securesms/ConversationListFragment.java  | 4 ++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/org/thoughtcrime/securesms/ConversationListAdapter.java b/src/org/thoughtcrime/securesms/ConversationListAdapter.java
index 7349b04413..54ecc2ed6f 100644
--- a/src/org/thoughtcrime/securesms/ConversationListAdapter.java
+++ b/src/org/thoughtcrime/securesms/ConversationListAdapter.java
@@ -69,8 +69,12 @@ public class ConversationListAdapter extends CursorAdapter implements AbsListVie
     }
   }
 
-  public void addToBatchSet(long threadId) {
-    batchSet.add(threadId);
+  public void toggleThreadInBatchSet(long threadId) {
+    if (batchSet.contains(threadId)) {
+      batchSet.remove(threadId);
+    } else {
+      batchSet.add(threadId);
+    }
   }
 
   public Set<Long> getBatchSelections() {
diff --git a/src/org/thoughtcrime/securesms/ConversationListFragment.java b/src/org/thoughtcrime/securesms/ConversationListFragment.java
index 6ffa43d8a9..2e516279b5 100644
--- a/src/org/thoughtcrime/securesms/ConversationListFragment.java
+++ b/src/org/thoughtcrime/securesms/ConversationListFragment.java
@@ -106,7 +106,7 @@ public class ConversationListFragment extends SherlockListFragment
                                  headerView.getDistributionType());
       } else {
         ConversationListAdapter adapter = (ConversationListAdapter)getListAdapter();
-        adapter.addToBatchSet(headerView.getThreadId());
+        adapter.toggleThreadInBatchSet(headerView.getThreadId());
         adapter.notifyDataSetChanged();
       }
     }
@@ -146,7 +146,7 @@ public class ConversationListFragment extends SherlockListFragment
         batchMode  = true;
 
         adapter.initializeBatchMode(true);
-        adapter.addToBatchSet(((ConversationListItem) v).getThreadId());
+        adapter.toggleThreadInBatchSet(((ConversationListItem) v).getThreadId());
         adapter.notifyDataSetChanged();
 
         return true;