From 38e2fde7edbb9f1720528e565b09d8d4e9e7980d Mon Sep 17 00:00:00 2001
From: "Mike P. Sinn" <m@quantimodo.com>
Date: Tue, 15 Sep 2020 23:19:26 -0500
Subject: [PATCH] doneState

---
 src/js/controllers/reminderAddCtrl.js | 24 +++++++++++++++++-------
 1 file changed, 17 insertions(+), 7 deletions(-)

diff --git a/src/js/controllers/reminderAddCtrl.js b/src/js/controllers/reminderAddCtrl.js
index d4b1ff2f42..af24376ea6 100644
--- a/src/js/controllers/reminderAddCtrl.js
+++ b/src/js/controllers/reminderAddCtrl.js
@@ -99,6 +99,7 @@ angular.module('starter').controller('ReminderAddCtrl', ["$scope", "$state", "$s
                 setupEditReminder($scope.state.trackingReminder);
             }  // Needed to set dates
             qmLog.info("tracking reminder after setup: ", $scope.state.trackingReminder);
+            setTitle();
         });
         $scope.showMoreOptions = function(){
             $scope.state.showMoreOptions = true;
@@ -136,17 +137,22 @@ angular.module('starter').controller('ReminderAddCtrl', ["$scope", "$state", "$s
             }
         }
         $scope.oldOpenReminderStartTimePicker = function(order){
-            var defaultStartTimeInSecondsSinceMidnightLocal = qmService.getSecondsSinceMidnightLocalFromLocalString($rootScope.user.earliestReminderTime);
+            var defaultStartTimeInSecondsSinceMidnightLocal =
+                qmService.getSecondsSinceMidnightLocalFromLocalString($rootScope.user.earliestReminderTime);
             if(order === 'first' && $scope.state.firstReminderStartTimeLocal){
-                defaultStartTimeInSecondsSinceMidnightLocal = qmService.getSecondsSinceMidnightLocalFromLocalString($scope.state.firstReminderStartTimeLocal);
+                defaultStartTimeInSecondsSinceMidnightLocal =
+                    qmService.getSecondsSinceMidnightLocalFromLocalString($scope.state.firstReminderStartTimeLocal);
             }
             if(order === 'second' && $scope.state.secondReminderStartTimeLocal){
-                defaultStartTimeInSecondsSinceMidnightLocal = qmService.getSecondsSinceMidnightLocalFromLocalString($scope.state.secondReminderStartTimeLocal);
+                defaultStartTimeInSecondsSinceMidnightLocal =
+                    qmService.getSecondsSinceMidnightLocalFromLocalString($scope.state.secondReminderStartTimeLocal);
             }
             if(order === 'third' && $scope.state.thirdReminderStartTimeLocal){
-                defaultStartTimeInSecondsSinceMidnightLocal = qmService.getSecondsSinceMidnightLocalFromLocalString($scope.state.thirdReminderStartTimeLocal);
+                defaultStartTimeInSecondsSinceMidnightLocal =
+                    qmService.getSecondsSinceMidnightLocalFromLocalString($scope.state.thirdReminderStartTimeLocal);
             }
-            defaultStartTimeInSecondsSinceMidnightLocal = qmService.getSecondsSinceMidnightLocalRoundedToNearestFifteen(defaultStartTimeInSecondsSinceMidnightLocal);
+            defaultStartTimeInSecondsSinceMidnightLocal =
+                qmService.getSecondsSinceMidnightLocalRoundedToNearestFifteen(defaultStartTimeInSecondsSinceMidnightLocal);
             $scope.state.timePickerConfiguration = {
                 callback: function(val){
                     if(typeof (val) === 'undefined'){
@@ -356,7 +362,11 @@ angular.module('starter').controller('ReminderAddCtrl', ["$scope", "$state", "$s
             var toastMessage = getVariableName($scope) + ' saved';
             qmService.showInfoToast(toastMessage);
             qmService.hideLoader();
-            $scope.goBack(); // We can't go back until we get new notifications
+            if($stateParams.doneState){
+                qmService.goToState($stateParams.doneState, $stateParams);
+            } else {
+                $scope.goBack(); // We can't go back until we get new notifications
+            }
         };
         function getFrequencyNameFromFrequencySeconds(frequencyName){
             var reverseFrequencyChart = {
@@ -476,7 +486,7 @@ angular.module('starter').controller('ReminderAddCtrl', ["$scope", "$state", "$s
                 });
         }
         var setTitle = function(){
-            if($stateParams.favorite){
+            if($stateParams.favorite || $state.current.name === qm.stateNames.favoriteAdd){
                 $scope.state.selectedFrequencyName = 'As-Needed';
                 if($stateParams.reminder){
                     if(getVariableCategoryName() === 'Treatments'){