@yinuo-ngm/server 1.0.0

package/lib/app.d.ts

@@ -0,0 +1,2 @@
+import { FastifyBaseLogger } from "fastify";
+export declare function createServer(): Promise<import("fastify").FastifyInstance<import("node:http").Server<typeof import("node:http").IncomingMessage, typeof import("node:http").ServerResponse>, import("node:http").IncomingMessage, import("node:http").ServerResponse<import("node:http").IncomingMessage>, FastifyBaseLogger, import("fastify").FastifyTypeProviderDefault>>;

package/lib/app.js

@@ -0,0 +1,65 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createServer = createServer;
+const fastify_1 = __importDefault(require("fastify"));
+const core_plugin_1 = __importDefault(require("./plugins/core.plugin"));
+const error_handler_plugin_1 = require("./plugins/error-handler.plugin");
+const request_id_plugin_1 = __importDefault(require("./plugins/request-id.plugin"));
+const routes_plugin_1 = __importDefault(require("./plugins/routes.plugin"));
+const static_plugin_1 = __importDefault(require("./plugins/static.plugin"));
+const success_handle_plugin_1 = __importDefault(require("./plugins/success-handle.plugin"));
+const ws_plugin_1 = __importDefault(require("./plugins/ws/ws.plugin"));
+function normalizeLogLevel(v) {
+    const lv = (v ?? "").toLowerCase().trim();
+    if (!lv)
+        return undefined;
+    if (["fatal", "error", "warn", "info", "debug", "trace", "silent"].includes(lv))
+        return lv;
+    return "info";
+}
+function createFastifyLogger() {
+    const level = normalizeLogLevel(process.env.NGM_LOG_LEVEL);
+    if (!level)
+        return false;
+    const isDev = process.env.NODE_ENV !== "production";
+    if (isDev) {
+        return {
+            level,
+            transport: {
+                target: "pino-pretty",
+                options: {
+                    colorize: true,
+                    translateTime: "HH:MM:ss.l",
+                    ignore: "pid,hostname",
+                },
+            },
+        };
+    }
+    return { level };
+}
+async function createServer() {
+    const fastify = (0, fastify_1.default)({
+        logger: createFastifyLogger(),
+        genReqId: (req) => {
+            const hdr = req.headers["x-request-id"];
+            if (typeof hdr === "string" && hdr.trim())
+                return hdr.trim();
+            return crypto.randomUUID();
+        }
+    });
+    await fastify.register(request_id_plugin_1.default);
+    await fastify.register(error_handler_plugin_1.errorHandlerPlugin);
+    await fastify.register(success_handle_plugin_1.default);
+    await fastify.register(core_plugin_1.default);
+    await fastify.register(ws_plugin_1.default);
+    await fastify.register(routes_plugin_1.default);
+    await fastify.register(static_plugin_1.default);
+    fastify.addHook('onClose', async () => {
+        await fastify.core.dispose?.();
+        fastify.log.info('Server is closing, performing cleanup...');
+    });
+    return fastify;
+}

package/lib/common/api.d.ts

@@ -0,0 +1,2 @@
+import type { FastifyReply, FastifyRequest } from "fastify";
+export declare function ok<T>(req: FastifyRequest, reply: FastifyReply, data: T, status?: number): FastifyReply<import("fastify").RouteGenericInterface, import("fastify").RawServerDefault, import("node:http").IncomingMessage, import("node:http").ServerResponse<import("node:http").IncomingMessage>, unknown, import("fastify").FastifySchema, import("fastify").FastifyTypeProviderDefault, unknown>;

package/lib/common/api.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ok = ok;
+function ok(req, reply, data, status = 200) {
+    return reply.status(status).send({
+        ok: true,
+        data,
+        meta: {
+            requestId: req.id,
+            ts: Date.now(),
+        },
+    });
+}

package/lib/common/editor.d.ts

@@ -0,0 +1,5 @@
+export interface OpenFolderOptions {
+    editor?: "code" | "system";
+    file?: string;
+}
+export declare function openFolder(folder: string, opts?: OpenFolderOptions): Promise<void>;

