@made-by-moonlight/athene-web 0.9.2

package/LICENSE

@@ -0,0 +1,22 @@
+MIT License
+
+Copyright (c) 2025 Composio, Inc.
+Copyright (c) 2026 slievr (Athene fork)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist-server/direct-terminal-ws.js

@@ -0,0 +1,109 @@
+/**
+ * Direct WebSocket terminal server.
+ * Hosts the multiplexed /mux WebSocket endpoint for all terminal connections.
+ */
+import { createServer } from "node:http";
+import { findTmux } from "./tmux-utils.js";
+import { createMuxWebSocket } from "./mux-websocket.js";
+/**
+ * Create the direct terminal WebSocket server.
+ * Separated from listen() so tests can control lifecycle.
+ */
+export function createDirectTerminalServer(tmuxPath) {
+    const TMUX = tmuxPath ?? findTmux();
+    let muxWss = null;
+    const metrics = {
+        totalConnections: 0,
+        totalDisconnects: 0,
+        totalErrors: 0,
+    };
+    const server = createServer((req, res) => {
+        if (req.url === "/health") {
+            res.writeHead(200, { "Content-Type": "application/json" });
+            res.end(JSON.stringify({
+                status: "ok",
+                clients: muxWss?.clients.size ?? 0,
+                metrics,
+            }));
+            return;
+        }
+        res.writeHead(404);
+        res.end("Not found");
+    });
+    muxWss = createMuxWebSocket(TMUX);
+    if (muxWss) {
+        muxWss.on("connection", (ws) => {
+            metrics.totalConnections++;
+            ws.on("close", () => {
+                metrics.totalDisconnects++;
+            });
+            ws.on("error", () => {
+                metrics.totalErrors++;
+            });
+        });
+    }
+    // Manual upgrade routing — ws library doesn't support multiple WebSocketServer
+    // instances with different `path` options on the same HTTP server.
+    // `/ao-terminal-mux` is accepted as an alias of `/mux` so deployments fronted
+    // by a path-routing reverse proxy (e.g. cloudflared, nginx) can forward the
+    // dashboard's path-based mux URL straight at this port without needing a
+    // path-rewrite rule. The dashboard's MuxProvider already constructs that
+    // path when accessed on a standard HTTPS port; see `packages/web/src/providers/MuxProvider.tsx`.
+    server.on("upgrade", (request, socket, head) => {
+        const pathname = new URL(request.url ?? "/", "ws://localhost").pathname;
+        if ((pathname === "/mux" || pathname === "/ao-terminal-mux") && muxWss) {
+            muxWss.handleUpgrade(request, socket, head, (ws) => {
+                muxWss.emit("connection", ws, request);
+            });
+        }
+        else {
+            socket.destroy();
+        }
+    });
+    function shutdown() {
+        // Terminate all connected mux clients — this triggers their 'close' events
+        // which unsubscribe terminal callbacks and kill PTY processes.
+        if (muxWss) {
+            for (const client of muxWss.clients) {
+                client.terminate();
+            }
+            muxWss.close();
+        }
+        server.close();
+    }
+    return { server, shutdown };
+}
+// --- Run as standalone script ---
+// Only start the server when executed directly (not imported by tests)
+const isMainModule = process.argv[1]?.endsWith("direct-terminal-ws.ts") ||
+    process.argv[1]?.endsWith("direct-terminal-ws.js");
+if (isMainModule) {
+    const PORT = parseInt(process.env.DIRECT_TERMINAL_PORT ?? "14801", 10);
+    // On Windows, findTmux() returns null — mux-websocket.ts handles this by
+    // using named pipe relay to PTY hosts instead of tmux attach.
+    const TMUX = findTmux();
+    if (TMUX) {
+        console.log(`[DirectTerminal] Using tmux: ${TMUX}`);
+    }
+    else if (process.platform === "win32") {
+        console.log(`[DirectTerminal] Windows mode — using named pipe relay to PTY hosts`);
+    }
+    else {
+        console.log(`[DirectTerminal] No tmux available — terminal relay may be limited`);
+    }
+    const { server, shutdown } = createDirectTerminalServer(TMUX);
+    server.listen(PORT, () => {
+        console.log(`[DirectTerminal] WebSocket server listening on port ${PORT}`);
+    });
+    function handleShutdown(signal) {
+        console.log(`[DirectTerminal] Received ${signal}, shutting down...`);
+        shutdown();
+        const forceExitTimer = setTimeout(() => {
+            console.error("[DirectTerminal] Forced shutdown after timeout");
+            process.exit(1);
+        }, 5000);
+        forceExitTimer.unref();
+    }
+    process.on("SIGINT", () => handleShutdown("SIGINT"));
+    process.on("SIGTERM", () => handleShutdown("SIGTERM"));
+}