Updates

dsql-app/docker/web-socket/app/functions/events/client-requests/check-queue.ts

@@ -0,0 +1,34 @@
+import type { ServerWebSocket } from "bun";
+import type { WebSocketData } from "@WS/.";
+import { _n } from "@/client-exports";
+import sendData from "../../../utils/send-data";
+import getQueue from "@/package-shared/functions/backend/queues/get-queue";
+import { DSQL_DATASQUIREL_PROCESS_QUEUE } from "@/package-shared/types/dsql";
+
+type Param = {
+    ws: ServerWebSocket<WebSocketData>;
+};
+
+export default async function checkQueue({ ws }: Param) {
+    try {
+        const user = ws.data.user;
+        const queue = (await getQueue({
+            userId: user.id,
+            single: true,
+        })) as DSQL_DATASQUIREL_PROCESS_QUEUE | undefined;
+
+        sendData(ws, {
+            event: "server:queue",
+            data: {
+                queue,
+            },
+        });
+    } catch (error: any) {
+        sendData(ws, {
+            event: "server:queue",
+            data: {
+                queue: undefined,
+            },
+        });
+    }
+}

dsql-app/docker/web-socket/app/functions/events/client-requests/delete-queue.ts

@@ -0,0 +1,28 @@
+import type { ServerWebSocket } from "bun";
+import type { WebSocketData } from "@WS/.";
+import { _n } from "@/client-exports";
+import sendData from "../../../utils/send-data";
+import { WebSocketDataType } from "@/types";
+import deleteQueue from "@/package-shared/functions/backend/queues/delete-queue";
+
+type Param = {
+    ws: ServerWebSocket<WebSocketData>;
+    data?: WebSocketDataType;
+};
+
+export default async function webSocketDeleteQueue({ ws, data }: Param) {
+    try {
+        const user = ws.data.user;
+
+        await deleteQueue({
+            queueId: _n(data?.data?.queue?.id),
+            userId: user.id,
+        });
+
+        await Bun.sleep(2000);
+
+        sendData(ws, {
+            event: "server:queue-deleted",
+        });
+    } catch (error: any) {}
+}

dsql-app/docker/web-socket/app/functions/events/client-requests/send-dummy-queue.ts

@@ -0,0 +1,36 @@
+import type { ServerWebSocket } from "bun";
+import type { WebSocketData } from "@WS/.";
+import { _n } from "@/client-exports";
+import sendData from "@WS/utils/send-data";
+import addQueue from "@/package-shared/functions/backend/queues/add-queue";
+
+type Param = {
+    ws: ServerWebSocket<WebSocketData>;
+};
+
+export default async function webSocketSendDummyQueue({ ws }: Param) {
+    try {
+        console.log("Sending Dummy Queue ...");
+
+        const user = ws.data.user;
+        const dummyQueue = await addQueue({
+            queue: {
+                job_type: "dummy",
+                user_id: user.id,
+                title: "Running Dummy Queue ...",
+            },
+            userId: user.id,
+            dummy: true,
+        });
+
+        await Bun.sleep(2000);
+
+        sendData(ws, {
+            event: "server:dev:queue",
+        });
+    } catch (error: any) {
+        sendData(ws, {
+            event: "server:dev:queue",
+        });
+    }
+}

dsql-app/docker/web-socket/app/functions/socket-close.ts

@@ -0,0 +1,19 @@
+import { type ServerWebSocket } from "bun";
+import type { WebSocketData } from "@WS/.";
+
+type Param = {
+    ws: ServerWebSocket<WebSocketData>;
+};
+
+export default async function socketClose({ ws }: Param) {
+    const user = ws.data.user;
+    console.log(`Web Closed by ${user.first_name}`);
+
+    const userSessionIndex = global.ACTIVE_USERS.findIndex(
+        (session) => session.id == user.id
+    );
+
+    if (userSessionIndex >= 0) {
+        global.ACTIVE_USERS.splice(userSessionIndex, 1);
+    }
+}

dsql-app/docker/web-socket/app/functions/socket-init.ts

@@ -0,0 +1,38 @@
+import datasquirel from "@moduletrace/datasquirel";
+import type { DATASQUIREL_LoggedInUser } from "@moduletrace/datasquirel/dist/package-shared/types";
+
+type Param = {
+    req: Request;
+    debug?: boolean;
+};
+
+type Return = {
+    user: DATASQUIREL_LoggedInUser | null;
+};
+
+export default async function socketInit({
+    req,
+    debug,
+}: Param): Promise<Return> {
+    const cookieString = req.headers.get("Cookie") || undefined;
+
+    if (debug) {
+        console.log("DEBUG:::socketInit:cookieString", cookieString);
+    }
+
+    if (!cookieString)
+        return {
+            user: null,
+        };
+
+    const user = datasquirel.user.userAuth({
+        cookieString,
+        database: process.env.DB_NAME || "",
+    });
+
+    if (debug) {
+        console.log("DEBUG:::socketInit:user", user);
+    }
+
+    return { user: user.payload };
+}

dsql-app/docker/web-socket/app/functions/socket-message.ts

@@ -0,0 +1,55 @@
+import type { ServerWebSocket } from "bun";
+import type { WebSocketData } from "@WS/.";
+import type { WebSocketDataType } from "@/types";
+import { EJSON } from "@/client-exports";
+import checkQueue from "./events/client-requests/check-queue";
+import webSocketSendDummyQueue from "./events/client-requests/send-dummy-queue";
+import webSocketDeleteQueue from "./events/client-requests/delete-queue";
+
+type Param = {
+    ws: ServerWebSocket<WebSocketData>;
+    message: string | Buffer;
+};
+
+export type WebSocketMessageParam = {
+    ws: ServerWebSocket<WebSocketData>;
+    message?: string | Buffer;
+    data?: WebSocketDataType;
+};
+
+export default async function socketMessage({ ws, message }: Param) {
+    const user = ws.data.user;
+    const data = EJSON.parse(message.toString()) as
+        | WebSocketDataType
+        | undefined;
+
+    const websocketMessageParams: WebSocketMessageParam = {
+        ws,
+        data,
+        message,
+    };
+
+    switch (data?.event) {
+        /**
+         * Check Queue
+         */
+        case "client:check-queue":
+            checkQueue({ ws });
+            break;
+        /**
+         * Send Dummy Queue
+         */
+        case "client:dev:queue":
+            webSocketSendDummyQueue({ ws });
+            break;
+        /**
+         * Delete Queue
+         */
+        case "client:delete-queue":
+            webSocketDeleteQueue({ ws, data });
+            break;
+
+        default:
+            break;
+    }
+}

dsql-app/docker/web-socket/app/functions/socket-open.ts

@@ -0,0 +1,12 @@
+import type { ServerWebSocket } from "bun";
+import type { WebSocketData } from "@WS/.";
+
+type Param = {
+    ws: ServerWebSocket<WebSocketData>;
+};
+
+export default async function socketOpen({ ws }: Param) {
+    const user = ws.data.user;
+    console.log(`Web Socket Opened by ${user.first_name}`);
+    global.ACTIVE_USERS.push({ ...user, ws });
+}