dsh-perf-boost 0.0.0-stage → 0.2.1

package/lib/host/warmup.js

@@ -0,0 +1,146 @@
+// 大会话预热：把「最近用过的大会话」的文件先读一遍，让系统文件缓存里已经摊开，
+// 这样你点进去时读的是内存而不是硬盘。只读、分段、可中断，完全不碰会话数据本身。
+//
+// 为什么不做「直接热投影」：投影缓存属于宿主内部实现（sessionProjections /
+// sessionProjectionCache），上游没给预热接口。硬塞进去一旦签名变化就会在启动时
+// 抛错，违背「稳定性别变差」。这里只做没有任何耦合面的文件缓存预热。
+
+import { open, readdir, stat } from 'node:fs/promises';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+
+const DEFAULTS = {
+  minBytes: 8 * 1024 * 1024,   // 只预热 8MB 以上的会话（本机 39 个）
+  maxSessions: 10,             // 最多 10 个
+  maxBytes: 400 * 1024 * 1024, // 一次最多读 400MB，到量就停
+  maxWarmMs: 180 * 1000,       // 最多占用 3 分钟（可中断）
+  chunkBytes: 4 * 1024 * 1024,
+  pauseMs: 40,                 // 每读完一块歇一下，别跟正常使用抢 IO
+  delayMs: 15 * 1000,          // 启动后等 15 秒再开始，避开首屏高峰
+};
+
+function num(value, fallback) {
+  const n = Number(value);
+  return Number.isFinite(n) && n > 0 ? n : fallback;
+}
+
+function settingsFrom(config) {
+  return {
+    minBytes: num(config?.minBytes, DEFAULTS.minBytes),
+    maxSessions: num(config?.maxSessions, DEFAULTS.maxSessions),
+    maxBytes: num(config?.maxBytes, DEFAULTS.maxBytes),
+    maxWarmMs: num(config?.maxWarmMs, DEFAULTS.maxWarmMs),
+    chunkBytes: num(config?.chunkBytes, DEFAULTS.chunkBytes),
+    pauseMs: num(config?.pauseMs, DEFAULTS.pauseMs),
+    delayMs: num(config?.delayMs, DEFAULTS.delayMs),
+  };
+}
+
+const SESSION_FILE = /^session\.[\w.-]+\.(jsonl|zstd|jsonl\.zstd)$/;
+
+async function collectCandidates(root, minBytes, limit) {
+  const found = [];
+  const now = Date.now();
+  const projects = await readdir(root, { withFileTypes: true }).catch(() => []);
+  for (const project of projects) {
+    if (!project.isDirectory()) continue;
+    const projectDir = join(root, project.name);
+    const sessions = await readdir(projectDir, { withFileTypes: true }).catch(() => []);
+    for (const entry of sessions) {
+      if (!entry.isDirectory()) continue;
+      const dir = join(projectDir, entry.name);
+      const files = await readdir(dir, { withFileTypes: true }).catch(() => []);
+      for (const file of files) {
+        if (!file.isFile() || !SESSION_FILE.test(file.name)) continue;
+        const path = join(dir, file.name);
+        const info = await stat(path).catch(() => null);
+        if (!info || info.size < minBytes) continue;
+        // 最近十分钟还在写的文件就是当前会话：这时候读它只会和首屏、流式落盘抢磁盘。
+        if (now - info.mtimeMs < 10 * 60 * 1000) continue;
+        found.push({ path, size: info.size, mtimeMs: info.mtimeMs });
+      }
+    }
+  }
+  found.sort((a, b) => b.mtimeMs - a.mtimeMs || b.size - a.size);
+  return found.slice(0, limit);
+}
+
+async function warmOne(path, size, options, deadline) {
+  const handle = await open(path, 'r').catch(() => null);
+  if (!handle) return 0;
+  const buffer = Buffer.allocUnsafe(Math.min(options.chunkBytes, 8 * 1024 * 1024));
+  let read = 0;
+  try {
+    for (;;) {
+      if (Date.now() > deadline) break;
+      const result = await handle.read(buffer, 0, buffer.length, read).catch(() => null);
+      if (!result || result.bytesRead <= 0) break;
+      read += result.bytesRead;
+      if (read >= size) break;
+      await new Promise((resolve) => setTimeout(resolve, options.pauseMs));
+    }
+  } finally {
+    await handle.close().catch(() => undefined);
+  }
+  return read;
+}
+
+/**
+ * @param {{ ctx: any, logger: any, config: any }} options
+ */
+export function createWarmup({ ctx, logger, config }) {
+  const options = settingsFrom(config);
+  let timer = null;
+  let stopped = false;
+
+  function stop() {
+    stopped = true;
+    if (timer !== null) {
+      clearTimeout(timer);
+      timer = null;
+    }
+  }
+
+  async function run() {
+    if (stopped) return;
+    const root = join(homedir(), '.dsh', 'sessions');
+    const candidates = await collectCandidates(root, options.minBytes, options.maxSessions);
+    if (candidates.length === 0) {
+      logger.info?.('warmup: 没有需要预热的大会话');
+      return;
+    }
+    const deadline = Date.now() + options.maxWarmMs;
+    let total = 0;
+    let done = 0;
+    for (const candidate of candidates) {
+      if (stopped || Date.now() > deadline) break;
+      if (total >= options.maxBytes) break;
+      const bytes = await warmOne(candidate.path, candidate.size, options, deadline);
+      total += bytes;
+      if (bytes > 0) done += 1;
+    }
+    if (total > 0) {
+      logger.info?.(`warmup: 已预热 ${(total / 1024 / 1024).toFixed(0)}MB（${done} 个大会话）`);
+    }
+  }
+
+  return {
+    start() {
+      // 宿主关闭时清掉定时器，别留下悬空的句柄。
+      try {
+        if (ctx && typeof ctx.on === 'function') ctx.on('dispose', stop);
+      } catch (error) {
+        void error;
+      }
+      timer = setTimeout(() => {
+        timer = null;
+        run().catch((error) => logger.warn?.('warmup 失败:', error?.message ?? error));
+      }, options.delayMs);
+      if (typeof timer.unref === 'function') timer.unref();
+    },
+    stop,
+    options,
+  };
+}
+
+export const warmupDefaults = DEFAULTS;

