1 files changed, 27 insertions, 4 deletions
diff --git a/lib/model/connection/universal_ble_connection.dart b/lib/model/connection/universal_ble_connection.dart
index 6aa3c87..fa986a0 100644
--- a/lib/model/connection/universal_ble_connection.dart
+++ b/lib/model/connection/universal_ble_connection.dart
@@ -1,10 +1,28 @@
+// Copyright (C) 2025, uvok cheetah
+//
+// This program 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.
+//
+// This program 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/>.
+
+import 'package:flutter/foundation.dart';
 import 'package:universal_ble/universal_ble.dart';
 import 'package:uvok_epaper_badge/model/connection/device_connection.dart';
 import 'package:uvok_epaper_badge/model/device/universal_ble_device.dart';
 
 class UniversalBleConnection implements DeviceConnection {
   final UniversalBleDevice _device;
-  ConnectionStatus _status = ConnectionStatus.disconnected;
+  final ValueNotifier<ConnectionStatus> _status = ValueNotifier(
+    ConnectionStatus.disconnected,
+  );
 
   UniversalBleConnection(this._device);
 
@@ -12,17 +30,22 @@ class UniversalBleConnection implements DeviceConnection {
   Future<void> connect() async {
     await _device.device.connect();
 
-    _status = await _device.device.isConnected
+    _status.value = await _device.device.isConnected
         ? ConnectionStatus.connected
         : ConnectionStatus.disconnected;
+
+    if (_status.value == ConnectionStatus.connected) {
+      /* Ignore return value for now */
+      await _device.device.discoverServices();
+    }
   }
 
   @override
   Future<void> disconnect() async {
     await _device.device.disconnect();
-    _status = ConnectionStatus.disconnected;
+    _status.value = ConnectionStatus.disconnected;
   }
 
   @override
-  ConnectionStatus get status => _status;
+  ValueNotifier<ConnectionStatus> get status => _status;
 }