48 lines
959 B
TypeScript
48 lines
959 B
TypeScript
import b4a from "b4a";
|
|
|
|
export interface Peer {
|
|
host: string;
|
|
port: number;
|
|
}
|
|
|
|
export type PeerSource = (pubkey: Buffer) => Promise<boolean | Peer>;
|
|
|
|
export class PeerDiscovery {
|
|
private _sources: Map<string, PeerSource> = new Map<string, PeerSource>();
|
|
|
|
public registerSource(name: string, source: PeerSource): boolean {
|
|
if (this._sources.has(name)) {
|
|
return false;
|
|
}
|
|
this._sources.set(name, source);
|
|
|
|
return true;
|
|
}
|
|
|
|
public removeSource(name: string): boolean {
|
|
if (!this._sources.has(name)) {
|
|
return false;
|
|
}
|
|
|
|
this._sources.delete(name);
|
|
|
|
return true;
|
|
}
|
|
|
|
public async discover(pubkey: string | Buffer): Promise<Peer | boolean> {
|
|
if (!b4a.isBuffer(pubkey)) {
|
|
pubkey = b4a.from(pubkey, "hex") as Buffer;
|
|
}
|
|
|
|
for (const source of this._sources.values()) {
|
|
const result = await source(pubkey);
|
|
|
|
if (result) {
|
|
return result;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|