package/lib/common/editor.js

@@ -0,0 +1,69 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.openFolder = openFolder;
+const core_1 = require("@yinuo-ngm/core");
+const launch_editor_1 = __importDefault(require("launch-editor"));
+const path = __importStar(require("path"));
+async function openFolder(folder, opts = {}) {
+    const editor = opts.editor ?? "code";
+    const file = opts.file;
+    const target = file
+        ? path.resolve(folder, file)
+        : path.resolve(folder);
+    return new Promise((resolve, reject) => {
+        let settled = false;
+        const timer = setTimeout(() => {
+            if (settled)
+                return;
+            settled = true;
+            resolve();
+        }, 200);
+        (0, launch_editor_1.default)(target, editor === "system" ? undefined : editor, (fileName, errorMsg) => {
+            if (settled)
+                return;
+            settled = true;
+            clearTimeout(timer);
+            if (errorMsg) {
+                reject(new core_1.AppError("EDITOR_NOT_FOUND", errorMsg, { fileName, editor, folder, file, target }));
+                return;
+            }
+            resolve();
+        });
+    });
+}

package/lib/env.d.ts

@@ -0,0 +1,7 @@
+export declare const env: {
+    port: number;
+    host: string;
+    dataDir: string;
+    logLevel: string;
+    sysLogCapacity: number;
+};

package/lib/env.js

@@ -0,0 +1,16 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.env = void 0;
+const path_1 = __importDefault(require("path"));
+const os_1 = __importDefault(require("os"));
+exports.env = {
+    port: Number(process.env.NGM_SERVER_PORT || 3210),
+    host: process.env.NGM_SERVER_HOST || "127.0.0.1",
+    dataDir: process.env.NGM_DATA_DIR ||
+        path_1.default.join(os_1.default.homedir(), ".ng-manager"),
+    logLevel: process.env.NGM_LOG_LEVEL || "info",
+    sysLogCapacity: 3000,
+};

package/lib/index.d.ts

@@ -0,0 +1 @@
+export {};

package/lib/index.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const app_1 = require("./app");
+const env_1 = require("./env");
+async function start() {
+    const { port, host } = env_1.env;
+    const app = await (0, app_1.createServer)();
+    await app.listen({ port, host });
+    app.log.info(`local server listening on http://${host}:${port}`);
+    app.core.sysLog.append({
+        level: "info",
+        text: `Server started at http://${host}:${port}`,
+        source: "server",
+        scope: "system",
+    });
+}
+start().catch((err) => {
+    console.error("Failed to start server:", err);
+    process.exit(1);
+});

package/lib/plugins/core.plugin.d.ts

@@ -0,0 +1,9 @@
+import type { FastifyInstance } from "fastify";
+import { type CoreApp } from "@yinuo-ngm/core";
+declare module "fastify" {
+    interface FastifyInstance {
+        core: CoreApp;
+    }
+}
+declare const _default: (fastify: FastifyInstance) => Promise<void>;
+export default _default;

package/lib/plugins/core.plugin.js

@@ -0,0 +1,16 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fastify_plugin_1 = __importDefault(require("fastify-plugin"));
+const core_1 = require("@yinuo-ngm/core");
+const env_1 = require("../env");
+exports.default = (0, fastify_plugin_1.default)(async function corePlugin(fastify) {
+    const coreApp = await (0, core_1.createCoreApp)({
+        sysLogCapacity: env_1.env.sysLogCapacity,
+        dataDir: env_1.env.dataDir,
+    });
+    fastify.decorate("core", coreApp);
+    fastify.log.info("[core] core app initialized");
+});

package/lib/plugins/error-handler.plugin.d.ts

@@ -0,0 +1,6 @@
+import type { FastifyPluginAsync } from "fastify";
+import { type ErrorCode } from "@yinuo-ngm/core";
+export declare const ERROR_STATUS: Record<ErrorCode, number>;
+export declare function mapStatus(code: ErrorCode): number;
+export declare const errorHandlerPlugin: FastifyPluginAsync;
+export default errorHandlerPlugin;

