@next-open-ai/openbot 0.1.1

package/dist/agent/config-manager.js

@@ -0,0 +1,84 @@
+import { readFileSync, writeFileSync, existsSync } from "node:fs";
+import { join } from "node:path";
+import { AuthStorage } from "@mariozechner/pi-coding-agent";
+import { getOpenbotAgentDir, ensureDefaultAgentDir } from "./agent-dir.js";
+/**
+ * Manage agent configuration (auth.json, models.json)
+ */
+export class ConfigManager {
+    agentDir;
+    authStorage;
+    constructor(agentDir = getOpenbotAgentDir()) {
+        this.agentDir = agentDir;
+        ensureDefaultAgentDir(this.agentDir);
+        const authPath = join(this.agentDir, "auth.json");
+        this.authStorage = new AuthStorage(authPath);
+    }
+    /**
+     * Save API key for a provider
+     */
+    async login(provider, apiKey) {
+        this.authStorage.set(provider, {
+            type: "api_key",
+            key: apiKey
+        });
+        console.log(`[config] Successfully saved API key for provider: ${provider}`);
+    }
+    /**
+     * Update default model for a provider in models.json
+     */
+    async setModel(provider, modelId) {
+        const modelsPath = join(this.agentDir, "models.json");
+        if (!existsSync(modelsPath)) {
+            throw new Error(`models.json not found at ${modelsPath}`);
+        }
+        const config = JSON.parse(readFileSync(modelsPath, "utf-8"));
+        if (!config.providers[provider]) {
+            throw new Error(`Provider "${provider}" not found in models.json`);
+        }
+        // Check if model exists in that provider's list
+        const providerConfig = config.providers[provider];
+        const modelExists = providerConfig.models?.some((m) => m.id === modelId);
+        if (!modelExists) {
+            console.warn(`[config] Warning: Model "${modelId}" not found in the defined list for "${provider}". Adding it anyway.`);
+            if (!providerConfig.models)
+                providerConfig.models = [];
+            providerConfig.models.push({
+                id: modelId,
+                name: modelId,
+                contextWindow: 128000,
+                supportsTools: true
+            });
+        }
+        // In this simple implementation, we assume the first model in the list
+        // might be intended as default by some logic, although the registry 
+        // usually needs explicit selection.
+        // For now, let's just reorder so the selected model is first.
+        const modelIndex = providerConfig.models.findIndex((m) => m.id === modelId);
+        if (modelIndex > 0) {
+            const [model] = providerConfig.models.splice(modelIndex, 1);
+            providerConfig.models.unshift(model);
+        }
+        writeFileSync(modelsPath, JSON.stringify(config, null, 2), "utf-8");
+        console.log(`[config] Updated default model for "${provider}" to "${modelId}"`);
+    }
+    /**
+     * List current configurations
+     */
+    list() {
+        const modelsPath = join(this.agentDir, "models.json");
+        if (!existsSync(modelsPath))
+            return [];
+        const config = JSON.parse(readFileSync(modelsPath, "utf-8"));
+        const results = [];
+        for (const [providerId, providerConfig] of Object.entries(config.providers)) {
+            const firstModel = providerConfig.models?.[0]?.id || "none";
+            results.push({
+                provider: providerId,
+                model: firstModel,
+                hasKey: this.authStorage.has(providerId)
+            });
+        }
+        return results;
+    }
+}

package/dist/agent/desktop-config.d.ts

@@ -0,0 +1,15 @@
+export interface DesktopAgentConfig {
+    provider: string;
+    model: string;
+    apiKey?: string;
+}
+/**
+ * 从 config.json 读取缺省智能体 id（defaultAgentId）。
+ * 若未配置、或该 id 在 agents.json 中不存在，则返回 default，即使用主智能体配置创建 session。
+ */
+export declare function getBoundAgentIdForCli(): Promise<string>;
+/**
+ * 根据 workspace（agent id）从桌面配置中读取该 agent 的 provider、model 及 API Key。
+ * 若文件不存在或解析失败返回 null，调用方使用环境变量或默认值。
+ */
+export declare function loadDesktopAgentConfig(workspace: string): Promise<DesktopAgentConfig | null>;

