You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
2.5 KiB
TypeScript
112 lines
2.5 KiB
TypeScript
import {Event, Filter, relayInit, Relay, Sub} from 'nostr-tools';
|
|
|
|
type SubCallback = (
|
|
event: Readonly<Event>,
|
|
relay: Readonly<string>,
|
|
) => void;
|
|
|
|
type Subscribe = {
|
|
cb: SubCallback;
|
|
filter: Filter;
|
|
unsub?: boolean;
|
|
};
|
|
|
|
const subList: Array<Sub> = [];
|
|
const currentSubList: Array<Subscribe> = [];
|
|
const relayMap = new Map<string, Relay>();
|
|
|
|
export const addRelay = async (url: string) => {
|
|
const relay = relayInit(url);
|
|
relay.on('connect', () => {
|
|
console.info(`connected to ${relay.url}`);
|
|
});
|
|
relay.on('error', () => {
|
|
console.warn(`failed to connect to ${relay.url}`);
|
|
});
|
|
try {
|
|
await relay.connect();
|
|
currentSubList.forEach(({cb, filter}) => subscribe(cb, filter, relay));
|
|
relayMap.set(url, relay);
|
|
} catch {
|
|
console.warn(`could not connect to ${url}`);
|
|
}
|
|
};
|
|
|
|
export const unsubscribe = (sub: Sub) => {
|
|
sub.unsub();
|
|
subList.splice(subList.indexOf(sub), 1);
|
|
};
|
|
|
|
const subscribe = (
|
|
cb: SubCallback,
|
|
filter: Filter,
|
|
relay: Relay,
|
|
unsub?: boolean
|
|
) => {
|
|
const sub = relay.sub([filter]);
|
|
subList.push(sub);
|
|
sub.on('event', (event: Event) => {
|
|
cb(event, relay.url);
|
|
});
|
|
if (unsub) {
|
|
sub.on('eose', () => {
|
|
// console.log('eose', relay.url);
|
|
unsubscribe(sub);
|
|
});
|
|
}
|
|
return sub;
|
|
};
|
|
|
|
export const sub = (obj: Subscribe) => {
|
|
currentSubList.push(obj);
|
|
relayMap.forEach((relay) => subscribe(obj.cb, obj.filter, relay, obj.unsub));
|
|
};
|
|
|
|
export const subOnce = (
|
|
obj: Subscribe & {relay: string}
|
|
) => {
|
|
const relay = relayMap.get(obj.relay);
|
|
if (relay) {
|
|
const sub = subscribe(obj.cb, obj.filter, relay);
|
|
sub.on('eose', () => {
|
|
// console.log('eose', obj.relay);
|
|
unsubscribe(sub);
|
|
});
|
|
}
|
|
};
|
|
|
|
export const unsubAll = () => {
|
|
subList.forEach(unsubscribe);
|
|
currentSubList.length = 0;
|
|
};
|
|
|
|
type PublishCallback = (
|
|
relay: string,
|
|
errorMessage?: string,
|
|
) => void;
|
|
|
|
export const publish = (
|
|
event: Event,
|
|
cb: PublishCallback,
|
|
) => {
|
|
relayMap.forEach((relay, url) => {
|
|
const pub = relay.publish(event);
|
|
pub.on('ok', () => {
|
|
console.info(`${relay.url} has accepted our event`);
|
|
cb(relay.url);
|
|
});
|
|
pub.on('failed', (reason: any) => {
|
|
console.error(`failed to publish to ${relay.url}: ${reason}`);
|
|
cb(relay.url, reason);
|
|
});
|
|
});
|
|
};
|
|
|
|
|
|
addRelay('wss://relay.snort.social');
|
|
addRelay('wss://nostr.bitcoiner.social');
|
|
addRelay('wss://nostr.mom');
|
|
addRelay('wss://relay.nostr.bg');
|
|
addRelay('wss://nos.lol');
|
|
// addRelay('wss://relay.nostr.ch');
|