From 7f1553a4a1a19ba2543624da8954d5d1aae8c96f Mon Sep 17 00:00:00 2001
From: Alinson Xavier <axavier@uwaterloo.ca>
Date: Sat, 27 Feb 2016 14:09:02 -0500
Subject: [PATCH] Toggle checkmarks from widget

---
 app/src/main/AndroidManifest.xml              |  2 +-
 ...eiver.java => HabitBroadcastReceiver.java} | 53 ++++++++++++-------
 .../isoron/uhabits/SmallWidgetProvider.java   |  3 ++
 .../uhabits/helpers/ReminderHelper.java       |  8 +--
 .../java/org/isoron/uhabits/models/Habit.java |  6 +++
 5 files changed, 47 insertions(+), 25 deletions(-)
 rename app/src/main/java/org/isoron/uhabits/{ReminderAlarmReceiver.java => HabitBroadcastReceiver.java} (83%)

diff --git a/app/src/main/AndroidManifest.xml b/app/src/main/AndroidManifest.xml
index f1c20a8d3..e1191f8f5 100644
--- a/app/src/main/AndroidManifest.xml
+++ b/app/src/main/AndroidManifest.xml
@@ -39,7 +39,7 @@
         </activity>
 
         <receiver
-            android:name=".ReminderAlarmReceiver" />
+            android:name=".HabitBroadcastReceiver" />
 
         <activity
             android:name=".ShowHabitActivity"
diff --git a/app/src/main/java/org/isoron/uhabits/ReminderAlarmReceiver.java b/app/src/main/java/org/isoron/uhabits/HabitBroadcastReceiver.java
similarity index 83%
rename from app/src/main/java/org/isoron/uhabits/ReminderAlarmReceiver.java
rename to app/src/main/java/org/isoron/uhabits/HabitBroadcastReceiver.java
index 8dac22b31..bb08eadeb 100644
--- a/app/src/main/java/org/isoron/uhabits/ReminderAlarmReceiver.java
+++ b/app/src/main/java/org/isoron/uhabits/HabitBroadcastReceiver.java
@@ -38,12 +38,11 @@ import org.isoron.uhabits.models.Habit;
 
 import java.util.Date;
 
