UDN/index.ts

#!/usr/bin/env bun

import {
  connectServers,
  forgetConnection,
  getWebSocketStats,
  processMessage,
  trackConnection,
} from "./websocket-handler";
import {
  getVersionNumber,
  writeStatNumber,
  writeStatString,
  writeSuccess,
} from "./utility";

import Colors from "colors";
import IP from "ip";
import Path from "path";
import { ServerWebSocket } from "bun";
import { getConfig } from "./config-handler";

// CONFIG
export const CODE_DIR = Path.dirname(import.meta.path);
export const STATIC_DIR = Path.join(CODE_DIR, "static/dist");
export const PACKAGE_FILE = Path.join(CODE_DIR, "package.json");
export const DEFAULT_PAGE = "index.html";

// TYPES
export type WS = ServerWebSocket<unknown> | WebSocket;

// MAIN
async function main() {
  // CONFIG
  const config = await getConfig();
  const versionNumber = await getVersionNumber();

  // SERVER
  const server = Bun.serve({
    port: config.port,

    fetch(request, server) {
      // allow websockets
      server.upgrade(request);

      // respond
      const requestPath = new URL(request.url).pathname;
      switch (requestPath) {
        case "/":
          return createFileResponse(DEFAULT_PAGE);
        case "/stats":
          return new Response(JSON.stringify(getAllStats()));
        default:
          return createFileResponse(requestPath);
      }
    },

    websocket: {
      open(ws) {
        trackConnection(ws);
      },
      message(ws, message) {
        processMessage(ws, message.toString(), config);
      },
      close(ws) {
        forgetConnection(ws);
      },
    },

    tls: await (async () => {
      const keyFile = Bun.file("./key.pem");
      const certFile = Bun.file("./cert.pem");

      if (
        (await keyFile.exists()) == false ||
        (await certFile.exists()) == false
      )
        return undefined;

      return {
        key: keyFile,
        cert: certFile,
        passphrase: process.env.TLS_PASSPHRASE,
      };
    })(),
  });

  function createFileResponse(requestPath: string): Response {
    const fullPath = Path.join(STATIC_DIR, requestPath);
    return new Response(Bun.file(fullPath));
  }

  // OTHER SERVERS
  connectServers(config);

  // AUDIT
  function getServerStats(): [string, string][] {
    return [
      ["server url", server.url.toString()],
      ["server ip address", IP.address()],
      ["server version", versionNumber],
    ];
  }

  function getAllStats(): [string, string | number][] {
    return [...getServerStats(), ...getWebSocketStats()];
  }

  // CLI
  writeSuccess(`###\nstarted up ${new Date().toLocaleString()}`);

  function updateCLI() {
    console.clear();

    console.log(Colors.bold.bgWhite("UNIVERSAL DECENTRALIZED NETWORK"));
    console.log(
      Colors.yellow("relevant output is available in the 'logs' file")
    );
    console.log();

    console.log(new Date().toLocaleString());
    console.log();

    getServerStats().forEach((entry) => writeStatString(...entry));
    console.log();

    getWebSocketStats().forEach((entry) => writeStatNumber(...entry));
    console.log();
  }

  setTimeout(updateCLI, 1000);
  setInterval(updateCLI, 5000);
}

main();