first bang implementation

This commit is contained in:
Fabian Freund
2024-06-10 03:08:23 +02:00
parent 9ae08409fe
commit 388eba93b7
47 changed files with 3809 additions and 34 deletions
@@ -0,0 +1,154 @@
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/features/bangs/data/services/bang_source.dart';
import 'package:bang_navigator/utils/image_helper.dart';
import 'package:drift/isolate.dart';
import 'package:exceptions/exceptions.dart';
import 'package:riverpod/riverpod.dart';
import 'package:riverpod_annotation/riverpod_annotation.dart';
part 'bang.g.dart';
Future<Result<void>> _syncBangGroupStatic(
BangSource source,
BangDatabase db,
Uri url,
BangGroup? group,
) async {
final result = await source.getBangs(url, group);
return result.flatMapAsync(
(value) async {
final sourcedBangMap =
Map.fromEntries(value.map((e) => MapEntry(e.trigger, e)));
final localBangMap =
await db.bangDao.getBangs(groups: [group]).get().then(
(bangs) =>
Map.fromEntries(bangs.map((e) => MapEntry(e.trigger, e))),
);
final sourcedBangTriggers = sourcedBangMap.keys.toSet();
final localBangTriggers = localBangMap.keys.toSet();
final removedBangs = localBangTriggers.difference(sourcedBangTriggers);
final addedBangs = sourcedBangTriggers.difference(localBangTriggers).map(
(e) => sourcedBangMap[e]!,
);
final changedBangs = sourcedBangTriggers
.intersection(localBangTriggers)
.where((e) => sourcedBangMap[e] != localBangMap[e])
.map(
(e) => sourcedBangMap[e]!,
);
await db.bangDao.deleteBangs(removedBangs);
await db.bangDao.insertBangs(addedBangs);
await db.bangDao.replaceBangs(changedBangs);
await db.optimizeFtsIndex();
},
);
}
@Riverpod(keepAlive: true)
class BangRepository extends _$BangRepository {
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<List<BangData>> watchBangs({Iterable<BangGroup?>? groups}) {
return _db.bangDao.getBangDataList(groups: groups).watch();
}
Stream<List<BangData>> watchFrequentBangs({Iterable<BangGroup?>? groups}) {
return _db.bangDao.getFrequentBangDataList(groups: groups).watch();
}
Future<void> increaseFrequency(String trigger) {
return _db.bangDao.increaseFrequency(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.upsertIcon(bang.trigger, faviconBytes);
return bang.copyWith.iconData(faviconBytes);
}
return bang;
},
);
}
Future<Result<void>> syncBangGroup(
Uri url,
BangGroup? group,
) =>
_syncBangGroupStatic(
ref.read(bangSourceProvider.notifier),
_db,
url,
group,
);
Future<void> syncAllBangGroups() async {
await _db.computeWithDatabase(
connect: BangDatabase.new,
computation: (db) async {
final ref = ProviderContainer();
final source = ref.read(bangSourceProvider.notifier);
await _syncBangGroupStatic(
source,
db,
Uri.parse(
'https://raw.githubusercontent.com/kagisearch/bangs/main/data/bangs.json',
),
null,
);
await _syncBangGroupStatic(
source,
db,
Uri.parse(
'https://raw.githubusercontent.com/kagisearch/bangs/main/data/kagi_bangs.json',
),
BangGroup.kagi,
);
await _syncBangGroupStatic(
source,
db,
Uri.parse(
'https://raw.githubusercontent.com/kagisearch/bangs/main/data/assistant_bangs.json',
),
BangGroup.assistant,
);
},
);
}
}
@@ -0,0 +1,25 @@
// GENERATED CODE - DO NOT MODIFY BY HAND
part of 'bang.dart';
// **************************************************************************
// RiverpodGenerator
// **************************************************************************
String _$bangRepositoryHash() => r'd405550e2ad11c57f3644f18390348efb71bdf0f';
/// See also [BangRepository].
@ProviderFor(BangRepository)
final bangRepositoryProvider = NotifierProvider<BangRepository, void>.internal(
BangRepository.new,
name: r'bangRepositoryProvider',
debugGetCreateSourceHash: const bool.fromEnvironment('dart.vm.product')
? null
: _$bangRepositoryHash,
dependencies: null,
allTransitiveDependencies: null,
);
typedef _$BangRepository = 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,32 @@
import 'dart:async';
import 'package:bang_navigator/features/bangs/data/database/database.dart';
import 'package:bang_navigator/features/bangs/data/models/bang_data.dart';
import 'package:bang_navigator/features/bangs/data/providers.dart';
import 'package:riverpod_annotation/riverpod_annotation.dart';
part 'bang_search.g.dart';
@Riverpod()
class BangSearch extends _$BangSearch {
late StreamController<List<BangData>> _streamController;
late BangDatabase _db;
Future<void> search(String input) async {
if (input.isNotEmpty) {
await _db.bangDao.queryBangs(input).get().then(_streamController.add);
}
}
@override
Stream<List<BangData>> build() {
_db = ref.watch(bangDatabaseProvider);
_streamController = StreamController();
ref.onDispose(() async {
await _streamController.close();
});
return _streamController.stream;
}
}
@@ -0,0 +1,25 @@
// GENERATED CODE - DO NOT MODIFY BY HAND
part of 'bang_search.dart';
// **************************************************************************
// RiverpodGenerator
// **************************************************************************
String _$bangSearchHash() => r'abd71f10288ac1f559e66fc11035f9f2b027c353';
/// See also [BangSearch].
@ProviderFor(BangSearch)
final bangSearchProvider =
AutoDisposeStreamNotifierProvider<BangSearch, List<BangData>>.internal(
BangSearch.new,
name: r'bangSearchProvider',
debugGetCreateSourceHash:
const bool.fromEnvironment('dart.vm.product') ? null : _$bangSearchHash,
dependencies: null,
allTransitiveDependencies: null,
);
typedef _$BangSearch = AutoDisposeStreamNotifier<List<BangData>>;
// ignore_for_file: type=lint
// ignore_for_file: subtype_of_sealed_class, invalid_use_of_internal_member, invalid_use_of_visible_for_testing_member