import { createId } from "@paralleldrive/cuid2";
import { WebSocketHandler } from "bun";
import { WSData } from "../../../pkgs/core/server/create";
import { WS_MSG } from "../../web/src/utils/types/ws";
import { eg } from "./edit/edit-global";
import { decompress } from "lz-string";
import { getPage } from "./edit/action/get-page";
import { getComp } from "./edit/action/get-comp";
import { svLocal } from "./edit/action/sv-local";
import { diffLocal } from "./edit/action/diff-local";
import { svdiffRemote } from "./edit/action/svdiff-remote";
import { redo, undo } from "./edit/action/undo-redo";

eg.edit = {
  site: {},
  comp: {},
  page: {},
  ws: new WeakMap(),
};
const site = {
  saveTimeout: null as any,
};

export const wsHandler: Record<string, WebSocketHandler<WSData>> = {
  "/edit": {
    open(ws) {
      eg.edit.ws.set(ws, {
        clientID: createId(),
      });
    },
    async message(ws, raw) {
      if (typeof raw === "string") {
        try {
          const msg = JSON.parse(raw) as WS_MSG;

          if (msg.type === "ping") {
            ws.send(JSON.stringify({ type: "pong" }));
            return;
          }

          switch (msg.type) {
            case "site-js":
              clearTimeout(site.saveTimeout);
              site.saveTimeout = setTimeout(async () => {
                const js = JSON.parse(decompress(msg.src));
                await db.site.update({
                  where: {
                    id: msg.id_site,
                  },
                  data: {
                    js: js.src,
                    js_compiled: js.compiled,
                  },
                });
              }, 1000);
              break;
            case "get_page":
              await getPage(ws, msg);
              break;
            case "get_comp":
              await getComp(ws, msg);
              break;
            case "sv_local":
              await svLocal(ws, msg);
              break;
            case "diff_local":
              await diffLocal(ws, msg);
              break;
            case "svd_remote":
              await svdiffRemote(ws, msg);
              break;
            case "undo":
              undo(ws, msg);
              break;
            case "redo":
              redo(ws, msg);
              break;
          }
        } catch (e) {
          console.log(e);
        }
      }
    },
    close(ws, code, reason) {
      eg.edit.ws.delete(ws);

      for (const page of Object.values(eg.edit.page)) {
        page.ws.delete(ws);
      }

      for (const site of Object.values(eg.edit.site)) {
        site.ws.delete(ws);
      }

      for (const comp of Object.values(eg.edit.comp)) {
        comp.ws.delete(ws);
      }
    },
    drain(ws) {},
  },
};