package/lib/index.js

@@ -0,0 +1,278 @@
+// lib/index.js —— 宿主半边。
+//
+// 插件的能力绝大部分在客户端（渲染进程）实现：帧率上限、长对话省渲染、动画开关。
+// 宿主侧只做两件事：
+//
+//   1. 提供一个挂载点，让 cordis.patch.yml 把插件插进 profile，从而把 lib/client.js
+//      作为客户端 bundle 下发；
+//   2. 提供一条同源路由 `POST /api/dsh-perf-boost/update-check`，代替页面去问 npm
+//      上的最新版本（页面自己请求跨域会被 CSP 拦掉，读不到 npm registry）。
+//
+// 路由只读、无副作用：只取 `latest` 标签的版本号，不下载、不安装、不落盘。
+// 网络不可用时返回 `latest: null`，页面显示「未查到」，其余功能不受影响。
+import { readFile, writeFile, mkdir } from 'node:fs/promises';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+
+import { createWarmup } from './host/warmup.js';
+
+export const name = 'dsh-perf-boost';
+
+// 不声明 inject：宿主侧只读文件、不依赖任何服务，少一个依赖就少一种启动失败的方式。
+
+const ROUTE = '/api/dsh-perf-boost/update-check';
+const LAST_ROUTE = '/api/dsh-perf-boost/last-run';
+// 页面每次启动端口都不一样（localhost:随机端口），localStorage 跟着端口走，
+// 所以要跨启动留一份「上次实测」只能落到宿主侧的文件里。
+const LAST_FILE = join(homedir(), '.dsh', 'perf-boost-last.json');
+// 大会话预热开关：客户端只在设置页里改，宿主要能持久化，重启后仍然是你的选择。
+const WARM_ROUTE = '/api/dsh-perf-boost/warmup';
+const WARM_FILE = join(homedir(), '.dsh', 'perf-boost-warmup.json');
+const PACKAGE_NAME = 'dsh-perf-boost';
+// npmmirror 优先（国内快），失败再问官方源。
+const REGISTRIES = ['https://registry.npmmirror.com', 'https://registry.npmjs.org'];
+const CACHE_MS = 10 * 60 * 1000;
+const TIMEOUT_MS = 8000;
+
+let cache = { at: 0, latest: null, source: '', error: '' };
+
+async function readInstalledVersion() {
+  try {
+    const text = await readFile(new URL('../package.json', import.meta.url), 'utf8');
+    const pkg = JSON.parse(text);
+    return typeof pkg.version === 'string' ? pkg.version : '';
+  } catch {
+    return '';
+  }
+}
+
+export function parseVersion(value) {
+  if (typeof value !== 'string') return null;
+  const match = /^(\d+)\.(\d+)\.(\d+)/.exec(value.trim());
+  if (match === null) return null;
+  return [Number(match[1]), Number(match[2]), Number(match[3])];
+}
+
+export function isNewer(candidate, current) {
+  const a = parseVersion(candidate);
+  const b = parseVersion(current);
+  if (a === null || b === null) return false;
+  for (let i = 0; i < 3; i += 1) {
+    if (a[i] !== b[i]) return a[i] > b[i];
+  }
+  return false;
+}
+
+async function fetchLatest(registry) {
+  const url = `${registry}/${PACKAGE_NAME}/latest`;
+  const response = await fetch(url, {
+    headers: { accept: 'application/json' },
+    signal: AbortSignal.timeout(TIMEOUT_MS),
+  });
+  if (!response.ok) throw new Error(`HTTP ${response.status}`);
+  const body = await response.json();
+  const version = body?.version;
+  if (typeof version !== 'string' || version === '') throw new Error('响应里没有 version');
+  return { version, source: registry };
+}
+
+async function resolveLatest() {
+  const now = Date.now();
+  if (cache.at !== 0 && cache.at !== undefined && now - cache.at < CACHE_MS) return cache;
+  let lastError = '未连接';
+  let notPublished = false;
+  let best = null;
+  let bestSource = '';
+  for (const registry of REGISTRIES) {
+    try {
+      const { version, source } = await fetchLatest(registry);
+      // 镜像可能比官方慢半拍，两边都问，取版本更高的那个，别被慢的镜像带偏。
+      if (best === null || isNewer(version, best)) {
+        best = version;
+        bestSource = source;
+      }
+    } catch (error) {
+      lastError = error?.message ?? String(error);
+      // 所有镜像都 404 = 这个包还没发到 npm，不算「网络失败」。
+      if (lastError === 'HTTP 404') notPublished = true;
+      else notPublished = false;
+    }
+  }
+  if (best !== null) {
+    cache = { at: now, latest: best, source: bestSource, error: '' };
+    return cache;
+  }
+  cache = { at: now, latest: null, source: '', error: notPublished ? 'not-published' : lastError };
+  return cache;
+}
+
+async function readWarmSetting() {
+  try {
+    const parsed = JSON.parse(await readFile(WARM_FILE, 'utf8'));
+    if (typeof parsed?.enabled === 'boolean') return parsed.enabled;
+  } catch {
+    // 没有文件 / 文件坏了：用默认值（开）
+  }
+  return true;
+}
+
+async function writeWarmSetting(enabled) {
+  try {
+    await mkdir(join(homedir(), '.dsh'), { recursive: true });
+    await writeFile(WARM_FILE, JSON.stringify({ enabled: !!enabled }), 'utf8');
+  } catch {
+    // 写不进去只影响「下次启动还记得吗」，本次开关照旧生效
+  }
+}
+
+export function apply(ctx, config) {
+  // 大会话预热（宿主侧，可关）：把投影冷折叠从首屏挪到后台分片做。
+  // 结构不认识时它会自己静默退出，不影响下面这条路由与客户端功能。
+  let warm = null;
+  let warmSetting = true; // 落盘的开关状态，和 warm 实例分开记
+  const warmBlocked = config?.warmup === false;
+  const warmLogger = (() => {
+    try {
+      return typeof ctx.logger === 'function' ? ctx.logger('perf-boost') : { info() {}, warn: console.warn };
+    } catch {
+      return { info() {}, warn: console.warn };
+    }
+  })();
+
+  function startWarm() {
+    if (warmBlocked || warm !== null) return;
+    try {
+      warm = createWarmup({ ctx, logger: warmLogger, config: config || {} });
+      warm.start();
+    } catch (error) {
+      warm = null;
+      try { console.warn('[perf-boost] warmup 装配失败:', error?.message ?? error); } catch { void 0; }
+    }
+  }
+
+  function stopWarm() {
+    if (warm === null) return;
+    try { warm.stop(); } catch { void 0; }
+    warm = null;
+  }
+
+  if (!warmBlocked) {
+    readWarmSetting()
+      .then((on) => { warmSetting = on; if (on) startWarm(); })
+      .catch(() => { void 0; });
+  }
+
+  ctx.inject(['connection'], (c) => {
+    const connection = c.get('connection');
+    const register = typeof connection?.fetch?.register === 'function'
+      ? connection.fetch.register.bind(connection.fetch)
+      : undefined;
+    if (register === undefined) return;
+
+    // 读/写上一条实测结果。写只接受四个数字字段，不落任何别的东西。
+    const lastHandler = async (request) => {
+      let body = null;
+      try {
+        const text = typeof request?.text === 'function' ? await request.text() : '';
+        if (text) body = JSON.parse(text);
+      } catch {
+        body = null;
+      }
+      const wantWrite = body && typeof body.pct === 'number';
+      if (wantWrite) {
+        const round = (n) => (Number.isFinite(n) ? Math.round(n) : 0);
+        const record = {
+          pct: Math.max(0, Math.min(100, round(body.pct))),
+          saved: Math.max(0, round(body.saved)),
+          busy: Math.max(0, round(body.busy)),
+          at: Number.isFinite(body.at) ? Math.round(body.at) : Date.now(),
+        };
+        try {
+          await mkdir(join(homedir(), '.dsh'), { recursive: true });
+          await writeFile(LAST_FILE, JSON.stringify(record), 'utf8');
+        } catch {
+          // 写不进去不影响使用，下一次再试。
+        }
+        return Response.json({ ok: true, value: record }, { headers: { 'cache-control': 'no-store' } });
+      }
+      let value = null;
+      try {
+        const text = await readFile(LAST_FILE, 'utf8');
+        const parsed = JSON.parse(text);
+        if (typeof parsed?.pct === 'number') value = parsed;
+      } catch {
+        value = null;
+      }
+      return Response.json({ ok: true, value }, { headers: { 'cache-control': 'no-store' } });
+    };
+
+    try {
+      c.effect(
+        () => register({ path: LAST_ROUTE, methods: ['POST'], requestBody: 'buffered', fetch: lastHandler }),
+        'dsh-perf-boost: /api/dsh-perf-boost/last-run',
+      );
+    } catch {
+      // 注册失败只是少一条缓存读取，面板会退回到只显示本次数据。
+    }
+
+    // 预热开关：读/写都走这一条，写会落盘，重启后保持。
+    const warmHandler = async (request) => {
+      let body = null;
+      try {
+        const text = typeof request?.text === 'function' ? await request.text() : '';
+        if (text) body = JSON.parse(text);
+      } catch {
+        body = null;
+      }
+      if (body && typeof body.enabled === 'boolean') {
+        warmSetting = body.enabled;
+        await writeWarmSetting(body.enabled);
+        if (body.enabled) startWarm();
+        else stopWarm();
+      } else {
+        warmSetting = await readWarmSetting();
+        if (!warmBlocked && warmSetting && warm === null) startWarm();
+      }
+      return Response.json({ ok: true, value: { enabled: warmSetting && !warmBlocked } }, { headers: { 'cache-control': 'no-store' } });
+    };
+
+    try {
+      c.effect(
+        () => register({ path: WARM_ROUTE, methods: ['POST'], requestBody: 'buffered', fetch: warmHandler }),
+        'dsh-perf-boost: /api/dsh-perf-boost/warmup',
+      );
+    } catch {
+      // 注册失败只是设置页那个开关不出现，预热仍按配置文件走。
+    }
+
+    const handler = async () => {
+      const [installed, resolved] = await Promise.all([readInstalledVersion(), resolveLatest()]);
+      const latest = resolved.latest;
+      return Response.json(
+        {
+          ok: true,
+          value: {
+            installed,
+            latest,
+            updateAvailable: latest === null ? false : isNewer(latest, installed),
+            source: resolved.source,
+            checkedAt: new Date(resolved.at === 0 ? Date.now() : resolved.at).toISOString(),
+            error: latest === null ? resolved.error : '',
+          },
+        },
+        { headers: { 'cache-control': 'no-store' } },
+      );
+    };
+
+    try {
+      c.effect(
+        () => register({ path: ROUTE, methods: ['POST'], requestBody: 'buffered', fetch: handler }),
+        'dsh-perf-boost: /api/dsh-perf-boost/update-check',
+      );
+    } catch {
+      // 注册失败只影响版本检查，插件其余功能照旧。
+    }
+  });
+}
+
+export default { name, apply };

