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,
                ),
            'userAgent': settings['userAgent']?.readAs(
              DriftSqlType.string,
              db.typeMapping,
            ),
          });
        });

    ref.onDispose(() async {
      await watchSub.cancel();
    });

    return EngineSettings.withDefaults();
  }
}