first bang implementation
This commit is contained in:
@@ -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
|
||||
Reference in New Issue
Block a user