properly check for idle gear

OpenTails · Jun 28, 2024 · 45fc2ff · 45fc2ff
1 parent 9f0ef41
commit 45fc2ff
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/lib/Backend/device_registry.dart b/lib/Backend/device_registry.dart
@@ -102,9 +102,24 @@ BuiltSet<BaseStatefulDevice> getByAction(GetByActionRef ref, BaseAction baseActi
 }
 
 @Riverpod(keepAlive: true)
-BuiltList<BaseStatefulDevice> getAvailableIdleGear(GetAvailableIdleGearRef ref) {
-  final BuiltList<BaseStatefulDevice> watch = ref.watch(getAvailableGearProvider);
-  return watch.where((element) => element.deviceState.value == DeviceState.standby).toBuiltList();
+class GetAvailableIdleGear extends _$GetAvailableIdleGear {
+  @override
+  BuiltList<BaseStatefulDevice> build() {
+    for (BaseStatefulDevice baseStatefulDevice in ref.watch(knownDevicesProvider).values) {
+      baseStatefulDevice.deviceState
+        ..removeListener(_listener)
+        ..addListener(_listener);
+    }
+    return getState();
+  }
+
+  BuiltList<BaseStatefulDevice> getState() {
+    return ref.read(getAvailableGearProvider).where((element) => element.deviceState.value == DeviceState.standby).toBuiltList();
+  }
+
+  void _listener() {
+    state = getState();
+  }
 }
 
 @Riverpod(keepAlive: true)