1 files changed, 7 insertions, 2 deletions
diff --git a/src/Components/Ble/NimbleController.h b/src/Components/Ble/NimbleController.h
index 2a39694..c4eab52 100644
--- a/src/Components/Ble/NimbleController.h
+++ b/src/Components/Ble/NimbleController.h
@@ -1,7 +1,7 @@
 #pragma once
 
 #include <cstdint>
-
+#include "AlertNotificationClient.h"
 #include "DeviceInformationService.h"
 #include "CurrentTimeClient.h"
 #include <host/ble_gap.h>
@@ -11,16 +11,21 @@ namespace Pinetime {
     class DateTime;
     class NimbleController {
       public:
-        NimbleController(DateTime& dateTimeController);
+        NimbleController(Pinetime::System::SystemTask& systemTask, DateTime& dateTimeController, Pinetime::Controllers::NotificationManager& notificationManager);
         void Init();
         void StartAdvertising();
         int OnGAPEvent(ble_gap_event *event);
 
+        int OnDiscoveryEvent(uint16_t i, const ble_gatt_error *pError, const ble_gatt_svc *pSvc);
       private:
         static constexpr char* deviceName = "Pinetime-JF";
+        Pinetime::System::SystemTask& systemTask;
         DateTime& dateTimeController;
+        Pinetime::Controllers::NotificationManager& notificationManager;
+
         DeviceInformationService deviceInformationService;
         CurrentTimeClient currentTimeClient;
+        AlertNotificationClient alertNotificationClient;
         uint8_t addrType;
         uint16_t connectionHandle;
     };