dsh-shutdown 0.1.0 → 0.2.0

package/README.md

@@ -2,9 +2,9 @@
 
 <!-- Hero -->
 <div align="center">
-  <b style="font-size: 1.15em;">一键结束 DSH 服务</b><br /><br />
-  Settings「关机」页 · 二次确认 · 优雅退出<br /><br />
-  <b>关掉浏览器不再留下后台进程</b>。
+  <b style="font-size: 1.15em;">一键关闭 / 重启 DSH 服务</b><br /><br />
+  Settings「服务开关」页 · 二次确认 · 优雅退出 / 自动重启<br /><br />
+  <b>关掉浏览器不再留下后台进程，也能一键重启服务</b>。
 </div>
 
 <div align="center">
@@ -13,13 +13,14 @@
 
 ## ✨ 为什么需要
 
-关闭浏览器**不会**结束 dsh 的后台服务进程——服务继续占用端口与内存。本插件在页面内提供一键关机入口，无需命令行。
+关闭浏览器**不会**结束 dsh 的后台服务进程——服务继续占用端口与内存。本插件在页面内提供一键关机/重启入口，无需命令行。
 
 ## ✨ 功能一览
 
-- **🖥️ Settings「关机」页**：侧栏最底部新增「关机」页签
-- **🛡️ 二次确认**：点「关闭 DSH 服务」→ 出现「确认关闭 / 取消」双按钮，防误触
-- **✅ 成功反馈**：确认后页面立即显示绿色「关闭成功」面板（并尝试自动关闭标签页）
+- **🖥️ Settings「服务开关」页**：侧栏最底部新增「服务开关」页签
+- **🛡️ 二次确认**：点「关闭 DSH 服务」或「重启 DSH 服务」→ 出现「确认 / 取消」双按钮，防误触
+- **✅ 成功反馈**：确认后页面立即显示绿色「关闭成功 / 重启中」面板（并尝试自动关闭标签页）
+- **🔁 自动重启**：重启请求会让宿主进程以特殊退出码退出，由 `launcher/dsh-web.ps1` 启动脚本捕获后自动重新拉起（根目录 `dsh-web.cmd`/`dsh-web.ps1` 为快捷入口）
 - **🎨 原生风格**：按钮与配色全部引用 DSH 主题 token（`--dsw-alias-*`），明暗自适应
 
 ## 📦 安装
@@ -32,8 +33,9 @@ dsh plugin --profile web add <本仓库 / npm 包名>
 ## 🧠 设计要点
 
 - 关机端点：`/_dsh/shutdown`（仅 POST）——服务先返回「正在关闭」响应，500ms 后 `process.exit(0)` 优雅退出
-- 会话日志为逐事件落盘，空闲时退出安全；正在进行的对话请等待其完成后再关闭
-- 浏览器安全策略下脚本只能关闭自己打开的窗口——手动打开的标签页会保留「关闭成功」面板，两种情况下你都能确认结果
+- 重启端点：`/_dsh/restart`（仅 POST）——服务先返回「正在重启」响应，500ms 后以退出码 42 退出；`dsh-web.ps1` 检测到该退出码会自动重新启动 `dsh web`
+- 会话日志为逐事件落盘，空闲时退出安全；正在进行的对话请等待其完成后再操作
+- 浏览器安全策略下脚本只能关闭自己打开的窗口——手动打开的标签页会保留成功面板，两种情况下你都能确认结果
 
 ## 🧩 生态
 
@@ -42,3 +44,20 @@ dsh plugin --profile web add <本仓库 / npm 包名>
 ## 📄 许可
 
 [MIT](./LICENSE)
+
+## 🔒 安全说明
+
+关机端点会终止整个服务，是破坏性最强的操作，因此端点自身带有跨站请求（CSRF）防护：
+
+- **浏览器发起的请求**：校验 `Origin` 头。来自外部站点的一律返回 `403`，防止任意网页在后台关闭你本机的 DSH——浏览器不会为这类跨源简单 POST 做预检，所以端点必须自己把关
+- **同源访问**：`127.0.0.1` / `localhost` / `::1` 直接放行；通过局域网 IP 或自定义主机名访问（`dsh web --host 0.0.0.0`）时，要求 `Origin` 与 `Host` 一致
+- **无 `Origin` 的请求**：`curl`、启动脚本等非浏览器客户端放行，不影响脚本化使用
+
+## 📝 更新日志
+
+### v0.2.0
+
+- **新增** CSRF 防护（详见上方安全说明）
+- **修复** 前端在端点返回 500 等错误时仍显示「关闭成功」——此前只要 `fetch` 未抛异常就当作成功，现在区分网络层失败、响应解析失败与 HTTP 错误三类情况
+- **修复** 插件卸载 / 热重载时未撤销 `slots.inject` 注册，可能导致设置页残留重复条目
+- **新增** 17 个行为测试（host 10 + client 7），覆盖 CSRF 守卫、退出码语义、卸载清理与响应判定

