30 lines
848 B
Dart
30 lines
848 B
Dart
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();
|
|
}
|
|
}
|