*Refactor to use new MultiSocketProxy from libhyperproxy

This commit is contained in:
Derrick Hammer 2023-04-16 20:19:14 -04:00
parent fb729e6a1d
commit 20d163bcf2
Signed by: pcfreak30
GPG Key ID: C997C339BE476FF2
1 changed files with 4 additions and 63 deletions

View File

@ -3,87 +3,28 @@ import type { Plugin, PluginAPI } from "@lumeweb/relay-types";
import rand from "random-key";
// @ts-ignore
import FullNode from "hsd/lib/node/fullnode.js";
import { Proxy, Socket } from "@lumeweb/libhyperproxy";
import { MultiSocketProxy } from "@lumeweb/libhyperproxy";
let server: FullNode;
let api: PluginAPI;
const PROTOCOL = "lumeweb.proxy.handshake";
async function abort(err: any) {
const timeout = setTimeout(() => {
api.logger.error("Shutdown is taking a long time. Exiting.");
process.exit(3);
}, 5000);
timeout.unref();
try {
api.logger.error("Shutting down...");
await server.close();
clearTimeout(timeout);
api.logger.error((err as Error).stack);
process.exit(2);
} catch (e: any) {
api.logger.error(`Error occurred during shutdown: ${(e as Error).message}`);
process.exit(3);
}
}
async function boot(api: PluginAPI) {
const { pluginConfig: config } = api;
const apiKey = rand.generate();
if (!config.bool("external")) {
server = new FullNode({
config: false,
argv: false,
env: true,
noDns: false,
memory: false,
httpHost: "127.0.0.1",
apiKey,
logFile: true,
logConsole: true,
logLevel: "info",
workers: false,
network: "main",
bip37: true,
});
server.on("abort", abort);
api.logger.info("API Key %s", apiKey);
try {
await server.ensure();
await server.open();
await server.connect();
server.startSync();
} catch (e: any) {
api.logger.error((e as Error).stack);
}
}
}
const plugin: Plugin = {
name: "handshake",
async plugin(_api: PluginAPI): Promise<void> {
api = _api;
boot(api);
const proxy = new Proxy({
const proxy = new MultiSocketProxy({
swarm: api.swarm,
protocol: PROTOCOL,
allowedPorts: [44806, 12038],
server: true,
});
api.swarm.join(api.util.crypto.createHash(PROTOCOL));
api.protocols.register(PROTOCOL, (peer: any, muxer: any) => {
proxy.handlePeer({
peer,
muxer,
onopen(socket: Socket) {
server.pool.server.emit("connection", socket);
},
});
});
},