package/dist/agent/desktop-config.js

@@ -0,0 +1,91 @@
+/**
+ * 从桌面端配置目录（~/.openbot/desktop）读取 config.json 与 agents.json，
+ * 供 CLI / Gateway 等与桌面设置一致的 provider、model、apiKey。
+ */
+import { readFile } from "fs/promises";
+import { join } from "path";
+import { existsSync } from "fs";
+import { homedir } from "os";
+function getDesktopDir() {
+    const home = process.env.HOME || process.env.USERPROFILE || homedir();
+    return join(home, ".openbot", "desktop");
+}
+const DEFAULT_AGENT_ID = "default";
+/**
+ * 从 config.json 读取缺省智能体 id（defaultAgentId）。
+ * 若未配置、或该 id 在 agents.json 中不存在，则返回 default，即使用主智能体配置创建 session。
+ */
+export async function getBoundAgentIdForCli() {
+    const desktopDir = getDesktopDir();
+    const configPath = join(desktopDir, "config.json");
+    const agentsPath = join(desktopDir, "agents.json");
+    let boundId = DEFAULT_AGENT_ID;
+    if (existsSync(configPath)) {
+        try {
+            const raw = await readFile(configPath, "utf-8");
+            const config = JSON.parse(raw);
+            const id = config.defaultAgentId ? String(config.defaultAgentId).trim() : "";
+            if (id)
+                boundId = id;
+        }
+        catch {
+            // ignore, use default
+        }
+    }
+    if (boundId === DEFAULT_AGENT_ID)
+        return DEFAULT_AGENT_ID;
+    if (!existsSync(agentsPath))
+        return DEFAULT_AGENT_ID;
+    try {
+        const raw = await readFile(agentsPath, "utf-8");
+        const data = JSON.parse(raw);
+        const agents = Array.isArray(data.agents) ? data.agents : [];
+        const found = agents.some((a) => a.id === boundId);
+        return found ? boundId : DEFAULT_AGENT_ID;
+    }
+    catch {
+        return DEFAULT_AGENT_ID;
+    }
+}
+/**
+ * 根据 workspace（agent id）从桌面配置中读取该 agent 的 provider、model 及 API Key。
+ * 若文件不存在或解析失败返回 null，调用方使用环境变量或默认值。
+ */
+export async function loadDesktopAgentConfig(workspace) {
+    const desktopDir = getDesktopDir();
+    const configPath = join(desktopDir, "config.json");
+    const agentsPath = join(desktopDir, "agents.json");
+    let config = {};
+    if (existsSync(configPath)) {
+        try {
+            const raw = await readFile(configPath, "utf-8");
+            config = JSON.parse(raw);
+        }
+        catch {
+            return null;
+        }
+    }
+    const agentId = workspace === "default" ? "default" : workspace;
+    let provider = config.defaultProvider ?? "deepseek";
+    let model = config.defaultModel ?? "deepseek-chat";
+    if (existsSync(agentsPath)) {
+        try {
+            const raw = await readFile(agentsPath, "utf-8");
+            const data = JSON.parse(raw);
+            const agents = Array.isArray(data.agents) ? data.agents : [];
+            const agent = agents.find((a) => a.id === agentId);
+            if (agent?.provider)
+                provider = agent.provider;
+            if (agent?.model)
+                model = agent.model;
+        }
+        catch {
+            // ignore
+        }
+    }
+    const provConfig = config.providers?.[provider];
+    const apiKey = provConfig?.apiKey && typeof provConfig.apiKey === "string" && provConfig.apiKey.trim()
+        ? provConfig.apiKey.trim()
+        : undefined;
+    return { provider, model, apiKey: apiKey ?? undefined };
+}

package/dist/agent/run.d.ts

