Simplify MotorController timer code

1 files changed, 8 insertions, 1 deletions

diff --git a/src/components/motor/MotorController.h b/src/components/motor/MotorController.h
index 3c6cbd2..99a4e64 100644
--- a/src/components/motor/MotorController.h
+++ b/src/components/motor/MotorController.h
@@ -5,18 +5,25 @@
 #include <cstdint>
 
 namespace Pinetime {
+  namespace System {
+    class SystemTask;
+  }
   namespace Controllers {
 
     class MotorController {
     public:
       MotorController() = default;
 
-      void Init();
       void RunForDuration(uint8_t motorDuration);
       void StartRinging();
       void StopRinging();
 
+    protected:
+      friend class Pinetime::System::SystemTask;
+      void Init(System::SystemTask* systemTask);
+
     private:
+      System::SystemTask* systemTask = nullptr;
       static void Ring(TimerHandle_t xTimer);
       static void StopMotor(TimerHandle_t xTimer);
       TimerHandle_t shortVibTimer;