package/README_EN.md

@@ -2,9 +2,9 @@
 
 <!-- Hero -->
 <div align="center">
-  <b style="font-size: 1.15em;">One-click shutdown for DeepSeek Harness</b><br /><br />
-  Settings page · double confirmation · graceful exit<br /><br />
-  <b>Closing the browser no longer leaves a background process behind</b>.
+  <b style="font-size: 1.15em;">One-click shutdown / restart for DeepSeek Harness</b><br /><br />
+  Settings page · double confirmation · graceful exit / auto restart<br /><br />
+  <b>Closing the browser no longer leaves a background process behind, and you can restart with one click</b>.
 </div>
 
 <div align="center">
@@ -13,13 +13,14 @@
 
 ## ✨ Why
 
-Closing the browser does **not** stop the dsh background service——it keeps holding the port and memory. This plugin adds an in-page shutdown entry, no CLI needed.
+Closing the browser does **not** stop the dsh background service——it keeps holding the port and memory. This plugin adds an in-page shutdown/restart entry, no CLI needed.
 
 ## ✨ Features
 
-- **🖥️ Settings page**: a new "Shutdown" section at the bottom of the sidebar
-- **🛡️ Double confirmation**: click "Shut down DSH service" → "Confirm / Cancel" buttons appear, preventing accidents
-- **✅ Success feedback**: a green "shut down successfully" panel renders immediately (and tries to close the tab)
+- **🖥️ Settings page**: a new "Service Control" section at the bottom of the sidebar
+- **🛡️ Double confirmation**: click "Shut down" or "Restart" → "Confirm / Cancel" buttons appear, preventing accidents
+- **✅ Success feedback**: a green "shut down successfully / restarting" panel renders immediately (and tries to close the tab)
+- **🔁 Auto restart**: the restart request makes the host process exit with a special code; the `launcher/dsh-web.ps1` launcher catches it (root `dsh-web.cmd`/`dsh-web.ps1` are shortcuts) and relaunches `dsh web`
 - **🎨 Native look**: all colors reference DSH theme tokens (`--dsw-alias-*`), light/dark adaptive
 
 ## 📦 Install
@@ -32,6 +33,7 @@ dsh plugin --profile web add <this repo / npm package>
 ## 🧠 Design notes
 
 - Shutdown endpoint: `/_dsh/shutdown` (POST only) —— responds "shutting down" first, then `process.exit(0)` after 500ms
+- Restart endpoint: `/_dsh/restart` (POST only) —— responds "restarting" first, then exits with code 42 after 500ms; `dsh-web.ps1` catches that code and relaunches `dsh web`
 - Session logs are appended per event; exiting while idle is safe. Let an in-flight turn finish first
 - Browsers only allow scripts to close windows they opened——manually opened tabs keep the success panel, so the result is always visible
 
@@ -42,3 +44,20 @@ Tag your own plugins with <a href="https://github.com/topics/dsh-shutdown"><code
 ## 📄 License
 
 [MIT](./LICENSE)
+
+## 🔒 Security
+
+The shutdown endpoint terminates the whole service, so it guards itself against cross-site requests:
+
+- **Browser requests**: the `Origin` header is validated. Requests from external sites receive `403`, preventing any web page from shutting down your local DSH in the background — browsers do not preflight such cross-origin simple POSTs, so the endpoint has to check for itself
+- **Same-origin access**: `127.0.0.1` / `localhost` / `::1` pass directly; when accessed via a LAN IP or a custom hostname (`dsh web --host 0.0.0.0`), `Origin` must equal `Host`
+- **Requests without `Origin`**: non-browser clients such as `curl` and launch scripts pass, so scripting keeps working
+
+## 📝 Changelog
+
+### v0.2.0
+
+- **Added** CSRF protection (see Security above)
+- **Fixed** the UI reporting "shut down successfully" when the endpoint returned an error such as 500 — previously any non-throwing `fetch` counted as success; network failure, response parsing failure and HTTP errors are now distinguished
+- **Fixed** `slots.inject` registration not being disposed on plugin unload or hot reload, which could leave duplicate entries in the Settings page
+- **Added** 17 behavioural tests (10 host + 7 client) covering the CSRF guard, exit-code semantics, teardown and response handling

package/lib/client.js