package/lib/plugins/error-handler.plugin.js

@@ -0,0 +1,113 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.errorHandlerPlugin = exports.ERROR_STATUS = void 0;
+exports.mapStatus = mapStatus;
+const fastify_plugin_1 = __importDefault(require("fastify-plugin"));
+const core_1 = require("@yinuo-ngm/core");
+exports.ERROR_STATUS = {
+    PROJECT_NOT_FOUND: 404,
+    PROJECT_ROOT_INVALID: 400,
+    PROJECT_ALREADY_EXISTS: 409,
+    PROJECT_IMPORT_NOT_EXISTS: 404,
+    PROJECT_IMPORT_NOT_DIR: 400,
+    PROJECT_IMPORT_ALREADY_REGISTERED: 409,
+    PROJECT_IMPORT_NOT_RECOGNIZED: 422,
+    PROJECT_IMPORT_SCAN_FAILED: 500,
+    INVALID_NAME: 400,
+    TARGET_EXISTS: 409,
+    INVALID_REPO_URL: 400,
+    INVALID_PARENT_DIR: 400,
+    GIT_CHECKOUT_FAILED: 500,
+    BOOTSTRAP_NOT_IN_PICK_STATE: 400,
+    BOOTSTRAP_CTX_NOT_FOUND: 404,
+    BOOTSTRAP_INVALID_PICKED_ROOT: 400,
+    BOOTSTRAP_NOT_WAITING_PICK: 400,
+    PROJECT_ANGULAR_JSON_INVALID: 400,
+    PROJECT_ANGULAR_JSON_NOT_FOUND: 404,
+    PROJECT_VITE_CONFIG_INVALID: 400,
+    PROJECT_VUE_CONFIG_NOT_FOUND: 404,
+    CONFIG_BACKUP_NOT_FOUND: 404,
+    CONFIG_READ_FAILED: 500,
+    CONFIG_WRITE_FAILED: 500,
+    CONFIG_CONFLICT: 409,
+    CONFIG_OPEN_FAILED: 500,
+    CONFIG_SCHEMA_NOT_FOUND: 404,
+    CONFIG_DOMAIN_NOT_FOUND: 404,
+    CONFIG_DOC_NOT_FOUND: 404,
+    TASK_NOT_FOUND: 404,
+    TASK_ALREADY_RUNNING: 409,
+    PROCESS_SPAWN_FAILED: 500,
+    TASK_SPEC_NOT_FOUND: 404,
+    TASK_NOT_RUNNABLE: 400,
+    RUN_NOT_FOUND: 404,
+    TASK_ID_REQUIRED: 400,
+    COMMAND_NOT_FOUND: 404,
+    BAD_JSON: 400,
+    BAD_MSG: 400,
+    OP_NOT_SUPPORTED: 400,
+    HANDLER_FAILED: 500,
+    TOPIC_NOT_FOUND: 404,
+    OP_NOT_FOUND: 400,
+    FS_PATH_NOT_FOUND: 404,
+    FS_ALREADY_EXISTS: 409,
+    FS_PERMISSION_DENIED: 403,
+    FS_EXISTS_FAILED: 500,
+    FS_INVALID_NAME: 400,
+    FS_MKDIR_FAILED: 500,
+    EDITOR_NOT_FOUND: 404,
+    EDITOR_LAUNCH_FAILED: 500,
+    STORAGE_IO_ERROR: 500,
+    UNAUTHORIZED: 401,
+    INVALID_TIMESTAMP: 400,
+    DEP_INSTALL_FAILED: 500,
+    DEP_UNINSTALL_FAILED: 500,
+    DEP_NOT_FOUND: 404,
+    DASHBOARD_CONFLICT: 409,
+    WIDGET_NOT_FOUND: 404,
+    WIDGET_LOCKED: 423,
+    RSS_FETCH_FAILED: 500,
+    INVALID_RSS_URL: 400,
+    UNKNOWN_ERROR: 500,
+    BAD_REQUEST: 400,
+    NOT_IMPLEMENTED: 501,
+};
+function mapStatus(code) {
+    return exports.ERROR_STATUS[code] ?? 400;
+}
+function buildErrorBody(reqId, code, message, details) {
+    return {
+        ok: false,
+        error: {
+            code,
+            message,
+            ...(details !== undefined ? { details } : {}),
+        },
+        meta: {
+            requestId: reqId,
+            ts: Date.now(),
+        },
+    };
+}
+exports.errorHandlerPlugin = (0, fastify_plugin_1.default)(async (app) => {
+    app.setErrorHandler((err, req, reply) => {
+        const requestId = req.id;
+        if (err instanceof core_1.AppError || isCoreAppError(err)) {
+            const status = mapStatus(err.code);
+            return reply.status(status).send(buildErrorBody(requestId, err.code, err.message, err.meta));
+        }
+        if (err?.validation) {
+            return reply
+                .status(400)
+                .send(buildErrorBody(requestId, "VALIDATION_ERROR", "参数不合法", err.validation));
+        }
+        req.log.error({ err, requestId }, "Unhandled error");
+        return reply.status(500).send(buildErrorBody(requestId, "INTERNAL_ERROR", "服务异常，请稍后重试"));
+    });
+});
+function isCoreAppError(err) {
+    return err && typeof err === "object" && typeof err.code === "string" && typeof err.message === "string";
+}
+exports.default = exports.errorHandlerPlugin;

