*Add addRelay interface method
This commit is contained in:
parent
9b4ac8ff93
commit
afc332b75e
133
src/index.ts
133
src/index.ts
|
|
@ -1,10 +1,10 @@
|
|||
import {
|
||||
callModule as callModuleKernel,
|
||||
connectModule as connectModuleKernel,
|
||||
callModule as callModuleKernel,
|
||||
connectModule as connectModuleKernel,
|
||||
} from "libkernel";
|
||||
import {
|
||||
callModule as callModuleModule,
|
||||
connectModule as connectModuleModule,
|
||||
callModule as callModuleModule,
|
||||
connectModule as connectModuleModule,
|
||||
} from "libkmodule";
|
||||
import { EventEmitter } from "events";
|
||||
import { DataFn, ErrTuple } from "libskynet";
|
||||
|
|
@ -13,82 +13,89 @@ import { Buffer } from "buffer";
|
|||
const DHT_MODULE = "AQD1IgE4lTZkq1fqdoYGojKRNrSk0YQ_wrHbRtIiHDrnow";
|
||||
|
||||
let callModule: typeof callModuleModule,
|
||||
connectModule: typeof connectModuleModule;
|
||||
connectModule: typeof connectModuleModule;
|
||||
|
||||
if (window.document) {
|
||||
callModule = callModuleKernel;
|
||||
connectModule = connectModuleKernel;
|
||||
callModule = callModuleKernel;
|
||||
connectModule = connectModuleKernel;
|
||||
} else {
|
||||
callModule = callModuleModule;
|
||||
connectModule = connectModuleModule;
|
||||
callModule = callModuleModule;
|
||||
connectModule = connectModuleModule;
|
||||
}
|
||||
|
||||
export class DHT {
|
||||
public async connect(pubkey: string): Promise<Socket> {
|
||||
const [resp, err] = await callModule(DHT_MODULE, "connect", { pubkey });
|
||||
if (err) {
|
||||
throw err;
|
||||
}
|
||||
return new Socket(resp.id);
|
||||
public async connect(pubkey: string): Promise<Socket> {
|
||||
const [resp, err] = await callModule(DHT_MODULE, "connect", { pubkey });
|
||||
if (err) {
|
||||
throw err;
|
||||
}
|
||||
return new Socket(resp.id);
|
||||
}
|
||||
|
||||
async ready(): Promise<ErrTuple> {
|
||||
return callModule(DHT_MODULE, "ready");
|
||||
async ready(): Promise<ErrTuple> {
|
||||
return callModule(DHT_MODULE, "ready");
|
||||
}
|
||||
|
||||
public async addRelay(pubkey: string): Promise<void> {
|
||||
const [, err] = await callModule(DHT_MODULE, "addRelay", { pubkey });
|
||||
if (err) {
|
||||
throw err;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
export class Socket extends EventEmitter {
|
||||
private id: number;
|
||||
private eventUpdates: { [event: string]: DataFn[] } = {};
|
||||
private id: number;
|
||||
private eventUpdates: { [event: string]: DataFn[] } = {};
|
||||
|
||||
constructor(id: number) {
|
||||
super();
|
||||
this.id = id;
|
||||
constructor(id: number) {
|
||||
super();
|
||||
this.id = id;
|
||||
}
|
||||
|
||||
on(eventName: string, listener: (...args: any[]) => void): this {
|
||||
const [update, promise] = connectModule(
|
||||
DHT_MODULE,
|
||||
"listenSocketEvent",
|
||||
{ id: this.id, event: eventName },
|
||||
(data: any) => {
|
||||
this.emit(eventName, data);
|
||||
}
|
||||
);
|
||||
this.trackEvent(eventName, update);
|
||||
|
||||
promise.then(() => {
|
||||
this.off(eventName, listener);
|
||||
});
|
||||
|
||||
return super.on(eventName, listener);
|
||||
}
|
||||
|
||||
off(type: string, listener: any): this {
|
||||
const updates = [...this.eventUpdates[type]];
|
||||
this.eventUpdates[type] = [];
|
||||
for (const func of updates) {
|
||||
func({ action: "off" });
|
||||
}
|
||||
return super.off(type, listener);
|
||||
}
|
||||
|
||||
on(eventName: string, listener: (...args: any[]) => void): this {
|
||||
const [update, promise] = connectModule(
|
||||
DHT_MODULE,
|
||||
"listenSocketEvent",
|
||||
{ id: this.id, event: eventName },
|
||||
(data: any) => {
|
||||
this.emit(eventName, data);
|
||||
}
|
||||
);
|
||||
this.trackEvent(eventName, update);
|
||||
write(message: string | Buffer): void {
|
||||
callModule(DHT_MODULE, "write", { id: this.id, message });
|
||||
}
|
||||
|
||||
promise.then(() => {
|
||||
this.off(eventName, listener);
|
||||
});
|
||||
end(): void {
|
||||
callModule(DHT_MODULE, "close", { id: this.id });
|
||||
}
|
||||
|
||||
return super.on(eventName, listener);
|
||||
private ensureEvent(event: string): void {
|
||||
if (!(event in this.eventUpdates)) {
|
||||
this.eventUpdates[event] = [];
|
||||
}
|
||||
}
|
||||
|
||||
off(type: string, listener: any): this {
|
||||
const updates = [...this.eventUpdates[type]];
|
||||
this.eventUpdates[type] = [];
|
||||
for (const func of updates) {
|
||||
func({ action: "off" });
|
||||
}
|
||||
return super.off(type, listener);
|
||||
}
|
||||
|
||||
write(message: string | Buffer): void {
|
||||
callModule(DHT_MODULE, "write", { id: this.id, message });
|
||||
}
|
||||
|
||||
end(): void {
|
||||
callModule(DHT_MODULE, "close", { id: this.id });
|
||||
}
|
||||
|
||||
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);
|
||||
}
|
||||
private trackEvent(event: string, update: DataFn): void {
|
||||
this.ensureEvent(event as string);
|
||||
this.eventUpdates[event].push(update);
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue