@reclaimprotocol/attestor-core 5.0.1-beta.2 → 5.0.1-beta.22

package/lib/server/create-server.js

@@ -1,105 +1,103 @@
-import { createServer as createHttpServer } from "http";
-import serveStatic from "serve-static";
-import { WebSocketServer } from "ws";
+import { createServer as createHttpServer } from 'http';
+import serveStatic from 'serve-static';
+import { WebSocketServer } from 'ws';
 import { API_SERVER_PORT, ATTESTOR_ADDRESS_PATHNAME, BROWSER_RPC_PATHNAME, WS_PATHNAME } from "../config/index.js";
-import { AttestorServerSocket } from "../server/socket.js";
-import { getAttestorAddress } from "../server/utils/generics.js";
-import { addKeepAlive } from "../server/utils/keep-alive.js";
+import { AttestorServerSocket } from "./socket.js";
+import { getAttestorAddress } from "./utils/generics.js";
+import { addKeepAlive } from "./utils/keep-alive.js";
 import { createBgpListener } from "../utils/bgp-listener.js";
 import { getEnvVariable } from "../utils/env.js";
 import { logger as LOGGER } from "../utils/index.js";
 import { SelectedServiceSignatureType } from "../utils/signatures/index.js";
 import { promisifySend } from "../utils/ws.js";
-const PORT = +(getEnvVariable("PORT") || API_SERVER_PORT);
-const DISABLE_BGP_CHECKS = getEnvVariable("DISABLE_BGP_CHECKS") === "1";
+const PORT = +(getEnvVariable('PORT') || API_SERVER_PORT);
+const DISABLE_BGP_CHECKS = getEnvVariable('DISABLE_BGP_CHECKS') === '1';
 const ATTESTOR_ADDRESS_JSON_RES = JSON.stringify({
-  address: getAttestorAddress(SelectedServiceSignatureType),
-  signatureType: SelectedServiceSignatureType
+    address: getAttestorAddress(SelectedServiceSignatureType),
+    signatureType: SelectedServiceSignatureType
 });
