dsh-cuigengji-plugin 0.4.2 → 0.4.4

package/src/adapters/mcp/client.js

@@ -1,172 +1,172 @@
-import http from 'node:http';
-import { randomBytes } from 'node:crypto';
-import { fileURLToPath } from 'node:url';
-import { Client } from '@modelcontextprotocol/sdk/client/index.js';
-import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
-import { assertMemoryAction, toolName } from './actions.js';
-
-const MAX_BODY_BYTES = 1024 * 1024;
-
-export function memoryRuntime(options = {}, runtime = process) {
-  const command = options.nodeCommand ?? runtime.env.CUIGENGJI_DSH_NODE ?? runtime.execPath;
-  const env = {};
-  if (runtime.env.NODE_OPTIONS) env.NODE_OPTIONS = runtime.env.NODE_OPTIONS;
-  // Desktop hosts use the Electron executable as Node. The SDK's default
-  // environment omits this switch, which otherwise launches a second GUI.
-  if (runtime.versions.electron || runtime.env.ELECTRON_RUN_AS_NODE === '1') {
-    env.ELECTRON_RUN_AS_NODE = '1';
-  }
-  return { command, env };
-}
-
-export class MemoryBridge {
-  constructor(store, options = {}) {
-    this.store = store;
-    this.dispatch = options.dispatch ?? ((action, args, actor) => store.dispatch(action, args, actor));
-    this.resolveNovel = options.resolveNovel;
-    this.timeoutMs = options.timeoutMs ?? 30_000;
-    const runtime = memoryRuntime(options);
-    this.nodeCommand = runtime.command;
-    this.runtimeEnv = runtime.env;
-    this.sessions = new Map();
-    this.tokens = new Map();
-    this.closed = false;
-  }
-
-  async start() {
-    if (this.closed) throw new Error('记忆服务已关闭');
-    if (this.starting) return this.starting;
-    this.starting = (async () => {
-      this.server = http.createServer((request, response) => { void this.handle(request, response); });
-      this.server.requestTimeout = this.timeoutMs;
-      await new Promise((resolve, reject) => {
-        this.server.once('error', reject);
-        this.server.listen(0, '127.0.0.1', resolve);
-      });
-      this.endpoint = `http://127.0.0.1:${this.server.address().port}/memory`;
-      return this;
-    })();
-    return this.starting;
-  }
-
-  async handle(request, response) {
-    const reply = (status, data) => {
-      if (response.destroyed) return;
-      response.writeHead(status, { 'content-type': 'application/json; charset=utf-8', 'cache-control': 'no-store' });
-      response.end(JSON.stringify(data));
-    };
-    try {
-      if (request.method !== 'POST' || request.url !== '/memory') return reply(404, { error: '未知接口' });
-      const sessionId = this.tokens.get((request.headers.authorization ?? '').replace(/^Bearer /, ''));
-      if (!sessionId) return reply(401, { error: '会话凭据失效' });
-      let size = 0;
-      const chunks = [];
-      for await (const chunk of request) {
-        size += chunk.length;
-        if (size > MAX_BODY_BYTES) return reply(413, { error: '记忆请求过大' });
-        chunks.push(chunk);
-      }
-      const { action, args = {} } = JSON.parse(Buffer.concat(chunks).toString('utf8'));
-      assertMemoryAction(action);
-      if (!args || typeof args !== 'object' || Array.isArray(args)) throw new Error('参数必须是对象');
-      const actor = { kind: 'agent', sessionId };
-      const novelId = await this.boundNovel(sessionId);
-      if (!novelId) throw new Error('当前会话尚未绑定小说');
-      if (args.novelId && args.novelId !== novelId) throw new Error('不能访问其他小说');
-      // Never forward model-supplied session/actor identities.
-      const { sessionId: _session, actor: _actor, ...safeArgs } = args;
-      const result = await this.dispatch(action, { ...safeArgs, novelId }, actor);
-      reply(200, { result });
-    } catch (error) {
-      reply(400, { error: error.message });
-    }
-  }
-
-  async boundNovel(sessionId) {
-    if (this.resolveNovel) return this.resolveNovel(sessionId);
-    const binding = await this.store.dispatch('binding.get', {}, { kind: 'agent', sessionId });
-    return binding?.novelId;
-  }
-
-  async connection(sessionId) {
-    if (!sessionId || typeof sessionId !== 'string') throw new Error('缺少会话 ID');
-    await this.start();
-    const existing = this.sessions.get(sessionId);
-    if (existing) return existing.pending ?? existing;
-    const entry = { pending: null };
-    entry.pending = this.connect(sessionId, entry);
-    this.sessions.set(sessionId, entry);
-    try { return await entry.pending; }
-    catch (error) {
-      if (this.sessions.get(sessionId) === entry) this.sessions.delete(sessionId);
-      throw error;
-    }
-  }
-
-  async connect(sessionId, entry) {
-    const token = randomBytes(32).toString('hex');
-    this.tokens.set(token, sessionId);
-    const env = { ...this.runtimeEnv, CUIGENGJI_MEMORY_ENDPOINT: this.endpoint, CUIGENGJI_MEMORY_TOKEN: token };
-    const transport = new StdioClientTransport({
-      command: this.nodeCommand,
-      args: [fileURLToPath(new URL('./server.js', import.meta.url))],
-      env,
-      stderr: 'pipe',
-    });
-    const client = new Client({ name: 'cuigengji-host', version: '0.1.0' });
-    // Keep child diagnostics out of model output; drain to avoid pipe backpressure.
-    transport.stderr?.on('data', () => {});
-    client.onclose = () => {
-      this.tokens.delete(token);
-      // A reconnect may have replaced this entry. Never tear down the new
-      // connection when an older MCP child exits asynchronously.
-      if (this.sessions.get(sessionId) === entry) this.sessions.delete(sessionId);
-    };
-    try {
-      await client.connect(transport);
-      return { client, transport, token };
-    } catch (error) {
-      this.tokens.delete(token);
-      await transport.close().catch(() => {});
-      throw error;
-    }
-  }
-
-  async call(sessionId, action, args = {}, { signal } = {}) {
-    assertMemoryAction(action);
-    if (signal?.aborted) throw signal.reason ?? new Error('已取消');
-    const { client } = await this.connection(sessionId);
-    const result = await client.callTool({ name: toolName(action), arguments: args }, undefined, {
-      timeout: this.timeoutMs, signal,
-    });
-    const output = result.content?.find(item => item.type === 'text')?.text;
-    if (result.isError) throw new Error(output ?? '记忆工具失败');
-    return JSON.parse(output ?? 'null');
-  }
-
-  async disconnect(sessionId) {
-    const entry = this.sessions.get(sessionId);
-    this.sessions.delete(sessionId);
-    if (!entry) return;
-    const connection = await (entry.pending ?? entry).catch(() => null);
-    if (!connection) return;
-    this.tokens.delete(connection.token);
-    await connection.client.close();
-  }
-
-  async close() {
-    this.closed = true;
-    await Promise.allSettled([...this.sessions.keys()].map(sessionId => this.disconnect(sessionId)));
-    this.tokens.clear();
-    if (this.server) {
-      this.server.closeAllConnections();
-      await new Promise(resolve => this.server.close(resolve));
-    }
-  }
-}
-
-export class MemoryClient {
-  constructor(bridge, sessionId) { this.bridge = bridge; this.sessionId = sessionId; }
-  call(action, args, options) { return this.bridge.call(this.sessionId, action, args, options); }
-  close() { return this.bridge.disconnect(this.sessionId); }
-}
+import http from 'node:http';
+import { randomBytes } from 'node:crypto';
+import { fileURLToPath } from 'node:url';
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
+import { assertMemoryAction, toolName } from './actions.js';
+
+const MAX_BODY_BYTES = 1024 * 1024;
+
+export function memoryRuntime(options = {}, runtime = process) {
+  const command = options.nodeCommand ?? runtime.env.CUIGENGJI_DSH_NODE ?? runtime.execPath;
+  const env = {};
+  if (runtime.env.NODE_OPTIONS) env.NODE_OPTIONS = runtime.env.NODE_OPTIONS;
+  // Desktop hosts use the Electron executable as Node. The SDK's default
+  // environment omits this switch, which otherwise launches a second GUI.
+  if (runtime.versions.electron || runtime.env.ELECTRON_RUN_AS_NODE === '1') {
+    env.ELECTRON_RUN_AS_NODE = '1';
+  }
+  return { command, env };
+}
+
+export class MemoryBridge {
+  constructor(store, options = {}) {
+    this.store = store;
+    this.dispatch = options.dispatch ?? ((action, args, actor) => store.dispatch(action, args, actor));
+    this.resolveNovel = options.resolveNovel;
+    this.timeoutMs = options.timeoutMs ?? 30_000;
+    const runtime = memoryRuntime(options);
+    this.nodeCommand = runtime.command;
+    this.runtimeEnv = runtime.env;
+    this.sessions = new Map();
+    this.tokens = new Map();
+    this.closed = false;
+  }
+
+  async start() {
+    if (this.closed) throw new Error('记忆服务已关闭');
+    if (this.starting) return this.starting;
+    this.starting = (async () => {
+      this.server = http.createServer((request, response) => { void this.handle(request, response); });
+      this.server.requestTimeout = this.timeoutMs;
+      await new Promise((resolve, reject) => {
+        this.server.once('error', reject);
+        this.server.listen(0, '127.0.0.1', resolve);
+      });
+      this.endpoint = `http://127.0.0.1:${this.server.address().port}/memory`;
+      return this;
+    })();
+    return this.starting;
+  }
+
+  async handle(request, response) {
+    const reply = (status, data) => {
+      if (response.destroyed) return;
+      response.writeHead(status, { 'content-type': 'application/json; charset=utf-8', 'cache-control': 'no-store' });
+      response.end(JSON.stringify(data));
+    };
+    try {
+      if (request.method !== 'POST' || request.url !== '/memory') return reply(404, { error: '未知接口' });
+      const sessionId = this.tokens.get((request.headers.authorization ?? '').replace(/^Bearer /, ''));
+      if (!sessionId) return reply(401, { error: '会话凭据失效' });
+      let size = 0;
+      const chunks = [];
+      for await (const chunk of request) {
+        size += chunk.length;
+        if (size > MAX_BODY_BYTES) return reply(413, { error: '记忆请求过大' });
+        chunks.push(chunk);
+      }
+      const { action, args = {} } = JSON.parse(Buffer.concat(chunks).toString('utf8'));
+      assertMemoryAction(action);
+      if (!args || typeof args !== 'object' || Array.isArray(args)) throw new Error('参数必须是对象');
+      const actor = { kind: 'agent', sessionId };
+      const novelId = await this.boundNovel(sessionId);
+      if (!novelId) throw new Error('当前会话尚未绑定小说');
+      if (args.novelId && args.novelId !== novelId) throw new Error('不能访问其他小说');
+      // Never forward model-supplied session/actor identities.
+      const { sessionId: _session, actor: _actor, ...safeArgs } = args;
+      const result = await this.dispatch(action, { ...safeArgs, novelId }, actor);
+      reply(200, { result });
+    } catch (error) {
+      reply(400, { error: error.message });
+    }
+  }
+
+  async boundNovel(sessionId) {
+    if (this.resolveNovel) return this.resolveNovel(sessionId);
+    const binding = await this.store.dispatch('binding.get', {}, { kind: 'agent', sessionId });
+    return binding?.novelId;
+  }
+
+  async connection(sessionId) {
+    if (!sessionId || typeof sessionId !== 'string') throw new Error('缺少会话 ID');
+    await this.start();
+    const existing = this.sessions.get(sessionId);
+    if (existing) return existing.pending ?? existing;
+    const entry = { pending: null };
+    entry.pending = this.connect(sessionId, entry);
+    this.sessions.set(sessionId, entry);
+    try { return await entry.pending; }
+    catch (error) {
+      if (this.sessions.get(sessionId) === entry) this.sessions.delete(sessionId);
+      throw error;
+    }
+  }
+
+  async connect(sessionId, entry) {
+    const token = randomBytes(32).toString('hex');
+    this.tokens.set(token, sessionId);
+    const env = { ...this.runtimeEnv, CUIGENGJI_MEMORY_ENDPOINT: this.endpoint, CUIGENGJI_MEMORY_TOKEN: token };
+    const transport = new StdioClientTransport({
+      command: this.nodeCommand,
+      args: [fileURLToPath(new URL('./server.js', import.meta.url))],
+      env,
+      stderr: 'pipe',
+    });
+    const client = new Client({ name: 'cuigengji-host', version: '0.1.0' });
+    // Keep child diagnostics out of model output; drain to avoid pipe backpressure.
+    transport.stderr?.on('data', () => {});
+    client.onclose = () => {
+      this.tokens.delete(token);
+      // A reconnect may have replaced this entry. Never tear down the new
+      // connection when an older MCP child exits asynchronously.
+      if (this.sessions.get(sessionId) === entry) this.sessions.delete(sessionId);
+    };
+    try {
+      await client.connect(transport);
+      return { client, transport, token };
+    } catch (error) {
+      this.tokens.delete(token);
+      await transport.close().catch(() => {});
+      throw error;
+    }
+  }
+
+  async call(sessionId, action, args = {}, { signal } = {}) {
+    assertMemoryAction(action);
+    if (signal?.aborted) throw signal.reason ?? new Error('已取消');
+    const { client } = await this.connection(sessionId);
+    const result = await client.callTool({ name: toolName(action), arguments: args }, undefined, {
+      timeout: this.timeoutMs, signal,
+    });
+    const output = result.content?.find(item => item.type === 'text')?.text;
+    if (result.isError) throw new Error(output ?? '记忆工具失败');
+    return JSON.parse(output ?? 'null');
+  }
+
+  async disconnect(sessionId) {
+    const entry = this.sessions.get(sessionId);
+    this.sessions.delete(sessionId);
+    if (!entry) return;
+    const connection = await (entry.pending ?? entry).catch(() => null);
+    if (!connection) return;
+    this.tokens.delete(connection.token);
+    await connection.client.close();
+  }
+
+  async close() {
+    this.closed = true;
+    await Promise.allSettled([...this.sessions.keys()].map(sessionId => this.disconnect(sessionId)));
+    this.tokens.clear();
+    if (this.server) {
+      this.server.closeAllConnections();
+      await new Promise(resolve => this.server.close(resolve));
+    }
+  }
+}
+
+export class MemoryClient {
+  constructor(bridge, sessionId) { this.bridge = bridge; this.sessionId = sessionId; }
+  call(action, args, options) { return this.bridge.call(this.sessionId, action, args, options); }
+  close() { return this.bridge.disconnect(this.sessionId); }
+}

