refactored chat archive and added search

2024-06-16 13:07:16 +02:00
parent 9daca26261
commit d7a65a5770
18 changed files with 1150 additions and 53 deletions
@@ -0,0 +1,120 @@
+import 'dart:async';
+
+import 'package:bang_navigator/core/logger.dart';
+import 'package:bang_navigator/features/chat_archive/data/database/database.dart';
+import 'package:bang_navigator/features/chat_archive/data/providers.dart';
+import 'package:bang_navigator/features/chat_archive/data/services/file.dart';
+import 'package:bang_navigator/features/chat_archive/domain/entities/chat_entity.dart';
+import 'package:bang_navigator/features/chat_archive/domain/repositories/archive.dart';
+import 'package:bang_navigator/features/chat_archive/utils/markdown_to_text.dart';
+import 'package:collection/collection.dart';
+import 'package:path/path.dart' as path;
+import 'package:riverpod_annotation/riverpod_annotation.dart';
+import 'package:watcher/watcher.dart';
+
+part 'search.g.dart';
+
+@Riverpod()
+class ChatArchiveSearchRepository extends _$ChatArchiveSearchRepository {
+  late Completer<void> _populatedCompleter;
+  late StreamController<List<ChatQueryResult>> _streamController;
+  late ChatSearchDatabase _searchDatabase;
+
+  Future<ChatCompanion?> _readChat(ChatEntity chat) async {
+    final contentResult = await ref
+        .read(chatArchiveRepositoryProvider.notifier)
+        .readChat(chat.fileName);
+
+    return contentResult.fold(
+      (content) => ChatCompanion.insert(
+        fileName: chat.fileName,
+        title: chat.name!,
+        content: markdownToText(content),
+      ),
+      onFailure: (errorMessage) {
+        logger.e(
+          'Error reading file "${chat.fileName}": ${errorMessage.message}',
+          error: errorMessage.details,
+          stackTrace: errorMessage.stackTrace,
+        );
+
+        return null;
+      },
+    );
+  }
+
+  Future<Iterable<ChatCompanion>> _availableChats() async {
+    final availableChats = await ref
+        .read(chatArchiveRepositoryProvider.notifier)
+        .listArchivedChats()
+        .then(
+          (chats) => Future.wait(
+            chats.where((chat) => chat.name != null).map(_readChat),
+          ),
+        );
+
+    return availableChats.whereNotNull();
+  }
+
+  Future<void> search(
+    String input, {
+    int snippetLength = 120,
+    String matchPrefix = '***',
+    String matchSuffix = '***',
+    String ellipsis = '…',
+  }) async {
+    if (input.isNotEmpty) {
+      await _populatedCompleter.future;
+      await _searchDatabase.searchDao
+          .queryChats(
+            searchString: input,
+            snippetLength: snippetLength,
+            matchPrefix: matchPrefix,
+            matchSuffix: matchSuffix,
+            ellipsis: ellipsis,
+          )
+          .get()
+          .then(_streamController.add);
+    }
+  }
+
+  @override
+  Stream<List<ChatQueryResult>> build() async* {
+    _populatedCompleter = Completer();
+    _streamController = StreamController();
+    _searchDatabase = ref.watch(chatSearchDatabaseProvider);
+
+    ref.onDispose(() async {
+      await _streamController.close();
+    });
+
+    // populate with initial chats
+    await _searchDatabase.searchDao.deleteAllChats();
+    await _searchDatabase.searchDao.indexChats(await _availableChats());
+    _populatedCompleter.complete();
+
+    final changeStreamSubscription =
+        ref.watch(chatArchiveFileServiceProvider).listen((event) async {
+      final chat = ChatEntity.fromFileName(path.basename(event.path));
+
+      switch (event.type) {
+        case ChangeType.ADD:
+        case ChangeType.MODIFY:
+          if (chat.name != null) {
+            final companion = await _readChat(chat);
+            if (companion != null) {
+              await _searchDatabase.searchDao.upsertChat(companion);
+            }
+          }
+        case ChangeType.REMOVE:
+          await _searchDatabase.searchDao.deleteChat(chat.fileName);
+      }
+    });
+
+    ref.onDispose(() async {
+      await changeStreamSubscription.cancel();
+    });
+
+    yield* _streamController.stream;
+  }
+}