Jon Staab
138 lines
3.4 KiB
TypeScript
138 lines
3.4 KiB
TypeScript
import {last, Emitter, normalizeUrl, sleep, stripProtocol} from "@welshman/lib"
|
|
import {matchFilters} from "./Filters.js"
|
|
import type {Repository} from "./Repository.js"
|
|
import type {Filter} from "./Filters.js"
|
|
import type {HashedEvent, TrustedEvent} from "./Events.js"
|
|
|
|
// Constants and types
|
|
|
|
export const LOCAL_RELAY_URL = "local://welshman.relay/"
|
|
|
|
export const BOGUS_RELAY_URL = "bogus://welshman.relay/"
|
|
|
|
export type RelayProfile = {
|
|
url: string
|
|
icon?: string
|
|
banner?: string
|
|
name?: string
|
|
pubkey?: string
|
|
contact?: string
|
|
software?: string
|
|
version?: string
|
|
description?: string
|
|
supported_nips?: number[]
|
|
limitation?: {
|
|
min_pow_difficulty?: number
|
|
payment_required?: boolean
|
|
auth_required?: boolean
|
|
}
|
|
}
|
|
|
|
// Utils related to bare urls
|
|
|
|
export const isRelayUrl = (url: string) => {
|
|
if (!url.includes("://")) {
|
|
url = "wss://" + url
|
|
}
|
|
|
|
try {
|
|
new URL(url)
|
|
} catch (e) {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
export const isShareableRelayUrl = (url: string) =>
|
|
Boolean(
|
|
isRelayUrl(url) &&
|
|
// Is it actually a websocket url and has a dot
|
|
url.match(/^wss:\/\/.+\..+/) &&
|
|
// Don't match stuff with a port number
|
|
!url.slice(6).match(/:\d+/) &&
|
|
// Don't match stuff with a numeric tld
|
|
!url.slice(6).match(/\.\d+\b/) &&
|
|
// Don't match raw ip addresses
|
|
!url.slice(6).match(/\d+\.\d+\.\d+\.\d+/) &&
|
|
// Skip nostr.wine's virtual relays
|
|
!url.slice(6).match(/\/npub/),
|
|
)
|
|
|
|
export const normalizeRelayUrl = (url: string) => {
|
|
const prefix = url.match(/^wss?:\/\//)?.[0] || "wss://"
|
|
|
|
// Use our library to normalize
|
|
url = normalizeUrl(url, {stripHash: true, stripAuthentication: false})
|
|
|
|
// Strip the protocol, lowercase
|
|
url = stripProtocol(url).toLowerCase()
|
|
|
|
// Urls without pathnames are supposed to have a trailing slash
|
|
if (!url.includes("/")) {
|
|
url += "/"
|
|
}
|
|
|
|
return prefix + url
|
|
}
|
|
|
|
export const displayRelayUrl = (url: string) => last(url.split("://")).replace(/\/$/, "")
|
|
|
|
export const displayRelayProfile = (profile?: RelayProfile, fallback = "") =>
|
|
profile?.name || fallback
|
|
|
|
// In-memory relay implementation backed by Repository
|
|
|
|
export class Relay<E extends HashedEvent = TrustedEvent> extends Emitter {
|
|
subs = new Map<string, Filter[]>()
|
|
|
|
constructor(readonly repository: Repository<E>) {
|
|
super()
|
|
}
|
|
|
|
send(type: string, ...message: any[]) {
|
|
switch (type) {
|
|
case "EVENT":
|
|
return this.handleEVENT(message as [E])
|
|
case "CLOSE":
|
|
return this.handleCLOSE(message as [string])
|
|
case "REQ":
|
|
return this.handleREQ(message as [string, ...Filter[]])
|
|
}
|
|
}
|
|
|
|
handleEVENT([event]: [E]) {
|
|
this.repository.publish(event)
|
|
|
|
// Callers generally expect async relays
|
|
void sleep(1).then(() => {
|
|
this.emit("OK", event.id, true, "")
|
|
|
|
if (!this.repository.isDeleted(event)) {
|
|
for (const [subId, filters] of this.subs.entries()) {
|
|
if (matchFilters(filters, event)) {
|
|
this.emit("EVENT", subId, event)
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
handleCLOSE([subId]: [string]) {
|
|
this.subs.delete(subId)
|
|
}
|
|
|
|
handleREQ([subId, ...filters]: [string, ...Filter[]]) {
|
|
this.subs.set(subId, filters)
|
|
|
|
// Callers generally expect async relays
|
|
void sleep(1).then(() => {
|
|
for (const event of this.repository.query(filters)) {
|
|
this.emit("EVENT", subId, event)
|
|
}
|
|
|
|
this.emit("EOSE", subId)
|
|
})
|
|
}
|
|
}
|