@nodesecure/cli/src/http-server/websocket/index.js

Node.js security CLI

// Import Third-party Dependencies
import { WebSocketServer } from "ws";
import { match } from "ts-pattern";

// Import Internal Dependencies
import { appCache } from "../../cache.js";
import { logger } from "../../logger.js";
import { search } from "./commands/search.js";
import { remove } from "./commands/remove.js";

export class WebSocketServerInstanciator {
  constructor() {
    const websocket = new WebSocketServer({
      port: 1338
    });
    websocket.on("connection", this.onConnectionHandler.bind(this));
  }

  async onConnectionHandler(socket) {
    socket.on("message", (rawData) => {
      logger.info(`[ws](message: ${rawData})`);

      this.onMessageHandler(socket, JSON.parse(rawData))
        .catch(console.error);
    });

    const data = await this.initializeServer();
    sendSocketResponse(socket, data);
  }

  async onMessageHandler(
    socket,
    message
  ) {
    const ctx = { socket, cache: appCache, logger };

    const socketMessages = await match(message.action)
      .with("SEARCH", () => search(message.pkg, ctx))
      .with("REMOVE", () => remove(message.pkg, ctx))
      .exhaustive();

    for await (const message of socketMessages) {
      sendSocketResponse(socket, message);
    }
  }

  async initializeServer(
    stopInitializationOnError = false
  ) {
    try {
      const { current, mru, lru, availables, root } = await appCache.payloadsList();
      logger.info(`[ws|init](mru: ${mru}|lru: ${lru}|availables: ${availables}|current: ${current}|root: ${root})`);

      if (mru === void 0 || current === void 0) {
        throw new Error("Payloads list not found in cache.");
      }

      return {
        status: "INIT",
        current,
        mru,
        lru,
        availables,
        root
      };
    }
    catch {
      if (stopInitializationOnError) {
        return null;
      }

      logger.error("[ws|init](No cache yet. Creating one...)");
      await appCache.initPayloadsList();

      return this.initializeServer(true);
    }
  }
}

function sendSocketResponse(
  socket,
  message
) {
  if (message !== null) {
    socket.send(JSON.stringify(message));
  }
}