package/src/adapters/mcp/server.js

@@ -1,40 +1,40 @@
-import { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
-import { MEMORY_ACTIONS, toolName } from './actions.js';
-
-// This process owns no project files. Its capability is one session-bound token.
-const endpoint = process.env.CUIGENGJI_MEMORY_ENDPOINT;
-const token = process.env.CUIGENGJI_MEMORY_TOKEN;
-if (!endpoint || !token) throw new Error('缺少 cuigengji 宿主连接信息');
-const address = new URL(endpoint);
-if (address.protocol !== 'http:' || address.hostname !== '127.0.0.1') {
-  throw new Error('记忆服务必须为本机 loopback 地址');
-}
-
-const server = new Server({ name: 'cuigengji-memory', version: '0.1.0' }, { capabilities: { tools: {} } });
-server.setRequestHandler(ListToolsRequestSchema, async () => ({
-  tools: MEMORY_ACTIONS.map(action => ({
-    name: toolName(action),
-    description: `当前小说记忆操作 ${action}。小说由会话绑定；修改须携带读取到的版本。`,
-    inputSchema: { type: 'object', additionalProperties: true },
-  })),
-}));
-server.setRequestHandler(CallToolRequestSchema, async (request, extra) => {
-  const action = MEMORY_ACTIONS.find(item => toolName(item) === request.params.name);
-  if (!action) return { isError: true, content: [{ type: 'text', text: '未知记忆工具' }] };
-  try {
-    const response = await fetch(endpoint, {
-      method: 'POST',
-      headers: { authorization: `Bearer ${token}`, 'content-type': 'application/json' },
-      body: JSON.stringify({ action, args: request.params.arguments ?? {} }),
-      signal: AbortSignal.any([extra.signal, AbortSignal.timeout(30_000)]),
-    });
-    const body = await response.json();
-    if (!response.ok) return { isError: true, content: [{ type: 'text', text: body.error ?? '记忆操作失败' }] };
-    return { content: [{ type: 'text', text: JSON.stringify(body.result ?? null) }] };
-  } catch (error) {
-    return { isError: true, content: [{ type: 'text', text: error.message }] };
-  }
-});
-await server.connect(new StdioServerTransport());
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+import { MEMORY_ACTIONS, toolName } from './actions.js';
+
+// This process owns no project files. Its capability is one session-bound token.
+const endpoint = process.env.CUIGENGJI_MEMORY_ENDPOINT;
+const token = process.env.CUIGENGJI_MEMORY_TOKEN;
+if (!endpoint || !token) throw new Error('缺少 cuigengji 宿主连接信息');
+const address = new URL(endpoint);
+if (address.protocol !== 'http:' || address.hostname !== '127.0.0.1') {
+  throw new Error('记忆服务必须为本机 loopback 地址');
+}
+
+const server = new Server({ name: 'cuigengji-memory', version: '0.1.0' }, { capabilities: { tools: {} } });
+server.setRequestHandler(ListToolsRequestSchema, async () => ({
+  tools: MEMORY_ACTIONS.map(action => ({
+    name: toolName(action),
+    description: `当前小说记忆操作 ${action}。小说由会话绑定；修改须携带读取到的版本。`,
+    inputSchema: { type: 'object', additionalProperties: true },
+  })),
+}));
+server.setRequestHandler(CallToolRequestSchema, async (request, extra) => {
+  const action = MEMORY_ACTIONS.find(item => toolName(item) === request.params.name);
+  if (!action) return { isError: true, content: [{ type: 'text', text: '未知记忆工具' }] };
+  try {
+    const response = await fetch(endpoint, {
+      method: 'POST',
+      headers: { authorization: `Bearer ${token}`, 'content-type': 'application/json' },
+      body: JSON.stringify({ action, args: request.params.arguments ?? {} }),
+      signal: AbortSignal.any([extra.signal, AbortSignal.timeout(30_000)]),
+    });
+    const body = await response.json();
+    if (!response.ok) return { isError: true, content: [{ type: 'text', text: body.error ?? '记忆操作失败' }] };
+    return { content: [{ type: 'text', text: JSON.stringify(body.result ?? null) }] };
+  } catch (error) {
+    return { isError: true, content: [{ type: 'text', text: error.message }] };
+  }
+});
+await server.connect(new StdioServerTransport());

package/src/application/bindings/session.js

@@ -1,72 +1,72 @@
-import { isReadAction } from '../../contracts/actions.js';
-import {novelLifecycleActions, recyclesNovel} from '../novels/lifecycle.ts';
-import {createHash} from 'node:crypto';
-const sessionKey = id => createHash('sha256').update(id || '').digest('hex');
-
-export function createSessionPolicy({ store, audit, ready = async () => undefined }) {
-  const active = new Set();
-  const resolveBinding = async sessionId => {
-    await ready();
-    return store.dispatch('binding.get', {}, { kind: 'agent', sessionId });
-  };
-
-  async function dispatch(action, args = {}, actor = { kind: 'agent' }) {
-    await ready();
-    if (action === 'binding.set' && active.has(actor.sessionId)) throw new Error('会话正在运行，请停止后切换小说或章节');
-    if (novelLifecycleActions.includes(action) && actor.kind !== 'human') throw new Error('整部作品的回收、恢复与永久删除需要作者操作');
-    if (actor.kind === 'agent') {
-      const binding = await resolveBinding(actor.sessionId);
-      // Inspecting an unbound session is valid and lets the agent explain how
-      // to begin; all novel data operations still require an author binding.
-      if (action === 'binding.get') return binding;
-      if (!binding) throw new Error('请先在催更姬面板绑定小说');
-      if (args.novelId && args.novelId !== binding.novelId) throw new Error('不能访问另一部小说');
-      args = { ...args, novelId: binding.novelId };
-
-    }
-    const guard = action === 'binding.set' || recyclesNovel(action, args) ? state => {
-      if (action === 'binding.set' && active.has(actor.sessionId)) throw new Error('会话正在运行，请停止后切换小说或章节');
-      if (recyclesNovel(action, args)) {
-        const novelId = args.novelId || state.bindings[sessionKey(actor.sessionId)]?.novelId;
-        for (const sessionId of active) if (novelId && state.bindings[sessionKey(sessionId)]?.novelId === novelId) {
-          throw Object.assign(new Error('这部作品仍有 AI 写作任务在运行，请先停止相关会话，再移入回收站'), {code: 'NOVEL_BUSY'});
-        }
-      }
-    } : undefined;
-    const result = await store.dispatch(action, args, actor, guard);
-    if (!isReadAction(action)) await audit(actor.sessionId, {
-      action, actor: actor.kind, entityId: result?.id, revision: result?.revision, reason: args.reason || '',
-    });
-    return result;
-  }
-
-  function register(ctx, memory) {
-    const disconnect = sessionId => {
-      // Status/disposal are emit events; do not leave an unhandled rejection.
-      void memory.disconnect(sessionId).catch(error => ctx.logger?.warn?.(`cuigengji memory cleanup: ${error.message}`));
-    };
-    ctx.on('agent/pre-step', async ({ agent }, next) => {
-      const sessionId = String(agent.id);
-      // DSH only emits pre-step for running agents. Record this before any I/O
-      // so a simultaneous recycle sees the active session inside the lock.
-      active.add(sessionId);
-      try {await resolveBinding(sessionId);}
-      catch (error) {if (error.code !== 'UPGRADE_REQUIRED') throw error;}
-      return next();
-    });
-    ctx.on('agent/status', ({ agent, status }) => {
-      const sessionId = String(agent.id);
-      if (status === 'running') active.add(sessionId);
-      else {
-        active.delete(sessionId);
-        disconnect(sessionId);
-      }
-    });
-    ctx.on('agent/disposed', ({ agent }) => {
-      const sessionId = String(agent.id);
-      active.delete(sessionId);
-      disconnect(sessionId);
-    });
-  }
-  return { dispatch, resolveBinding, register };
-}
+import { isReadAction } from '../../contracts/actions.js';
+import {novelLifecycleActions, recyclesNovel} from '../novels/lifecycle.ts';
+import {createHash} from 'node:crypto';
+const sessionKey = id => createHash('sha256').update(id || '').digest('hex');
+
+export function createSessionPolicy({ store, audit, ready = async () => undefined }) {
+  const active = new Set();
+  const resolveBinding = async sessionId => {
+    await ready();
+    return store.dispatch('binding.get', {}, { kind: 'agent', sessionId });
+  };
+
+  async function dispatch(action, args = {}, actor = { kind: 'agent' }) {
+    await ready();
+    if (action === 'binding.set' && active.has(actor.sessionId)) throw new Error('会话正在运行，请停止后切换小说或章节');
+    if (novelLifecycleActions.includes(action) && actor.kind !== 'human') throw new Error('整部作品的回收、恢复与永久删除需要作者操作');
+    if (actor.kind === 'agent') {
+      const binding = await resolveBinding(actor.sessionId);
+      // Inspecting an unbound session is valid and lets the agent explain how
+      // to begin; all novel data operations still require an author binding.
+      if (action === 'binding.get') return binding;
+      if (!binding) throw new Error('请先在催更姬面板绑定小说');
+      if (args.novelId && args.novelId !== binding.novelId) throw new Error('不能访问另一部小说');
+      args = { ...args, novelId: binding.novelId };
+
+    }
+    const guard = action === 'binding.set' || recyclesNovel(action, args) ? state => {
+      if (action === 'binding.set' && active.has(actor.sessionId)) throw new Error('会话正在运行，请停止后切换小说或章节');
+      if (recyclesNovel(action, args)) {
+        const novelId = args.novelId || state.bindings[sessionKey(actor.sessionId)]?.novelId;
+        for (const sessionId of active) if (novelId && state.bindings[sessionKey(sessionId)]?.novelId === novelId) {
+          throw Object.assign(new Error('这部作品仍有 AI 写作任务在运行，请先停止相关会话，再移入回收站'), {code: 'NOVEL_BUSY'});
+        }
+      }
+    } : undefined;
+    const result = await store.dispatch(action, args, actor, guard);
+    if (!isReadAction(action)) await audit(actor.sessionId, {
+      action, actor: actor.kind, entityId: result?.id, revision: result?.revision, reason: args.reason || '',
+    });
+    return result;
+  }
+
+  function register(ctx, memory) {
+    const disconnect = sessionId => {
+      // Status/disposal are emit events; do not leave an unhandled rejection.
+      void memory.disconnect(sessionId).catch(error => ctx.logger?.warn?.(`cuigengji memory cleanup: ${error.message}`));
+    };
+    ctx.on('agent/pre-step', async ({ agent }, next) => {
+      const sessionId = String(agent.id);
+      // DSH only emits pre-step for running agents. Record this before any I/O
+      // so a simultaneous recycle sees the active session inside the lock.
+      active.add(sessionId);
+      try {await resolveBinding(sessionId);}
+      catch (error) {if (error.code !== 'UPGRADE_REQUIRED') throw error;}
+      return next();
+    });
+    ctx.on('agent/status', ({ agent, status }) => {
+      const sessionId = String(agent.id);
+      if (status === 'running') active.add(sessionId);
+      else {
+        active.delete(sessionId);
+        disconnect(sessionId);
+      }
+    });
+    ctx.on('agent/disposed', ({ agent }) => {
+      const sessionId = String(agent.id);
+      active.delete(sessionId);
+      disconnect(sessionId);
+    });
+  }
+  return { dispatch, resolveBinding, register };
+}

package/src/application/queries/context.js

@@ -1,18 +1,18 @@
-// Select complete sections; the plugin does not impose a character budget.
-export function selectReference(novel, chapterId) {
-  const current = chapterId ? novel.chapters[chapterId] : null;
-  const previous = Object.values(novel.chapters).filter(c=>!c.deleted && c.id!==current?.id && (!current || c.order<current.order))
-    .sort((a,b)=>b.order-a.order || a.id.localeCompare(b.id)).slice(0,2).reverse();
-  const items=[];
-  const add=(kind,entity)=>{
-    if (!entity?.content || entity.deleted) return;
-    items.push({kind,id:entity.id,revision:entity.revision,title:entity.title||'情节规划',content:entity.content,truncated:false});
-  };
-  for(const chapter of previous)add('previous_chapter',chapter);
-  add('current_chapter',current);
-  return {items,omitted:[],usedChars:items.reduce((sum,i)=>sum+i.content.length,0)};
-}
-
+// Select complete sections; the plugin does not impose a character budget.
+export function selectReference(novel, chapterId) {
+  const current = chapterId ? novel.chapters[chapterId] : null;
+  const previous = Object.values(novel.chapters).filter(c=>!c.deleted && c.id!==current?.id && (!current || c.order<current.order))
+    .sort((a,b)=>b.order-a.order || a.id.localeCompare(b.id)).slice(0,2).reverse();
+  const items=[];
+  const add=(kind,entity)=>{
+    if (!entity?.content || entity.deleted) return;
+    items.push({kind,id:entity.id,revision:entity.revision,title:entity.title||'情节规划',content:entity.content,truncated:false});
+  };
+  for(const chapter of previous)add('previous_chapter',chapter);
+  add('current_chapter',current);
+  return {items,omitted:[],usedChars:items.reduce((sum,i)=>sum+i.content.length,0)};
+}
+
 export function renderReference(material) {
   const names={previous_chapter:'前文章节',current_chapter:'当前参考章节',memory:'按需读取的设定'};
   const sections=material.items.map(item=>`## ${names[item.kind]||'参考资料'}：${item.title}\n[ID: ${item.id} | 版本: ${item.revision}${item.truncated?` | 已截断，仅含${item.position||'部分'}片段，可用工具读取完整内容`:''}]\n\n${item.content}`);