Compare commits

..

2 Commits

Author SHA1 Message Date
Derrick Hammer 79bbf5c144
*Update dist 2023-04-08 21:02:16 -04:00
Derrick Hammer 9740402e82
*Update module hash 2023-04-08 21:02:10 -04:00
2 changed files with 166 additions and 157 deletions

321
dist/index.js vendored
View File

@ -6,174 +6,183 @@ import b4a from "b4a";
import Backoff from "backoff.js"; import Backoff from "backoff.js";
import { Mutex } from "async-mutex"; import { Mutex } from "async-mutex";
export class SwarmClient extends Client { export class SwarmClient extends Client {
useDefaultSwarm; useDefaultSwarm;
id = 0; id = 0;
_autoReconnect; _autoReconnect;
_connectBackoff; _connectBackoff;
_ready; _ready;
_connectionListener; _connectionListener;
_topics = new Set(); _topics = new Set();
_sockets = new Map(); _sockets = new Map();
get dht() { get dht() {
const self = this; const self = this;
return { return {
async ready() { async ready() {
return self.ready(); return self.ready();
}, },
}; };
}
constructor(useDefaultDht = true, autoReconnect = false) {
super();
this.useDefaultSwarm = useDefaultDht;
this._autoReconnect = autoReconnect;
this._connectBackoff = new Backoff({
strategy: "fibo",
maxAttempts: Number.MAX_SAFE_INTEGER,
});
this._connectBackoff.on("retry", (error) => {
this.logErr(error);
});
}
get swarm() {
return this.useDefaultSwarm ? undefined : this.id;
}
async connect(pubkey) {
if (typeof pubkey === "string") {
const buf = hexToBuf(pubkey);
pubkey = this.handleErrorOrReturn(buf);
} }
constructor(useDefaultDht = true, autoReconnect = false) { let existing = Array.from(this._sockets.values()).filter((socket) => {
super(); return b4a.equals(socket.remotePublicKey, pubkey);
this.useDefaultSwarm = useDefaultDht; });
this._autoReconnect = autoReconnect; if (existing.length) {
this._connectBackoff = new Backoff({ return existing[0];
strategy: "fibo",
maxAttempts: Number.MAX_SAFE_INTEGER,
});
this._connectBackoff.on("retry", (error) => {
this.logErr(error);
});
} }
get swarm() { throw new Error("not implemented");
return this.useDefaultSwarm ? undefined : this.id; }
async init() {
return await this.callModuleReturn("init", { swarm: this.swarm });
}
async ready() {
if (this._ready) {
return this._ready;
} }
async connect(pubkey) { this._listen();
if (typeof pubkey === "string") { this._ready = this.callModuleReturn("ready", { swarm: this.swarm });
const buf = hexToBuf(pubkey); await this._ready;
pubkey = this.handleErrorOrReturn(buf); this._ready = undefined;
for (const topic of this._topics) {
await this.join(topic);
}
}
async start() {
await this._connectBackoff.run(() => this.init());
await this.ready();
}
async _listen() {
if (!this._connectionListener) {
this._connectionListener = this.connectModule(
"listenConnections",
{ swarm: this.swarm },
async (socketId) => {
const socket =
this._sockets.get(socketId) ?? (await createSocket(socketId, this));
socket.on("close", () => {
this._sockets.delete(socketId);
});
if (!this._sockets.has(socketId)) {
this._sockets.set(socketId, socket);
}
this.emit("connection", socket);
} }
let existing = Array.from(this._sockets.values()).filter((socket) => { );
return b4a.equals(socket.remotePublicKey, pubkey);
});
if (existing.length) {
return existing[0];
}
throw new Error("not implemented");
} }
async init() { await this._connectionListener[1];
return await this.callModuleReturn("init", { swarm: this.swarm }); this._connectionListener = undefined;
} this.start();
async ready() { }
if (this._ready) { async addRelay(pubkey) {
return this._ready; return this.callModuleReturn("addRelay", { pubkey, swarm: this.swarm });
} }
this._listen(); async removeRelay(pubkey) {
this._ready = this.callModuleReturn("ready", { swarm: this.swarm }); return this.callModuleReturn("removeRelay", { pubkey, swarm: this.swarm });
await this._ready; }
this._ready = undefined; async clearRelays() {
for (const topic of this._topics) { return this.callModuleReturn("clearRelays", { swarm: this.swarm });
await this.join(topic); }
} async getRelays() {
} return this.callModuleReturn("getRelays", { swarm: this.swarm });
async start() { }
await this._connectBackoff.run(() => this.init()); async join(topic) {
await this.ready(); if (typeof topic === "string") {
} topic = blake2b(topic, { dkLen: 32 });
async _listen() {
if (!this._connectionListener) {
this._connectionListener = this.connectModule("listenConnections", { swarm: this.swarm }, async (socketId) => {
const socket = this._sockets.get(socketId) ?? (await createSocket(socketId, this));
socket.on("close", () => {
this._sockets.delete(socketId);
});
if (!this._sockets.has(socketId)) {
this._sockets.set(socketId, socket);
}
this.emit("connection", socket);
});
}
await this._connectionListener[1];
this._connectionListener = undefined;
this.start();
}
async addRelay(pubkey) {
return this.callModuleReturn("addRelay", { pubkey, swarm: this.swarm });
}
async removeRelay(pubkey) {
return this.callModuleReturn("removeRelay", { pubkey, swarm: this.swarm });
}
async clearRelays() {
return this.callModuleReturn("clearRelays", { swarm: this.swarm });
}
async getRelays() {
return this.callModuleReturn("getRelays", { swarm: this.swarm });
}
async join(topic) {
if (typeof topic === "string") {
topic = blake2b(topic, { dkLen: 32 });
}
this._topics.add(topic);
this.callModule("join", { id: this.id, topic });
} }
this._topics.add(topic);
this.callModule("join", { id: this.id, topic });
}
} }
export class Socket extends Client { export class Socket extends Client {
id; id;
eventUpdates = {}; eventUpdates = {};
syncMutex = new Mutex(); syncMutex = new Mutex();
swarm; swarm;
userData = null; userData = null;
constructor(id, swarm) { constructor(id, swarm) {
super(); super();
this.id = id; this.id = id;
this.swarm = swarm; this.swarm = swarm;
}
_remotePublicKey;
get remotePublicKey() {
return this._remotePublicKey;
}
_rawStream;
get rawStream() {
return this._rawStream;
}
async setup() {
let info = await this.callModuleReturn("socketGetInfo", { id: this.id });
this._remotePublicKey = info.remotePublicKey;
this._rawStream = info.rawStream;
await this.swarm.emitAsync("setup", this);
}
on(event, listener, options) {
const [update, promise] = this.connectModule(
"socketListenEvent",
{ id: this.id, event: event },
(data) => {
this.emit(event, data);
}
);
this.trackEvent(event, update);
promise.then(() => {
this.off(event, listener);
});
return super.on(event, listener, options);
}
off(event, listener) {
const updates = [...this.eventUpdates[event]];
this.eventUpdates[event] = [];
for (const func of updates) {
func();
} }
_remotePublicKey; return super.off(event, listener);
get remotePublicKey() { }
return this._remotePublicKey; write(message) {
} this.callModule("socketWrite", { id: this.id, message });
_rawStream; }
get rawStream() { end() {
return this._rawStream; this.callModule("socketExists", { id: this.id }).then(([exists]) => {
} if (exists) {
async setup() { this.callModule("socketClose", { id: this.id });
let info = await this.callModuleReturn("socketGetInfo", { id: this.id }); }
this._remotePublicKey = info.remotePublicKey; });
this._rawStream = info.rawStream; }
await this.swarm.emitAsync("setup", this); ensureEvent(event) {
} if (!(event in this.eventUpdates)) {
on(event, listener, options) { this.eventUpdates[event] = [];
const [update, promise] = this.connectModule("socketListenEvent", { id: this.id, event: event }, (data) => {
this.emit(event, data);
});
this.trackEvent(event, update);
promise.then(() => {
this.off(event, listener);
});
return super.on(event, listener, options);
}
off(event, listener) {
const updates = [...this.eventUpdates[event]];
this.eventUpdates[event] = [];
for (const func of updates) {
func();
}
return super.off(event, listener);
}
write(message) {
this.callModule("socketWrite", { id: this.id, message });
}
end() {
this.callModule("socketExists", { id: this.id }).then(([exists]) => {
if (exists) {
this.callModule("socketClose", { id: this.id });
}
});
}
ensureEvent(event) {
if (!(event in this.eventUpdates)) {
this.eventUpdates[event] = [];
}
}
trackEvent(event, update) {
this.ensureEvent(event);
this.eventUpdates[event].push(update);
} }
}
trackEvent(event, update) {
this.ensureEvent(event);
this.eventUpdates[event].push(update);
}
} }
export const MODULE = "_AVKgzVYC8Sb_qiTA6kw5BDzQ4Ch-8D4sldQJl8dXF9oTw"; export const MODULE = "_AVbH2uOdd4s4hypKB3QqaP1kxJO_Q5m6y4hsEro6dpVJQ";
export const createClient = factory(SwarmClient, MODULE); export const createClient = factory(SwarmClient, MODULE);
const socketFactory = factory(Socket, MODULE); const socketFactory = factory(Socket, MODULE);
const createSocket = async (...args) => { const createSocket = async (...args) => {
const socket = socketFactory(...args); const socket = socketFactory(...args);
await socket.setup(); await socket.setup();
return socket; return socket;
}; };

View File

@ -255,7 +255,7 @@ export class Socket extends Client {
} }
} }
export const MODULE = "_AVKgzVYC8Sb_qiTA6kw5BDzQ4Ch-8D4sldQJl8dXF9oTw"; export const MODULE = "_AVbH2uOdd4s4hypKB3QqaP1kxJO_Q5m6y4hsEro6dpVJQ";
export const createClient = factory<SwarmClient>(SwarmClient, MODULE); export const createClient = factory<SwarmClient>(SwarmClient, MODULE);