Jon Staab
79 lines
2.4 KiB
TypeScript
79 lines
2.4 KiB
TypeScript
import {ctx, isNil} from "@welshman/lib"
|
|
import {Repository, Relay, LOCAL_RELAY_URL, getFilterResultCardinality} from "@welshman/util"
|
|
import type {TrustedEvent, Filter} from "@welshman/util"
|
|
import {Tracker, subscribe as baseSubscribe} from "@welshman/net"
|
|
import type {SubscribeRequestWithHandlers} from "@welshman/net"
|
|
import {custom} from "@welshman/store"
|
|
|
|
export const repository = new Repository<TrustedEvent>()
|
|
|
|
export const relay = new Relay(repository)
|
|
|
|
export const tracker = new Tracker()
|
|
|
|
export const trackerStore = custom(setter => {
|
|
const onUpdate = () => setter(tracker)
|
|
|
|
onUpdate()
|
|
tracker.on('update', onUpdate)
|
|
|
|
return () => tracker.off('update', onUpdate)
|
|
}, {
|
|
set: (other: Tracker) => tracker.load(other.data),
|
|
})
|
|
|
|
export type PartialSubscribeRequest = Partial<SubscribeRequestWithHandlers> & {filters: Filter[]}
|
|
|
|
export const subscribe = (request: PartialSubscribeRequest) => {
|
|
const events: TrustedEvent[] = []
|
|
|
|
// If we already have all results for any filter, don't send the filter to the network
|
|
if (request.closeOnEose) {
|
|
for (const filter of request.filters.splice(0)) {
|
|
const cardinality = getFilterResultCardinality(filter)
|
|
|
|
if (!isNil(cardinality)) {
|
|
const results = repository.query([filter])
|
|
|
|
if (results.length === cardinality) {
|
|
for (const event of results) {
|
|
events.push(event)
|
|
}
|
|
|
|
break
|
|
}
|
|
}
|
|
|
|
request.filters.push(filter)
|
|
}
|
|
}
|
|
|
|
// Make sure to query our local relay too
|
|
const delay = ctx.app.requestDelay
|
|
const authTimeout = ctx.app.authTimeout
|
|
const timeout = request.closeOnEose ? ctx.app.requestTimeout : 0
|
|
const sub = baseSubscribe({delay, timeout, authTimeout, relays: [], ...request})
|
|
|
|
sub.emitter.on("event", (url: string, e: TrustedEvent) => {
|
|
repository.publish(e)
|
|
})
|
|
|
|
// Keep cached results async so the caller can set up handlers
|
|
setTimeout(() => {
|
|
for (const event of events) {
|
|
sub.emitter.emit("event", LOCAL_RELAY_URL, event)
|
|
}
|
|
})
|
|
|
|
return sub
|
|
}
|
|
|
|
export const load = (request: PartialSubscribeRequest) =>
|
|
new Promise<TrustedEvent[]>(resolve => {
|
|
const sub = subscribe({closeOnEose: true, timeout: ctx.app.requestTimeout, ...request})
|
|
const events: TrustedEvent[] = []
|
|
|
|
sub.emitter.on("event", (url: string, e: TrustedEvent) => events.push(e))
|
|
sub.emitter.on("complete", () => resolve(events))
|
|
})
|