implemented content blocking

2024-06-19 21:08:18 +02:00
parent 2f6ae2684d
commit 2bacbbffd2
18 changed files with 1301 additions and 0 deletions
@@ -0,0 +1,29 @@
+import 'package:bang_navigator/features/content_block/data/database/database.dart';
+import 'package:bang_navigator/features/content_block/data/models/host.dart';
+import 'package:bang_navigator/features/content_block/data/providers.dart';
+import 'package:riverpod_annotation/riverpod_annotation.dart';
+
+part 'host.g.dart';
+
+@Riverpod(keepAlive: true)
+class HostRepository extends _$HostRepository {
+  late HostDatabase _db;
+
+  @override
+  void build() {
+    _db = ref.watch(hostDatabaseProvider);
+  }
+
+  Stream<List<String>> watchHosts({Iterable<HostSource>? sources}) {
+    return _db.hostDao
+        .getHostList(sources: sources)
+        .map(
+          (hostData) => hostData.hostname,
+        )
+        .watch();
+  }
+
+  Stream<int> watchHostCount(HostSource source) {
+    return _db.hostDao.getHostCount(sources: [source]).watchSingle();
+  }
+}
@@ -0,0 +1,25 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+part of 'host.dart';
+
+// **************************************************************************
+// RiverpodGenerator
+// **************************************************************************
+
+String _$hostRepositoryHash() => r'b28362146b75391f55fb353fc6fde72dc1d2fcb9';
+
+/// See also [HostRepository].
+@ProviderFor(HostRepository)
+final hostRepositoryProvider = NotifierProvider<HostRepository, void>.internal(
+  HostRepository.new,
+  name: r'hostRepositoryProvider',
+  debugGetCreateSourceHash: const bool.fromEnvironment('dart.vm.product')
+      ? null
+      : _$hostRepositoryHash,
+  dependencies: null,
+  allTransitiveDependencies: null,
+);
+
+typedef _$HostRepository = Notifier<void>;
+// ignore_for_file: type=lint
+// ignore_for_file: subtype_of_sealed_class, invalid_use_of_internal_member, invalid_use_of_visible_for_testing_member
@@ -0,0 +1,105 @@
+import 'package:bang_navigator/features/content_block/data/database/database.dart';
+import 'package:bang_navigator/features/content_block/data/models/host.dart';
+import 'package:bang_navigator/features/content_block/data/providers.dart';
+import 'package:bang_navigator/features/content_block/data/services/source.dart';
+import 'package:drift/isolate.dart';
+import 'package:exceptions/exceptions.dart';
+import 'package:riverpod/riverpod.dart';
+import 'package:riverpod_annotation/riverpod_annotation.dart';
+
+part 'sync.g.dart';
+
+@Riverpod(keepAlive: true)
+class HostSyncRepository extends _$HostSyncRepository {
+  late HostDatabase _db;
+
+  HostSyncRepository();
+
+  static Future<Result<void>> _fetchAndSync({
+    required HostSourceService sourceService,
+    required HostDatabase db,
+    required Uri url,
+    required HostSource source,
+    required Duration? syncInterval,
+  }) async {
+    if (syncInterval != null) {
+      final lastSync =
+          await db.syncDao.lastSyncOfSource(source).getSingleOrNull();
+
+      if (lastSync != null &&
+          lastSync.difference(DateTime.now()) < syncInterval) {
+        return Result.success(null);
+      }
+    }
+
+    final result = await sourceService.getHosts(url);
+    return result.flatMapAsync(
+      (remoteHosts) async {
+        await db.syncDao.syncHosts(
+          source: source,
+          remoteHosts: remoteHosts,
+          syncTime: DateTime.now(),
+        );
+      },
+    );
+  }
+
+  Future<Result<void>> syncHostSource(
+    HostSource source,
+    Duration? syncInterval,
+  ) async {
+    try {
+      return Result.success(
+        await _db.computeWithDatabase(
+          connect: HostDatabase.new,
+          computation: (db) async {
+            final ref = ProviderContainer();
+            final result = await _fetchAndSync(
+              sourceService: ref.read(hostSourceServiceProvider.notifier),
+              db: db,
+              url: Uri.parse(source.url),
+              source: source,
+              syncInterval: syncInterval,
+            );
+
+            //Throw if necessary
+            return result.value;
+          },
+        ),
+      );
+    } catch (e) {
+      return Result.failure(
+        ErrorMessage(
+          message: "Failed to sync Hosts (${source.name})",
+          source: 'HostSync',
+          details: e,
+        ),
+      );
+    }
+  }
+
+  Stream<DateTime?> watchLastSyncOfSource(HostSource source) {
+    return _db.syncDao.lastSyncOfSource(source).watchSingleOrNull();
+  }
+
+  Future<Map<HostSource, Result<void>>> syncHostSources({
+    Set<HostSource>? sources,
+    Duration? syncInterval,
+  }) async {
+    //Default to all sources
+    sources ??= HostSource.values.toSet();
+
+    //Run isolated operations
+    final futures = sources.map(
+      (source) => syncHostSource(source, syncInterval)
+          .then((result) => MapEntry(source, result)),
+    );
+
+    return Map.fromEntries(await Future.wait(futures));
+  }
+
+  @override
+  void build() {
+    _db = ref.watch(hostDatabaseProvider);
+  }
+}
@@ -0,0 +1,27 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+part of 'sync.dart';
+
+// **************************************************************************
+// RiverpodGenerator
+// **************************************************************************
+
+String _$hostSyncRepositoryHash() =>
+    r'22b89d3687c415441f99caa4d77dabaeb4969534';
+
+/// See also [HostSyncRepository].
+@ProviderFor(HostSyncRepository)
+final hostSyncRepositoryProvider =
+    NotifierProvider<HostSyncRepository, void>.internal(
+  HostSyncRepository.new,
+  name: r'hostSyncRepositoryProvider',
+  debugGetCreateSourceHash: const bool.fromEnvironment('dart.vm.product')
+      ? null
+      : _$hostSyncRepositoryHash,
+  dependencies: null,
+  allTransitiveDependencies: null,
+);
+
+typedef _$HostSyncRepository = Notifier<void>;
+// ignore_for_file: type=lint
+// ignore_for_file: subtype_of_sealed_class, invalid_use_of_internal_member, invalid_use_of_visible_for_testing_member