push

2025-02-03 15:09:04 +01:00
parent 36acd46318
commit 025ecaf2e3
132 changed files with 4431 additions and 1644 deletions
@@ -0,0 +1,78 @@
+import 'dart:async';
+
+import 'package:drift/drift.dart';
+import 'package:lensai/features/user/data/database/database.dart';
+import 'package:lensai/features/user/data/models/engine_settings.dart';
+import 'package:lensai/features/user/data/providers.dart';
+import 'package:riverpod_annotation/riverpod_annotation.dart';
+
+part 'engine_settings.g.dart';
+
+typedef UpdateEngineSettingsFunc = EngineSettings Function(
+  EngineSettings currentSettings,
+);
+
+@Riverpod(keepAlive: true)
+class EngineSettingsRepository extends _$EngineSettingsRepository {
+  final _partitionKey = 'engine';
+
+  Future<void> updateSettings(UpdateEngineSettingsFunc updateWithCurrent) {
+    final oldJson = state.toJson();
+    final newJson = updateWithCurrent(state).toJson();
+
+    return ref.read(userDatabaseProvider).transaction(() async {
+      for (final MapEntry(:key, :value) in newJson.entries) {
+        if (oldJson[key] != value) {
+          await ref
+              .read(userDatabaseProvider)
+              .settingDao
+              .updateSetting(key, _partitionKey, value);
+        }
+      }
+    });
+  }
+
+  @override
+  EngineSettings build() {
+    final db = ref.watch(userDatabaseProvider);
+
+    final watchSub = db.settingDao
+        .allSettingsOfPartitionKey(_partitionKey)
+        .watch()
+        .listen((entries) {
+      final settings = Map.fromEntries(entries);
+
+      state = EngineSettings.fromJson({
+        'incognitoMode': settings['incognitoMode']
+            ?.readAs(DriftSqlType.bool, db.typeMapping),
+        'javascriptEnabled': settings['javascriptEnabled']
+            ?.readAs(DriftSqlType.bool, db.typeMapping),
+        'trackingProtectionPolicy': settings['trackingProtectionPolicy']
+            ?.readAs(DriftSqlType.string, db.typeMapping),
+        'httpsOnlyMode': settings['httpsOnlyMode']
+            ?.readAs(DriftSqlType.string, db.typeMapping),
+        'globalPrivacyControlEnabled': settings['globalPrivacyControlEnabled']
+            ?.readAs(DriftSqlType.bool, db.typeMapping),
+        'cookieBannerHandlingMode': settings['cookieBannerHandlingMode']
+            ?.readAs(DriftSqlType.string, db.typeMapping),
+        'cookieBannerHandlingModePrivateBrowsing':
+            settings['cookieBannerHandlingModePrivateBrowsing']
+                ?.readAs(DriftSqlType.string, db.typeMapping),
+        'cookieBannerHandlingGlobalRules':
+            settings['cookieBannerHandlingGlobalRules']
+                ?.readAs(DriftSqlType.bool, db.typeMapping),
+        'cookieBannerHandlingGlobalRulesSubFrames':
+            settings['cookieBannerHandlingGlobalRulesSubFrames']
+                ?.readAs(DriftSqlType.bool, db.typeMapping),
+        'webContentIsolationStrategy': settings['webContentIsolationStrategy']
+            ?.readAs(DriftSqlType.string, db.typeMapping),
+      });
+    });
+
+    ref.onDispose(() {
+      unawaited(watchSub.cancel());
+    });
+
+    return EngineSettings.withDefaults();
+  }
+}