dsh-cuigengji-plugin 0.4.3 → 0.4.4

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}`);