-async function createServer(port = PORT) {
-  const http = createHttpServer();
-  const serveBrowserRpc = serveStatic(
-    "browser",
-    {
-      index: ["index.html"],
-      setHeaders(res) {
-        res.setHeader("Access-Control-Allow-Origin", "*");
-      }
-    }
-  );
-  const bgpListener = !DISABLE_BGP_CHECKS ? createBgpListener(LOGGER.child({ service: "bgp-listener" })) : void 0;
-  const wss = new WebSocketServer({ noServer: true });
-  http.on("upgrade", handleUpgrade.bind(wss));
-  http.on("request", (req, res) => {
-    const url = URL.parse(req.url || "", "http://localhost");
-    if (!url) {
-      res.statusCode = 422;
-      res.end("Invalid URL");
-      return;
-    }
-    if (url.pathname === ATTESTOR_ADDRESS_PATHNAME) {
-      res.writeHead(200, { "Content-Type": "application/json" });
-      res.end(ATTESTOR_ADDRESS_JSON_RES);
-      return;
-    }
-    if (!url.pathname?.startsWith(BROWSER_RPC_PATHNAME)) {
-      res.statusCode = 404;
-      res.end("Not found");
-      return;
-    }
-    req.url = req.url.slice(BROWSER_RPC_PATHNAME.length) || "/";
-    serveBrowserRpc(req, res, (err) => {
-      if (err) {
-        LOGGER.error({ err, url: req.url }, "Failed to serve file");
-      }
-      res.statusCode = err?.statusCode ?? 404;
-      res.end(err?.message ?? "Not found");
+/**
+ * Creates the WebSocket API server,
+ * creates a fileserver to serve the browser RPC client,
+ * and listens on the given port.
+ */
+export async function createServer(port = PORT) {
+    const http = createHttpServer();
+    const serveBrowserRpc = serveStatic('browser', {
+        index: ['index.html'],
+        setHeaders(res) {
+            res.setHeader('Access-Control-Allow-Origin', '*');
+        },
+    });
+    const bgpListener = !DISABLE_BGP_CHECKS
+        ? createBgpListener(LOGGER.child({ service: 'bgp-listener' }))
+        : undefined;
+    const wss = new WebSocketServer({ noServer: true });
+    http.on('upgrade', handleUpgrade.bind(wss));
+    http.on('request', (req, res) => {
+        const url = URL.parse(req.url || '', 'http://localhost');
+        if (!url) {
+            res.statusCode = 422;
+            res.end('Invalid URL');
+            return;
+        }
+        if (url.pathname === ATTESTOR_ADDRESS_PATHNAME) {
+            res.writeHead(200, { 'Content-Type': 'application/json' });
+            res.end(ATTESTOR_ADDRESS_JSON_RES);
+            return;
+        }
+        // simple way to serve files at the browser RPC path
+        if (!url.pathname?.startsWith(BROWSER_RPC_PATHNAME)) {
+            res.statusCode = 404;
+            res.end('Not found');
+            return;
+        }
+        req.url = req.url.slice(BROWSER_RPC_PATHNAME.length) || '/';
+        serveBrowserRpc(req, res, (err) => {
+            if (err) {
+                LOGGER.error({ err, url: req.url }, 'Failed to serve file');
+            }
+            res.statusCode = err?.statusCode ?? 404;
+            res.end(err?.message ?? 'Not found');
+        });
     });
-  });
-  http.listen(port);
-  await new Promise((resolve, reject) => {
-    http.once("listening", () => resolve());
-    http.once("error", reject);
-  });
-  wss.on("connection", (ws, req) => handleNewClient(ws, req, bgpListener));
-  LOGGER.info(
-    {
-      port,
-      apiPath: WS_PATHNAME,
-      browserRpcPath: BROWSER_RPC_PATHNAME,
-      signerAddress: getAttestorAddress(SelectedServiceSignatureType)
-    },
-    "WS server listening"
-  );
-  const wssClose = wss.close.bind(wss);
-  wss.close = (cb) => {
-    wssClose(() => http.close(cb));
-    bgpListener?.close();
-  };
-  return wss;
+    // wait for us to start listening
+    http.listen(port);
+    await new Promise((resolve, reject) => {
+        http.once('listening', () => resolve());
+        http.once('error', reject);
+    });
+    wss.on('connection', (ws, req) => handleNewClient(ws, req, bgpListener));
+    LOGGER.info({
+        port,
+        apiPath: WS_PATHNAME,
+        browserRpcPath: BROWSER_RPC_PATHNAME,
+        signerAddress: getAttestorAddress(SelectedServiceSignatureType)
+    }, 'WS server listening');
+    const wssClose = wss.close.bind(wss);
+    wss.close = (cb) => {
+        wssClose(() => http.close(cb));
+        bgpListener?.close();
+    };
+    return wss;
 }
 async function handleNewClient(ws, req, bgpListener) {
-  promisifySend(ws);
-  const client = await AttestorServerSocket.acceptConnection(
-    ws,
-    { req, bgpListener, logger: LOGGER }
-  );
-  if (!client) {
-    return;
-  }
-  ws.serverSocket = client;
-  addKeepAlive(ws, LOGGER.child({ sessionId: client.sessionId }));
+    promisifySend(ws);
+    const client = await AttestorServerSocket.acceptConnection(ws, { req, bgpListener, logger: LOGGER });
+    // if initialisation fails, don't store the client
+    if (!client) {
+        return;
+    }
+    ws.serverSocket = client;
+    addKeepAlive(ws, LOGGER.child({ sessionId: client.sessionId }));
 }
 function handleUpgrade(request, socket, head) {
-  const { pathname } = new URL(request.url, "wss://base.url");
-  if (pathname === WS_PATHNAME) {
-    this.handleUpgrade(request, socket, head, (ws) => {
-      this.emit("connection", ws, request);
-    });
-    return;
-  }
-  socket.destroy();
+    const { pathname } = new URL(request.url, 'wss://base.url');
+    if (pathname === WS_PATHNAME) {
+        this.handleUpgrade(request, socket, head, (ws) => {
+            this.emit('connection', ws, request);
+        });
+        return;
+    }
+    socket.destroy();
 }
-export {
-  createServer
-};