DBT/clawgo
1
0
Fork 0
mirror of https://github.com/YspCoder/clawgo.git synced 2026-05-07 16:47:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: unify websocket runtime and harden node control

Browse Source
This commit is contained in:
lpf
2026-03-08 22:22:49 +08:00
parent 7e67619826
commit 4172a57b39
15 changed files with 2082 additions and 124 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
webui/server.ts
View File

@@ -3,10 +3,14 @@ import { createServer as createViteServer } from "vite";
import { EventEmitter } from "events";
import multer from "multer";
import fs from "fs";
import http from "http";
import { WebSocketServer } from "ws";
const app = express();
const PORT = 3000;
const logEmitter = new EventEmitter();
const server = http.createServer(app);
const wss = new WebSocketServer({ server });
// In-memory only for local dev fallback (no sqlite persistence)
const mem = {
@@ -118,6 +122,8 @@ app.post("/webui/api/skills", (req, res) => {
});
app.get("/webui/api/logs/stream", (req, res) => {
res.setHeader("Deprecation", "true");
res.setHeader("X-Clawgo-Replaced-By", "/webui/api/logs/live");
res.setHeader("Content-Type", "application/x-ndjson");
res.setHeader("Cache-Control", "no-cache");
res.setHeader("Connection", "keep-alive");
@@ -130,6 +136,8 @@ app.get("/webui/api/logs/stream", (req, res) => {
});
app.post("/webui/api/chat/stream", async (req, res) => {
res.setHeader("Deprecation", "true");
res.setHeader("X-Clawgo-Replaced-By", "/webui/api/chat/live");
const { message } = req.body || {};
res.setHeader("Content-Type", "text/plain");
res.setHeader("Transfer-Encoding", "chunked");
@@ -141,6 +149,48 @@ app.post("/webui/api/chat/stream", async (req, res) => {
res.end();
});
wss.on("connection", (socket, req) => {
if (!req.url) return;
if (req.url.startsWith("/webui/api/logs/live")) {
const onLog = (entry: any) => {
if (socket.readyState === socket.OPEN) {
socket.send(JSON.stringify({ ok: true, type: "log_entry", entry }));
}
};
logEmitter.on("log", onLog);
onLog({ time: new Date().toISOString(), level: "INFO", msg: "Log stream connected" });
socket.on("close", () => {
logEmitter.off("log", onLog);
});
return;
}
if (req.url.startsWith("/webui/api/chat/live")) {
socket.on("message", async (payload) => {
let message = "";
try {
const body = JSON.parse(String(payload || "{}"));
message = String(body?.message || "");
} catch {
if (socket.readyState === socket.OPEN) {
socket.send(JSON.stringify({ ok: false, type: "chat_error", error: "invalid json" }));
}
return;
}
const words = `Simulated streaming response: ${message}`.split(" ");
for (const word of words) {
if (socket.readyState !== socket.OPEN) return;
socket.send(JSON.stringify({ ok: true, type: "chat_chunk", delta: `${word} ` }));
await new Promise((r) => setTimeout(r, 40));
}
if (socket.readyState === socket.OPEN) {
socket.send(JSON.stringify({ ok: true, type: "chat_done" }));
}
});
}
});
if (process.env.NODE_ENV !== "production") {
const vite = await createViteServer({ server: { middlewareMode: true }, appType: "spa" });
app.use(vite.middlewares);
@@ -151,7 +201,7 @@ if (process.env.NODE_ENV !== "production") {
});
}
app.listen(PORT, "0.0.0.0", () => {
server.listen(PORT, "0.0.0.0", () => {
console.log(`Server running on http://localhost:${PORT}`);
addLog("INFO", "Gateway WebUI Server started");
});