mcp-shrimp-task-manager/dist/web/webServer.js

Shrimp Task Manager is a task tool built for AI Agents, emphasizing chain-of-thought, reflection, and style consistency. It converts natural language into structured dev tasks with dependency tracking and iterative refinement, enabling agent-like develope

import express from "express";
import getPort from "get-port";
import path from "path";
import fs from "fs";
import fsPromises from "fs/promises";
import { fileURLToPath } from "url";
import { getDataDir, getTasksFilePath, getWebGuiFilePath, } from "../utils/paths.js";
export async function createWebServer() {
    // 創建 Express 應用
    const app = express();
    // 儲存 SSE 客戶端的列表
    let sseClients = [];
    // 發送 SSE 事件的輔助函數
    function sendSseUpdate() {
        sseClients.forEach((client) => {
            // 檢查客戶端是否仍然連接
            if (!client.writableEnded) {
                client.write(`event: update\ndata: ${JSON.stringify({
                    timestamp: Date.now(),
                })}\n\n`);
            }
        });
        // 清理已斷開的客戶端 (可選，但建議)
        sseClients = sseClients.filter((client) => !client.writableEnded);
    }
    // 設置靜態文件目錄
    const __filename = fileURLToPath(import.meta.url);
    const __dirname = path.dirname(__filename);
    const publicPath = path.join(__dirname, "..", "..", "src", "public");
    const TASKS_FILE_PATH = await getTasksFilePath(); // 使用工具函數取得檔案路徑
    app.use(express.static(publicPath));
    // 設置 API 路由
    app.get("/api/tasks", async (req, res) => {
        try {
            // 使用 fsPromises 保持異步讀取
            const tasksData = await fsPromises.readFile(TASKS_FILE_PATH, "utf-8");
            res.json(JSON.parse(tasksData));
        }
        catch (error) {
            // 確保檔案不存在時返回空任務列表
            if (error.code === "ENOENT") {
                res.json({ tasks: [] });
            }
            else {
                res.status(500).json({ error: "Failed to read tasks data" });
            }
        }
    });
    // 新增：SSE 端點
    app.get("/api/tasks/stream", (req, res) => {
        res.writeHead(200, {
            "Content-Type": "text/event-stream",
            "Cache-Control": "no-cache",
            Connection: "keep-alive",
            // 可選: CORS 頭，如果前端和後端不在同一個 origin
            // "Access-Control-Allow-Origin": "*",
        });
        // 發送一個初始事件或保持連接
        res.write("data: connected\n\n");
        // 將客戶端添加到列表
        sseClients.push(res);
        // 當客戶端斷開連接時，將其從列表中移除
        req.on("close", () => {
            sseClients = sseClients.filter((client) => client !== res);
        });
    });
    // 定義 writeWebGuiFile 函數
    async function writeWebGuiFile(port) {
        try {
            // 讀取 TEMPLATES_USE 環境變數並轉換為語言代碼
            const templatesUse = process.env.TEMPLATES_USE || "en";
            const getLanguageFromTemplate = (template) => {
                if (template === "zh")
                    return "zh-TW";
                if (template === "en")
                    return "en";
                // 自訂範本預設使用英文
                return "en";
            };
            const language = getLanguageFromTemplate(templatesUse);
            const websiteUrl = `[Task Manager UI](http://localhost:${port}?lang=${language})`;
            const websiteFilePath = await getWebGuiFilePath();
            const DATA_DIR = await getDataDir();
            try {
                await fsPromises.access(DATA_DIR);
            }
            catch (error) {
                await fsPromises.mkdir(DATA_DIR, { recursive: true });
            }
            await fsPromises.writeFile(websiteFilePath, websiteUrl, "utf-8");
        }
        catch (error) { }
    }
    return {
        app,
        sendSseUpdate,
        async startServer() {
            // 獲取可用埠
            const port = process.env.WEB_PORT || (await getPort());
            // 啟動 HTTP 伺服器
            const httpServer = app.listen(port, () => {
                // 在伺服器啟動後開始監聽檔案變化
                try {
                    // 檢查檔案是否存在，如果不存在則不監聽 (避免 watch 報錯)
                    if (fs.existsSync(TASKS_FILE_PATH)) {
                        fs.watch(TASKS_FILE_PATH, (eventType, filename) => {
                            if (filename &&
                                (eventType === "change" || eventType === "rename")) {
                                // 稍微延遲發送，以防短時間內多次觸發 (例如編輯器保存)
                                // debounce sendSseUpdate if needed
                                sendSseUpdate();
                            }
                        });
                    }
                }
                catch (watchError) { }
                // 將 URL 寫入 WebGUI.md
                writeWebGuiFile(port).catch((error) => { });
            });
            // 設置進程終止事件處理 (確保移除 watcher)
            const shutdownHandler = async () => {
                // 關閉所有 SSE 連接
                sseClients.forEach((client) => client.end());
                sseClients = [];
                // 關閉 HTTP 伺服器
                await new Promise((resolve) => httpServer.close(() => resolve()));
            };
            process.on("SIGINT", shutdownHandler);
            process.on("SIGTERM", shutdownHandler);
            return httpServer;
        },
    };
}
//# sourceMappingURL=webServer.js.map