agentworkshop 0.7.5 → 0.7.7

package/server/plugins/workshop.ts

@@ -1,153 +1,156 @@
-/**
- * Workshop 启动插件(设计文档 §8)。
- * Nitro 启动时:创建 data/ 目录 → openWorkshopDb(data/workshop.sqlite)→ 装配 5 个 repo
- * + createAgentImpl 工厂 → createAgentChannelManager → restore() 恢复运行时
- * → 为有 lead 的 channel 启动 SchedulerLoop → 挂 globalThis.__workshopManager。
- *
- * 导出:
- *  - getWorkshopManager():读取进程级 manager 单例(先检查已设置;未初始化抛 503)
- *  - ensureLeadSchedulerLoop():为 channel 装配并启动 lead 的 SchedulerLoop(幂等)
- *
- * 注意:不静态 import 'nitropack/runtime/plugin'——defineNitroPlugin 是恒等函数,
- * 插件 default 直接导出普通函数即可,避免把 nitropack 依赖引入可被 tsx 直跑测试
- * 导入的模块图(顶层 node_modules 无 nitropack 符号链接)。
- */
-import { mkdirSync } from 'node:fs'
-import { resolve } from 'node:path'
-import { ensureDataDir } from '@/shared/config/home.mjs'
-import { openWorkshopDb, initWorkshopDb } from '../services/workshop/db/database'
-import { createChannelRepo } from '../services/workshop/db/channel.repo'
-import { createAgentRepo } from '../services/workshop/db/agent.repo'
-import { createChannelAgentRepo } from '../services/workshop/db/channel-agent.repo'
-import { createTeamRepo } from '../services/workshop/db/team.repo'
-import { createTeamMemberRepo } from '../services/workshop/db/team-member.repo'
-import { createChannelTemplateRepo } from '../services/workshop/db/channel-template.repo'
-import { createTaskRepo } from '../services/workshop/db/task.repo'
-import { createMessageRepo } from '../services/workshop/db/message.repo'
-import { createSubscriptionRepo } from '../services/workshop/db/subscription.repo'
-import { createMemoryRepo } from '../services/workshop/db/memory.repo'
-import { createUserRepo } from '../services/workshop/db/user.repo'
-import { createChannelEventRepo } from '../services/workshop/db/channel-event.repo'
-import { createApprovalHistoryRepo, createAlarmEventRepo, createAuditRepo, createApprovalRequestRepo } from '../services/workshop/db/ops.repo'
-import { bindOpsRepos } from '../services/workshop/ops/ops'
-import { configureHitlResolver } from '../services/workshop/agents/hitl-registry'
-import { ensureAllEventRecorders } from '../api/workshop/ws'
-import { createAgentImpl } from '../services/workshop/agents/factory'
-import {
-  createAgentChannelManager,
-  type AgentChannelManager,
-  type AllRepos,
-  type ManagerDeps,
-} from '../services/workshop/runtime/manager'
-import { AppError } from '../utils/errors'
-import { workshopSettings } from '../services/workshop/settings'
-
-declare global {
-
-  var __workshopManager: AgentChannelManager | undefined
-}
-
-/** 读取进程级 manager 单例(先检查已设置;未初始化抛 503) */
-export function getWorkshopManager(): AgentChannelManager {
-  const manager = globalThis.__workshopManager
-  if (!manager) {
-    throw new AppError(503, 'WORKSHOP_NOT_READY', 'workshop 尚未初始化(plugin 未执行)')
-  }
-  return manager
-}
-
-/**
- * 为 channel 装配并启动 lead 的 SchedulerLoop(幂等)。
- * 委托 manager.ensureChannelActive(懒加载:装配 lead + 调度循环)。
- */
-export function ensureLeadSchedulerLoop(
-  manager: AgentChannelManager,
-  channelId: string,
-  options?: { tickMs?: number, stallMs?: number },
-): void {
-  manager.ensureChannelActive(channelId, options)
-}
-
-/**
- * Nitro 启动插件(默认导出普通函数;defineNitroPlugin 为恒等包装)。
- * 注意:runNitroPlugins 同步调用 plugin,不 await 返回值;manager.restore() 方法体全同步
- * (无 await 语句),直接调用即同步完成恢复,故此处不 await。
- * 幂等:globalThis.__workshopManager 已设置(如测试预置)则跳过初始化。
- */
-export default function workshopPlugin(nitroApp: {
-  hooks: { hook(name: string, fn: (...args: unknown[]) => void | Promise<void>): void }
-}): void {
-  // 进程级未处理拒绝兜底(Node ≥23 默认对 unhandledRejection 抛错可击穿整个服务)。
-  // 环境型连接复位(休眠/断网后陈旧 socket read ECONNRESET、omp 子进程管道抖动等)
-  // 不应让服务崩溃;记录上下文后继续运行(正常业务错误仍走各自显式处理路径)。
-  process.on('unhandledRejection', (reason) => {
-    console.error('[workshop] 未处理 Promise rejection(已兜底,不影响服务):', reason)
-  })
-
-  // 先检查已设置:避免重复装配覆盖既有单例
-  if (globalThis.__workshopManager) return
-
-  // 数据目录经配置根解析（repo: <repo>/.AgentWorkShop/data;home: ~/.AgentWorkShop/data;
-  // ensureDataDir 自动迁移旧 cwd/data 位置,目录不存在则创建）
-  const dataDir = ensureDataDir()
-  mkdirSync(dataDir, { recursive: true })
-
-  // 打开(或创建)数据库 → 初始化(建表 + PRAGMA;openWorkshopDb 已内置,显式再调一次保持幂等)
-  const db = openWorkshopDb(resolve(dataDir, 'workshop.sqlite'))
-  initWorkshopDb(db)
-
-  // 装配 repos + Agent impl 工厂
-  const repos: AllRepos = {
-    users: createUserRepo(db),
-    channelEvents: createChannelEventRepo(db),
-    channels: createChannelRepo(db),
-    agents: createAgentRepo(db),
-    teams: createTeamRepo(db),
-    teamMembers: createTeamMemberRepo(db),
-    channelTemplates: createChannelTemplateRepo(db),
-    channelAgents: createChannelAgentRepo(db),
-    messages: createMessageRepo(db),
-    subscriptions: createSubscriptionRepo(db),
-    tasks: createTaskRepo(db),
-    memories: createMemoryRepo(db),
-  }
-  // S4/S5/R1/R3:合规三表仓储接线(审批历史/报警事件/审计日志/高危复核)
-  bindOpsRepos({
-    approvalHistory: createApprovalHistoryRepo(db),
-    alarmEvents: createAlarmEventRepo(db),
-    audit: createAuditRepo(db),
-    approvalRequests: createApprovalRequestRepo(db),
-  })
-  const deps: ManagerDeps = { repos, implFactory: createAgentImpl, db }
-
-  // 创建 manager → 挂全局单例(后续 REST/A2A/WS 经 getWorkshopManager() 读取)
-  const manager = createAgentChannelManager(deps)
-  globalThis.__workshopManager = manager
-
-  // HITL 全局待办:agentId → channelId/agentName 解析器(避免 registry 反向依赖 manager)
-  configureHitlResolver((agentId) => {
-    const row = repos.channelAgents.findById(agentId)
-    return row ? { channelId: row.channelId, agentName: row.name } : null
-  })
-
-  // 懒加载恢复:仅激活有待办任务的 channel(装配 lead + 调度循环);其余纯持久化
-  manager.restore()
-
-  // 全时事件录制:为全部存量 channel 建立常驻流(事件 server 驱动落库,与 client 无关)
-  void ensureAllEventRecorders(manager).catch(err => console.error('[workshop] 事件录制器初始化失败:', err))
-
-  // idle sweeper:空闲 agent 超时自动卸载(释放 omp 子进程与内存)。
-  // busy 成员有三重守卫(state!==idle / 信箱 pending / lead 活跃任务)不会被卸载,
-  // "作业期间进程不断开"由守卫保证;空闲宽限可经 workshop.idle_grace_ms 调整 ——
-  // 需要进程更长常驻(免冷启动 30~90s)时调大(env WORKSHOP_IDLE_GRACE_MS 兼容)。
-  const idleSweepMs = workshopSettings().idle_sweep_ms
-  const idleGraceMs = workshopSettings().idle_grace_ms
-  const stopSweeper = manager.startIdleSweeper({ intervalMs: idleSweepMs, graceMs: idleGraceMs })
-
-  // 关机:完整关闭 manager(sweeper + 调度循环 + agent 运行时),关闭数据库
-  nitroApp.hooks.hook('close', async () => {
-    stopSweeper()
-    await manager.shutdown()
-    db.close()
-  })
-}
+/**
+ * Workshop 启动插件(设计文档 §8)。
+ * Nitro 启动时:创建 data/ 目录 → openWorkshopDb(data/workshop.sqlite)→ 装配 5 个 repo
+ * + createAgentImpl 工厂 → createAgentChannelManager → restore() 恢复运行时
+ * → 为有 lead 的 channel 启动 SchedulerLoop → 挂 globalThis.__workshopManager。
+ *
+ * 导出:
+ *  - getWorkshopManager():读取进程级 manager 单例(先检查已设置;未初始化抛 503)
+ *  - ensureLeadSchedulerLoop():为 channel 装配并启动 lead 的 SchedulerLoop(幂等)
+ *
+ * 注意:不静态 import 'nitropack/runtime/plugin'——defineNitroPlugin 是恒等函数,
+ * 插件 default 直接导出普通函数即可,避免把 nitropack 依赖引入可被 tsx 直跑测试
+ * 导入的模块图(顶层 node_modules 无 nitropack 符号链接)。
+ */
+import { mkdirSync } from 'node:fs'
+import { resolve } from 'node:path'
+import { ensureDataDir } from '@/shared/config/home.mjs'
+import { openWorkshopDb, initWorkshopDb } from '../services/workshop/db/database'
+import { backupRegistry } from '../services/workshop/db/backup-registry'
+import { createChannelRepo } from '../services/workshop/db/channel.repo'
+import { createAgentRepo } from '../services/workshop/db/agent.repo'
+import { createChannelAgentRepo } from '../services/workshop/db/channel-agent.repo'
+import { createTeamRepo } from '../services/workshop/db/team.repo'
+import { createTeamMemberRepo } from '../services/workshop/db/team-member.repo'
+import { createChannelTemplateRepo } from '../services/workshop/db/channel-template.repo'
+import { createTaskRepo } from '../services/workshop/db/task.repo'
+import { createMessageRepo } from '../services/workshop/db/message.repo'
+import { createSubscriptionRepo } from '../services/workshop/db/subscription.repo'
+import { createMemoryRepo } from '../services/workshop/db/memory.repo'
+import { createUserRepo } from '../services/workshop/db/user.repo'
+import { createChannelEventRepo } from '../services/workshop/db/channel-event.repo'
+import { createApprovalHistoryRepo, createAlarmEventRepo, createAuditRepo, createApprovalRequestRepo } from '../services/workshop/db/ops.repo'
+import { bindOpsRepos } from '../services/workshop/ops/ops'
+import { configureHitlResolver } from '../services/workshop/agents/hitl-registry'
+import { ensureAllEventRecorders } from '../api/workshop/ws'
+import { createAgentImpl } from '../services/workshop/agents/factory'
+import {
+  createAgentChannelManager,
+  type AgentChannelManager,
+  type AllRepos,
+  type ManagerDeps,
+} from '../services/workshop/runtime/manager'
+import { AppError } from '../utils/errors'
+import { workshopSettings } from '../services/workshop/settings'
+
+declare global {
+
+  var __workshopManager: AgentChannelManager | undefined
+}
+
+/** 读取进程级 manager 单例(先检查已设置;未初始化抛 503) */
+export function getWorkshopManager(): AgentChannelManager {
+  const manager = globalThis.__workshopManager
+  if (!manager) {
+    throw new AppError(503, 'WORKSHOP_NOT_READY', 'workshop 尚未初始化(plugin 未执行)')
+  }
+  return manager
+}
+
+/**
+ * 为 channel 装配并启动 lead 的 SchedulerLoop(幂等)。
+ * 委托 manager.ensureChannelActive(懒加载:装配 lead + 调度循环)。
+ */
+export function ensureLeadSchedulerLoop(
+  manager: AgentChannelManager,
+  channelId: string,
+  options?: { tickMs?: number, stallMs?: number },
+): void {
+  manager.ensureChannelActive(channelId, options)
+}
+
+/**
+ * Nitro 启动插件(默认导出普通函数;defineNitroPlugin 为恒等包装)。
+ * 注意:runNitroPlugins 同步调用 plugin,不 await 返回值;manager.restore() 方法体全同步
+ * (无 await 语句),直接调用即同步完成恢复,故此处不 await。
+ * 幂等:globalThis.__workshopManager 已设置(如测试预置)则跳过初始化。
+ */
+export default function workshopPlugin(nitroApp: {
+  hooks: { hook(name: string, fn: (...args: unknown[]) => void | Promise<void>): void }
+}): void {
+  // 进程级未处理拒绝兜底(Node ≥23 默认对 unhandledRejection 抛错可击穿整个服务)。
+  // 环境型连接复位(休眠/断网后陈旧 socket read ECONNRESET、omp 子进程管道抖动等)
+  // 不应让服务崩溃;记录上下文后继续运行(正常业务错误仍走各自显式处理路径)。
+  process.on('unhandledRejection', (reason) => {
+    console.error('[workshop] 未处理 Promise rejection(已兜底,不影响服务):', reason)
+  })
+
+  // 先检查已设置:避免重复装配覆盖既有单例
+  if (globalThis.__workshopManager) return
+
+  // 数据目录经配置根解析（repo: <repo>/.AgentWorkShop/data;home: ~/.AgentWorkShop/data;
+  // ensureDataDir 自动迁移旧 cwd/data 位置,目录不存在则创建）
+  const dataDir = ensureDataDir()
+  mkdirSync(dataDir, { recursive: true })
+
+  // 打开(或创建)数据库 → 初始化(建表 + PRAGMA;openWorkshopDb 已内置,显式再调一次保持幂等)
+  const db = openWorkshopDb(resolve(dataDir, 'workshop.sqlite'))
+  initWorkshopDb(db)
+  // 登记主连接(备份插件 serialize 跑在主连接上,杜绝第二连接失效语句的事故)
+  backupRegistry.register(resolve(dataDir, 'workshop.sqlite'), db)
+
+  // 装配 repos + Agent impl 工厂
+  const repos: AllRepos = {
+    users: createUserRepo(db),
+    channelEvents: createChannelEventRepo(db),
+    channels: createChannelRepo(db),
+    agents: createAgentRepo(db),
+    teams: createTeamRepo(db),
+    teamMembers: createTeamMemberRepo(db),
+    channelTemplates: createChannelTemplateRepo(db),
+    channelAgents: createChannelAgentRepo(db),
+    messages: createMessageRepo(db),
+    subscriptions: createSubscriptionRepo(db),
+    tasks: createTaskRepo(db),
+    memories: createMemoryRepo(db),
+  }
+  // S4/S5/R1/R3:合规三表仓储接线(审批历史/报警事件/审计日志/高危复核)
+  bindOpsRepos({
+    approvalHistory: createApprovalHistoryRepo(db),
+    alarmEvents: createAlarmEventRepo(db),
+    audit: createAuditRepo(db),
+    approvalRequests: createApprovalRequestRepo(db),
+  })
+  const deps: ManagerDeps = { repos, implFactory: createAgentImpl, db }
+
+  // 创建 manager → 挂全局单例(后续 REST/A2A/WS 经 getWorkshopManager() 读取)
+  const manager = createAgentChannelManager(deps)
+  globalThis.__workshopManager = manager
+
+  // HITL 全局待办:agentId → channelId/agentName 解析器(避免 registry 反向依赖 manager)
+  configureHitlResolver((agentId) => {
+    const row = repos.channelAgents.findById(agentId)
+    return row ? { channelId: row.channelId, agentName: row.name } : null
+  })
+
+  // 懒加载恢复:仅激活有待办任务的 channel(装配 lead + 调度循环);其余纯持久化
+  manager.restore()
+
+  // 全时事件录制:为全部存量 channel 建立常驻流(事件 server 驱动落库,与 client 无关)
+  void ensureAllEventRecorders(manager).catch(err => console.error('[workshop] 事件录制器初始化失败:', err))
+
+  // idle sweeper:空闲 agent 超时自动卸载(释放 omp 子进程与内存)。
+  // busy 成员有三重守卫(state!==idle / 信箱 pending / lead 活跃任务)不会被卸载,
+  // "作业期间进程不断开"由守卫保证;空闲宽限可经 workshop.idle_grace_ms 调整 ——
+  // 需要进程更长常驻(免冷启动 30~90s)时调大(env WORKSHOP_IDLE_GRACE_MS 兼容)。
+  const idleSweepMs = workshopSettings().idle_sweep_ms
+  const idleGraceMs = workshopSettings().idle_grace_ms
+  const stopSweeper = manager.startIdleSweeper({ intervalMs: idleSweepMs, graceMs: idleGraceMs })
+
+  // 关机:完整关闭 manager(sweeper + 调度循环 + agent 运行时),关闭数据库
+  nitroApp.hooks.hook('close', async () => {
+    stopSweeper()
+    await manager.shutdown()
+    db.close()
+  })
+}