feat: add vnc app

backend/.env.example

@@ -8,3 +8,4 @@ AUTH_PASSWORD=
 PC_MAC_ADDR=
 TERMINAL_SHELL=
 FILE_DIRS="/some/path;/another/path"
+VNC_PORT=5901

backend/lib/websockify.ts

@@ -0,0 +1,50 @@
+import type { WebSocket } from "ws";
+import net from "node:net";
+
+// const log = console.log;
+const log = (..._args: any[]) => null;
+
+export const websockify = (
+  client: WebSocket,
+  targetHost: string,
+  targetPort: number
+) => {
+  const target = net.createConnection(targetPort, targetHost, function () {
+    log("connected to target");
+  });
+
+  target.on("data", function (data) {
+    try {
+      client.send(data);
+    } catch (e) {
+      log("Client closed, cleaning up target");
+      target.end();
+    }
+  });
+
+  target.on("end", function () {
+    log("target disconnected");
+    client.close();
+  });
+
+  target.on("error", function () {
+    log("target connection error");
+    target.end();
+    client.close();
+  });
+
+  client.on("message", function (msg) {
+    // log("CLIENT message: " + msg);
+    target.write(msg as never);
+  });
+
+  client.on("close", function (code, reason) {
+    log("WebSocket client disconnected: " + code + " [" + reason + "]");
+    target.end();
+  });
+
+  client.on("error", function (a) {
+    log("WebSocket client error: " + a);
+    target.end();
+  });
+};

backend/websocket.ts

@@ -1,6 +1,9 @@
 import { WebSocketServer } from "ws";
 import { verifyToken } from "./lib/jwt";
 import { createTerminalSession } from "./lib/terminal";
+import { websockify } from "./lib/websockify";
+
+const VNC_PORT = parseInt(process.env.VNC_PORT || "") || 5901;
 
 const createWsServer = (server: any) => {
   const wss = new WebSocketServer({ server: server as never });
@@ -12,7 +15,7 @@ const createWsServer = (server: any) => {
     try {
       await verifyToken(token || "");
     } catch (err) {
-      console.log(err);
+      // console.log(err);
       ws.close();
       return;
     }
@@ -21,6 +24,10 @@ const createWsServer = (server: any) => {
       createTerminalSession(ws);
     }
 
+    if (url.pathname === "/vnc") {
+      websockify(ws, "localhost", VNC_PORT);
+    }
+
     ws.on("error", console.error);
   });
 };