refactored bang layers

2024-06-16 13:05:30 +02:00
parent 2007fd2809
commit 9daca26261
13 changed files with 14 additions and 14 deletions
@@ -0,0 +1,96 @@
+import 'dart:convert';
+
+import 'package:bang_navigator/domain/services/generic_website.dart';
+import 'package:bang_navigator/features/bangs/data/database/database.dart';
+import 'package:bang_navigator/features/bangs/data/models/bang.dart';
+import 'package:bang_navigator/features/bangs/data/models/bang_data.dart';
+import 'package:bang_navigator/features/bangs/data/providers.dart';
+import 'package:bang_navigator/utils/image_helper.dart';
+import 'package:drift/drift.dart';
+import 'package:exceptions/exceptions.dart';
+import 'package:riverpod_annotation/riverpod_annotation.dart';
+
+part 'data.g.dart';
+
+@Riverpod(keepAlive: true)
+class BangDataRepository extends _$BangDataRepository {
+  late BangDatabase _db;
+
+  @override
+  void build() {
+    _db = ref.watch(bangDatabaseProvider);
+  }
+
+  Stream<BangData?> watchBang(String? trigger) {
+    if (trigger != null) {
+      return _db.bangDao.getBangData(trigger).watchSingleOrNull();
+    } else {
+      return Stream.value(null);
+    }
+  }
+
+  Stream<Map<String, List<String>>> watchCategories() {
+    return _db.categoriesJson().watchSingle().map((json) {
+      final decoded = jsonDecode(json) as Map<String, dynamic>;
+      return decoded.map(
+        (key, value) => MapEntry(key, (value as List<dynamic>).cast()),
+      );
+    });
+  }
+
+  Stream<int> watchBangCount(BangGroup group) {
+    return _db.bangDao.getBangCount(groups: [group]).watchSingle();
+  }
+
+  Stream<List<BangData>> watchBangs({
+    Iterable<BangGroup>? groups,
+    String? domain,
+    ({String category, String? subCategory})? categoryFilter,
+  }) {
+    return _db.bangDao
+        .getBangDataList(
+          groups: groups,
+          domain: domain,
+          category: categoryFilter?.category,
+          subCategory: categoryFilter?.subCategory,
+        )
+        .watch();
+  }
+
+  Stream<List<BangData>> watchFrequentBangs({Iterable<BangGroup>? groups}) {
+    return _db.bangDao.getFrequentBangDataList(groups: groups).watch();
+  }
+
+  Future<void> increaseFrequency(String trigger) {
+    return _db.bangDao.increaseBangFrequency(trigger);
+  }
+
+  Future<Result<BangData>> ensureIconAvailable(BangData bang) async {
+    if (bang.iconData != null) {
+      return Result.success(bang);
+    }
+
+    final result = await ref
+        .read(genericWebsiteServiceProvider.notifier)
+        .getFaviconBytes(Uri.parse(bang.getUrl('').origin));
+
+    return result.flatMapAsync(
+      (faviconBytes) async {
+        if (faviconBytes != null && await isImageValid(faviconBytes)) {
+          await _db.bangDao.upsertBangIcon(bang.trigger, faviconBytes);
+          return bang.copyWith.iconData(faviconBytes);
+        }
+
+        return bang;
+      },
+    );
+  }
+
+  Future<int> clearIconData() {
+    return _db.bangIcon.deleteAll();
+  }
+
+  Future<int> clearFrequency() {
+    return _db.bangFrequency.deleteAll();
+  }
+}