pi-web-ui 0.20.1 → 0.21.0

package/dist/server/agent-service.js

@@ -12,7 +12,7 @@
  */
 import { spawn } from "node:child_process";
 import { existsSync, readFileSync, statSync, writeFileSync, mkdirSync, watch, } from "node:fs";
-import { dirname, join, relative, resolve, sep } from "node:path";
+import { basename, dirname, join, relative, resolve, sep } from "node:path";
 import { fileURLToPath } from "node:url";
 import { createAgentSessionFromServices, createAgentSessionRuntime, createAgentSessionServices, createBashTool, createLocalBashOperations, defineTool, getAgentDir, ModelRuntime, SessionManager, } from "@earendil-works/pi-coding-agent";
 import { Type } from "typebox";
@@ -750,6 +750,15 @@ async function readDirForUI(abs, rel) {
         out.length = MAX;
     return { entries: out, truncated };
 }
+/** Stable identity of an extension for the enable/disable toggle: the npm
+ *  spec for packages (survives version bumps), the resolved entry path
+ *  otherwise. */
+function extensionKey(e) {
+    const src = e.sourceInfo;
+    if (src?.origin === "package" && src.source)
+        return src.source;
+    return src?.path ?? e.path;
+}
 /**
  * Persists which workspace each browser client last used + which workspaces it
  * has opened, so a server restart / page reload restores the same project and
@@ -820,6 +829,40 @@ class ClientStateStore {
         };
         this.save();
     }
+    /** Last-used settings-panel state for a client, or defaults. */
+    getSettings(clientId) {
+        const s = this.load()[clientId];
+        return {
+            promptMode: s?.settings?.promptMode === "replace" ? "replace" : "append",
+            customSystemPrompt: s?.settings?.customSystemPrompt ?? "",
+            disabledSkills: s?.settings?.disabledSkills ?? [],
+            disabledExtensions: s?.settings?.disabledExtensions ?? [],
+        };
+    }
+    /** Persist the client's settings-panel state (partial merge). */
+    saveSettings(clientId, settings) {
+        const all = this.load();
+        const state = (all[clientId] ??= { projects: [] });
+        const cur = state.settings ?? {};
+        state.settings = {
+            promptMode: settings.promptMode ?? cur.promptMode ?? "append",
+            customSystemPrompt: settings.customSystemPrompt ?? cur.customSystemPrompt ?? "",
+            disabledSkills: settings.disabledSkills ?? cur.disabledSkills ?? [],
+            disabledExtensions: settings.disabledExtensions ?? cur.disabledExtensions ?? [],
+        };
+        this.save();
+    }
+    /** Named settings presets for a client (empty if never saved). */
+    getPresets(clientId) {
+        return this.load()[clientId]?.presets ?? [];
+    }
+    /** Persist the client's named settings presets. */
+    savePresets(clientId, presets) {
+        const all = this.load();
+        const state = (all[clientId] ??= { projects: [] });
+        state.presets = presets;
+        this.save();
+    }
 }
 /** Hard cap on how long ONE tool call may run before the watchdog aborts the
  *  session. The SDK bash tool has NO default timeout, so a command that never
@@ -914,6 +957,20 @@ export class ClientSession {
     /** Guard: the goal wizard and the review loop are mutually exclusive — a
      *  wizard in flight stops review triggers (and vice versa). */
     goalWizardRunning = false;
+    /** Settings-panel state (system prompt + disabled skills/extensions). The
+     *  resource-loader overrides in makeRuntimeFactory() read this at every
+     *  reload(), so session.reload() applies changes to the running runtime. */
+    settings;
+    /** Named settings presets (saved combos the user can re-apply). */
+    presets = [];
+    /** Settings changed while the run was streaming — the runtime reload is
+     *  deferred to the next agent_end so an in-flight run is never torn down. */
+    pendingSettingsReload = false;
+    /** Full (incl. disabled) skill/extension lists seen so far — disabled
+     *  entries disappear from the loader after reload, so this cache keeps them
+     *  visible (and re-enableable) in the settings panel. */
+    knownSkills = new Map();
+    knownExtensions = new Map();
     /** Aborts the currently-running goal wizard (user clicked ✗ / timed out). Drives
      *  the in-flight goal_ask dialog to resolve as cancelled and (via the run
      *  signal) stops the wizard session's agent run. Recreated per wizard. */
