prasi-bun/app/srv/ws/edit/edit-handler.ts

94 lines
2.4 KiB
TypeScript

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 { diffLocal } from "./action/diff-local";
import { getComp } from "./action/get-comp";
import { getPage } from "./action/get-page";
import { svLocal } from "./action/sv-local";
import { svdiffRemote } from "./action/svdiff-remote";
import { redo, undo } from "./action/undo-redo";
import { eg } from "./edit-global";
import { sendWS } from "./send";
const site = {
saveTimeout: null as any,
};
export const editHandler: WebSocketHandler<WSData> = {
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") {
sendWS(ws, JSON.stringify({ type: "pong" }));
return;
}
switch (msg.type) {
case "site-js":
clearTimeout(site.saveTimeout);
site.saveTimeout = setTimeout(async () => {
const js = JSON.parse(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) {},
};