pi-web-ui 0.34.3 → 0.35.1

package/dist/server/agent-service.js

@@ -2212,8 +2212,14 @@ export class ClientSession {
      * keeps everything up to (but not including) that question, then sends the
      * edited text there. The original thread is untouched and stays in the
      * session list, so nothing is ever lost.
+     *
+     * Attachments (attachments) travel through the SAME pipeline as prompt()
+     * — the fork intentionally drops the original attachment asides because
+     * they live on the old branch past the fork point, so the browser re-sends
+     * the images it kept in the edit composer (original image blocks + any
+     * newly pasted/dropped ones). Text-only edits pass undefined.
      */
-    async editMessage(messageId, text) {
+    async editMessage(messageId, text, attachments) {
         if (this.quiesceBlocked())
             return;
         const trimmed = text.trim();
@@ -2248,7 +2254,7 @@ export class ClientSession {
                 return;
             }
             await this.bindSession();
-            await this.prompt(trimmed);
+            await this.prompt(trimmed, attachments);
             this.emit({
                 type: "notice",
                 level: "info",

package/dist/server/index.js

@@ -521,7 +521,7 @@ wss.on("connection", (ws) => {
                 void cs.newChat();
                 break;
             case "edit_message":
-                void cs.editMessage(msg.messageId, msg.text);
+                void cs.editMessage(msg.messageId, msg.text, msg.attachments);
                 break;
             case "cycle_model":
                 void cs.cycleModel();
@@ -748,7 +748,12 @@ wss.on("connection", (ws) => {
                 // freshly dropped plugins show up without a server restart.
                 pluginMgr
                     .ensureLoaded()
-                    .then((plugins) => send({ type: "plugins", plugins, epoch: pluginMgr.epoch }))
+                    .then((plugins) => {
+                    send({ type: "plugins", plugins, epoch: pluginMgr.epoch });
+                    // 让各插件向新接入的客户端推送自身初始状态（onAttach 钩子）——
+                    // 插件不要依赖客户端挂载后自己拉（见 plugins.ts onAttach 注释）。
+                    pluginMgr.notifyAttach(cid);
+                })
                     .catch(() => { });
                 // Replay anything that arrived while the session was starting.
                 const queued = pending;

package/dist/server/plugins.js

@@ -101,12 +101,33 @@ export class PluginManager {
         this.deliverAll({ type: "plugins", plugins: list, epoch: this.epochCounter });
     }
     /** 服务端热重载：反激活全部 → 清缓存 → 重扫重激活 → epoch+1。
-     *  返回新目录清单（含激活结果）。 */
+     *  返回新目录清单（含激活结果）。重激活后的插件实例是新模块，
+         *  内存状态为初始值——逐个客户端触发 onAttach 让它们重推自身状态。 */
     async reload() {
         this.dispose();
         this.attempted.clear();
         this.epochCounter += 1;
-        return this.ensureLoaded();
+        const list = await this.ensureLoaded();
+        for (const s of this.senders) {
+            const cid = s.cid();
+            if (cid)
+                this.notifyAttach(cid);
+        }
+        return list;
+    }
+    /** 每个客户端 attach 后调用：让各插件向该客户端推送自身完整状态。
+     *  异常隔离——单个插件钩子报错不影响其他插件与其他钩子。 */
+    notifyAttach(clientId) {
+        for (const [id, p] of this.loaded) {
+            for (const h of p.attachHandlers) {
+                try {
+                    h(clientId);
+                }
+                catch (err) {
+                    console.error(`[plugin:${id}] onAttach handler failed:`, err);
+                }
+            }
+        }
     }
     /** agent-service 调：把 SDK 工具执行事件扇出给所有插件（异常隔离）。 */
     emitToolEvent(ev) {
@@ -265,6 +286,19 @@ export class PluginManager {
                     icon: typeof m.icon === "string" && m.icon.trim() ? m.icon.trim() : undefined,
                     hasClient: existsSync(join(dir, "client", "entry.mjs")),
                     error: this.loaded.get(name)?.info.error,
+                    // 安装来源（pi-web-ui install 写入的 .pi-source.json）——
+                    // 设置面板据此显示「更新」按钮；手工拷入的插件没有此文件。
+                    source: await readFile(join(dir, ".pi-source.json"), "utf8")
+                        .then((raw) => {
+                        try {
+                            const s = JSON.parse(raw);
+                            return typeof s.source === "string" && s.source ? s.source : undefined;
+                        }
+                        catch {
+                            return undefined;
+                        }
+                    })
+                        .catch(() => undefined),
                 });
             }
             catch {
@@ -279,7 +313,9 @@ export class PluginManager {
         const handlers = new Set();
         this.messageHandlers.set(info.id, handlers);
         const toolHandlers = new Set();
+        const attachHandlers = new Set();
         const unregisterTools = [];
+        const p = { info, toolHandlers, attachHandlers };
         const host = {
             broadcast: (payload) => this.broadcast(info.id, payload),
             notify: (level, text) => this.notifyAll(level, text),
@@ -292,6 +328,10 @@ export class PluginManager {
                 toolHandlers.add(h);
                 return () => toolHandlers.delete(h);
             },
+            onAttach: (h) => {
+                attachHandlers.add(h);
+                return () => attachHandlers.delete(h);
+            },
             // 包一层：插件反激活时自动注销它注册的全部 AI 工具，不留悬挂项。
             registerAgentTool: (tool) => {
                 const off = this.registerAgentTool(info.id, tool);
@@ -317,6 +357,7 @@ export class PluginManager {
                 info: { ...info },
                 deactivate: typeof ret === "function" ? ret : undefined,
                 toolHandlers,
+                attachHandlers,
                 agentToolUnsubscribers: unregisterTools,
             });
             console.log(`[plugin:${info.id}] activated (v${info.version ?? "?"})`);
@@ -325,6 +366,7 @@ export class PluginManager {
             this.loaded.set(info.id, {
                 info: { ...info, error: err.message },
                 toolHandlers,
+                attachHandlers,
             });
             console.error(`[plugin:${info.id}] activate failed:`, err);
         }

package/dist/server/protocol-version.js

@@ -8,4 +8,4 @@
  * its own copy in web/src/protocol-version.ts; scripts/check-protocol-sync.mjs
  * verifies the two never drift.
  */
-export const PROTOCOL_VERSION = 7;
+export const PROTOCOL_VERSION = 9;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-web-ui",
-  "version": "0.34.3",
+  "version": "0.35.1",
   "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",