@@ -993,6 +1050,8 @@ export class ClientSession {
         this.cwd = cwd;
         this.agentDir = agentDir;
         this.stateStore = stateStore;
+        this.settings = stateStore.getSettings(clientId);
+        this.presets = stateStore.getPresets(clientId);
     }
     static async create(clientId, cwd, stateStore) {
         const agentDir = process.env.PI_CODING_AGENT_DIR ?? getAgentDir();
@@ -1040,16 +1099,41 @@ export class ClientSession {
             const services = await createAgentSessionServices({
                 cwd: effectiveCwd,
                 modelRuntime: this.sharedModelRuntime,
-                ...(process.platform === "win32"
-                    ? {
-                        // Windows 专属 persona：bash 工具跑 Git Bash 且无默认超时、终端是
-                        // 交互式 TTY——注入约束避免 heredoc/交互/长驻命令挂死整个会话；
-                        // GBK 老中文文件让模型改用终端按正确编码读（iconv/chcp/Get-Content）。
-                        resourceLoaderOptions: {
-                            systemPromptOverride: (base) => base ? `${base}\n\n${WINDOWS_PERSONA}` : WINDOWS_PERSONA,
-                        },
-                    }
-                    : {}),
+                // 设置面板钩子（官方 SDK 的 resourceLoader overrides）：三个 override
+                // 在每次 resourceLoader.reload() 时重放，且读取 this.settings 的当前
+                // 值——因此 session.reload() 即可让系统提示词 / 技能 / 插件开关生效，
+                // 新对话（新 runtime）也会自动带上当前设置。
+                resourceLoaderOptions: {
+                    // 系统提示词：replace 模式整体替换；append 模式追加到提示词末尾。
+                    systemPromptOverride: (base) => this.settings.promptMode === "replace" &&
+                        this.settings.customSystemPrompt.trim()
+                        ? this.settings.customSystemPrompt
+                        : base,
+                    appendSystemPromptOverride: (base) => {
+                        const out = [...base];
+                        const custom = this.settings.customSystemPrompt.trim();
+                        if (this.settings.promptMode === "append" && custom) {
+                            out.push(custom);
+                        }
+                        if (process.platform === "win32") {
+                            // Windows 专属 persona：bash 工具跑 Git Bash 且无默认超时、终端
+                            // 是交互式 TTY——注入约束避免 heredoc/交互/长驻命令挂死整个会话；
+                            // GBK 老中文文件让模型改用终端按正确编码读（iconv/chcp/Get-Content）。
+                            out.push(WINDOWS_PERSONA);
+                        }
+                        return out;
+                    },
+                    // 技能开关：禁用的技能从系统提示词和 /skill: 目录中剔除。
+                    skillsOverride: (res) => ({
+                        ...res,
+                        skills: res.skills.filter((s) => !this.settings.disabledSkills.includes(s.name)),
+                    }),
+                    // 插件开关：禁用的扩展整个卸载（工具 / 命令随之消失）。
+                    extensionsOverride: (res) => ({
+                        ...res,
+                        extensions: res.extensions.filter((e) => !this.settings.disabledExtensions.includes(extensionKey(e))),
+                    }),
+                },
             });
             return {
                 ...(await createAgentSessionFromServices({
@@ -1117,6 +1201,9 @@ export class ClientSession {
         // Reconnect: push the remembered goal prefs (model choice, rounds cap,
         // locked) so the goal bar restores them on reload — "全局记忆".
         this.emitGoalStatus();
+        // Reconnect: push the settings panel state (prompt text/mode, skill &
+        // extension toggles, saved presets).
+        this.pushSettings();
     }
     detachSink(send) {
         this.sinks.delete(send);
@@ -1327,6 +1414,13 @@ export class ClientSession {
                     !this.disposed) {
                     void this.runGoalReview(conv);
                 }
+                // Deferred settings reload: settings (system prompt / skills /
+                // extensions) changed while the run was streaming — applying now
+                // would have torn down the in-flight run.
+                if (this.pendingSettingsReload && !this.disposed) {
+                    this.pendingSettingsReload = false;
+                    void this.applySettingsReload();
+                }
                 break;
             }
             case "entry_appended":
@@ -2275,6 +2369,174 @@ export class ClientSession {
         this.emit({ type: "slash_commands", commands });
     }
     // ---------------------------------------------------------------------------
+    // Settings (system prompt / skills / extensions / presets)
+    // ---------------------------------------------------------------------------
+    /** Push the full settings state (current settings + loaded skills/extensions
+     *  with enabled flags + saved presets). Pushed on attach and after every
+     *  settings change. */
+    pushSettings() {
+        const disabledSkills = new Set(this.settings.disabledSkills);
+        const disabledExts = new Set(this.settings.disabledExtensions);
+        try {
+            // Refresh the cache with the CURRENTLY loaded set (post-filter).
+            for (const s of this.session.resourceLoader.getSkills().skills) {
+                this.knownSkills.set(s.name, {
+                    name: s.name,
+                    description: s.description,
+                    enabled: true,
+                });
+            }
+            for (const e of this.session.resourceLoader.getExtensions().extensions) {
+                const id = extensionKey(e);
+                const p = e.sourceInfo?.path ?? e.path;
+                this.knownExtensions.set(id, {
+                    id,
+                    name: e.sourceInfo?.origin === "package" && e.sourceInfo.source
+                        ? e.sourceInfo.source
+                        : basename(p),
+                    path: p,
+                    enabled: true,
+                });
+            }
+        }
+        catch {
+            // Session not ready yet — keep whatever we already know.
+        }
+        // Disabled entries are filtered out of the loader — keep them in the
+        // panel (with the last-known description) so they can be re-enabled.
+        for (const name of this.settings.disabledSkills) {
+            if (!this.knownSkills.has(name)) {
+                this.knownSkills.set(name, { name, description: "", enabled: false });
+            }
+        }
+        for (const id of this.settings.disabledExtensions) {
+            if (!this.knownExtensions.has(id)) {
+                this.knownExtensions.set(id, {
+                    id,
+                    name: id.startsWith("npm:") ? id : basename(id),
+                    path: "",
+                    enabled: false,
+                });
+            }
+        }
+        const skills = [...this.knownSkills.values()]
+            .map((s) => ({ ...s, enabled: !disabledSkills.has(s.name) }))
+            .sort((a, b) => a.name.localeCompare(b.name));
+        const extensions = [...this.knownExtensions.values()]
+            .map((e) => ({ ...e, enabled: !disabledExts.has(e.id) }))
+            .sort((a, b) => a.name.localeCompare(b.name));
+        this.emit({
+            type: "settings_state",
+            settings: {
+                promptMode: this.settings.promptMode,
+                customSystemPrompt: this.settings.customSystemPrompt,
+                disabledSkills: [...this.settings.disabledSkills],
+                disabledExtensions: [...this.settings.disabledExtensions],
+                skills,
+                extensions,
+                presets: this.presets.map((p) => ({ ...p })),
+            },
+        });
+    }
+    /** Persist + apply a partial settings update (prompt text/mode, toggles). */
+    async setSettings(partial) {
+        if (partial.promptMode !== undefined)
+            this.settings.promptMode = partial.promptMode;
+        if (partial.customSystemPrompt !== undefined) {
+            this.settings.customSystemPrompt = partial.customSystemPrompt;
+        }
+        if (partial.disabledSkills !== undefined) {
+            this.settings.disabledSkills = partial.disabledSkills;
+        }
+        if (partial.disabledExtensions !== undefined) {
+            this.settings.disabledExtensions = partial.disabledExtensions;
+        }
+        this.stateStore.saveSettings(this.clientId, this.settings);
+        this.pushSettings();
+        await this.applyRuntimeSettings();
+    }
+    /** Save the CURRENT settings as a named preset (overwrites if exists). */
+    async savePreset(name) {
+        const n = name.trim();
+        if (!n) {
+            this.emit({ type: "notice", level: "error", text: "预设名称不能为空" });
+            return;
+        }
+        const preset = {
+            name: n,
+            promptMode: this.settings.promptMode,
+            customSystemPrompt: this.settings.customSystemPrompt,
+            disabledSkills: [...this.settings.disabledSkills],
+            disabledExtensions: [...this.settings.disabledExtensions],
+        };
+        const existing = this.presets.findIndex((p) => p.name === n);
+        if (existing >= 0)
+            this.presets[existing] = preset;
+        else
+            this.presets.push(preset);
+        this.stateStore.savePresets(this.clientId, this.presets);
+        this.pushSettings();
+    }
+    /** Replace the current settings with the named preset and apply it. */
+    async applyPreset(name) {
+        const p = this.presets.find((x) => x.name === name);
+        if (!p) {
+            this.emit({ type: "notice", level: "error", text: `预设不存在：${name}` });
+            return;
+        }
+        this.settings = {
+            promptMode: p.promptMode,
+            customSystemPrompt: p.customSystemPrompt,
+            disabledSkills: [...p.disabledSkills],
+            disabledExtensions: [...p.disabledExtensions],
+        };
+        this.stateStore.saveSettings(this.clientId, this.settings);
+        this.pushSettings();
+        await this.applyRuntimeSettings();
+    }
+    /** Remove a named preset. */
+    async deletePreset(name) {
+        this.presets = this.presets.filter((p) => p.name !== name);
+        this.stateStore.savePresets(this.clientId, this.presets);
+        this.pushSettings();
+    }
+    /**
+     * Make settings changes effective in the running runtime. The resource-loader
+     * overrides read this.settings at call time, so a reload re-applies them.
+     * Reloading mid-stream would tear down the in-flight run — defer instead.
+     */
+    async applyRuntimeSettings() {
+        if (this.disposed)
+            return;
+        if (this.session.isStreaming) {
+            this.pendingSettingsReload = true;
+            this.emit({
+                type: "notice",
+                level: "info",
+                text: "当前回复进行中，设置将在回复结束后自动应用",
+            });
+            return;
+        }
+        await this.applySettingsReload();
+    }
+    /** session.reload() + refresh the slash-command catalog + push state. */
+    async applySettingsReload() {
+        try {
+            await this.session.reload();
+            await this.pushSlashCommands();
+            this.pushSettings();
+            this.flushSnapshot();
+            this.emit({ type: "notice", level: "info", text: "设置已应用" });
+        }
+        catch (err) {
+            this.emit({
+                type: "notice",
+                level: "error",
+                text: `设置应用失败：${err.message}`,
+            });
+        }
+    }
+    // ---------------------------------------------------------------------------
     // Commands
     // ---------------------------------------------------------------------------
     async prompt(text, attachments) {

package/dist/server/index.js

@@ -363,6 +363,26 @@ wss.on("connection", (ws) => {
                     locked: msg.locked,
                 });
                 break;
+            case "get_settings":
+                cs.pushSettings();
+                break;
+            case "set_settings":
+                void cs.setSettings({
+                    promptMode: msg.promptMode,
+                    customSystemPrompt: msg.customSystemPrompt,
+                    disabledSkills: msg.disabledSkills,
+                    disabledExtensions: msg.disabledExtensions,
+                });
+                break;
+            case "save_preset":
+                void cs.savePreset(msg.name);
+                break;
+            case "apply_preset":
+                void cs.applyPreset(msg.name);
+                break;
+            case "delete_preset":
+                void cs.deletePreset(msg.name);
+                break;
             default:
                 break;
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-web-ui",
-  "version": "0.20.1",
+  "version": "0.21.0",
   "description": "Web chat interface for the pi coding agent, powered by the pi SDK (@earendil-works/pi-coding-agent) — one-command run, Docker/systemd/launchd deployable",
   "license": "MIT",
   "type": "module",