package/lib/plugins/request-id.plugin.d.ts

@@ -0,0 +1,3 @@
+import type { FastifyPluginAsync } from "fastify";
+export declare const requestIdPlugin: FastifyPluginAsync;
+export default requestIdPlugin;

package/lib/plugins/request-id.plugin.js

@@ -0,0 +1,14 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.requestIdPlugin = void 0;
+const fastify_plugin_1 = __importDefault(require("fastify-plugin"));
+exports.requestIdPlugin = (0, fastify_plugin_1.default)(async (app) => {
+    app.addHook("onSend", async (req, reply, payload) => {
+        reply.header("X-Request-Id", req.id);
+        return payload;
+    });
+});
+exports.default = exports.requestIdPlugin;

package/lib/plugins/routes.plugin.d.ts

@@ -0,0 +1,3 @@
+import type { FastifyInstance } from "fastify";
+declare const _default: (fastify: FastifyInstance) => Promise<void>;
+export default _default;

package/lib/plugins/routes.plugin.js

@@ -0,0 +1,17 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fastify_plugin_1 = __importDefault(require("fastify-plugin"));
+const routes_1 = require("../routes");
+exports.default = (0, fastify_plugin_1.default)(async function routesPlugin(fastify) {
+    await fastify.register(routes_1.systemRoutes);
+    await fastify.register(routes_1.taskRoutes, { prefix: '/api/tasks' });
+    await fastify.register(routes_1.projectRoutes, { prefix: '/api/projects' });
+    await fastify.register(routes_1.depsRoutes, { prefix: '/api/deps' });
+    await fastify.register(routes_1.fsRoutes, { prefix: '/api/fs' });
+    await fastify.register(routes_1.configRoutes, { prefix: '/api/config' });
+    await fastify.register(routes_1.dashboardRoutes, { prefix: '/api/dashboard' });
+    await fastify.register(routes_1.rssRoutes, { prefix: '/api/rss' });
+});

package/lib/plugins/static.plugin.d.ts

@@ -0,0 +1,2 @@
+declare const _default: (fastify: import("fastify").FastifyInstance<import("fastify").RawServerDefault, import("node:http").IncomingMessage, import("node:http").ServerResponse<import("node:http").IncomingMessage>, import("fastify").FastifyBaseLogger, import("fastify").FastifyTypeProviderDefault>) => Promise<void>;
+export default _default;

package/lib/plugins/static.plugin.js

