From 20d163bcf2309bebfa01ddc38d25b54d7ff8442e Mon Sep 17 00:00:00 2001
From: Derrick Hammer <derrick@derrickhammer.com>
Date: Sun, 16 Apr 2023 20:19:14 -0400
Subject: [PATCH] *Refactor to use new MultiSocketProxy from libhyperproxy

---
 src/index.ts | 67 ++++------------------------------------------------
 1 file changed, 4 insertions(+), 63 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index 68a705b..4885a65 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -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);
-        },
       });
     });
   },