From ea47169f96abac7b5498627e7333370d45425c5c Mon Sep 17 00:00:00 2001
From: Alex Baker <baker.alex@gmail.com>
Date: Fri, 17 Apr 2015 16:49:10 -0500
Subject: [PATCH] Register/deregister geofences on completion change

---
 src/main/java/com/todoroo/astrid/dao/TaskDao.java | 15 ++++++++++++---
 .../java/org/tasks/location/GeofenceService.java  | 14 ++++++++++----
 2 files changed, 22 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/todoroo/astrid/dao/TaskDao.java b/src/main/java/com/todoroo/astrid/dao/TaskDao.java
index f4da51569..089662c28 100644
--- a/src/main/java/com/todoroo/astrid/dao/TaskDao.java
+++ b/src/main/java/com/todoroo/astrid/dao/TaskDao.java
@@ -26,6 +26,7 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.tasks.Broadcaster;
 import org.tasks.R;
+import org.tasks.location.GeofenceService;
 import org.tasks.notifications.NotificationManager;
 import org.tasks.preferences.Preferences;
 
@@ -52,11 +53,13 @@ public class TaskDao {
     private final ReminderService reminderService;
     private final NotificationManager notificationManager;
     private final Preferences preferences;
+    private GeofenceService geofenceService;
 
     @Inject
 	public TaskDao(Database database, MetadataDao metadataDao, Broadcaster broadcaster,
                    ReminderService reminderService, NotificationManager notificationManager,
-                   Preferences preferences) {
+                   Preferences preferences, GeofenceService geofenceService) {
+        this.geofenceService = geofenceService;
         dao = new RemoteModelDao<>(database, Task.class);
         this.preferences = preferences;
         this.metadataDao = metadataDao;
@@ -355,9 +358,13 @@ public class TaskDao {
         }
 
         task.markSaved();
-        if(values.containsKey(Task.COMPLETION_DATE.name) && task.isCompleted()) {
+        boolean completionDateModified = values.containsKey(Task.COMPLETION_DATE.name);
+        if(completionDateModified && task.isCompleted()) {
             afterComplete(task);
         } else {
+            if (completionDateModified) {
+                geofenceService.setupGeofences(task.getId());
+            }
             if(values.containsKey(Task.DUE_DATE.name) ||
                     values.containsKey(Task.REMINDER_FLAGS.name) ||
                     values.containsKey(Task.REMINDER_PERIOD.name) ||
@@ -399,7 +406,9 @@ public class TaskDao {
      * Called after the task was just completed
      */
     private void afterComplete(Task task) {
-        notificationManager.cancel((int) task.getId());
+        long taskId = task.getId();
+        notificationManager.cancel(taskId);
+        geofenceService.cancelGeofences(taskId);
     }
 }
 
diff --git a/src/main/java/org/tasks/location/GeofenceService.java b/src/main/java/org/tasks/location/GeofenceService.java
index 5326a472f..a2ae9569b 100644
--- a/src/main/java/org/tasks/location/GeofenceService.java
+++ b/src/main/java/org/tasks/location/GeofenceService.java
@@ -53,6 +53,16 @@ public class GeofenceService {
         geofenceApi.register(getActiveGeofences());
     }
 
+    public void setupGeofences(long taskId) {
+        geofenceApi.register(getGeofencesForTask(taskId));
+    }
+
+    public void cancelGeofences(long taskId) {
+        for (Geofence geofence : getGeofences(taskId)) {
+            geofenceApi.cancel(geofence);
+        }
+    }
+
     public boolean synchronizeGeofences(final long taskId, Set<Geofence> geofences) {
         List<Metadata> metadata = newArrayList(transform(geofences, new Function<Geofence, Metadata>() {
             @Override
@@ -89,10 +99,6 @@ public class GeofenceService {
         }));
     }
 
-    private void setupGeofences(long taskId) {
-        geofenceApi.register(getGeofencesForTask(taskId));
-    }
-
     private List<Geofence> getActiveGeofences() {
         return toGeofences(metadataDao.toList(Query.select(Metadata.ID, Metadata.TASK, GeofenceFields.PLACE, GeofenceFields.LATITUDE, GeofenceFields.LONGITUDE, GeofenceFields.RADIUS).
                 join(Join.inner(Task.TABLE, Metadata.TASK.eq(Task.ID))).