@@ -0,0 +1,26 @@
+export interface RunOptions {
+    /** Workspace name (e.g. "default") */
+    workspace?: string;
+    /** Additional skill paths */
+    skillPaths?: string[];
+    /** 用户提示词 */
+    userPrompt: string;
+    /** 是否仅打印组装后的内容，不调 LLM */
+    dryRun?: boolean;
+    /** API key，不传则从 pi 的 auth 或 OPENAI_API_KEY 读取 */
+    apiKey?: string;
+    /** 模型 ID（如 deepseek-chat、qwen-max） */
+    model?: string;
+    /** Provider（如 deepseek、dashscope、openai） */
+    provider?: string;
+    /** Agent 配置目录，默认 ~/.openbot/agent */
+    agentDir?: string;
+}
+export interface RunResult {
+    systemPrompt: string;
+    userPrompt: string;
+    /** 若调用了 API，则为助手回复 */
+    assistantContent?: string;
+    dryRun: boolean;
+}
+export declare function run(options: RunOptions): Promise<RunResult>;

package/dist/agent/run.js

@@ -0,0 +1,65 @@
+import { getOpenbotAgentDir } from "./agent-dir.js";
+import { AgentManager } from "./agent-manager.js";
+export async function run(options) {
+    const { workspace = "default", skillPaths = [], userPrompt, dryRun: explicitDryRun, apiKey, model: modelId = "deepseek-chat", provider = "deepseek", agentDir = getOpenbotAgentDir(), } = options;
+    const manager = new AgentManager({ agentDir, workspace, skillPaths });
+    const { systemPrompt } = await manager.getContext();
+    // Determine dry run
+    const dryRun = !!explicitDryRun; // Ensure boolean
+    const result = {
+        systemPrompt,
+        userPrompt,
+        dryRun,
+    };
+    if (dryRun) {
+        return result;
+    }
+    // Create a temporary session for this run
+    const sessionId = `cli-${Date.now()}`;
+    const session = await manager.getOrCreateSession(sessionId, {
+        provider,
+        modelId,
+        apiKey,
+    });
+    // Collect assistant response
+    let assistantContent = "";
+    session.subscribe((event) => {
+        switch (event.type) {
+            case "message_update": {
+                const evt = event.assistantMessageEvent;
+                if (evt.type === "text_delta") {
+                    assistantContent += evt.delta;
+                    process.stdout.write(evt.delta);
+                }
+                else if (evt.type === "thinking_delta") {
+                    process.stdout.write(evt.delta);
+                }
+                else if (evt.type === "error") {
+                    console.error(`\n[model:error] ${evt.error.errorMessage || "Unknown error"}`);
+                }
+                break;
+            }
+            case "message_end": {
+                if (event.message.role === "assistant" && event.message.errorMessage) {
+                    console.error(`\n[message:error] ${event.message.errorMessage}`);
+                }
+                break;
+            }
+            case "tool_execution_start": {
+                console.log(`\n[tool:start] ${event.toolName}(${JSON.stringify(event.args)})`);
+                break;
+            }
+            case "tool_execution_end": {
+                const status = event.isError ? "failed" : "ok";
+                console.log(`[tool:end] ${event.toolName} -> ${status}`);
+                break;
+            }
+        }
+    });
+    // Send prompt and wait for completion
+    await session.prompt(userPrompt);
+    // Clean up session
+    manager.deleteSession(sessionId);
+    result.assistantContent = assistantContent.trim();
+    return result;
+}

package/dist/agent/skills.d.ts

