relay/src/modules/relay.ts

// @ts-ignore
import DHT from "@hyperswarm/dht";
// @ts-ignore
import { relay } from "@hyperswarm/dht-relay";
// @ts-ignore
import Stream from "@hyperswarm/dht-relay/ws";
import express, { Express } from "express";
import config from "../config.js";
import * as http from "http";
import * as https from "https";
import { get as getDHT } from "./dht.js";
import WS from "ws";
// @ts-ignore
import log from "loglevel";
import { AddressInfo } from "net";
// @ts-ignore
import promiseRetry from "promise-retry";
import { getSslContext } from "./ssl.js";

export async function start() {
  const relayPort = config.uint("port");

  const dht = await getDHT();

  const statusCodeServer = http.createServer(function (req, res) {
    // @ts-ignore
    res.writeHead(req.headers["x-status"] ?? 200, {
      "Content-Type": "text/plain",
    });
    res.end();
  });

  await new Promise((resolve) => {
    statusCodeServer.listen(25252, "0.0.0.0", function () {
      const address = statusCodeServer.address() as AddressInfo;
      log.info(
        "Status Code Server started on ",
        `${address.address}:${address.port}`
      );
      resolve(null);
    });
  });

  let relayServer: https.Server | http.Server;

  if (config.bool("ssl")) {
    relayServer = https.createServer({
      SNICallback(servername, cb) {
        cb(null, getSslContext());
      },
    });
  } else {
    relayServer = http.createServer();
  }

  let wsServer = new WS.Server({ server: relayServer });

  wsServer.on("connection", (socket: any) => {
    relay(dht, new Stream(false, socket));
  });

  await new Promise((resolve) => {
    relayServer.listen(relayPort, "0.0.0.0", function () {
      const address = relayServer.address() as AddressInfo;
      log.info(
        "DHT Relay Server started on ",
        `${address.address}:${address.port}`
      );
      resolve(null);
    });
  });
}