Compare commits
No commits in common. "v0.0.2-develop.1" and "v0.0.1" have entirely different histories.
v0.0.2-dev
...
v0.0.1
|
|
@ -1,13 +0,0 @@
|
||||||
name: Build/Publish
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
branches:
|
|
||||||
- master
|
|
||||||
- develop
|
|
||||||
- develop-*
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
main:
|
|
||||||
uses: lumeweb/github-node-deploy-workflow/.github/workflows/main.yml@master
|
|
||||||
secrets: inherit
|
|
||||||
|
|
@ -1,8 +0,0 @@
|
||||||
{
|
|
||||||
"preset": [
|
|
||||||
"@lumeweb/presetter-kernel-module-preset"
|
|
||||||
],
|
|
||||||
"config": {
|
|
||||||
"official": true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1 +0,0 @@
|
||||||
## [0.0.2-develop.1](https://git.lumeweb.com/LumeWeb/kernel-swarm/compare/v0.0.1...v0.0.2-develop.1) (2023-07-01)
|
|
||||||
2
LICENSE
2
LICENSE
|
|
@ -1,6 +1,6 @@
|
||||||
MIT License
|
MIT License
|
||||||
|
|
||||||
Copyright (c) 2022 Hammer Technologies LLC
|
Copyright (c) 2022 Lume Web
|
||||||
|
|
||||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
of this software and associated documentation files (the "Software"), to deal
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
|
|
||||||
|
|
@ -1,2 +1,2 @@
|
||||||
# kernel-swarm
|
# kernel-dht
|
||||||
Skynet kernel module for DHT requests using the @lumeweb/swarm-rpc-client package
|
Skynet kernel module for DHT requests using the @lumeweb/dht-rpc-client package
|
||||||
|
|
|
||||||
File diff suppressed because it is too large
Load Diff
38
package.json
38
package.json
|
|
@ -1,38 +0,0 @@
|
||||||
{
|
|
||||||
"name": "@lumeweb/kernel-swarm",
|
|
||||||
"author": {
|
|
||||||
"name": "Hammer Technologies LLC",
|
|
||||||
"email": "contact@lumeweb.com"
|
|
||||||
},
|
|
||||||
"repository": {
|
|
||||||
"type": "git",
|
|
||||||
"url": "gitea@git.lumeweb.com:LumeWeb/kernel-swarm.git"
|
|
||||||
},
|
|
||||||
"scripts": {
|
|
||||||
"prepare": "presetter bootstrap",
|
|
||||||
"build": "run build",
|
|
||||||
"semantic-release": "semantic-release"
|
|
||||||
},
|
|
||||||
"type": "module",
|
|
||||||
"dependencies": {
|
|
||||||
"@lumeweb/hyperswarm-web": "0.0.2-develop.4",
|
|
||||||
"@lumeweb/libkernel": "^0.1.0-develop.7",
|
|
||||||
"@lumeweb/rpc": "0.0.2-develop.1",
|
|
||||||
"@noble/curves": "^1.1.0",
|
|
||||||
"@peculiar/webcrypto": "^1.4.3",
|
|
||||||
"async-mutex": "^0.4.0",
|
|
||||||
"b4a": "^1.6.4",
|
|
||||||
"eventemitter2": "^6.4.9",
|
|
||||||
"hyperswarm": "^4.5.1",
|
|
||||||
"noise-handshake": "^3.0.3",
|
|
||||||
"p-defer": "^4.0.0",
|
|
||||||
"protomux": "^3.5.0"
|
|
||||||
},
|
|
||||||
"devDependencies": {
|
|
||||||
"@lumeweb/presetter-kernel-module-preset": "^0.1.0-develop.20",
|
|
||||||
"@types/b4a": "^1.6.0",
|
|
||||||
"presetter": "*"
|
|
||||||
},
|
|
||||||
"version": "0.0.2-develop.1",
|
|
||||||
"readme": "ERROR: No README data found!"
|
|
||||||
}
|
|
||||||
|
|
@ -1,45 +0,0 @@
|
||||||
diff --git a/node_modules/protomux/index.js b/node_modules/protomux/index.js
|
|
||||||
index 07bf0ad..c8db59c 100644
|
|
||||||
--- a/node_modules/protomux/index.js
|
|
||||||
+++ b/node_modules/protomux/index.js
|
|
||||||
@@ -195,34 +195,34 @@ class Channel {
|
|
||||||
type,
|
|
||||||
encoding,
|
|
||||||
onmessage,
|
|
||||||
- recv (state, session) {
|
|
||||||
+ async recv (state, session) {
|
|
||||||
session._track(m.onmessage(encoding.decode(state), session))
|
|
||||||
},
|
|
||||||
- send (m, session = s) {
|
|
||||||
+ async send (m, session = s) {
|
|
||||||
if (session.closed === true) return false
|
|
||||||
|
|
||||||
const mux = session._mux
|
|
||||||
const state = { buffer: null, start: 0, end: typeLen }
|
|
||||||
|
|
||||||
if (mux._batch !== null) {
|
|
||||||
- encoding.preencode(state, m)
|
|
||||||
+ await encoding.preencode(state, m)
|
|
||||||
state.buffer = mux._alloc(state.end)
|
|
||||||
|
|
||||||
c.uint.encode(state, type)
|
|
||||||
- encoding.encode(state, m)
|
|
||||||
+ await encoding.encode(state, m)
|
|
||||||
|
|
||||||
mux._pushBatch(session._localId, state.buffer)
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
c.uint.preencode(state, session._localId)
|
|
||||||
- encoding.preencode(state, m)
|
|
||||||
+ await encoding.preencode(state, m)
|
|
||||||
|
|
||||||
state.buffer = mux._alloc(state.end)
|
|
||||||
|
|
||||||
c.uint.encode(state, session._localId)
|
|
||||||
c.uint.encode(state, type)
|
|
||||||
- encoding.encode(state, m)
|
|
||||||
+ await encoding.encode(state, m)
|
|
||||||
|
|
||||||
mux.drained = mux.stream.write(state.buffer)
|
|
||||||
|
|
||||||
72
src/addr.ts
72
src/addr.ts
|
|
@ -1,72 +0,0 @@
|
||||||
/*
|
|
||||||
The following is based on https://github.com/yggdrasil-network/yggdrasil-go/blob/develop/src/address/address.go, which is licensed LGPL3. Full credit to them for the idea and original algorithm
|
|
||||||
*/
|
|
||||||
|
|
||||||
export function pubKeyToIpv6(publicKey: Uint8Array) {
|
|
||||||
const keySize = 32;
|
|
||||||
if (publicKey.length !== keySize) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
const buf = new Uint8Array(keySize);
|
|
||||||
for (let i = 0; i < keySize; i++) {
|
|
||||||
buf[i] = buf[i] = publicKey[i] ^ 0xff;
|
|
||||||
}
|
|
||||||
|
|
||||||
const prefix = [0x02];
|
|
||||||
const ones = getLeadingOnes(buf);
|
|
||||||
const nodeId = getTruncatedNodeID(buf);
|
|
||||||
|
|
||||||
const addr = new Uint8Array(prefix.length + 1 + nodeId.length);
|
|
||||||
addr.set(prefix, 0);
|
|
||||||
addr[prefix.length] = ones;
|
|
||||||
addr.set(nodeId, prefix.length + 1);
|
|
||||||
|
|
||||||
const result: string[] = [];
|
|
||||||
for (let i = 0; i < 8; i++) {
|
|
||||||
const num1 = addr[i * 2].toString(16).padStart(2, "0");
|
|
||||||
const num2 = addr[i * 2 + 1].toString(16).padStart(2, "0");
|
|
||||||
result.push(`${num1}${num2}`);
|
|
||||||
}
|
|
||||||
return result.join(":");
|
|
||||||
}
|
|
||||||
|
|
||||||
function getLeadingOnes(buf: Uint8Array) {
|
|
||||||
let done = false;
|
|
||||||
let ones = 0;
|
|
||||||
for (let i = 0; i < buf.length * 8; i++) {
|
|
||||||
const bit = (buf[i >>> 3] & (0x80 >> (i & 7))) >> (7 - (i & 7));
|
|
||||||
if (!done && bit !== 0) {
|
|
||||||
ones++;
|
|
||||||
} else if (!done && bit === 0) {
|
|
||||||
done = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return ones;
|
|
||||||
}
|
|
||||||
|
|
||||||
function getTruncatedNodeID(buf: Uint8Array) {
|
|
||||||
const result: number[] = [];
|
|
||||||
let done = false;
|
|
||||||
let bits = 0;
|
|
||||||
let nBits = 0;
|
|
||||||
for (let i = 0; i < buf.length * 8; i++) {
|
|
||||||
const bit = (buf[i >>> 3] & (0x80 >> (i & 7))) >> (7 - (i & 7));
|
|
||||||
if (!done && bit !== 0) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!done && bit === 0) {
|
|
||||||
done = true;
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
bits = (bits << 1) | bit;
|
|
||||||
nBits++;
|
|
||||||
if (nBits === 8) {
|
|
||||||
nBits = 0;
|
|
||||||
result.push(bits);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
627
src/index.ts
627
src/index.ts
|
|
@ -1,627 +0,0 @@
|
||||||
// @ts-ignore
|
|
||||||
import Hyperswarm from "@lumeweb/hyperswarm-web";
|
|
||||||
import type { ActiveQuery } from "@lumeweb/libkernel/module";
|
|
||||||
import {
|
|
||||||
addHandler,
|
|
||||||
getKey,
|
|
||||||
handleMessage,
|
|
||||||
handlePresentKey as handlePresentKeyModule,
|
|
||||||
} from "@lumeweb/libkernel/module";
|
|
||||||
import { Buffer } from "buffer";
|
|
||||||
import { ed25519 } from "@noble/curves/ed25519";
|
|
||||||
import b4a from "b4a";
|
|
||||||
import { pubKeyToIpv6 } from "./addr.js";
|
|
||||||
import { EventEmitter2 as EventEmitter } from "eventemitter2";
|
|
||||||
import { logErr } from "@lumeweb/libkernel";
|
|
||||||
// @ts-ignore
|
|
||||||
import Protomux from "protomux";
|
|
||||||
import defer, { DeferredPromise } from "p-defer";
|
|
||||||
|
|
||||||
const MAX_PEER_LISTENERS = 20;
|
|
||||||
|
|
||||||
interface SwarmConnection {
|
|
||||||
swarm: number;
|
|
||||||
conn: any;
|
|
||||||
channels: Map<number, Protomux>;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface SwarmEvents {
|
|
||||||
swarm: number;
|
|
||||||
events: EventEmitter;
|
|
||||||
}
|
|
||||||
|
|
||||||
const connections = new Map<number, SwarmConnection>();
|
|
||||||
const swarmInstances = new Map<number, Hyperswarm>();
|
|
||||||
const swarmEvents = new Map<number, SwarmEvents>();
|
|
||||||
|
|
||||||
let defaultSwarm: Hyperswarm;
|
|
||||||
|
|
||||||
let moduleReadyResolve: Function;
|
|
||||||
let moduleReady: Promise<void> = new Promise((resolve) => {
|
|
||||||
moduleReadyResolve = resolve;
|
|
||||||
});
|
|
||||||
|
|
||||||
onmessage = handleMessage;
|
|
||||||
|
|
||||||
function idFactory(start = 1) {
|
|
||||||
let id = start;
|
|
||||||
|
|
||||||
return function nextId() {
|
|
||||||
const nextId = id;
|
|
||||||
id += 1;
|
|
||||||
return nextId;
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
const getSwarmId = idFactory();
|
|
||||||
const getSocketId = idFactory();
|
|
||||||
const getChannelId = idFactory();
|
|
||||||
const getMessageId = idFactory();
|
|
||||||
|
|
||||||
addHandler("presentKey", handlePresentKey);
|
|
||||||
addHandler("join", handleJoin);
|
|
||||||
addHandler("getPeerByPubkey", handleGetPeerByPubkey);
|
|
||||||
|
|
||||||
addHandler("addRelay", handleAddRelay);
|
|
||||||
addHandler("removeRelay", handleRemoveRelay);
|
|
||||||
addHandler("clearRelays", handleClearRelays);
|
|
||||||
addHandler("getRelays", handleGetRelays);
|
|
||||||
addHandler("init", handleInit);
|
|
||||||
addHandler("ready", handleReady);
|
|
||||||
addHandler("listenConnections", handleListenConnections, {
|
|
||||||
receiveUpdates: true,
|
|
||||||
});
|
|
||||||
addHandler("socketGetInfo", handleGetSocketInfo);
|
|
||||||
|
|
||||||
addHandler("socketExists", handleSocketExists);
|
|
||||||
addHandler("socketListenEvent", handleSocketListenEvent, {
|
|
||||||
receiveUpdates: true,
|
|
||||||
});
|
|
||||||
addHandler("socketWrite", handleWriteSocketEvent);
|
|
||||||
addHandler("socketClose", handleCloseSocketEvent);
|
|
||||||
addHandler("createProtomuxChannel", handleCreateProtomuxChannel, {
|
|
||||||
receiveUpdates: true,
|
|
||||||
});
|
|
||||||
addHandler("createProtomuxMessage", handleCreateProtomuxMessage, {
|
|
||||||
receiveUpdates: true,
|
|
||||||
});
|
|
||||||
addHandler("createSwarm", handleCreateSwarm);
|
|
||||||
|
|
||||||
async function handlePresentKey(aq: ActiveQuery) {
|
|
||||||
const pubkey = ed25519.getPublicKey(aq.callerInput.rootKey);
|
|
||||||
handlePresentKeyModule({
|
|
||||||
callerInput: {
|
|
||||||
seed: {
|
|
||||||
publicKey: ed25519.getPublicKey(aq.callerInput.rootKey),
|
|
||||||
secretKey: b4a.concat([aq.callerInput.rootKey, pubkey]),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
} as ActiveQuery);
|
|
||||||
|
|
||||||
if (!defaultSwarm) {
|
|
||||||
defaultSwarm = swarmInstances.get(await createSwarm()) as Hyperswarm;
|
|
||||||
}
|
|
||||||
moduleReadyResolve();
|
|
||||||
}
|
|
||||||
|
|
||||||
async function createSwarm(): Promise<number> {
|
|
||||||
const privateKey = await getKey();
|
|
||||||
const swarmInstance = new Hyperswarm({
|
|
||||||
keyPair: {
|
|
||||||
publicKey: ed25519.getPublicKey(privateKey),
|
|
||||||
secretKey: privateKey,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
const id = getSwarmId();
|
|
||||||
swarmInstances.set(id, swarmInstance);
|
|
||||||
|
|
||||||
swarmInstance.onSelf("init", () => {
|
|
||||||
const swarmInstanceEvents = new EventEmitter();
|
|
||||||
swarmInstanceEvents.setMaxListeners(MAX_PEER_LISTENERS);
|
|
||||||
swarmEvents.set(id, { swarm: id, events: swarmInstanceEvents });
|
|
||||||
swarmInstance.on("connection", (peer: any) => {
|
|
||||||
const socketId = getSocketId();
|
|
||||||
connections.set(socketId, {
|
|
||||||
swarm: id,
|
|
||||||
conn: peer,
|
|
||||||
channels: new Map<number, Protomux>(),
|
|
||||||
});
|
|
||||||
|
|
||||||
peer.once("close", () => {
|
|
||||||
connections.delete(socketId);
|
|
||||||
});
|
|
||||||
|
|
||||||
swarmInstanceEvents.emit("connection", peer);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
swarmInstance.onSelf("close", (...args) => {
|
|
||||||
swarmEvents.get(id)?.events.emit("close", ...args);
|
|
||||||
swarmEvents.get(id)?.events.removeAllListeners();
|
|
||||||
swarmEvents.delete(id);
|
|
||||||
});
|
|
||||||
|
|
||||||
return id;
|
|
||||||
}
|
|
||||||
|
|
||||||
function handleSocketListenEvent(aq: ActiveQuery) {
|
|
||||||
const { event = null } = aq.callerInput;
|
|
||||||
|
|
||||||
const socket = validateConnection(aq);
|
|
||||||
|
|
||||||
if (!socket) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!event) {
|
|
||||||
aq.reject("Invalid event");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
let responded = false;
|
|
||||||
const respond = () => {
|
|
||||||
if (responded) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
responded = true;
|
|
||||||
aq.respond();
|
|
||||||
};
|
|
||||||
|
|
||||||
const cb = async (data: Buffer) => {
|
|
||||||
await socket.mutex?.waitForUnlock();
|
|
||||||
if (responded) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
aq.sendUpdate(data);
|
|
||||||
};
|
|
||||||
|
|
||||||
socket.on(event, cb);
|
|
||||||
socket.once("close", () => {
|
|
||||||
socket.off(event, cb);
|
|
||||||
respond();
|
|
||||||
});
|
|
||||||
|
|
||||||
aq.setReceiveUpdate?.(() => {
|
|
||||||
socket.off(event, cb);
|
|
||||||
respond();
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleSocketExists(aq: ActiveQuery) {
|
|
||||||
const { id = null } = aq.callerInput;
|
|
||||||
|
|
||||||
aq.respond(connections.has(Number(id)));
|
|
||||||
}
|
|
||||||
|
|
||||||
function handleCloseSocketEvent(aq: ActiveQuery) {
|
|
||||||
const socket = validateConnection(aq);
|
|
||||||
|
|
||||||
if (!socket) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
socket.end();
|
|
||||||
|
|
||||||
aq.respond();
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleWriteSocketEvent(aq: ActiveQuery) {
|
|
||||||
const socket = validateConnection(aq);
|
|
||||||
|
|
||||||
if (!socket) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
const { message = null } = aq.callerInput;
|
|
||||||
|
|
||||||
if (!message) {
|
|
||||||
aq.reject("empty message");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
await socket.mutex?.waitForUnlock();
|
|
||||||
|
|
||||||
socket.write(message);
|
|
||||||
|
|
||||||
aq.respond();
|
|
||||||
}
|
|
||||||
|
|
||||||
function validateConnection(aq: ActiveQuery): any | boolean {
|
|
||||||
const { id = null } = aq.callerInput;
|
|
||||||
|
|
||||||
if (!id || !connections.has(id)) {
|
|
||||||
aq.reject("Invalid connection id");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return connections.get(id)?.conn;
|
|
||||||
}
|
|
||||||
|
|
||||||
async function getSwarm(aq: ActiveQuery): Promise<Hyperswarm> {
|
|
||||||
await moduleReady;
|
|
||||||
let swarm;
|
|
||||||
if ("callerInput" in aq && aq.callerInput) {
|
|
||||||
swarm = aq.callerInput.swarm ?? null;
|
|
||||||
|
|
||||||
if (swarm && !swarmInstances.has(swarm)) {
|
|
||||||
const error = "Invalid swarm id";
|
|
||||||
aq.reject(error);
|
|
||||||
throw new Error(error);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!swarm) {
|
|
||||||
if (defaultSwarm.activeRelay && defaultSwarm.ready) {
|
|
||||||
await defaultSwarm.activeRelay.dht._protocol.opened;
|
|
||||||
}
|
|
||||||
return defaultSwarm;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (swarm.activeRelay && swarm.ready) {
|
|
||||||
await swarm.activeRelay.dht._protocol.opened;
|
|
||||||
}
|
|
||||||
|
|
||||||
return swarmInstances.get(swarm) as Hyperswarm;
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleAddRelay(aq: ActiveQuery) {
|
|
||||||
const { pubkey = null } = aq.callerInput;
|
|
||||||
|
|
||||||
if (!pubkey) {
|
|
||||||
aq.reject("invalid pubkey");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const swarm = await getSwarm(aq);
|
|
||||||
|
|
||||||
aq.respond(await swarm.addRelay(pubkey));
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleRemoveRelay(aq: ActiveQuery) {
|
|
||||||
const { pubkey = null } = aq.callerInput;
|
|
||||||
|
|
||||||
if (!pubkey) {
|
|
||||||
aq.reject("invalid pubkey");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const swarm = await getSwarm(aq);
|
|
||||||
|
|
||||||
aq.respond(swarm.removeRelay(pubkey));
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleClearRelays(aq: ActiveQuery) {
|
|
||||||
const swarm = await getSwarm(aq);
|
|
||||||
|
|
||||||
swarm.clearRelays();
|
|
||||||
|
|
||||||
aq.respond();
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleGetRelays(aq: ActiveQuery) {
|
|
||||||
aq.respond(await (await getSwarm(aq)).relays);
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleJoin(aq: ActiveQuery) {
|
|
||||||
const { topic = null } = aq.callerInput;
|
|
||||||
|
|
||||||
const swarm = await getSwarm(aq);
|
|
||||||
|
|
||||||
if (!topic) {
|
|
||||||
aq.reject("invalid topic");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (!b4a.isBuffer(topic)) {
|
|
||||||
aq.reject("topic must be a buffer");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// @ts-ignore
|
|
||||||
swarm.join(topic, { server: false });
|
|
||||||
aq.respond();
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleGetPeerByPubkey(aq: ActiveQuery) {
|
|
||||||
const { pubkey = null } = aq.callerInput;
|
|
||||||
|
|
||||||
const swarm = await getSwarm(aq);
|
|
||||||
|
|
||||||
if (!pubkey) {
|
|
||||||
aq.reject("invalid topic");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!b4a.isBuffer(pubkey)) {
|
|
||||||
aq.reject("pubkey must be a buffer");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// @ts-ignore
|
|
||||||
if (!swarm._allConnections.has(pubkey)) {
|
|
||||||
aq.reject("peer does not exist");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// @ts-ignore
|
|
||||||
const peer = swarm._allConnections.get(pubkey);
|
|
||||||
|
|
||||||
aq.respond(getSwarmToSocketConnectionId(peer));
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleInit(aq: ActiveQuery) {
|
|
||||||
const swarm = await getSwarm(aq);
|
|
||||||
try {
|
|
||||||
await swarm.init();
|
|
||||||
} catch (e) {
|
|
||||||
aq.reject((e as Error).message);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
aq.respond();
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleReady(aq: ActiveQuery) {
|
|
||||||
const swarm = await getSwarm(aq);
|
|
||||||
|
|
||||||
if (swarm.activeRelay && swarm.ready) {
|
|
||||||
aq.respond();
|
|
||||||
await swarm.activeRelay.dht._protocol.opened;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
swarm.once("ready", async () => {
|
|
||||||
await swarm.activeRelay.dht._protocol.opened;
|
|
||||||
aq.respond();
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleListenConnections(aq: ActiveQuery) {
|
|
||||||
const swarm = await getSwarm(aq);
|
|
||||||
const swarmId = getSwarmToSwarmId(swarm);
|
|
||||||
|
|
||||||
const listener = (peer: any) => {
|
|
||||||
aq.sendUpdate(getSwarmToSocketConnectionId(peer));
|
|
||||||
};
|
|
||||||
|
|
||||||
const swarmEvent = swarmEvents.get(swarmId as number)?.events;
|
|
||||||
|
|
||||||
if (!swarmEvent) {
|
|
||||||
logErr("swarm event object is missing");
|
|
||||||
}
|
|
||||||
|
|
||||||
swarmEvent?.on("connection", listener);
|
|
||||||
|
|
||||||
aq.setReceiveUpdate?.(() => {
|
|
||||||
swarmEvent?.off("connection", listener);
|
|
||||||
aq.respond();
|
|
||||||
});
|
|
||||||
|
|
||||||
for (const conn of connections) {
|
|
||||||
if (conn[1].swarm === swarmId) {
|
|
||||||
listener(conn[1].conn);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const closeCb = () => {
|
|
||||||
swarmEvent?.off("connection", listener);
|
|
||||||
swarmEvent?.emit("close");
|
|
||||||
aq.respond();
|
|
||||||
};
|
|
||||||
|
|
||||||
const hookClose = () => {
|
|
||||||
swarmEvent?.once("close", closeCb);
|
|
||||||
};
|
|
||||||
|
|
||||||
if (swarm.activeRelay) {
|
|
||||||
hookClose();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
swarm.onceSelf("ready", hookClose);
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleGetSocketInfo(aq: ActiveQuery) {
|
|
||||||
const socket = validateConnection(aq);
|
|
||||||
|
|
||||||
if (!socket) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
aq.respond({
|
|
||||||
remotePublicKey: socket.remotePublicKey,
|
|
||||||
publicKey: socket.publicKey,
|
|
||||||
rawStream: {
|
|
||||||
remoteHost: pubKeyToIpv6(socket.remotePublicKey),
|
|
||||||
remotePort: 0,
|
|
||||||
remoteFamily: "IPv6",
|
|
||||||
},
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleCreateProtomuxChannel(aq: ActiveQuery) {
|
|
||||||
const socket = validateConnection(aq);
|
|
||||||
|
|
||||||
if (!socket) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!("data" in aq.callerInput)) {
|
|
||||||
aq.reject("data required");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const mux = Protomux.from(socket);
|
|
||||||
const data = aq.callerInput.data;
|
|
||||||
|
|
||||||
const handleCallback = (name: string, enabled: boolean) => {
|
|
||||||
if (!enabled && name !== "destroy") {
|
|
||||||
return undefined;
|
|
||||||
}
|
|
||||||
return (...args: any) => {
|
|
||||||
args = args.filter(
|
|
||||||
(item: any) => item?.constructor.name.toLowerCase() !== "channel",
|
|
||||||
);
|
|
||||||
|
|
||||||
if (name === "destroy") {
|
|
||||||
connections.get(aq.callerInput.id)?.channels.delete(channelId);
|
|
||||||
aq.respond();
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!enabled) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
aq.sendUpdate({
|
|
||||||
action: name,
|
|
||||||
args,
|
|
||||||
});
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
let channel = mux.createChannel({
|
|
||||||
protocol: data?.protocol,
|
|
||||||
id: data?.id,
|
|
||||||
handshake: data?.handshake,
|
|
||||||
onopen: handleCallback("onopen", data?.onopen ?? undefined),
|
|
||||||
onclose: handleCallback("onclose", data?.onclose ?? undefined),
|
|
||||||
ondestroy: handleCallback("ondestroy", data?.ondestroy ?? undefined),
|
|
||||||
});
|
|
||||||
|
|
||||||
if (channel === null) {
|
|
||||||
aq.reject("duplicate channel");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
channel.open();
|
|
||||||
|
|
||||||
const channelId = getChannelId();
|
|
||||||
|
|
||||||
connections.get(aq.callerInput.id)?.channels.set(channelId, channel);
|
|
||||||
|
|
||||||
aq.sendUpdate(channelId);
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleCreateProtomuxMessage(aq: ActiveQuery) {
|
|
||||||
const socket = validateConnection(aq);
|
|
||||||
|
|
||||||
if (!socket) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!("data" in aq.callerInput)) {
|
|
||||||
aq.reject("action required");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!("channelId" in aq.callerInput)) {
|
|
||||||
aq.reject("channel id required");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const channel = connections
|
|
||||||
.get(aq.callerInput.id)
|
|
||||||
?.channels.get(aq.callerInput.channelId);
|
|
||||||
|
|
||||||
if (!channel) {
|
|
||||||
aq.reject("invalid channel");
|
|
||||||
}
|
|
||||||
|
|
||||||
const data = aq.callerInput.data;
|
|
||||||
|
|
||||||
const defers = new Map<number, DeferredPromise<any>>();
|
|
||||||
|
|
||||||
const handleEncoding = (enabled: boolean) => {
|
|
||||||
if (!enabled) {
|
|
||||||
return undefined;
|
|
||||||
}
|
|
||||||
|
|
||||||
const update = async (action: string, args: any) => {
|
|
||||||
const messageId = getMessageId();
|
|
||||||
const d = defer();
|
|
||||||
defers.set(messageId, d);
|
|
||||||
aq.sendUpdate({
|
|
||||||
id: messageId,
|
|
||||||
action,
|
|
||||||
args,
|
|
||||||
});
|
|
||||||
|
|
||||||
const ret = (await d.promise) as any;
|
|
||||||
if (ret[1]) {
|
|
||||||
if (ret[1].buffer) {
|
|
||||||
args[0].buffer = b4a.from(ret[1].buffer);
|
|
||||||
}
|
|
||||||
|
|
||||||
args[0].start = ret[1].start;
|
|
||||||
args[0].end = ret[1].end;
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret[0];
|
|
||||||
};
|
|
||||||
|
|
||||||
return {
|
|
||||||
async preencode(...args: any) {
|
|
||||||
return update("preencode", args);
|
|
||||||
},
|
|
||||||
async encode(...args: any) {
|
|
||||||
return update("encode", args);
|
|
||||||
},
|
|
||||||
async decode(...args: any) {
|
|
||||||
return update("decode", args);
|
|
||||||
},
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
aq.setReceiveUpdate?.((data: any) => {
|
|
||||||
if (data.action === "send") {
|
|
||||||
message.send(...data.args);
|
|
||||||
}
|
|
||||||
|
|
||||||
defers.get(data.id)?.resolve(data.args);
|
|
||||||
defers.delete(data.id);
|
|
||||||
});
|
|
||||||
|
|
||||||
if (data.onmessage) {
|
|
||||||
data.onmessage = (...args: any) => {
|
|
||||||
args = args.filter(
|
|
||||||
(item: any) => item?.constructor.name.toLowerCase() !== "channel",
|
|
||||||
);
|
|
||||||
aq.sendUpdate({
|
|
||||||
action: "onmessage",
|
|
||||||
args,
|
|
||||||
});
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
const message = channel.addMessage({
|
|
||||||
encoding: handleEncoding(data.encoding ?? false),
|
|
||||||
onmessage: data.onmessage ?? noop,
|
|
||||||
});
|
|
||||||
|
|
||||||
aq.sendUpdate({
|
|
||||||
action: "created",
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleCreateSwarm(aq: ActiveQuery) {
|
|
||||||
aq.respond(await createSwarm());
|
|
||||||
}
|
|
||||||
|
|
||||||
function getSwarmToSocketConnectionId(socket: any) {
|
|
||||||
for (const conn of connections) {
|
|
||||||
if (conn[1].conn === socket) {
|
|
||||||
return conn[0];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
function getSwarmToSwarmId(swarm: any) {
|
|
||||||
for (const swarmInstance of swarmInstances) {
|
|
||||||
if (swarmInstance[1] === swarm) {
|
|
||||||
return swarmInstance[0];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
function noop() {}
|
|
||||||
Loading…
Reference in New Issue