dsh-comfyui-canvas 0.1.4 → 0.1.5

package/README.md

@@ -3,7 +3,7 @@
 > [中文](README.zh.md) · English
 
 [![License: MIT](https://img.shields.io/badge/license-MIT-blue.svg)](LICENSE)
-[![Version](https://img.shields.io/badge/version-0.1.4-brightgreen.svg)](https://github.com/wbin0001/dsh-comfyui-canvas/releases)
+[![Version](https://img.shields.io/badge/version-0.1.5-brightgreen.svg)](https://github.com/wbin0001/dsh-comfyui-canvas/releases)
 [![GitHub Stars](https://img.shields.io/github/stars/wbin0001/dsh-comfyui-canvas.svg?style=social)](https://github.com/wbin0001/dsh-comfyui-canvas)
 [![DSH](https://img.shields.io/badge/DSH-v0.1.x-blueviolet.svg)](https://github.com/DeepSeek-Harness/DSH)
 [![ComfyUI](https://img.shields.io/badge/ComfyUI-0.34+-orange.svg)](https://github.com/comfyanonymous/ComfyUI)
@@ -12,9 +12,9 @@
 > **✅ DSH version compatibility (since v0.1.4)**: the split-screen layout is **fully self-contained** in the plugin — it uses only official DSH slots (`conversation.session.header.utilities`) and DOM `data-*` anchors, with **zero core modifications**. Works on any official DSH **v0.1.x** (including v0.1.2+ with the breaking client refresh) without patches. The earlier split-rail implementation depended on private core patches; v0.1.4 removes that dependency entirely.
 > - ❌ **Non-official desktop wrappers** (e.g. the community `dsh-desktop`) are not guaranteed compatible — they bundle an upstream version that may be ahead of or behind this plugin's baseline; rely on official DSH.
 
-![dsh-comfyui-canvas demo — agent drives a live ComfyUI workflow and fetches the output grid back into the chat](docs/screenshots/03-workflow-output.png)
+![dsh-comfyui-canvas demo — agent drives a live ComfyUI workflow and fetches the output grid back into the chat](docs/screenshots/03-output-grid.png)
 
-Embed your **ComfyUI** (local or cloud) as a split-screen canvas tab inside [DeepSeek Harness](https://github.com/DeepSeek-Harness/DSH) Web, and merge DSH's LLM power with ComfyUI's generation into **one visual creation platform** — the agent sparks ideas, writes prompts and scripts right in the chat, applies them live to the canvas in front of you, and produces images, music, video, and 3D. From idea to finished output without ever leaving the conversation or switching front-ends:
+**From chat to canvas to artwork — drive ComfyUI as a visual workflow IDE inside DSH.** Embed **ComfyUI** (local or cloud) as a split-screen canvas in [DeepSeek Harness](https://github.com/DeepSeek-Harness/DSH) Web: the agent sparks ideas, writes prompts and scripts right in the chat, applies them live to the canvas in front of you, and produces images, music, video, and 3D. From idea to finished output without ever leaving the conversation or switching front-ends:
 
 - **Canvas ops** — compose and arrange pipelines, read/write workflows, edit nodes, wire links, run, tune parameters, and debug errors, all live and WYSIWYG on the exact canvas you are looking at
 - **Production tasks** — batch parameter sweeps (`batch_run`) and automatic output-image retrieval back into the chat (`get_outputs`), powering multi-modal creative and batch generation across images, music, video, and 3D
@@ -37,7 +37,7 @@ This package is the DSH-side plugin, and it ships the ComfyUI-side bridge node t
 | **Upkeep tool** | `comfyui_upgrade` one-click updates the ComfyUI core and every git-backed custom node (concurrent, dirty-safe); `comfyui_config` reports the active connection, canvas focus, project directory, and a bridge-auth handshake check (`bridgeAuthEffective`). |
 | **Node dev tools** | `comfyui_read_source` / `comfyui_edit_source` / `comfyui_reload` — read and edit custom-node source under custom_nodes/ and restart ComfyUI from the conversation, then verify on the canvas. |
 | **Canvas focus mode** | The agent can tell (via `comfyui_config`) whether canvas split mode is active for the current session, and focus on canvas work only then. Session-isolated. |
-| **Settings page** | ComfyUI base URL / port / network mode / bridge token / launch command / project directory / rail width. Changes apply live. Customized nav icon with ComfyUI logo. |
+| **Settings page** | ComfyUI base URL / port / network mode / bridge token / launch command / project directory / rail width. Changes apply live. Shows the plugin version and supports check / one-click update (installs the latest npm release; a DSH restart is required). Customized nav icon with ComfyUI logo. |
 | **Split layout** | The **ComfyUI** button in the session header opens canvas-on-the-left + official chat rail on the right (state is session-isolated). Only data-* anchors and CSS variables are used — no DSH core class names, so the layout survives upstream styling changes. |
 
 ## Install

package/README.zh.md

@@ -3,7 +3,7 @@
 > 中文 · [English](README.md)
 
 [![License: MIT](https://img.shields.io/badge/license-MIT-blue.svg)](LICENSE)
-[![Version](https://img.shields.io/badge/version-0.1.4-brightgreen.svg)](https://github.com/wbin0001/dsh-comfyui-canvas/releases)
+[![Version](https://img.shields.io/badge/version-0.1.5-brightgreen.svg)](https://github.com/wbin0001/dsh-comfyui-canvas/releases)
 [![GitHub Stars](https://img.shields.io/github/stars/wbin0001/dsh-comfyui-canvas.svg?style=social)](https://github.com/wbin0001/dsh-comfyui-canvas)
 [![DSH](https://img.shields.io/badge/DSH-v0.1.x-blueviolet.svg)](https://github.com/DeepSeek-Harness/DSH)
 [![ComfyUI](https://img.shields.io/badge/ComfyUI-0.34+-orange.svg)](https://github.com/comfyanonymous/ComfyUI)
@@ -12,9 +12,9 @@
 > **✅ DSH 版本兼容（v0.1.4 起）**：分屏布局已完全**自包含**在插件内——只用官方 DSH 插槽（`conversation.session.header.utilities`）与 DOM `data-*` 锚点，**零核心改动**。任何官方 DSH **v0.1.x**（含带破坏性 client 更新的 v0.1.2+）都开箱即用、无需补丁。此前分屏 rail 依赖 DSH 私有核心补丁，v0.1.4 已彻底移除该依赖。
 > - ❌ **非官方桌面封装**（如 `dsh-desktop` 社区版）不保证兼容——它内部跑的上游版本可能超前/滞后于本插件基线，请以官方 DSH 为准。
 
-![dsh-comfyui-canvas 演示 —— agent 驱动实时 ComfyUI 工作流，把出图网格直接带回对话](docs/screenshots/03-workflow-output.png)
+![dsh-comfyui-canvas 演示 —— agent 驱动实时 ComfyUI 工作流，把出图网格直接带回对话](docs/screenshots/03-output-grid.png)
 
-把 **ComfyUI**（本地或云端）以画布分屏嵌入 [DeepSeek Harness](https://github.com/DeepSeek-Harness/DSH) Web，把 DSH 的 LLM 能力与 ComfyUI 的生成能力合成**一个可视化创作平台**——agent 在对话里激发创意、书写提示词与脚本，实时落到你眼前的画布上，产出图像、音乐、视频、3D。从灵感到成品，全程不离开对话，不用切换任何前端工具：
+**从对话到画布再到作品——DSH 里驾驭 ComfyUI 的可视化工作流 IDE。** 把 **ComfyUI**（本地或云端）以画布分屏嵌入 [DeepSeek Harness](https://github.com/DeepSeek-Harness/DSH) Web，agent 在对话里激发创意、书写提示词与脚本，实时落到你眼前的画布上，产出图像、音乐、视频、3D。从灵感到成品，全程不离开对话，不用切换任何前端工具：
 
 - **画布操作**：搭建编排、读写工作流、修改节点、连线、运行、调整参数、工作流查错——所见即所得，实时落在你眼前的画布上
 - **生产任务**：批量扫参（`batch_run`）、自动取回出图（`get_outputs`）带回对话，实现图像、音乐、视频、3D 等多任务智能创作与批量生产
@@ -39,7 +39,7 @@
 | **节点开发工具** | `comfyui_read_source` / `comfyui_edit_source` / `comfyui_reload`——对话里直接读写 `custom_nodes/` 下的节点源码并重启 ComfyUI，再到画布上验证。 |
 | **画布专注模式（会话隔离）** | agent 通过 `comfyui_config` 感知当前会话是否开启画布分屏，只在画布场景专注画布操作，且**按会话隔离**——多个会话互不干扰。 |
 | **ComfyUI 报错处理** | `debug` 校验工作流并高亮报错节点（纯校验，不触发执行），agent 帮你定位/修复画布错误。 |
-| **设置页** | ComfyUI 地址 / 端口 / 网络模式 / 桥接 Token / 启动命令 / 项目目录 / 右侧面板宽度，实时生效。导航栏已自定义为 ComfyUI logo 图标。 |
+| **设置页** | ComfyUI 地址 / 端口 / 网络模式 / 桥接 Token / 启动命令 / 项目目录 / 右侧面板宽度，实时生效；显示插件版本号并支持检查 / 一键更新（更新到 npm 最新版后需重启 DSH）。导航栏已自定义为 ComfyUI logo 图标。 |
 | **对话栏** | 分屏 rail 复用官方完整对话（消息、输入、发送、贴图、授权弹窗全在），不再需要插件自绘迷你输入框与授权 overlay。 |
 
 ---

package/lib/client.js

@@ -74,6 +74,32 @@ function buildCanvas() {
   var actions = document.createElement("div");
   actions.style.cssText = "margin-left:auto;display:flex;align-items:center;gap:12px;";
 
+  // 重启 / 关闭 ComfyUI 控制按钮（放在状态圆点前面）：点击派发自定义事件，
+  // apply 里的监听器把它写进 settings（comfyControl），host watcher 执行
+  // kill / restart；结果由画布的可达性探测自动反映到状态圆点。
+  var controlButton = function (title, inner, action) {
+    var btn = document.createElement("button");
+    btn.type = "button";
+    btn.title = title;
+    btn.setAttribute("aria-label", title);
+    btn.innerHTML = inner;
+    btn.style.cssText = "background:transparent;border:none;padding:0;color:var(--dsw-alias-state-business-primary,#8ab4f8);cursor:pointer;display:inline-flex;align-items:center;";
+    btn.addEventListener("click", function () {
+      document.dispatchEvent(new CustomEvent("dsh-comfyui-control", { bubbles: true, detail: { action: action } }));
+    });
+    return btn;
+  };
+  var restartBtn = controlButton(
+    "重启 ComfyUI",
+    '<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" aria-hidden="true"><path d="M21 2v6h-6"/><path d="M21 13a9 9 0 1 1-3-7.7L21 8"/></svg>',
+    "restart"
+  );
+  var stopBtn = controlButton(
+    "关闭 ComfyUI",
+    '<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" aria-hidden="true"><path d="M12 2v10"/><path d="M18.4 6.6a9 9 0 1 1-12.77.04"/></svg>',
+    "stop"
+  );
+
   var status = document.createElement("span");
   status.id = "dsh-comfy-status";
   status.title = "检测中…";
@@ -127,6 +153,8 @@ function buildCanvas() {
     document.dispatchEvent(new CustomEvent("dsh-comfyui-toggle-rail", { bubbles: true }));
   });
 
+  actions.appendChild(restartBtn);
+  actions.appendChild(stopBtn);
   actions.appendChild(status);
   actions.appendChild(refresh);
   actions.appendChild(link);
@@ -286,6 +314,8 @@ function ComfyUIControl(props) {
     background: "var(--dsw-alias-bg-layer-2, #1f1f23)", color: "var(--dsw-alias-label-primary, #e8e8e8)",
     border: "1px solid var(--dsw-alias-border-l1, rgba(128,128,128,.25))",
     borderRadius: "12px", boxShadow: "0 8px 32px rgba(0,0,0,.4)", fontFamily: "inherit",
+    // 内容纵向排列、全部居中：标题/说明/错误/按钮都在卡片中间，按钮与画布左右距离相等。
+    display: "flex", flexDirection: "column", alignItems: "center", textAlign: "center",
   };
   var btnStyle = {
     border: "none", borderRadius: "8px", padding: "10px 18px", fontSize: "13px",
@@ -776,6 +806,22 @@ function ComfyUISettingsRow(props) {
     scope.set(field, next).catch(function () { /* recover handled by scope */ });
   }
 
+  // 插件更新请求：settings 页「检查更新 / 更新」按钮把 updateRequest 写进
+  // settings，host watcher 执行 registry 检查或安装，并把结果写回
+  // pluginUpdate（status / latestVersion / message / checkedAt / updatedAt）。
+  function requestUpdate(action) {
+    if (!writable || !scope.set) return;
+    scope.set("updateRequest", { requested: true, action: action }).catch(function () { /* recover handled by scope */ });
+  }
+  var upStatus = value.pluginUpdate || {};
+  var btnRowStyle = {
+    border: "1px solid var(--dsw-alias-border-l2,#3a3a40)",
+    borderRadius: "6px", padding: "6px 12px",
+    fontSize: "12px", cursor: "pointer",
+    background: "transparent",
+    color: "var(--dsw-alias-label-primary,#e8e8e8)",
+  };
+
   // 桥接鉴权握手（非启动路径）：设置页打开 / token 变化时向 host 请求一次
   // 探测（probeRequested=true → host 后台探测 → 结果写回 bridgeProbe）。
   // host 代理是因为浏览器跨域读不到 ComfyUI 的状态码（CORS），而 host 端
@@ -863,6 +909,46 @@ function ComfyUISettingsRow(props) {
     probeText ? React.createElement("div", {
       style: { fontSize: "12px", color: probeColor, marginTop: "10px", lineHeight: "1.5" },
     }, probeText) : null,
+    React.createElement("div", {
+      style: {
+        marginTop: "14px", paddingTop: "12px",
+        borderTop: "1px solid var(--dsw-alias-border-l1,rgba(128,128,128,.25))",
+        display: "flex", flexDirection: "column", gap: "8px", alignItems: "flex-start",
+      },
+    },
+      React.createElement("div", { style: { fontSize: "13px", fontWeight: "600", color: "var(--dsw-alias-label-primary,#e8e8e8)" } }, "插件版本与更新"),
+      React.createElement("div", { style: { fontSize: "12px", color: "var(--dsw-alias-label-secondary,#9a9aa2)", lineHeight: 1.5 } },
+        "当前版本 v" + (value.pluginVersion || "?") +
+        (upStatus.latestVersion ? " · 最新 v" + upStatus.latestVersion : "")
+      ),
+      upStatus.status && upStatus.status !== "idle" ? React.createElement("div", {
+        style: {
+          fontSize: "12px", lineHeight: 1.5, whiteSpace: "pre-wrap",
+          color: upStatus.status === "error"
+            ? "var(--dsw-alias-state-error-primary,#f85149)"
+            : upStatus.status === "available"
+              ? "var(--dsw-alias-state-warn-primary,#d29922)"
+              : "var(--dsw-alias-label-secondary,#9a9aa2)",
+        },
+      }, upStatus.message || "") : null,
+      upStatus.status === "done" ? React.createElement("div", {
+        style: { fontSize: "12px", color: "var(--dsw-alias-state-warn-primary,#d29922)", lineHeight: 1.5 },
+      }, "⚠️ 重启 DSH 后更新才会生效") : null,
+      React.createElement("button", {
+        type: "button",
+        disabled: !writable || upStatus.status === "checking" || upStatus.status === "updating",
+        onClick: function () { requestUpdate("check"); },
+        style: btnRowStyle,
+      }, upStatus.status === "checking" ? "检测中…" : "检查更新"),
+      upStatus.status === "available" ? React.createElement("button", {
+        type: "button",
+        disabled: !writable || upStatus.status === "updating",
+        onClick: function () { requestUpdate("update"); },
+        style: Object.assign({}, btnRowStyle, {
+          background: "var(--dsw-alias-state-business-primary,#3b82f6)", color: "#fff",
+        }),
+      }, upStatus.status === "updating" ? "更新中…" : "更新到 v" + (upStatus.latestVersion || "")) : null,
+    ),
   );
 }
 
@@ -870,6 +956,14 @@ var inject = ["slots", "settingsScope"];
 
 function apply(ctx) {
   var scope = ctx.settingsScope.bind({ namespace: NAMESPACE });
+  // 画布顶栏「重启 / 关闭 ComfyUI」按钮 → 写 settings.comfyControl，
+  // host watcher 执行 kill / restart（结果由可达性探测反映到状态圆点）。
+  document.addEventListener("dsh-comfyui-control", function (e) {
+    var action = e && e.detail && e.detail.action === "restart" ? "restart" : "stop";
+    if (scope && typeof scope.set === "function") {
+      scope.set("comfyControl", { requested: true, action: action }).catch(function () { /* recover handled by scope */ });
+    }
+  });
   // 让画布 iframe 使用设置里的地址（并跟随变更）。
   var syncConfig = function () {
     var snap = scope.getSnapshot();

package/lib/index.js

@@ -15,6 +15,8 @@
 import { defineTool } from '@deepseek-ai/dsh-tools'
 import z from '@deepseek-ai/schemastery'
 import path from 'node:path'
+import { readFileSync } from 'node:fs'
+import { fileURLToPath } from 'node:url'
 import { mediaTypeOf, extractExecutionError, nextStemNumber, upsertRun } from './utils.js'
 
 export const name = 'dsh-comfyui-canvas'
@@ -93,6 +95,46 @@ const ConfigSchema = z.object({
     bridgeAuthProbe: z.string(),
     checkedAt: z.number(),
   }).default({ bridgeAuthEffective: null, bridgeAuthProbe: 'not-probed', checkedAt: 0 }),
+  /**
+   * This plugin's own version, published to settings at startup so the
+   * settings page can show it next to the update button.
+   */
+  pluginVersion: z.string().default(''),
+  /**
+   * One-shot plugin-update request from the settings page: the client sets
+   * `{ requested: true, action: 'check' | 'update' }`, the host watcher runs
+   * the npm-registry check or the package install and writes the outcome into
+   * `pluginUpdate`, then clears `requested`. Installs require a DSH restart to
+   * take effect, which the client surfaces after a successful update.
+   */
+  updateRequest: z.object({
+    requested: z.boolean(),
+    action: z.union([z.const('check'), z.const('update')]),
+  }).default({ requested: false, action: 'check' }),
+  /** Plugin-update state, written by the host watcher for the settings page. */
+  pluginUpdate: z.object({
+    status: z.string(), // idle | checking | up-to-date | available | updating | done | error
+    latestVersion: z.string(),
+    message: z.string(),
+    checkedAt: z.number(),
+    updatedAt: z.number(),
+  }).default({ status: 'idle', latestVersion: '', message: '', checkedAt: 0, updatedAt: 0 }),
+  /**
+   * One-shot ComfyUI control request from the canvas header: the client sets
+   * `{ requested: true, action: 'stop' | 'restart' }`, the host watcher kills
+   * the recorded launch process (falling back to whatever listens on the
+   * configured port) and, for 'restart', re-enters the launch path.
+   */
+  comfyControl: z.object({
+    requested: z.boolean(),
+    action: z.union([z.const('stop'), z.const('restart')]),
+  }).default({ requested: false, action: 'stop' }),
+  /** Last control outcome, written by the host watcher. */
+  comfyControlResult: z.object({
+    status: z.string(), // idle | stopping | restarting | done | error
+    message: z.string(),
+    at: z.number(),
+  }).default({ status: 'idle', message: '', at: 0 }),
 })
 
 function renderJSON(_args, value) {
@@ -183,6 +225,9 @@ export function apply(ctx) {
   // Function-level handle to the settings scope (used by comfyui_config's
   // activeView read); undefined when the settings service is unavailable.
   let settingsScope = undefined
+  // The background process handle of the ComfyUI launched by this plugin, so
+  // the canvas header's stop/restart buttons can kill it directly.
+  let comfyProc = null
 
   // v0.1.3: in-memory run ledger for `.runs.json` traceability — keyed by
   // prompt_id, recording the overrides each run was started with so downloads
@@ -261,6 +306,7 @@ export function apply(ctx) {
                 ...(workdir ? { workdir } : {}),
                 sandboxPolicy: { mode: 'danger-full-access', workspaceRoot: process.cwd() },
               }))
+              comfyProc = proc
               console.log(`[dsh-comfyui-canvas] launching ComfyUI: ${cmd}`)
               // A healthy launch keeps python alive well past this grace
               // window; an immediate exit (bad command / sandbox denial /
@@ -309,6 +355,174 @@ export function apply(ctx) {
           }
         })()
       })
+
+      // Publish this plugin's version once so the settings page can show it
+      // next to the update button (best effort — never blocks startup).
+      try {
+        const pkgPath = fileURLToPath(new URL('../package.json', import.meta.url))
+        const version = String(JSON.parse(readFileSync(pkgPath, 'utf8')).version || '')
+        if (version) scope.update({ pluginVersion: version })
+      } catch { /* best effort */ }
+
+      // Plugin-update watcher: the settings page sets updateRequest with
+      // action 'check' (query npm registry for the latest version) or
+      // 'update' (install the latest package into the DSH web profile).
+      // Installs run unconfined (the profile lives outside the session
+      // workspace) and require a DSH restart to take effect.
+      let updatingPlugin = false
+      scope.watch(() => {
+        const value = scope.get()
+        const req = value?.updateRequest
+        if (!req?.requested || updatingPlugin) return
+        updatingPlugin = true
+        void (async () => {
+          const setState = (patch) => {
+            try { scope.update(patch) } catch { /* best effort */ }
+          }
+          try {
+            const action = req.action === 'update' ? 'update' : 'check'
+            const current = String(value?.pluginVersion || '')
+            setState({
+              updateRequest: { requested: false, action },
+              pluginUpdate: {
+                status: action === 'update' ? 'updating' : 'checking',
+                latestVersion: '',
+                message: action === 'update' ? '正在安装最新版本…' : '正在查询最新版本…',
+                checkedAt: Date.now(),
+                updatedAt: 0,
+              },
+            })
+            const res = await fetch('https://registry.npmjs.org/dsh-comfyui-canvas/latest', {
+              signal: AbortSignal.timeout(10000),
+            })
+            if (!res.ok) throw new Error(`npm registry HTTP ${res.status}`)
+            const meta = await res.json().catch(() => null)
+            const latest = String(meta?.version || '')
+            if (!latest) throw new Error('npm registry returned no version')
+            if (action === 'check') {
+              const upToDate = latest === current
+              setState({
+                pluginUpdate: {
+                  status: upToDate ? 'up-to-date' : 'available',
+                  latestVersion: latest,
+                  message: upToDate
+                    ? `已是最新版本 v${latest}`
+                    : `发现新版本 v${latest}（当前 v${current || '?'}）`,
+                  checkedAt: Date.now(),
+                  updatedAt: 0,
+                },
+              })
+              return
+            }
+            // Update: install the latest package into the DSH web profile.
+            const shell = ctx.get?.('shell')
+            if (!shell || typeof shell.start !== 'function') throw new Error('no shell service available to update the plugin')
+            const profileDir = process.env.DSH_HOME
+              ? path.join(process.env.DSH_HOME, 'profiles', 'web')
+              : path.dirname(path.dirname(path.dirname(fileURLToPath(import.meta.url))))
+            const proc = shell.start(shell.resolve({
+              command: 'pnpm add dsh-comfyui-canvas@latest',
+              workdir: profileDir,
+              sandboxPolicy: { mode: 'danger-full-access', workspaceRoot: process.cwd() },
+            }))
+            const outcome = await proc.done
+            if (!outcome || (outcome.exitCode ?? outcome.code) !== 0) {
+              throw new Error('install failed — see DSH logs')
+            }
+            setState({
+              pluginUpdate: {
+                status: 'done',
+                latestVersion: latest,
+                message: `已更新到 v${latest}，重启 DSH 后生效`,
+                checkedAt: Date.now(),
+                updatedAt: Date.now(),
+              },
+            })
+          } catch (e) {
+            setState({
+              pluginUpdate: {
+                status: 'error',
+                latestVersion: '',
+                message: `更新失败：${e instanceof Error ? e.message : String(e)}`,
+                checkedAt: Date.now(),
+                updatedAt: 0,
+              },
+            })
+          } finally {
+            updatingPlugin = false
+          }
+        })()
+      })
+
+      // ComfyUI control watcher: the canvas header's stop/restart buttons set
+      // comfyControl with action 'stop' (kill the recorded launch process, or
+      // whatever listens on the configured port as a fallback) or 'restart'
+      // (kill, then re-enter the launch path). The canvas status dot reflects
+      // the outcome through the existing reachability probe.
+      let controlling = false
+      scope.watch(() => {
+        const value = scope.get()
+        const ctrl = value?.comfyControl
+        if (!ctrl?.requested || controlling) return
+        controlling = true
+        void (async () => {
+          const setState = (patch) => {
+            try { scope.update(patch) } catch { /* best effort */ }
+          }
+          try {
+            const action = ctrl.action === 'restart' ? 'restart' : 'stop'
+            setState({
+              comfyControl: { requested: false, action },
+              comfyControlResult: {
+                status: action === 'restart' ? 'restarting' : 'stopping',
+                message: action === 'restart' ? '正在重启 ComfyUI…' : '正在关闭 ComfyUI…',
+                at: Date.now(),
+              },
+            })
+            const shell = ctx.get?.('shell')
+            let killed = false
+            if (comfyProc && typeof comfyProc.kill === 'function') {
+              killed = comfyProc.kill()
+              try { await comfyProc.done } catch { /* already settled */ }
+              comfyProc = null
+            }
+            if (!killed && shell && typeof shell.start === 'function') {
+              // Fallback: kill whatever listens on the configured port. Uses
+              // netstat + PID parsing because Get-NetTCPConnection can come
+              // back empty in some environments even when the port is
+              // listening (observed on this host). The command is plain
+              // PowerShell: the pwsh-local executor passes it as ONE argv
+              // element to `pwsh -Command`, so there is no intermediate shell
+              // to double-quote for — wrapping it in another
+              // `powershell -NoProfile -Command "..."` would let the outer
+              // pwsh interpolate $_ / $pids and break the script.
+              let port = '8188'
+              try { port = new URL(resolveBase()).port || '8188' } catch { /* keep default */ }
+              const killer = shell.start(shell.resolve({
+                command: `$pids = netstat -ano | Select-String ':${port}' | Select-String 'LISTENING' | ForEach-Object { ($_ -split '\\s+')[-1] } | Sort-Object -Unique; foreach ($procId in $pids) { Stop-Process -Id $procId -Force -ErrorAction SilentlyContinue }; Write-Output ('killed=' + ($pids | Measure-Object).Count)`,
+                sandboxPolicy: { mode: 'danger-full-access', workspaceRoot: process.cwd() },
+              }))
+              try { await killer.done } catch { /* best effort */ }
+              const out = (typeof killer.readOutput === 'function' ? killer.readOutput().delta : '') || ''
+              const matched = /killed=(\d+)/.exec(out)
+              killed = matched ? Number(matched[1]) > 0 : false
+            }
+            console.log(`[dsh-comfyui-canvas] comfyControl ${action}: killed=${killed}`)
+            if (action === 'restart') {
+              // Give the port a moment to release, then re-enter the launch path.
+              await new Promise((done) => setTimeout(done, 2500))
+              setState({ launchRequested: true })
+              setState({ comfyControlResult: { status: 'done', message: 'ComfyUI 已停止，正在重新启动…', at: Date.now() } })
+            } else {
+              setState({ comfyControlResult: { status: 'done', message: killed ? 'ComfyUI 已关闭' : '未找到运行中的 ComfyUI', at: Date.now() } })
+            }
+          } catch (e) {
+            setState({ comfyControlResult: { status: 'error', message: `操作失败：${e instanceof Error ? e.message : String(e)}`, at: Date.now() } })
+          } finally {
+            controlling = false
+          }
+        })()
+      })
     } catch {
       // Duplicate registration or unavailable provider — keep defaults.
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-comfyui-canvas",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "Give DeepSeek Harness agents live control of your ComfyUI canvas (local or cloud): embed it as a split-screen tab, then read/edit/run workflows, debug errors, fetch output images, and batch-sweep parameters straight from the chat. Ships the ComfyUI-side bridge node.",
   "type": "module",
   "main": "lib/index.js",

package/screenshots.json

@@ -1,5 +1,8 @@
 [
-  "docs/screenshots/01-startup-prompt.png",
-  "docs/screenshots/02-canvas-ready.png",
-  "docs/screenshots/03-workflow-output.png"
+  "docs/screenshots/01-split-canvas-chat.png",
+  "docs/screenshots/02-workflow-dunhuang.png",
+  "docs/screenshots/03-output-grid.png",
+  "docs/screenshots/04-gallery.png",
+  "docs/screenshots/05-batch-portraits.png",
+  "docs/screenshots/06-template-library.png"
 ]