@@ -0,0 +1,24 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fastify_plugin_1 = __importDefault(require("fastify-plugin"));
+const path_1 = __importDefault(require("path"));
+const static_1 = __importDefault(require("@fastify/static"));
+exports.default = (0, fastify_plugin_1.default)(async function staticPlugin(fastify) {
+    const webRoot = path_1.default.resolve(__dirname, "../../www/browser");
+    fastify.log.info(`[static] serving webapp from ${webRoot}`);
+    await fastify.register(static_1.default, {
+        root: webRoot,
+        index: "index.html",
+    });
+    fastify.setNotFoundHandler((req, reply) => {
+        if (req.method === "GET" &&
+            !req.url.startsWith("/api") &&
+            !req.url.startsWith("/ws")) {
+            return reply.sendFile("index.html");
+        }
+        reply.code(404).send({ ok: false, message: "Not Found" });
+    });
+});

package/lib/plugins/success-handle.plugin.d.ts

@@ -0,0 +1,3 @@
+import type { FastifyPluginAsync } from "fastify";
+export declare const successHandlerPlugin: FastifyPluginAsync;
+export default successHandlerPlugin;

package/lib/plugins/success-handle.plugin.js

@@ -0,0 +1,30 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.successHandlerPlugin = void 0;
+const fastify_plugin_1 = __importDefault(require("fastify-plugin"));
+exports.successHandlerPlugin = (0, fastify_plugin_1.default)(async (app) => {
+    app.addHook("preSerialization", async (req, reply, payload) => {
+        if (reply.statusCode < 200 || reply.statusCode >= 300) {
+            return payload;
+        }
+        if (payload === undefined || payload === null) {
+            return payload;
+        }
+        const type = reply.getHeader("content-type");
+        if (typeof type === "string" && !type.includes("application/json")) {
+            return payload;
+        }
+        return {
+            ok: true,
+            data: payload,
+            meta: {
+                requestId: req.id,
+                ts: Date.now(),
+            },
+        };
+    });
+});
+exports.default = exports.successHandlerPlugin;

package/lib/plugins/ws/topics/index.d.ts

@@ -0,0 +1 @@
+export * from "./task.ws";

package/lib/plugins/ws/topics/index.js

@@ -0,0 +1,17 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./task.ws"), exports);

package/lib/plugins/ws/topics/syslog.ws.d.ts

@@ -0,0 +1,9 @@
+import type { LogLine } from "@yinuo-ngm/core";
+import { WsContext } from "../ws.context";
+import type { TopicHandler } from "../ws.router";
+export type SyslogWsDeps = {
+    getSyslogTail: (tail: number) => LogLine[];
+};
+export declare function createSyslogTopicHandler(deps: SyslogWsDeps, getAllClients: () => Iterable<WsContext>): TopicHandler & {
+    push(entry: LogLine): void;
+};

package/lib/plugins/ws/topics/syslog.ws.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createSyslogTopicHandler = createSyslogTopicHandler;
+const KEY_ALL = "syslog:all";
+function createSyslogTopicHandler(deps, getAllClients) {
+    return {
+        topic: "syslog",
+        async sub(ctx, msg) {
+            const tail = Math.max(0, Number(msg?.tail ?? 0));
+            ctx.addSub("syslog", KEY_ALL);
+            if (tail > 0) {
+                const entries = await deps.getSyslogTail(tail);
+                const m = { op: "syslog.tail", entries: entries ?? [] };
+                ctx.send(m);
+            }
+        },
+        unsub(ctx) {
+            ctx.delSub("syslog", KEY_ALL);
+        },
+        push(entry) {
+            const m = { op: "syslog.append", entry };
+            for (const c of getAllClients()) {
+                if (c.hasSub("syslog", KEY_ALL))
+                    c.send(m);
+            }
+        },
+    };
+}

package/lib/plugins/ws/topics/task.ws.d.ts

