*Update dist

This commit is contained in:
Derrick Hammer 2022-07-26 19:28:53 -04:00
parent 685bc59894
commit a0ecea1dc9
3 changed files with 39 additions and 28 deletions

6
dist/index.d.ts vendored
View File

@ -1,8 +1,10 @@
import DhtNode from "@hyperswarm/dht-relay";
export default class DHT {
private _wsPool;
private _options;
private _relays;
private _activeRelays;
private _maxConnections;
private _inited;
constructor(opts?: {});
ready(): Promise<void>;
get relays(): string[];
@ -11,7 +13,7 @@ export default class DHT {
clearRelays(): void;
private isServerAvailable;
connect(pubkey: string, options?: {}): Promise<DhtNode>;
getAvailableRelay(): Promise<string | boolean>;
private fillConnections;
}
export declare function hashDataKey(dataKey: string): Uint8Array;
//# sourceMappingURL=index.d.ts.map

2
dist/index.d.ts.map vendored
View File

@ -1 +1 @@
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,OAAO,MAAM,uBAAuB,CAAC;AAkB5C,MAAM,CAAC,OAAO,OAAO,GAAG;IACpB,OAAO,CAAC,OAAO,CAA8B;IAC7C,OAAO,CAAC,QAAQ,CAAM;IACtB,OAAO,CAAC,OAAO,CAAoC;gBAEvC,IAAI,KAAK;IAOrB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAItB,IAAI,MAAM,IAAI,MAAM,EAAE,CAErB;IAEY,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAgChD,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAWpC,WAAW,IAAI,IAAI;YAKZ,iBAAiB;IAazB,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC;IAavD,iBAAiB,IAAI,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC;CAUvD;AAED,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,UAAU,CAEvD"}
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,OAAO,MAAM,uBAAuB,CAAC;AAkB5C,MAAM,CAAC,OAAO,OAAO,GAAG;IACpB,OAAO,CAAC,QAAQ,CAAM;IACtB,OAAO,CAAC,OAAO,CAAkC;IACjD,OAAO,CAAC,aAAa,CAA0C;IAC/D,OAAO,CAAC,eAAe,CAAM;IAC7B,OAAO,CAAC,OAAO,CAAS;gBAEZ,IAAI,KAAK;IAMrB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAKtB,IAAI,MAAM,IAAI,MAAM,EAAE,CAErB;IAEY,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAiChD,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAgBpC,WAAW,IAAI,IAAI;YAIZ,iBAAiB;IAazB,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC;YAU/C,eAAe;CAoBhC;AAED,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,UAAU,CAEvD"}

59
dist/index.js vendored
View File

@ -3,30 +3,32 @@ import DhtNode from "@hyperswarm/dht-relay";
// @ts-ignore
import Stream from "@hyperswarm/dht-relay/ws";
// @ts-ignore
import createRoundRobin from "@derhuerst/round-robin-scheduler";
// @ts-ignore
import { Buffer } from "buffer";
// @ts-ignore
// @ts-ignore
import { blake2b } from "libskynet";
// @ts-ignore
import { registryRead } from "libkmodule";
import { unpack } from "msgpackr";
import randomNumber from "random-number-csprng";
const REGISTRY_DHT_KEY = "lumeweb-dht-node";
export default class DHT {
_wsPool;
_options;
_relays = {};
_relays = new Map();
_activeRelays = new Map();
_maxConnections = 10;
_inited = false;
constructor(opts = {}) {
// @ts-ignore
opts.custodial = false;
this._options = opts;
this._wsPool = createRoundRobin();
}
ready() {
return Promise.resolve();
this._inited = true;
return this.fillConnections();
}
get relays() {
return Object.keys(this._relays);
return [...this._relays.keys()];
}
async addRelay(pubkey) {
let entry = await registryRead(Uint8Array.from(Buffer.from(pubkey, "hex")), hashDataKey(REGISTRY_DHT_KEY));
@ -44,22 +46,25 @@ export default class DHT {
if (isNaN(parseInt(port))) {
return false;
}
const connection = `wss://${domain}:${port}/`;
this._wsPool.add(connection);
this._relays[pubkey] = connection;
this._relays[pubkey] = `wss://${domain}:${port}/`;
if (this._inited) {
await this.fillConnections();
}
return true;
}
removeRelay(pubkey) {
if (!(pubkey in this._relays)) {
if (!this._relays.has(pubkey)) {
return false;
}
this._wsPool.remove(this._relays[pubkey]);
delete this._relays[pubkey];
if (this._activeRelays.has(pubkey)) {
this._activeRelays.get(pubkey).destroy();
this._activeRelays.delete(pubkey);
}
this._relays.delete(pubkey);
return true;
}
clearRelays() {
this._wsPool = createRoundRobin();
this._relays = {};
[...this._relays.keys()].forEach(this.removeRelay);
}
async isServerAvailable(connection) {
return new Promise((resolve) => {
@ -74,22 +79,26 @@ export default class DHT {
});
}
async connect(pubkey, options = {}) {
const relay = await this.getAvailableRelay();
if (!relay) {
if (this._activeRelays.size === 0) {
throw new Error("Failed to find an available relay");
}
const node = new DhtNode(new Stream(true, new WebSocket(relay)), this._options);
await node.ready();
const node = this._activeRelays.get([...this._activeRelays.keys()][await randomNumber(0, this._activeRelays.size - 1)]);
return node.connect(pubkey, options);
}
async getAvailableRelay() {
for (let i = 0; i < this._wsPool.length; i++) {
const relay = this._wsPool.get();
if (await this.isServerAvailable(relay)) {
return relay;
async fillConnections() {
let available = [...this._relays.keys()].filter(x => [...this._activeRelays.keys()].includes(x));
let relayPromises = [];
while (this._activeRelays.size <= Math.min(this._maxConnections, available.length + this._activeRelays.size)) {
const relayIndex = await randomNumber(0, available.length - 1);
const connection = available[relayIndex];
if (!this.isServerAvailable(connection)) {
continue;
}
const node = new DhtNode(new Stream(true, new WebSocket(this._activeRelays.get(connection))), this._options);
this._activeRelays.set(available[relayIndex], node);
relayPromises.push(node.ready());
}
return false;
return Promise.allSettled(relayPromises);
}
}
export function hashDataKey(dataKey) {