import { toJsonRpc } from "@valkyr/json-rpc";

import { Session } from "~libraries/auth/mod.ts";
import { logger } from "~libraries/logger/mod.ts";

import { sockets } from "./sockets.ts";

export function upgrade(request: Request, session?: Session) {
  const { socket, response } = Deno.upgradeWebSocket(request);

  socket.addEventListener("open", () => {
    logger.prefix("Socket").info("socket connected", { session });
    sockets.add(socket);
  });

  socket.addEventListener("close", () => {
    logger.prefix("Socket").info("socket disconnected", { session });
    sockets.del(socket);
  });

  socket.addEventListener("message", (event) => {
    if (event.data === "ping") {
      return;
    }

    const body = toJsonRpc(event.data);

    logger.prefix("Socket").info(body);

    asyncLocalStorage.run(
      {
        session,
        info: {
          method: body.method!,
          start: Date.now(),
        },
        socket,
        response: {
          headers: new Headers(),
        },
      },
      async () => {
        api
          .handleCommand(body)
          .then((response) => {
            if (response !== undefined) {
              logger.info({ response });
              socket.send(JSON.stringify(response));
            }
          })
          .catch((error) => {
            logger.info({ error });
            socket.send(JSON.stringify(error));
          });
      },
    );
  });

  return response;
}