kernel-swarm-client/src/index.ts

141 lines
3.7 KiB
TypeScript
Raw Normal View History

2022-08-03 16:05:10 +00:00
import { Buffer } from "buffer";
2023-02-01 12:54:20 +00:00
import { Client, factory } from "@lumeweb/libkernel-universal";
import { hexToBuf, DataFn, ErrTuple } from "@siaweb/libweb";
2022-07-20 01:38:29 +00:00
import type { EventEmitter } from "eventemitter3";
2023-02-01 12:54:20 +00:00
export class SwarmClient extends Client {
private useDefaultSwarm: boolean;
private id: number = 0;
2022-07-20 01:38:29 +00:00
2023-02-01 12:54:20 +00:00
get swarm(): number | undefined {
return this.useDefaultSwarm ? undefined : this.id;
2022-08-03 16:05:10 +00:00
}
2022-07-20 01:38:29 +00:00
2022-08-03 16:05:10 +00:00
constructor(useDefaultDht = true) {
2023-02-01 12:54:20 +00:00
super();
this.useDefaultSwarm = useDefaultDht;
2022-08-03 16:05:10 +00:00
}
2023-02-01 12:54:20 +00:00
public async connect(pubkey: string | Uint8Array): Promise<Socket> {
if (typeof pubkey === "string") {
const buf = hexToBuf(pubkey);
pubkey = this.handleErrorOrReturn(buf);
}
2023-02-01 12:54:20 +00:00
const resp = this.callModuleReturn("connect", {
pubkey,
swarm: this.swarm,
}) as any;
return createSocket(resp.id);
2022-08-03 16:05:10 +00:00
}
2023-02-01 13:47:19 +00:00
async init(): Promise<ErrTuple> {
return this.callModuleReturn("init", { swarm: this.swarm });
}
async ready(): Promise<void> {
await this.callModuleReturn("ready", { swarm: this.swarm });
this.connectModule(
"listenConnections",
{ swarm: this.swarm },
(socketId: any) => {
2023-02-06 08:26:06 +00:00
this.emit("connection", createSocket(socketId));
}
);
2022-08-03 16:05:10 +00:00
}
2022-08-03 16:05:10 +00:00
public async addRelay(pubkey: string): Promise<void> {
2023-02-01 12:54:20 +00:00
return this.callModuleReturn("addRelay", { pubkey, swarm: this.swarm });
2022-08-03 16:05:10 +00:00
}
2022-08-03 16:05:10 +00:00
public async removeRelay(pubkey: string): Promise<void> {
2023-02-01 12:54:20 +00:00
return this.callModuleReturn("removeRelay", { pubkey, swarm: this.swarm });
2022-08-03 16:05:10 +00:00
}
2022-08-03 16:05:10 +00:00
public async clearRelays(): Promise<void> {
2023-02-01 12:54:20 +00:00
return this.callModuleReturn("clearRelays", { swarm: this.swarm });
2022-08-03 16:05:10 +00:00
}
public async getRelays(): Promise<string[]> {
2023-02-01 12:54:20 +00:00
return this.callModuleReturn("getRelays", { swarm: this.swarm });
}
2023-02-01 19:06:13 +00:00
2023-02-01 19:08:11 +00:00
public async join(topic: Buffer): Promise<void> {
2023-02-01 19:06:13 +00:00
this.callModule("join", { id: this.id, topic });
}
2022-08-03 16:05:10 +00:00
}
2023-02-01 12:54:20 +00:00
export class Socket extends Client {
2022-08-03 16:05:10 +00:00
private id: number;
private eventUpdates: { [event: string]: DataFn[] } = {};
constructor(id: number) {
super();
this.id = id;
}
on<T extends EventEmitter.EventNames<string | symbol>>(
event: T,
fn: EventEmitter.EventListener<string | symbol, T>,
context?: any
): this {
2023-02-01 12:54:20 +00:00
const [update, promise] = this.connectModule(
"socketListenEvent",
{ id: this.id, event: event },
2022-08-03 16:05:10 +00:00
(data: any) => {
this.emit(event, data);
2022-08-03 16:05:10 +00:00
}
);
this.trackEvent(event as string, update);
2022-08-03 16:05:10 +00:00
promise.then(() => {
this.off(event as string, fn);
2022-08-03 16:05:10 +00:00
});
return super.on(event, fn, context) as this;
2022-08-03 16:05:10 +00:00
}
off<T extends EventEmitter.EventNames<string | symbol>>(
event: T,
fn?: EventEmitter.EventListener<string | symbol, T>,
context?: any,
once?: boolean
): this {
const updates = [...this.eventUpdates[event as string]];
this.eventUpdates[event as string] = [];
2022-08-03 16:05:10 +00:00
for (const func of updates) {
func();
2022-08-03 16:05:10 +00:00
}
return super.off(event, fn, context, once);
2022-08-03 16:05:10 +00:00
}
write(message: string | Buffer): void {
2023-02-06 08:33:44 +00:00
this.callModule("socketWrite", { id: this.id, message });
2022-08-03 16:05:10 +00:00
}
end(): void {
2023-02-01 12:54:20 +00:00
this.callModule("socketExists", { id: this.id }).then(
([exists]: ErrTuple) => {
if (exists) {
this.callModule("socketClose", { id: this.id });
}
}
);
2022-08-03 16:05:10 +00:00
}
private ensureEvent(event: string): void {
if (!(event in this.eventUpdates)) {
this.eventUpdates[event] = [];
}
}
private trackEvent(event: string, update: DataFn): void {
this.ensureEvent(event as string);
this.eventUpdates[event].push(update);
}
2022-07-20 01:38:29 +00:00
}
2023-02-01 12:54:20 +00:00
2023-02-01 17:33:17 +00:00
const MODULE = "_A7ClA0mSa1-Pg5c4V3C0H_fnhAFjgccITYT83Euc7t_9A";
2023-02-01 12:54:20 +00:00
export const createClient = factory<SwarmClient>(SwarmClient, MODULE);
const createSocket = factory<Socket>(Socket, MODULE);