d8f4dbb2aa
This makes negentropy optimizations available to the rest of the app via Subscription Manager. Changelog not needed because this should not have user-facing changes Changelog-None Signed-off-by: Daniel D’Aquino <daniel@daquino.me>
58 lines
2.2 KiB
Swift
58 lines
2.2 KiB
Swift
//
|
||
// Ndb+.swift
|
||
// damus
|
||
//
|
||
// Created by Daniel D’Aquino on 2025-04-04.
|
||
//
|
||
|
||
/// ## Implementation notes
|
||
///
|
||
/// 1. This was created as a separate file because it contains dependencies to damus-specific structures such as `NostrFilter`, which is not yet available inside the NostrDB codebase.
|
||
|
||
import Foundation
|
||
|
||
extension Ndb {
|
||
/// Subscribe to events matching the provided NostrFilters
|
||
/// - Parameters:
|
||
/// - filters: Array of NostrFilter objects
|
||
/// - maxSimultaneousResults: Maximum number of initial results to return
|
||
/// - Returns: AsyncStream of StreamItem events
|
||
/// - Throws: NdbStreamError if subscription fails
|
||
func subscribe(filters: [NostrFilter], maxSimultaneousResults: Int = 1000) throws -> AsyncStream<StreamItem> {
|
||
let ndbFilters: [NdbFilter]
|
||
do {
|
||
ndbFilters = try filters.toNdbFilters()
|
||
} catch {
|
||
throw NdbStreamError.cannotConvertFilter(error)
|
||
}
|
||
return try self.subscribe(filters: ndbFilters, maxSimultaneousResults: maxSimultaneousResults)
|
||
}
|
||
|
||
/// Determines if a given note was seen on any of the listed relay URLs
|
||
func was(noteKey: NoteKey, seenOnAnyOf relayUrls: [RelayURL]) throws -> Bool {
|
||
return try self.was(noteKey: noteKey, seenOnAnyOf: relayUrls.map({ $0.absoluteString }))
|
||
}
|
||
|
||
func processEvent(_ str: String, originRelayURL: RelayURL? = nil) -> Bool {
|
||
self.process_event(str, originRelayURL: originRelayURL?.absoluteString)
|
||
}
|
||
|
||
/// Adds a NostrEvent to the database by converting it to a push event and processing it.
|
||
/// - Parameter event: The NostrEvent to add
|
||
/// - Throws: NdbAddError.couldNotMakePushEvent if the event cannot be converted, or NdbAddError.processingFailed if processing fails
|
||
func add(event: NostrEvent) throws {
|
||
guard let nostrPushEvent = make_nostr_push_event(ev: event) else {
|
||
throw NdbAddError.couldNotMakePushEvent
|
||
}
|
||
let success = self.process_client_event(nostrPushEvent)
|
||
if !success {
|
||
throw NdbAddError.processingFailed
|
||
}
|
||
}
|
||
|
||
enum NdbAddError: Error {
|
||
case couldNotMakePushEvent
|
||
case processingFailed
|
||
}
|
||
}
|