@@ -5,9 +5,10 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.apply = apply;
 
 /**
- * dsh-shutdown 浏览器半部：Settings「关机」页面。
- * 确认模式：点「关闭 DSH 服务」→ 出现「确认关闭 / 取消」双按钮；
- * 确认后请求关机并渲染成功面板。样式引用 DSH 主题 token。
+ * dsh-shutdown 浏览器半部：Settings「服务开关」页面。
+ * 支持「关闭 DSH 服务」和「重启 DSH 服务」两个操作；
+ * 都使用相同的确认模式：点击操作 → 出现「确认 / 取消」双按钮。
+ * 确认后请求对应端点并渲染成功面板。样式引用 DSH 主题 token。
  */
 const react_1 = require("react");
 
@@ -28,81 +29,97 @@ const CSS = '' +
   '.sd-err{color:var(--dsw-alias-state-error-primary)}';
 
 function ShutdownSection() {
-  const [confirmOpen, setConfirmOpen] = react_1.useState(false);
-  const [closing, setClosing] = react_1.useState(false);
-  const [done, setDone] = react_1.useState(false);
+  const [action, setAction] = react_1.useState(null); // 'shutdown' | 'restart' | null
+  const [busy, setBusy] = react_1.useState(false);
+  const [done, setDone] = react_1.useState(null); // 'shutdown' | 'restart' | null
   const [error, setError] = react_1.useState(null);
 
   function handleCancel() {
-    setConfirmOpen(false);
+    setAction(null);
     setError(null);
   }
 
   async function handleConfirm() {
-    if (closing) return;
-    setClosing(true);
+    if (busy) return;
+    const currentAction = action;
+    setBusy(true);
     setError(null);
+    const path = currentAction === 'restart' ? '/_dsh/restart' : '/_dsh/shutdown';
+    let res;
     try {
-      const res = await fetch('/_dsh/shutdown', {
+      res = await fetch(path, {
         method: 'POST',
         headers: { accept: 'application/json' },
       });
-      const body = await res.json();
-      if (body && body.ok) {
-        setDone(true);
-        // 尝试关闭标签页（仅对脚本打开的窗口生效，手动打开的标签页静默失败）
-        window.setTimeout(function () { try { window.close(); } catch (e) { /* 忽略 */ } }, 800);
-        return;
-      }
-      setError(String((body && body.error) || '关机请求失败'));
-      setClosing(false);
-      setConfirmOpen(false);
     } catch (e) {
-      // 进程可能已退出导致 fetch 失败——这其实意味着关机成功
-      setDone(true);
+      // 只有网络层失败才说明进程已经退出——这恰恰意味着操作成功。
+      // 注意范围：响应解析失败不走这条分支，否则 500 错误页会被误报成「关闭成功」。
+      setDone(currentAction);
+      return;
     }
+
+    let body = null;
+    try {
+      body = await res.json();
+    } catch (e) {
+      body = null; // 非 JSON 响应（错误页等）：按失败处理
+    }
+
+    if (res.ok && body && body.ok) {
+      setDone(currentAction);
+      // 尝试关闭标签页（仅对脚本打开的窗口生效，手动打开的标签页静默失败）
+      window.setTimeout(function () { try { window.close(); } catch (e) { /* 忽略 */ } }, 800);
+      return;
+    }
+
+    const fallback = currentAction === 'restart' ? '重启请求失败' : '关机请求失败';
+    setError(String((body && body.error) || ('HTTP ' + res.status + '：' + fallback)));
+    setBusy(false);
+    setAction(null);
   }
 
   if (done) {
-    const okTitle = react_1.createElement('h3', { className: 'sd-title' }, '关机');
+    const isRestart = done === 'restart';
+    const okTitle = react_1.createElement('h3', { className: 'sd-title' }, isRestart ? '重启' : '关机');
     const okLine = react_1.createElement(
       'div',
       { className: 'sd-ok' },
-      '✅ 关闭成功——DSH 服务已退出。'
+      isRestart ? '✅ 重启中——DSH 服务正在重新启动。' : '✅ 关闭成功——DSH 服务已退出。'
     );
     const okDesc = react_1.createElement(
       'div',
       { className: 'sd-sub' },
-      '本标签页可以安全关闭。如需再次使用，请重新启动 dsh web。'
+      isRestart ? '稍后请刷新本页面以连接新的 DSH 服务。' : '本标签页可以安全关闭。如需再次使用，请重新启动 dsh web。'
     );
     return react_1.createElement('div', { className: 'sd-root' }, okTitle, okLine, okDesc);
   }
 
-  const title = react_1.createElement('h3', { className: 'sd-title' }, '关机');
+  const title = react_1.createElement('h3', { className: 'sd-title' }, '服务开关');
   const desc = react_1.createElement(
     'div',
     { className: 'sd-sub' },
-    '关闭 DSH 服务进程。关闭后需要重新启动 dsh web 才能继续使用。'
+    '关闭或重启 DSH 服务进程。重启后请稍等片刻再刷新页面。'
   );
 
   const errLine = error
     ? react_1.createElement('div', { className: 'sd-err' }, '错误：' + error)
     : null;
 
-  if (confirmOpen) {
+  if (action) {
+    const isRestart = action === 'restart';
     const confirmLine = react_1.createElement(
       'div',
       { className: 'sd-confirm' },
-      '确定要关闭 DSH 服务吗？正在进行的对话将被中断。'
+      isRestart ? '确定要重启 DSH 服务吗？正在进行的对话将被中断。' : '确定要关闭 DSH 服务吗？正在进行的对话将被中断。'
     );
     const confirmBtn = react_1.createElement(
       'button',
-      { className: 'sd-btn sd-btn-danger', onClick: handleConfirm, disabled: closing },
-      closing ? '正在关闭…' : '确认关闭'
+      { className: 'sd-btn sd-btn-danger', onClick: handleConfirm, disabled: busy },
+      busy ? (isRestart ? '正在重启…' : '正在关闭…') : (isRestart ? '确认重启' : '确认关闭')
     );
     const cancelBtn = react_1.createElement(
       'button',
-      { className: 'sd-btn sd-btn-plain', onClick: handleCancel, disabled: closing },
+      { className: 'sd-btn sd-btn-plain', onClick: handleCancel, disabled: busy },
       '取消'
     );
     const btnRow = react_1.createElement('div', { className: 'sd-row' }, confirmBtn, cancelBtn);
@@ -119,16 +136,22 @@ function ShutdownSection() {
 
   const shutdownBtn = react_1.createElement(
     'button',
-    { className: 'sd-btn sd-btn-danger', onClick: function () { setConfirmOpen(true); } },
+    { className: 'sd-btn sd-btn-danger', onClick: function () { setAction('shutdown'); } },
     '关闭 DSH 服务'
   );
+  const restartBtn = react_1.createElement(
+    'button',
+    { className: 'sd-btn sd-btn-plain', onClick: function () { setAction('restart'); } },
+    '重启 DSH 服务'
+  );
+  const btnRow = react_1.createElement('div', { className: 'sd-row' }, shutdownBtn, restartBtn);
 
   return react_1.createElement(
     'div',
     { className: 'sd-root' },
     title,
     desc,
-    shutdownBtn,
+    btnRow,
     errLine
   );
 }
@@ -142,14 +165,16 @@ function apply(ctx) {
   styleEl.textContent = CSS;
   document.head.appendChild(styleEl);
 
-  slots.inject('settings.section', () => slots.register(
-    { name: 'settings.section', id: 'dsh-shutdown', order: 999, label: () => '关机' },
+  const offSlot = slots.inject('settings.section', () => slots.register(
+    { name: 'settings.section', id: 'dsh-shutdown', order: 999, label: () => '服务开关' },
     () => react_1.createElement(ShutdownSection, null)
   ));
 
   ctx.effect(() => () => {
+    // slots.inject 返回 disposer，必须调用——否则插件卸载 / 热重载后设置页会残留条目
+    if (typeof offSlot === 'function') offSlot();
     if (styleEl.parentNode) styleEl.parentNode.removeChild(styleEl);
   }, NS + ':css');
 }
 
-return module.exports; } });
+return module.exports; } });

package/lib/index.js

@@ -1,40 +1,82 @@
 /**
  * dsh-shutdown — DeepSeek Harness 关机插件（Host 半部）。
  *
- * 提供 `/_dsh/shutdown` 端点：收到 POST 后先返回确认响应（让浏览器
- * 拿到"正在关闭"的反馈），500ms 后优雅退出宿主进程。会话日志为逐事件
- * 落盘，空闲时退出安全；正在进行的 turn 请等待其完成后再关闭。
+ * 提供 `/_dsh/shutdown` 与 `/_dsh/restart` 端点：收到 POST 后先返回确认响应
+ * （让浏览器拿到"正在关闭/重启"的反馈），500ms 后退出宿主进程。
+ * 关闭使用退出码 0；重启使用特殊退出码 42，由外部启动脚本捕获后自动重新拉起。
+ * 会话日志为逐事件落盘，空闲时退出安全；正在进行的 turn 请等待其完成后再操作。
  * @module dsh-shutdown
  */
 export const name = 'dsh-shutdown';
 export const inject = [];
 
 const EXIT_DELAY_MS = 500;
+const RESTART_EXIT_CODE = 42;
+
+/**
+ * 判断请求是否来自本机同源页面（CSRF 守卫）。
+ *
+ * 关机会终止整个服务，是破坏性最强的动作。任意网页都能向
+ * http://127.0.0.1:3080/_dsh/shutdown 发一个「简单 POST」，浏览器不会为此做预检，
+ * 所以端点自身必须把关：浏览器发 POST 一定会带 Origin，凡带外部 Origin 的一律拒绝；
+ * 无 Origin 的请求（curl、启动脚本）不是浏览器发起的，放行。
+ */
+const LOOPBACK_HOSTS = ['127.0.0.1', 'localhost', '::1', '[::1]'];
+
+function isTrustedOrigin(req) {
+  const origin = req.headers && req.headers.origin;
+  if (origin === undefined || origin === null || origin === '') return true;
+  let parsed;
+  try {
+    parsed = new URL(origin);
+  } catch {
+    return false; // Origin 无法解析：保守拒绝
+  }
+  if (LOOPBACK_HOSTS.includes(parsed.hostname)) return true;
+  // 通过局域网 IP / 自定义主机名访问（dsh web --host 0.0.0.0）时，
+  // Origin 与 Host 头一致即视为同源；不一致则说明是伪造的同源声明。
+  const host = req.headers && req.headers.host;
+  return typeof host === 'string' && host !== '' && parsed.host === host;
+}
+
+/** 通用动作端点注册。 */
+function registerAction(webServer, path, message, exitCode) {
+  return webServer.register({
+    kind: 'exact',
+    path,
+    handler: (req, res) => {
+      if (req.method !== 'POST') {
+        res.statusCode = 405;
+        res.setHeader('Content-Type', 'application/json; charset=utf-8');
+        res.end(JSON.stringify({ ok: false, error: 'method-not-allowed' }));
+        return;
+      }
+      if (!isTrustedOrigin(req)) {
+        res.statusCode = 403;
+        res.setHeader('Content-Type', 'application/json; charset=utf-8');
+        res.end(JSON.stringify({ ok: false, error: 'cross-origin-forbidden' }));
+        return;
+      }
+      res.statusCode = 200;
+      res.setHeader('Content-Type', 'application/json; charset=utf-8');
+      res.end(JSON.stringify({ ok: true, message }));
+      // 留出响应送达浏览器的时间，再退出进程
+      setTimeout(() => {
+        process.exit(exitCode);
+      }, EXIT_DELAY_MS);
+    },
+  });
+}
 
 /** Plugin 入口。 */
 export async function apply(ctx) {
   ctx.inject(['webServer'], (webCtx) => {
-    const disposeRoute = webCtx.webServer.register({
-      kind: 'exact',
-      path: '/_dsh/shutdown',
-      handler: (req, res) => {
-        if (req.method !== 'POST') {
-          res.statusCode = 405;
-          res.setHeader('Content-Type', 'application/json; charset=utf-8');
-          res.end(JSON.stringify({ ok: false, error: 'method-not-allowed' }));
-          return;
-        }
-        res.statusCode = 200;
-        res.setHeader('Content-Type', 'application/json; charset=utf-8');
-        res.end(JSON.stringify({ ok: true, message: '正在关闭 DSH 服务' }));
-        // 留出响应送达浏览器的时间，再退出进程
-        setTimeout(() => {
-          process.exit(0);
-        }, EXIT_DELAY_MS);
-      },
-    });
+    const disposers = [
+      registerAction(webCtx.webServer, '/_dsh/shutdown', '正在关闭 DSH 服务', 0),
+      registerAction(webCtx.webServer, '/_dsh/restart', '正在重启 DSH 服务', RESTART_EXIT_CODE),
+    ];
     return () => {
-      disposeRoute();
+      for (const dispose of disposers) dispose();
     };
   });
-}
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-shutdown",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "DeepSeek Harness 关机插件：Settings 增加「关机」页，一键优雅退出 DSH 服务（先响应页面再退出进程）。",
   "keywords": [
     "deepseek",
@@ -17,12 +17,17 @@
     "./package.json": "./package.json"
   },
   "files": [
-    "lib",
+    "lib/index.js",
+    "lib/client.js",
     "cordis.patch.yml",
     "README.md",
     "README_EN.md",
     "dsh.plugin.json"
   ],
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/knlght/DSH-shutdown.git"
+  },
   "engines": {
     "node": "^22.19.0 || >=24.0.0"
   },