@@ -0,0 +1,12 @@
+import type { TaskEventPayloadMap, TaskEventType, TaskOutputPayload, LogLine, TaskRuntime } from "@yinuo-ngm/core";
+import { WsContext } from "../ws.context";
+import type { TopicHandler } from "../ws.router";
+export type TaskWsDeps = {
+    getTaskSnapshotByTaskId?: (taskId: string) => Promise<TaskRuntime | null>;
+    getTaskTailLogsByRun?: (runId: string, tail: number) => Promise<LogLine[]>;
+    resizeRun?: (taskId: string, cols: number, rows: number) => void;
+};
+export declare function createTaskTopicHandler(deps: TaskWsDeps, getAllClients: () => Iterable<WsContext>): TopicHandler & {
+    pushOutput(payload: TaskOutputPayload): void;
+    pushEvent<K extends TaskEventType>(type: K, payload: TaskEventPayloadMap[K]): void;
+};

package/lib/plugins/ws/topics/task.ws.js

@@ -0,0 +1,75 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createTaskTopicHandler = createTaskTopicHandler;
+const keyOfTask = (taskId) => `task:${taskId}`;
+function createTaskTopicHandler(deps, getAllClients) {
+    return {
+        topic: "task",
+        async sub(ctx, msg) {
+            const taskId = String(msg?.taskId ?? "").trim();
+            const tail = Number(msg?.tail ?? 0);
+            if (!taskId) {
+                ctx.send({ op: "error", code: "TASK_ID_REQUIRED", message: "taskId is required", ts: Date.now() });
+                return;
+            }
+            ctx.addSub("task", keyOfTask(taskId));
+            let snap;
+            if (deps.getTaskSnapshotByTaskId) {
+                snap = await deps.getTaskSnapshotByTaskId(taskId) ?? undefined;
+                if (snap) {
+                    const m = {
+                        op: "task.event",
+                        type: "snapshot",
+                        payload: snap,
+                        ts: Date.now(),
+                    };
+                    ctx.send(m);
+                }
+            }
+            if (snap?.runId && deps.getTaskTailLogsByRun && tail > 0) {
+                const entries = await deps.getTaskTailLogsByRun(snap.runId, tail);
+                for (const e of entries ?? []) {
+                    const m = {
+                        op: "task.output",
+                        payload: {
+                            taskId: taskId,
+                            runId: snap.runId,
+                            stream: (e?.level === "warn" || e?.level === "error") ? "stderr" : "stdout",
+                            text: String(e?.text ?? ""),
+                        },
+                        ts: e?.ts ?? Date.now(),
+                    };
+                    ctx.send(m);
+                }
+            }
+        },
+        resize(ctx, msg) {
+            const taskId = msg.taskId;
+            const cols = Number(msg.cols);
+            const rows = Number(msg.rows);
+            if (!taskId || !Number.isFinite(cols) || !Number.isFinite(rows))
+                return;
+            const c = Math.max(2, Math.floor(cols));
+            const r = Math.max(1, Math.floor(rows));
+            deps.resizeRun?.(taskId, c, r);
+        },
+        unsub(ctx, msg) {
+            const taskId = String(msg?.taskId ?? "").trim();
+            if (!taskId)
+                return;
+            ctx.delSub("task", keyOfTask(taskId));
+        },
+        pushOutput(payload) {
+            const m = { op: "task.output", payload, ts: Date.now() };
+            for (const c of getAllClients())
+                if (c.hasSub("task", keyOfTask(payload.taskId)))
+                    c.send(m);
+        },
+        pushEvent(type, payload) {
+            const m = { op: "task.event", type, payload, ts: Date.now() };
+            for (const c of getAllClients())
+                if (c.hasSub("task", keyOfTask(payload.taskId)))
+                    c.send(m);
+        },
+    };
+}

package/lib/plugins/ws/ws.context.d.ts

@@ -0,0 +1,12 @@
+import type { WsServerMsg, WsTopic } from "@yinuo-ngm/core";
+export declare class WsContext {
+    private socket;
+    readonly connId: string;
+    private subs;
+    constructor(connId: string, socket: WebSocket);
+    send(msg: WsServerMsg): void;
+    addSub(topic: WsTopic, key: string): void;
+    delSub(topic: WsTopic, key: string): void;
+    hasSub(topic: WsTopic, key: string): boolean;
+    clearAll(): void;
+}