1 files changed, 20 insertions, 6 deletions

diff --git a/src/components/alarm/AlarmController.h b/src/components/alarm/AlarmController.h
index 22259da8..140c9c80 100644
--- a/src/components/alarm/AlarmController.h
+++ b/src/components/alarm/AlarmController.h
@@ -1,3 +1,16 @@
+/*  Copyright (C) 2021 JF, Adam Pigg, Avamander
+    This file is part of InfiniTime.
+    InfiniTime is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published
+    by the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+    InfiniTime is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+*/
 #pragma once
 
 #include <cstdint>
@@ -13,16 +26,14 @@ namespace Pinetime {
     public:
       AlarmController(Controllers::DateTime& dateTimeController);
 
-      void Init();
+      void Init(System::SystemTask* systemTask);
       void SetAlarm(uint8_t alarmHr, uint8_t alarmMin);
       void DisableAlarm();
       void SetOffAlarmNow();
       uint32_t SecondsToAlarm();
       void StopAlerting();
-      void Register(System::SystemTask* systemTask);
       enum class AlarmState { Not_Set, Set, Alerting };
       enum class RecurType { None, Daily, Weekdays };
-      void ToggleRecurrence();
       uint8_t Hours() const {
         return hours;
       }
@@ -35,6 +46,9 @@ namespace Pinetime {
       RecurType Recurrence() const {
         return recurrence;
       }
+      void SetRecurrence(RecurType recurType) {
+        recurrence = recurType;
+      }
 
     private:
       Controllers::DateTime& dateTimeController;
@@ -42,9 +56,9 @@ namespace Pinetime {
       uint8_t hours;
       uint8_t minutes;
       std::chrono::time_point<std::chrono::system_clock, std::chrono::nanoseconds> alarmTime;
-      AlarmState state = AlarmState::Not_Set; 
+      AlarmState state = AlarmState::Not_Set;
       RecurType recurrence = RecurType::None;
-      void scheduleAlarm();
+      void ScheduleAlarm();
     };
   }
-}
\ No newline at end of file
+}