#!/usr/bin/env bun

import {
  connectServers,
  forgetConnection,
  getWebSocketStats,
  processMessage,
  trackConnection,
} from "./websocket-handler";
import { 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
const STATIC_DIR = Path.join(Path.dirname(import.meta.path), "static/dist");
const DEFAULT_PAGE = "index.html";

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

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

  // 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);
      },
    },
  });

  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()],
    ];
  }

  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();