basic object box impl

leo-lox · Oct 29, 2024 · 63439f3 · 63439f3
1 parent ac173c0
commit 63439f3
Show file tree

Hide file tree

Showing 2 changed files with 182 additions and 79 deletions.
diff --git a/lib/data_layer/db/object_box/db_object_box.dart b/lib/data_layer/db/object_box/db_object_box.dart
@@ -4,7 +4,11 @@ import 'package:ndk/domain_layer/entities/user_relay_list.dart';
 import 'package:ndk/ndk.dart';
 import 'package:ndk/shared/nips/nip05/nip05.dart';
 
+import '../../../objectbox.g.dart';
 import 'db_init_object_box.dart';
+import 'schema/db_contact_list.dart';
+import 'schema/db_metadata.dart';
+import 'schema/db_nip_01_event.dart';
 
 class DbObjectBox implements CacheManager {
   Completer _initCompleter = Completer();
@@ -22,175 +26,273 @@ class DbObjectBox implements CacheManager {
   }
 
   @override
-  Future<ContactList?> loadContactList(String pubKey) {
-    // TODO: implement loadContactList
-    throw UnimplementedError();
+  Future<ContactList?> loadContactList(String pubKey) async {
+    await _dbRdy;
+    final contactListBox = _objectBox.store.box<DbContactList>();
+    final existingContact = contactListBox
+        .query(DbContactList_.pubKey.equals(pubKey))
+        .order(DbContactList_.createdAt, flags: Order.descending)
+        .build()
+        .findFirst();
+    if (existingContact == null) {
+      return null;
+    }
+    return existingContact.toNdk();
   }
 
   @override
-  Future<Nip01Event?> loadEvent(String id) {
-    // TODO: implement loadEvent
-    throw UnimplementedError();
+  Future<Nip01Event?> loadEvent(String id) async {
+    await _dbRdy;
+    final eventBox = _objectBox.store.box<DbNip01Event>();
+    final existingEvent =
+        eventBox.query(DbNip01Event_.nostrId.equals(id)).build().findFirst();
+    if (existingEvent == null) {
+      return null;
+    }
+    return existingEvent.toNdk();
   }
 
   @override
-  Future<List<Nip01Event>> loadEvents(
-      {List<String> pubKeys,
-      List<int> kinds,
-      String? pTag,
-      int? since,
-      int? until}) {
-    // TODO: implement loadEvents
-    throw UnimplementedError();
+  Future<List<Nip01Event>> loadEvents({
+    List<String>? pubKeys,
+    List<int>? kinds,
+    String? pTag,
+    int? since,
+    int? until,
+  }) async {
+    await _dbRdy;
+    final eventBox = _objectBox.store.box<DbNip01Event>();
+
+    var query = eventBox.query(DbNip01Event_.pubKey
+        .oneOf(pubKeys!)
+        .and(DbNip01Event_.kind.oneOf(kinds!)));
+
+    query = query.order(DbNip01Event_.createdAt, flags: Order.descending);
+
+    final foundDb = query.build().find();
+
+    final foundValid = foundDb.where((event) {
+      if (pTag != null && !event.pTags.contains(pTag)) {
+        return false;
+      }
+
+      if (since != null && event.createdAt < since) {
+        return false;
+      }
+
+      if (until != null && event.createdAt > until) {
+        return false;
+      }
+
+      return true;
+    }).toList();
+
+    return foundValid.map((dbEvent) => dbEvent.toNdk()).toList();
   }
 
   @override
-  Future<Metadata?> loadMetadata(String pubKey) {
-    // TODO: implement loadMetadata
-    throw UnimplementedError();
+  Future<Metadata?> loadMetadata(String pubKey) async {
+    await _dbRdy;
+    final metadataBox = _objectBox.store.box<DbMetadata>();
+    final existingMetadata = metadataBox
+        .query(DbMetadata_.pubKey.equals(pubKey))
+        .order(DbMetadata_.updatedAt, flags: Order.descending)
+        .build()
+        .findFirst();
+    if (existingMetadata == null) {
+      return null;
+    }
+    return existingMetadata.toNdk();
   }
 
   @override
-  Future<List<Metadata?>> loadMetadatas(List<String> pubKeys) {
-    // TODO: implement loadMetadatas
-    throw UnimplementedError();
+  Future<List<Metadata?>> loadMetadatas(List<String> pubKeys) async {
+    await _dbRdy;
+    final metadataBox = _objectBox.store.box<DbMetadata>();
+    final existingMetadatas = metadataBox
+        .query(DbMetadata_.pubKey.oneOf(pubKeys))
+        .order(DbMetadata_.updatedAt, flags: Order.descending)
+        .build()
+        .find();
+    return existingMetadatas.map((dbMetadata) => dbMetadata.toNdk()).toList();
   }
 
   @override
-  Future<Nip05?> loadNip05(String pubKey) {
-    // TODO: implement loadNip05
-    throw UnimplementedError();
+  Future<void> removeAllContactLists() async {
+    await _dbRdy;
+    final contactListBox = _objectBox.store.box<DbContactList>();
+    contactListBox.removeAll();
   }
 
   @override
-  Future<List<Nip05?>> loadNip05s(List<String> pubKeys) {
-    // TODO: implement loadNip05s
-    throw UnimplementedError();
+  Future<void> removeAllEvents() async {
+    await _dbRdy;
+    final eventBox = _objectBox.store.box<DbNip01Event>();
+    eventBox.removeAll();
   }
 
   @override
-  Future<RelaySet?> loadRelaySet(String name, String pubKey) {
-    // TODO: implement loadRelaySet
-    throw UnimplementedError();
+  Future<void> removeAllEventsByPubKey(String pubKey) async {
+    await _dbRdy;
+    final eventBox = _objectBox.store.box<DbNip01Event>();
+    final events =
+        eventBox.query(DbNip01Event_.pubKey.equals(pubKey)).build().find();
+    eventBox.removeMany(events.map((e) => e.dbId).toList());
   }
 
   @override
-  Future<UserRelayList?> loadUserRelayList(String pubKey) {
-    // TODO: implement loadUserRelayList
-    throw UnimplementedError();
+  Future<void> removeAllMetadatas() async {
+    await _dbRdy;
+    final metadataBox = _objectBox.store.box<DbMetadata>();
+    metadataBox.removeAll();
   }
 
   @override
-  Future<void> removeAllContactLists() {
-    // TODO: implement removeAllContactLists
-    throw UnimplementedError();
+  Future<void> saveContactList(ContactList contactList) async {
+    await _dbRdy;
+    final contactListBox = _objectBox.store.box<DbContactList>();
+    final existingContact = contactListBox
+        .query(DbContactList_.pubKey.equals(contactList.pubKey))
+        .order(DbContactList_.createdAt, flags: Order.descending)
+        .build()
+        .findFirst();
+    if (existingContact != null) {
+      contactListBox.remove(existingContact.dbId);
+    }
+    contactListBox.put(DbContactList.fromNdk(contactList));
   }
 
   @override
-  Future<void> removeAllEvents() {
-    // TODO: implement removeAllEvents
-    throw UnimplementedError();
+  Future<void> saveContactLists(List<ContactList> contactLists) async {
+    await _dbRdy;
+    final contactListBox = _objectBox.store.box<DbContactList>();
+    contactListBox
+        .putMany(contactLists.map((cl) => DbContactList.fromNdk(cl)).toList());
   }
 
   @override
-  Future<void> removeAllEventsByPubKey(String pubKey) {
-    // TODO: implement removeAllEventsByPubKey
-    throw UnimplementedError();
+  Future<void> saveEvent(Nip01Event event) async {
+    await _dbRdy;
+    final eventBox = _objectBox.store.box<DbNip01Event>();
+    final existingEvent = eventBox
+        .query(DbNip01Event_.nostrId.equals(event.id))
+        .build()
+        .findFirst();
+    if (existingEvent != null) {
+      eventBox.remove(existingEvent.dbId);
+    }
+    eventBox.put(DbNip01Event.fromNdk(event));
   }
 
   @override
-  Future<void> removeAllMetadatas() {
-    // TODO: implement removeAllMetadatas
-    throw UnimplementedError();
+  Future<void> saveEvents(List<Nip01Event> events) async {
+    await _dbRdy;
+    final eventBox = _objectBox.store.box<DbNip01Event>();
+    eventBox.putMany(events.map((e) => DbNip01Event.fromNdk(e)).toList());
   }
 
   @override
-  Future<void> removeAllNip05s() {
-    // TODO: implement removeAllNip05s
-    throw UnimplementedError();
+  Future<void> saveMetadata(Metadata metadata) async {
+    await _dbRdy;
+    final metadataBox = _objectBox.store.box<DbMetadata>();
+    final existingMetadatas = metadataBox
+        .query(DbMetadata_.pubKey.equals(metadata.pubKey))
+        .order(DbMetadata_.updatedAt, flags: Order.descending)
+        .build()
+        .find();
+    if (existingMetadatas.length > 1) {
+      metadataBox.removeMany(existingMetadatas.map((e) => e.dbId).toList());
+    }
+    if (existingMetadatas.isNotEmpty &&
+        metadata.updatedAt! < existingMetadatas[0].updatedAt!) {
+      return;
+    }
+    metadataBox.put(DbMetadata.fromNdk(metadata));
   }
 
   @override
-  Future<void> removeAllRelaySets() {
-    // TODO: implement removeAllRelaySets
-    throw UnimplementedError();
+  Future<void> saveMetadatas(List<Metadata> metadatas) async {
+    await _dbRdy;
+    for (final metadata in metadatas) {
+      await saveMetadata(metadata);
+    }
   }
 
   @override
-  Future<void> removeAllUserRelayLists() {
-    // TODO: implement removeAllUserRelayLists
+  Future<Nip05?> loadNip05(String pubKey) {
+    // TODO: implement loadNip05
     throw UnimplementedError();
   }
 
   @override
-  Future<void> removeContactList(String pubKey) {
-    // TODO: implement removeContactList
+  Future<List<Nip05?>> loadNip05s(List<String> pubKeys) {
+    // TODO: implement loadNip05s
     throw UnimplementedError();
   }
 
   @override
-  Future<void> removeEvent(String id) {
-    // TODO: implement removeEvent
+  Future<RelaySet?> loadRelaySet(String name, String pubKey) {
+    // TODO: implement loadRelaySet
     throw UnimplementedError();
   }
 
   @override
-  Future<void> removeMetadata(String pubKey) {
-    // TODO: implement removeMetadata
+  Future<UserRelayList?> loadUserRelayList(String pubKey) {
+    // TODO: implement loadUserRelayList
     throw UnimplementedError();
   }
 
   @override
-  Future<void> removeNip05(String pubKey) {
-    // TODO: implement removeNip05
+  Future<void> removeAllNip05s() {
+    // TODO: implement removeAllNip05s
     throw UnimplementedError();
   }
 
   @override
-  Future<void> removeRelaySet(String name, String pubKey) {
-    // TODO: implement removeRelaySet
+  Future<void> removeAllRelaySets() {
+    // TODO: implement removeAllRelaySets
     throw UnimplementedError();
   }
 
   @override
-  Future<void> removeUserRelayList(String pubKey) {
-    // TODO: implement removeUserRelayList
+  Future<void> removeAllUserRelayLists() {
+    // TODO: implement removeAllUserRelayLists
     throw UnimplementedError();
   }
 
   @override
-  Future<void> saveContactList(ContactList contactList) {
-    // TODO: implement saveContactList
+  Future<void> removeContactList(String pubKey) {
+    // TODO: implement removeContactList
     throw UnimplementedError();
   }
 
   @override
-  Future<void> saveContactLists(List<ContactList> contactLists) {
-    // TODO: implement saveContactLists
+  Future<void> removeEvent(String id) {
+    // TODO: implement removeEvent
     throw UnimplementedError();
   }
 
   @override
-  Future<void> saveEvent(Nip01Event event) {
-    // TODO: implement saveEvent
+  Future<void> removeMetadata(String pubKey) {
+    // TODO: implement removeMetadata
     throw UnimplementedError();
   }
 
   @override
-  Future<void> saveEvents(List<Nip01Event> events) {
-    // TODO: implement saveEvents
+  Future<void> removeNip05(String pubKey) {
+    // TODO: implement removeNip05
     throw UnimplementedError();
   }
 
   @override
-  Future<void> saveMetadata(Metadata metadata) {
-    // TODO: implement saveMetadata
+  Future<void> removeRelaySet(String name, String pubKey) {
+    // TODO: implement removeRelaySet
     throw UnimplementedError();
   }
 
   @override
-  Future<void> saveMetadatas(List<Metadata> metadatas) {
-    // TODO: implement saveMetadatas
+  Future<void> removeUserRelayList(String pubKey) {
+    // TODO: implement removeUserRelayList
     throw UnimplementedError();
   }
 

diff --git a/lib/presentation_layer/providers/ndk_provider.dart b/lib/presentation_layer/providers/ndk_provider.dart
@@ -1,20 +1,21 @@
 import 'package:ndk/ndk.dart';
 import 'package:riverpod/riverpod.dart';
 
+import 'db_object_box_provider.dart';
 import 'event_signer_provider.dart';
 import 'event_verifier.dart';
 
-
 final ndkProvider = Provider<Ndk>((ref) {
   final eventSigner = ref.watch(eventSignerProvider);
   final eventVerifier = ref.watch(eventVerifierProvider);
 
-
+  final dbObjectBox = ref.watch(dbObjectBoxProvider);
+
   final CacheManager memDb = MemCacheManager();
 
   final NdkConfig ndkConfig = NdkConfig(
     engine: NdkEngine.JIT,
-    cache: memDb,
+    cache: dbObjectBox,
     eventSigner: eventSigner,
     eventVerifier: eventVerifier,
   );