95 lines
3.2 KiB
Dart
95 lines
3.2 KiB
Dart
import 'dart:async';
|
|
|
|
import 'package:drift/drift.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
|
|
.getAllSettingsOfPartitionKey(_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();
|
|
}
|
|
}
|