update theme provider

Olivier-plessis · Aug 30, 2024 · af2bd34 · af2bd34
1 parent ff6cfa5
commit af2bd34
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/...scaffold_app/__brick__/{{name.snakeCase()}}/lib/src/core/theme/theme_mode_controller.dart b/...scaffold_app/__brick__/{{name.snakeCase()}}/lib/src/core/theme/theme_mode_controller.dart
@@ -6,13 +6,19 @@ part 'theme_mode_controller.g.dart';
 @riverpod
 class ThemeModeController extends _$ThemeModeController {
   @override
-  ThemeMode build() => ThemeMode.dark;
+  ThemeMode build() {
+    return switch (this) {
+      'light' => ThemeMode.light,
+      'dark' => ThemeMode.dark,
+      _ => ThemeMode.dark,
+    };
+  }
+
+  void setThemeMode(ThemeMode mode) {
+    state = mode;
+  }
 
   void toggleThemeMode() {
-    if (state == ThemeMode.light) {
-      state = ThemeMode.dark;
-    } else {
-      state = ThemeMode.light;
-    }
+    state = state == ThemeMode.light ? ThemeMode.dark : ThemeMode.light;
   }
 }