package/package.json

@@ -1,6 +1,67 @@
 {
   "name": "dsh-perf-boost",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "version": "0.2.1",
+  "description": "DSH 桌面端性能优化（纯客户端）：长对话省渲染、窗口隐藏不排帧默认开，关背景模糊/减少动画/空闲降帧按需打开，设置页显示实测省下的主线程时间。Client-side rendering optimizations for DeepSeek Harness Desktop.",
+  "type": "module",
+  "main": "lib/index.js",
+  "exports": {
+    ".": "./lib/index.js",
+    "./client": "./lib/client.js"
+  },
+  "license": "MIT",
+  "keywords": [
+    "deepseek-harness",
+    "dsh",
+    "dsh-plugin",
+    "performance",
+    "fps",
+    "requestanimationframe",
+    "content-visibility",
+    "性能优化",
+    "性能",
+    "优化",
+    "增强",
+    "流畅",
+    "加速",
+    "卡顿",
+    "省内存",
+    "faster",
+    "smooth",
+    "speed-up"
+  ],
+  "files": [
+    "lib",
+    "assets",
+    "cordis.patch.yml",
+    "README.md",
+    "LICENSE",
+    "CHANGELOG.md"
+  ],
+  "engines": {
+    "dsh": ">=0.2.0-0"
+  },
+  "peerDependencies": {
+    "@deepseek-ai/cordis": "^4.0.1"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/AlLHHH/dsh-perf-boost.git"
+  },
+  "homepage": "https://github.com/AlLHHH/dsh-perf-boost#readme",
+  "bugs": {
+    "url": "https://github.com/AlLHHH/dsh-perf-boost/issues"
+  },
+  "sideEffects": false,
+  "dsh": {
+    "bundle": {
+      "patch": "./cordis.patch.yml"
+    },
+    "client": {
+      "platform": "web",
+      "inject": [
+        "slots",
+        "sessions"
+      ]
+    }
+  }
+}