@@ -0,0 +1,20 @@
+import { type Skill } from "@mariozechner/pi-coding-agent";
+export type { Skill };
+export interface LoadSkillsFromDirOptions {
+    dir: string;
+    source: string;
+}
+/**
+ * 从目录加载 skills（委托给 pi-coding-agent，规则与其一致）
+ */
+export declare function loadSkillsFromDir(options: LoadSkillsFromDirOptions): Skill[];
+/**
+ * 从多个路径加载 skills（文件或目录），合并去重（按 name）
+ * 目录使用 pi-coding-agent 的 loadSkillsFromDir，单文件使用本地解析
+ */
+export declare function loadSkillsFromPaths(paths: string[], source?: string): Skill[];
+/**
+ * 将 Skill[] 格式化为系统提示中的一段（委托给 pi-coding-agent，对齐 agentskills.io）
+ */
+declare function formatSkillsForPrompt(skills: Skill[]): string;
+export { formatSkillsForPrompt };

package/dist/agent/skills.js

@@ -0,0 +1,86 @@
+import { existsSync, readFileSync, statSync } from "node:fs";
+import { join, dirname, basename } from "node:path";
+import { formatSkillsForPrompt as piFormatSkillsForPrompt, loadSkillsFromDir as piLoadSkillsFromDir, } from "@mariozechner/pi-coding-agent";
+/**
+ * 简单 frontmatter 解析：--- 与 --- 之间的 YAML 风格键值
+ */
+function parseFrontmatter(raw) {
+    const match = raw.match(/^---\s*\n([\s\S]*?)\n---/);
+    if (!match)
+        return {};
+    const block = match[1];
+    const out = {};
+    for (const line of block.split("\n")) {
+        const m = line.match(/^\s*([a-z-]+)\s*:\s*(.*)$/);
+        if (m)
+            out[m[1].trim()] = m[2].trim().replace(/^["']|["']$/g, "");
+    }
+    return {
+        name: out.name,
+        description: out.description,
+    };
+}
+/**
+ * 从单文件加载一个 Skill（与 pi-coding-agent Skill 结构一致）
+ */
+function loadSkillFromFile(filePath, source) {
+    try {
+        const raw = readFileSync(filePath, "utf-8");
+        const { name, description } = parseFrontmatter(raw);
+        const skillDir = dirname(filePath);
+        const parentDirName = basename(skillDir);
+        const finalName = (name || parentDirName).trim();
+        const finalDesc = (description || "").trim();
+        if (!finalDesc)
+            return null;
+        return {
+            name: finalName,
+            description: finalDesc,
+            filePath,
+            baseDir: skillDir,
+            source,
+            disableModelInvocation: false,
+        };
+    }
+    catch {
+        return null;
+    }
+}
+/**
+ * 从目录加载 skills（委托给 pi-coding-agent，规则与其一致）
+ */
+export function loadSkillsFromDir(options) {
+    const result = piLoadSkillsFromDir(options);
+    return result.skills ?? [];
+}
+/**
+ * 从多个路径加载 skills（文件或目录），合并去重（按 name）
+ * 目录使用 pi-coding-agent 的 loadSkillsFromDir，单文件使用本地解析
+ */
+export function loadSkillsFromPaths(paths, source = "user") {
+    const byName = new Map();
+    for (const p of paths) {
+        const resolved = p.startsWith("~") ? join(process.env.HOME || "", p.slice(1)) : p;
+        if (!existsSync(resolved))
+            continue;
+        const stat = statSync(resolved);
+        if (stat.isFile() && resolved.endsWith(".md")) {
+            const skill = loadSkillFromFile(resolved, source);
+            if (skill)
+                byName.set(skill.name, skill);
+        }
+        else if (stat.isDirectory()) {
+            for (const skill of loadSkillsFromDir({ dir: resolved, source })) {
+                byName.set(skill.name, skill);
+            }
+        }
+    }
+    return Array.from(byName.values());
+}
+/**
+ * 将 Skill[] 格式化为系统提示中的一段（委托给 pi-coding-agent，对齐 agentskills.io）
+ */
+function formatSkillsForPrompt(skills) {
+    return piFormatSkillsForPrompt(skills);
+}
+export { formatSkillsForPrompt };

package/dist/cli.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/dist/cli.js

@@ -0,0 +1,168 @@
+#!/usr/bin/env node
+import { createRequire } from "node:module";
+import { dirname, join } from "node:path";
+import { fileURLToPath } from "node:url";
+import { Command } from "commander";
+import { getOpenbotAgentDir } from "./agent/agent-dir.js";
+import { run } from "./agent/run.js";
+import { ConfigManager } from "./agent/config-manager.js";
+import { loadDesktopAgentConfig, getBoundAgentIdForCli } from "./agent/desktop-config.js";
+const require = createRequire(import.meta.url);
+const PKG = require("../package.json");
+// Based on distillate, the package root is one level up from dist/
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const PKG_ROOT = join(__dirname, "..");
+async function runAction(positionalPrompt, opts) {
+    const skillPaths = opts.skillPath || [];
+    // 指定智能体：-a/--agent 优先，其次 -w/--workspace（且非 default）；不提供则使用配置中的缺省智能体
+    const explicitAgent = (opts.agent && String(opts.agent).trim()) ||
+        (opts.workspace && opts.workspace !== "default" ? opts.workspace : "");
+    const workspace = explicitAgent ? explicitAgent.trim() : await getBoundAgentIdForCli();
+    const prompt = (opts.prompt ?? positionalPrompt ?? "").trim();
+    if (!prompt) {
+        console.error("Error: 请提供提示词（位置参数或 --prompt）");
+        process.exit(1);
+    }
+    console.error(`[openbot] Using agent: ${workspace}${!explicitAgent ? " (default from config)" : ""}`);
+    // 未传 --api-key / --provider / --model 时从桌面配置（~/.openbot/desktop）读取
+    const needDesktop = opts.apiKey === undefined || opts.provider === undefined || opts.model === undefined;
+    const desktopConfig = needDesktop ? await loadDesktopAgentConfig(workspace) : null;
+    const provider = opts.provider ?? desktopConfig?.provider ?? "deepseek";
+    const model = opts.model ?? desktopConfig?.model ?? "deepseek-chat";
+    const apiKey = opts.apiKey ?? process.env.OPENAI_API_KEY ?? desktopConfig?.apiKey ?? "";
+    if (desktopConfig && (desktopConfig.provider || desktopConfig.model)) {
+        console.error(`[openbot] Using model: ${provider}/${model} (from desktop config)`);
+    }
+    if (opts.timing)
+        process.env.OPENBOT_TIMING = "1";
+    try {
+        const result = await run({
+            workspace,
+            skillPaths,
+            userPrompt: prompt,
+            dryRun: opts.dryRun ?? false,
+            model,
+            provider,
+            agentDir: opts.agentDir ?? getOpenbotAgentDir(),
+            apiKey: apiKey || undefined,
+        });
+        if (result.dryRun) {
+            console.log("=== System prompt (skills) ===");
+            console.log(result.systemPrompt);
+            console.log("\n=== User prompt ===");
+            console.log(result.userPrompt);
+            console.log("\n(dry-run: 未调用 LLM，设置 OPENAI_API_KEY 可实际调用)");
+            return;
+        }
+        // Content is already streamed by run()
+        if (!result.dryRun &&
+            (result.assistantContent === undefined || result.assistantContent === "")) {
+            console.warn("（模型返回为空；若此前有工具调用提示，说明当前为单轮模式，模型会基于技能描述直接以文本回答。）");
+        }
+    }
+    catch (err) {
+        console.error(err);
+        process.exit(1);
+    }
+}
+const program = new Command();
+program
+    .name("openbot")
+    .description("CLI to run prompts with skill paths (Agent Skills style)")
+    .version(PKG.version, "-v, --version", "显示版本号")
+    .option("-s, --skill-path <paths...>", "Additional skill paths to load")
+    .option("-a, --agent <id>", "指定智能体 ID，不传则使用配置中的缺省智能体")
+    .option("-w, --workspace <name>", "Workspace/Agent 名称（同 --agent，兼容旧用法）", "default")
+    .option("-p, --prompt <text>", "用户提示词（与位置参数二选一）")
+    .option("--dry-run", "只输出组装的 system/user 内容，不调用 LLM")
+    .option("--model <id>", "模型 ID", "deepseek-chat")
+    .option("--provider <name>", "Provider（pi ModelRegistry）；可选 deepseek、dashscope、openai", "deepseek")
+    .option("--agent-dir <path>", "Agent 配置目录（默认 ~/.openbot/agent）", getOpenbotAgentDir())
+    .option("--api-key <key>", "API Key（不传则使用环境变量 OPENAI_API_KEY）")
+    .option("--timing", "打印每轮 LLM 与 tool 耗时到 stderr")
+    .option("--max-tool-turns <n>", "最大工具调用轮数（默认 30）；可设环境变量 OPENBOT_MAX_TOOL_TURNS", (v) => parseInt(v, 10) || 0, 0)
+    .argument("[prompt]", "用户提示词（与 --prompt 二选一）")
+    .action(async (positionalPrompt) => {
+    await runAction(positionalPrompt, program.opts());
+});
+program.addHelpText("after", `
+Environment:
+  DEEPSEEK_API_KEY          默认 provider 为 deepseek 时使用；不设时回退 OPENAI_API_KEY
+  OPENAI_API_KEY            通用 API Key（可被 --api-key 覆盖）
+  DASHSCOPE_API_KEY         provider=dashscope 时使用；不设时回退 OPENAI_API_KEY
+  OPENAI_BASE_URL           可选，在 pi 未找到模型时使用的 endpoint
+  OPENBOT_AGENT_DIR         缺省 agent 目录（默认 ~/.openbot/agent）
+  OPENBOT_TIMING=1          等同 --timing
+  OPENBOT_ALLOW_RUN_CODE    缺省 1（启用 run_python）；设为 0 关闭
+  OPENBOT_MAX_TOOL_TURNS    最大工具轮数（默认 30）
+
+Examples:
+  openbot "总结一下当前有哪些技能"
+  openbot -a my-agent "总结一下当前有哪些技能"  使用指定智能体
+  openbot -s ./skills "总结一下当前有哪些技能"
+  openbot -s ./my-skills --prompt "用 weather 技能查北京天气" --dry-run
+`);
+// Gateway server command
+program
+    .command("gateway")
+    .description("Start WebSocket gateway server")
+    .option("-p, --port <port>", "Port to listen on", "3000")
+    .action(async (options) => {
+    const port = parseInt(options.port, 10);
+    if (isNaN(port) || port <= 0 || port > 65535) {
+        console.error("Error: Invalid port number");
+        process.exit(1);
+    }
+    const { startGatewayServer } = await import("./gateway/index.js");
+    const { close } = await startGatewayServer(port);
+    // Handle graceful shutdown
+    const shutdown = async () => {
+        console.log("\nShutting down...");
+        await close();
+        process.exit(0);
+    };
+    process.on("SIGINT", shutdown);
+    process.on("SIGTERM", shutdown);
+});
+// Login command
+program
+    .command("login")
+    .description("Save API key for a provider persistently")
+    .argument("<provider>", "Provider name (e.g., deepseek, dashscope, openai)")
+    .argument("<apiKey>", "API Key")
+    .action(async (provider, apiKey) => {
+    const config = new ConfigManager(program.opts().agentDir);
+    await config.login(provider, apiKey);
+});
+// Config command
+const configCmd = program.command("config").description("Manage configurations");
+configCmd
+    .command("set-model")
+    .description("Set default model for a provider")
+    .argument("<provider>", "Provider name")
+    .argument("<modelId>", "Model ID")
+    .action(async (provider, modelId) => {
+    const cm = new ConfigManager(program.opts().agentDir);
+    await cm.setModel(provider, modelId);
+});
+configCmd
+    .command("list")
+    .description("List current configurations")
+    .action(() => {
+    const cm = new ConfigManager(program.opts().agentDir);
+    const results = cm.list();
+    if (results.length === 0) {
+        console.log("No providers found in models.json.");
+    }
+    else {
+        console.table(results.map(r => ({
+            Provider: r.provider,
+            "Default Model": r.model,
+            "Auth Configured": r.hasKey ? "✅ Yes" : "❌ No"
+        })));
+    }
+});
+program.parseAsync(process.argv).catch((err) => {
+    console.error(err);
+    process.exit(1);
+});

package/dist/gateway/backend-url.d.ts

@@ -0,0 +1,2 @@
+export declare function setBackendBaseUrl(url: string): void;
+export declare function getBackendBaseUrl(): string | null;

package/dist/gateway/backend-url.js

@@ -0,0 +1,11 @@
+/**
+ * Backend base URL for the Nest Desktop Server (e.g. http://localhost:3001).
+ * Set by startGatewayServer so agent-chat / usage reporting can POST to server-api.
+ */
+let backendBaseUrl = null;
+export function setBackendBaseUrl(url) {
+    backendBaseUrl = url;
+}
+export function getBackendBaseUrl() {
+    return backendBaseUrl;
+}

package/dist/gateway/clients.d.ts

@@ -0,0 +1,5 @@
+import type { GatewayClient } from "./types.js";
+/**
+ * Connected clients
+ */
+export declare const connectedClients: Set<GatewayClient>;

package/dist/gateway/clients.js

@@ -0,0 +1,4 @@
+/**
+ * Connected clients
+ */
+export const connectedClients = new Set();

package/dist/gateway/connection-handler.d.ts

@@ -0,0 +1,6 @@
+import type { WebSocket } from "ws";
+import type { IncomingMessage } from "http";
+/**
+ * Handle new WebSocket connection
+ */
+export declare function handleConnection(ws: WebSocket, req: IncomingMessage): void;

package/dist/gateway/connection-handler.js

@@ -0,0 +1,48 @@
+import { randomUUID } from "node:crypto";
+import { send, createEvent } from "./utils.js";
+import { handleMessage } from "./message-handler.js";
+import { connectedClients } from "./clients.js";
+/**
+ * Handle new WebSocket connection
+ */
+export function handleConnection(ws, req) {
+    const connId = randomUUID();
+    const remoteAddr = req.socket.remoteAddress;
+    console.log(`New connection: ${connId} from ${remoteAddr}`);
+    // Create client object
+    const client = {
+        id: connId,
+        ws,
+        authenticated: false,
+        connectedAt: Date.now(),
+    };
+    connectedClients.add(client);
+    // Send connection challenge
+    send(ws, createEvent("connect.challenge", {
+        nonce: randomUUID(),
+        ts: Date.now(),
+    }));
+    // Handle incoming messages
+    ws.on("message", async (data) => {
+        console.log(`Raw message received from ${connId}, type: ${typeof data}`);
+        try {
+            // Convert RawData to Buffer
+            const buffer = Buffer.isBuffer(data) ? data : Buffer.from(data.toString());
+            console.log(`Message buffer length: ${buffer.length}`);
+            await handleMessage(client, buffer);
+        }
+        catch (error) {
+            console.error(`Error handling message from ${connId}:`, error);
+        }
+    });
+    // Handle connection close
+    ws.on("close", (code, reason) => {
+        connectedClients.delete(client);
+        const duration = Date.now() - client.connectedAt;
+        console.log(`Connection closed: ${connId} (duration: ${duration}ms, code: ${code}, reason: ${reason.toString()})`);
+    });
+    // Handle errors
+    ws.on("error", (error) => {
+        console.error(`WebSocket error for ${connId}:`, error);
+    });
+}

package/dist/gateway/desktop-config.d.ts

@@ -0,0 +1,7 @@
+/**
+ * 读取桌面全局配置（与 Nest ConfigService 使用同一 config.json）。
+ * Gateway 进程内使用，用于获取 maxAgentSessions 等。
+ */
+export declare function getDesktopConfig(): {
+    maxAgentSessions: number;
+};