-public class ReminderAlarmReceiver extends BroadcastReceiver
+public class HabitBroadcastReceiver extends BroadcastReceiver
 {
     public static final String ACTION_CHECK = "org.isoron.uhabits.ACTION_CHECK";
     public static final String ACTION_DISMISS = "org.isoron.uhabits.ACTION_DISMISS";
-    public static final String ACTION_REMIND = "org.isoron.uhabits.ACTION_REMIND";
-    public static final String ACTION_REMOVE_REMINDER = "org.isoron.uhabits.ACTION_REMOVE_REMINDER";
+    public static final String ACTION_SHOW_REMINDER = "org.isoron.uhabits.ACTION_SHOW_REMINDER";
     public static final String ACTION_SNOOZE = "org.isoron.uhabits.ACTION_SNOOZE";
 
     @Override
@@ -51,7 +50,7 @@ public class ReminderAlarmReceiver extends BroadcastReceiver
     {
         switch (intent.getAction())
         {
-            case ACTION_REMIND:
+            case ACTION_SHOW_REMINDER:
                 createNotification(context, intent);
                 createReminderAlarms(context);
                 break;
@@ -148,21 +147,9 @@ public class ReminderAlarmReceiver extends BroadcastReceiver
         PendingIntent contentPendingIntent =
                 PendingIntent.getActivity(context, 0, contentIntent, 0);
 
-        Intent deleteIntent = new Intent(context, ReminderAlarmReceiver.class);
-        deleteIntent.setAction(ACTION_DISMISS);
-        PendingIntent deletePendingIntent = PendingIntent.getBroadcast(context, 0, deleteIntent, 0);
-
-        Intent checkIntent = new Intent(context, ReminderAlarmReceiver.class);
-        checkIntent.setData(data);
-        checkIntent.setAction(ACTION_CHECK);
-        checkIntent.putExtra("timestamp", timestamp);
-        PendingIntent checkIntentPending =
-                PendingIntent.getBroadcast(context, 0, checkIntent, PendingIntent.FLAG_ONE_SHOT);
-
-        Intent snoozeIntent = new Intent(context, ReminderAlarmReceiver.class);
-        snoozeIntent.setData(data);
-        snoozeIntent.setAction(ACTION_SNOOZE);
-        PendingIntent snoozeIntentPending = PendingIntent.getBroadcast(context, 0, snoozeIntent, 0);
+        PendingIntent dismissPendingIntent = buildDismissIntent(context);
+        PendingIntent checkIntentPending = buildCheckIntent(context, habit, timestamp);
+        PendingIntent snoozeIntentPending = buildSnoozeIntent(context, habit);
 
         Uri soundUri = RingtoneManager.getDefaultUri(RingtoneManager.TYPE_NOTIFICATION);
 
@@ -175,7 +162,7 @@ public class ReminderAlarmReceiver extends BroadcastReceiver
                         .setContentTitle(habit.name)
                         .setContentText(habit.description)
                         .setContentIntent(contentPendingIntent)
-                        .setDeleteIntent(deletePendingIntent)
+                        .setDeleteIntent(dismissPendingIntent)
                         .addAction(R.drawable.ic_action_check,
                                 context.getString(R.string.check), checkIntentPending)
                         .addAction(R.drawable.ic_action_snooze,
@@ -195,6 +182,32 @@ public class ReminderAlarmReceiver extends BroadcastReceiver
         notificationManager.notify(notificationId, notification);
     }
 
+    public static PendingIntent buildSnoozeIntent(Context context, Habit habit)
+    {
+        Uri data = habit.getUri();
+        Intent snoozeIntent = new Intent(context, HabitBroadcastReceiver.class);
+        snoozeIntent.setData(data);
+        snoozeIntent.setAction(ACTION_SNOOZE);
+        return PendingIntent.getBroadcast(context, 0, snoozeIntent, 0);
+    }
+
+    public static PendingIntent buildCheckIntent(Context context, Habit habit, Long timestamp)
+    {
+        Uri data = habit.getUri();
+        Intent checkIntent = new Intent(context, HabitBroadcastReceiver.class);
+        checkIntent.setData(data);
+        checkIntent.setAction(ACTION_CHECK);
+        if(timestamp != null) checkIntent.putExtra("timestamp", timestamp);
+        return PendingIntent.getBroadcast(context, 0, checkIntent, PendingIntent.FLAG_ONE_SHOT);
+    }
+
+    public static PendingIntent buildDismissIntent(Context context)
+    {
+        Intent deleteIntent = new Intent(context, HabitBroadcastReceiver.class);
+        deleteIntent.setAction(ACTION_DISMISS);
+        return PendingIntent.getBroadcast(context, 0, deleteIntent, 0);
+    }
+
     private boolean checkWeekday(Intent intent, Habit habit)
     {
         Long timestamp = intent.getLongExtra("timestamp", DateHelper.getStartOfToday());
diff --git a/app/src/main/java/org/isoron/uhabits/SmallWidgetProvider.java b/app/src/main/java/org/isoron/uhabits/SmallWidgetProvider.java
index 3a2466b8e..0569bebfb 100644
--- a/app/src/main/java/org/isoron/uhabits/SmallWidgetProvider.java
+++ b/app/src/main/java/org/isoron/uhabits/SmallWidgetProvider.java
@@ -64,6 +64,9 @@ public class SmallWidgetProvider extends AppWidgetProvider
 ;
         remoteViews.setTextViewText(R.id.tvName, habit.name);
         remoteViews.setImageViewBitmap(R.id.imageView, drawingCache);
+        remoteViews.setOnClickPendingIntent(R.id.imageView,
+                HabitBroadcastReceiver.buildCheckIntent(context, habit, null));
         manager.updateAppWidget(widgetId, remoteViews);
     }
+
 }
diff --git a/app/src/main/java/org/isoron/uhabits/helpers/ReminderHelper.java b/app/src/main/java/org/isoron/uhabits/helpers/ReminderHelper.java
index 94d10e5f2..cbc1bd9c1 100644
--- a/app/src/main/java/org/isoron/uhabits/helpers/ReminderHelper.java
+++ b/app/src/main/java/org/isoron/uhabits/helpers/ReminderHelper.java
@@ -25,7 +25,7 @@ import android.os.Build;
 import android.util.Log;
 
 import org.isoron.helpers.DateHelper;
-import org.isoron.uhabits.ReminderAlarmReceiver;
+import org.isoron.uhabits.HabitBroadcastReceiver;
 import org.isoron.uhabits.models.Habit;
 
 import java.text.DateFormat;
@@ -58,10 +58,10 @@ public class ReminderHelper
 
         long timestamp = DateHelper.getStartOfDay(DateHelper.toLocalTime(reminderTime));
 
-        Uri uri = Uri.parse(String.format("content://org.isoron.uhabits/habit/%d", habit.getId()));
+        Uri uri = habit.getUri();
 
-        Intent alarmIntent = new Intent(context, ReminderAlarmReceiver.class);
-        alarmIntent.setAction(ReminderAlarmReceiver.ACTION_REMIND);
+        Intent alarmIntent = new Intent(context, HabitBroadcastReceiver.class);
+        alarmIntent.setAction(HabitBroadcastReceiver.ACTION_SHOW_REMINDER);
         alarmIntent.setData(uri);
         alarmIntent.putExtra("timestamp", timestamp);
         alarmIntent.putExtra("reminderTime", reminderTime);
diff --git a/app/src/main/java/org/isoron/uhabits/models/Habit.java b/app/src/main/java/org/isoron/uhabits/models/Habit.java
index 9683b8b30..bad975652 100644
--- a/app/src/main/java/org/isoron/uhabits/models/Habit.java
+++ b/app/src/main/java/org/isoron/uhabits/models/Habit.java
@@ -19,6 +19,7 @@ package org.isoron.uhabits.models;
 import android.annotation.SuppressLint;
 import android.database.Cursor;
 import android.database.sqlite.SQLiteDatabase;
+import android.net.Uri;
 
 import com.activeandroid.ActiveAndroid;
 import com.activeandroid.Cache;
@@ -442,6 +443,11 @@ public class Habit extends Model
         deleteStreaksNewerThan(timestamp);
     }
 
+    public Uri getUri()
+    {
+        return Uri.parse(String.format("content://org.isoron.uhabits/habit/%d", getId()));
+    }
+
     public void archive()
     {
         archived = 1;