145 lines
4.2 KiB
Dart
145 lines
4.2 KiB
Dart
/*
|
|
* Copyright (c) 2024-2025 Fabian Freund.
|
|
*
|
|
* This file is part of WebLibre
|
|
* (see https://weblibre.eu).
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
import 'dart:convert';
|
|
|
|
import 'package:drift/drift.dart';
|
|
import 'package:riverpod_annotation/riverpod_annotation.dart';
|
|
import 'package:weblibre/features/bangs/data/models/bang_data.dart';
|
|
import 'package:weblibre/features/bangs/data/models/bang_group.dart';
|
|
import 'package:weblibre/features/bangs/data/models/search_history_entry.dart';
|
|
import 'package:weblibre/features/bangs/data/providers.dart';
|
|
|
|
part 'data.g.dart';
|
|
|
|
@Riverpod(keepAlive: true)
|
|
class BangDataRepository extends _$BangDataRepository {
|
|
@override
|
|
void build() {}
|
|
|
|
Stream<BangData?> watchBang(String? trigger) {
|
|
if (trigger != null) {
|
|
return ref
|
|
.read(bangDatabaseProvider)
|
|
.bangDao
|
|
.getBangData(trigger)
|
|
.watchSingleOrNull();
|
|
} else {
|
|
return Stream.value(null);
|
|
}
|
|
}
|
|
|
|
Stream<Map<String, List<String>>> watchCategories() {
|
|
return ref
|
|
.read(bangDatabaseProvider)
|
|
.definitionsDrift
|
|
.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 ref
|
|
.read(bangDatabaseProvider)
|
|
.bangDao
|
|
.getBangCount(groups: [group])
|
|
.watchSingle();
|
|
}
|
|
|
|
Stream<List<BangData>> watchBangs({
|
|
Iterable<String>? triggers,
|
|
Iterable<BangGroup>? groups,
|
|
String? domain,
|
|
({String category, String? subCategory})? categoryFilter,
|
|
bool? orderMostFrequentFirst,
|
|
}) {
|
|
return ref
|
|
.read(bangDatabaseProvider)
|
|
.bangDao
|
|
.getBangDataList(
|
|
triggers: triggers,
|
|
groups: groups,
|
|
domain: domain,
|
|
category: categoryFilter?.category,
|
|
subCategory: categoryFilter?.subCategory,
|
|
orderMostFrequentFirst: orderMostFrequentFirst,
|
|
)
|
|
.watch();
|
|
}
|
|
|
|
Stream<List<BangData>> watchFrequentBangs({Iterable<BangGroup>? groups}) {
|
|
return ref
|
|
.read(bangDatabaseProvider)
|
|
.bangDao
|
|
.getFrequentBangDataList(groups: groups)
|
|
.watch();
|
|
}
|
|
|
|
Stream<List<SearchHistoryEntry>> watchSearchHistory({required int limit}) {
|
|
return ref
|
|
.read(bangDatabaseProvider)
|
|
.definitionsDrift
|
|
.searchHistoryEntries(limit: limit)
|
|
.watch();
|
|
}
|
|
|
|
Future<void> increaseFrequency(String trigger) {
|
|
return ref
|
|
.read(bangDatabaseProvider)
|
|
.bangDao
|
|
.increaseBangFrequency(trigger);
|
|
}
|
|
|
|
Future<void> addSearchEntry(
|
|
String trigger,
|
|
String searchQuery, {
|
|
required int maxEntryCount,
|
|
}) {
|
|
final db = ref.read(bangDatabaseProvider);
|
|
//Pack in a transaction to bundle rebuilds of watch() queries
|
|
return db.transaction(() async {
|
|
await db.bangDao.addSearchEntry(trigger, searchQuery);
|
|
await db.definitionsDrift.evictHistoryEntries(limit: maxEntryCount);
|
|
});
|
|
}
|
|
|
|
Future<void> removeSearchEntry(String searchQuery) {
|
|
return ref
|
|
.read(bangDatabaseProvider)
|
|
.bangDao
|
|
.removeSearchEntry(searchQuery);
|
|
}
|
|
|
|
Future<int> resetFrequencies() {
|
|
return ref.read(bangDatabaseProvider).bangFrequency.deleteAll();
|
|
}
|
|
|
|
Future<int> resetFrequency(String trigger) {
|
|
return ref
|
|
.read(bangDatabaseProvider)
|
|
.bangFrequency
|
|
.deleteWhere((t) => t.trigger.equals(trigger));
|
|
}
|
|
}
|