zero-workspace 0.0.2

package/plugin/scripts/shared/binding.cjs

@@ -0,0 +1,98 @@
+/**
+ * 会话绑定逻辑
+ */
+
+const fs = require('node:fs');
+const { readJsonFile } = require('./utils.cjs');
+const { BINDINGS_PATH } = require('./config.cjs');
+
+// 提醒节流间隔（毫秒）
+const THROTTLE_INTERVAL = 3 * 60 * 1000; // 3 分钟
+
+/**
+ * 获取会话绑定信息（本地读取）
+ * @param {string} sessionId - 会话 ID（CC 的 session_id 或 Cursor 的 conversation_id）
+ * @returns {object|null} 绑定信息或 null
+ */
+function getSessionBinding(sessionId) {
+  const bindings = readJsonFile(BINDINGS_PATH);
+  if (!bindings || !bindings.bindings) {
+    return null;
+  }
+  return bindings.bindings[sessionId] || null;
+}
+
+/**
+ * 检测文本是否涉及工作区关键词
+ * @param {string} text - 要检测的文本
+ * @returns {boolean}
+ */
+function containsWorkspaceKeywords(text) {
+  const { WORKSPACE_KEYWORDS } = require('./config.cjs');
+  if (!text) return false;
+  const lowerText = text.toLowerCase();
+  return WORKSPACE_KEYWORDS.some(kw => lowerText.includes(kw.toLowerCase()));
+}
+
+/**
+ * 判断是否应该节流提醒
+ * @param {object} binding - 会话绑定信息
+ * @param {string} reminderType - 提醒类型
+ * @param {number} [customInterval] - 自定义节流间隔（毫秒），默认使用 THROTTLE_INTERVAL
+ * @returns {boolean} true 表示应该节流（不发送提醒）
+ */
+function shouldThrottle(binding, reminderType, customInterval) {
+  // P0 (problem) 类型不节流，每次都提醒
+  if (reminderType === 'problem') {
+    return false;
+  }
+
+  if (!binding || !binding.lastReminder) {
+    return false;
+  }
+
+  const lastReminder = binding.lastReminder;
+  const interval = customInterval || THROTTLE_INTERVAL;
+
+  // 同一类型的提醒需要节流
+  if (lastReminder.type === reminderType) {
+    const lastTime = new Date(lastReminder.time).getTime();
+    const now = Date.now();
+    return (now - lastTime) < interval;
+  }
+
+  return false;
+}
+
+/**
+ * 更新上次提醒记录
+ * @param {string} sessionId - 会话 ID
+ * @param {string} reminderType - 提醒类型
+ * @returns {boolean} 是否更新成功
+ */
+function updateLastReminder(sessionId, reminderType) {
+  try {
+    const bindings = readJsonFile(BINDINGS_PATH);
+    if (!bindings || !bindings.bindings || !bindings.bindings[sessionId]) {
+      return false;
+    }
+
+    bindings.bindings[sessionId].lastReminder = {
+      type: reminderType,
+      time: new Date().toISOString()
+    };
+
+    fs.writeFileSync(BINDINGS_PATH, JSON.stringify(bindings, null, 2), 'utf-8');
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+module.exports = {
+  getSessionBinding,
+  containsWorkspaceKeywords,
+  shouldThrottle,
+  updateLastReminder,
+  THROTTLE_INTERVAL
+};

package/plugin/scripts/shared/config.cjs

@@ -0,0 +1,65 @@
+/**
+ * 共享配置
+ */
+
+const fs = require('node:fs');
+const path = require('node:path');
+const os = require('node:os');
+
+// 判断是否为开发模式（通过环境变量）
+const IS_DEV_ENV = process.env.NODE_ENV === 'development' || process.env.ZERO_DEV === 'true';
+
+// 两个可能的目录路径
+const ZERO_HOME_DEV = path.join(os.homedir(), '.zero-workspace-dev');
+const ZERO_HOME_PROD = path.join(os.homedir(), '.zero-workspace');
+
+/**
+ * 自动检测实际使用的目录
+ * 优先检查 dev 目录是否存在绑定文件，如果存在则使用 dev 目录
+ * 这解决了 Hook 脚本没有 ZERO_DEV 环境变量但需要访问 dev 数据的问题
+ */
+function detectZeroHome() {
+  // 如果环境变量明确指定了 dev 模式，直接使用
+  if (IS_DEV_ENV) {
+    return ZERO_HOME_DEV;
+  }
+
+  // 否则，检查 dev 目录是否存在且有绑定文件（说明服务在 dev 模式运行）
+  const devBindingsPath = path.join(ZERO_HOME_DEV, 'session-bindings.json');
+  if (fs.existsSync(devBindingsPath)) {
+    return ZERO_HOME_DEV;
+  }
+
+  // 默认使用生产目录
+  return ZERO_HOME_PROD;
+}
+
+const ZERO_HOME = detectZeroHome();
+const IS_DEV = ZERO_HOME === ZERO_HOME_DEV;
+const DIR_SUFFIX = IS_DEV ? '-dev' : '';
+
+// 路径配置
+const BINDINGS_PATH = path.join(ZERO_HOME, 'session-bindings.json');
+const INDEX_PATH = path.join(ZERO_HOME, 'index.json');
+
+// HTTP 端口
+const HTTP_PORT = process.env.HTTP_PORT || process.env.PORT || (IS_DEV ? '3001' : '3000');
+const MCP_URL = `http://localhost:${HTTP_PORT}`;
+
+// 工作区相关关键词（用于检测用户消息是否涉及工作区）
+const WORKSPACE_KEYWORDS = [
+  '工作区', 'workspace', '任务', '节点', 'node',
+  '继续', 'continue', '进度', 'progress',
+  'zero', 'session_bind', 'workspace_'
+];
+
+module.exports = {
+  IS_DEV,
+  DIR_SUFFIX,
+  ZERO_HOME,
+  BINDINGS_PATH,
+  INDEX_PATH,
+  HTTP_PORT,
+  MCP_URL,
+  WORKSPACE_KEYWORDS
+};

package/plugin/scripts/shared/context.cjs

@@ -0,0 +1,120 @@
+/**
+ * 上下文生成逻辑
+ */
+
+/**
+ * 生成工作区上下文注入内容
+ * @param {object} binding - 会话绑定信息
+ * @param {object} config - 工作区配置
+ * @param {object} workspaceMdData - 工作区 Markdown 数据
+ * @param {object} graph - 节点图
+ * @param {object} focusedNodeInfo - 聚焦节点信息
+ * @returns {string} 上下文内容
+ */
+function generateWorkspaceContext(binding, config, workspaceMdData, graph, focusedNodeInfo) {
+  let context = `
+<zero-workspace-context>
+## 当前工作区: ${config.name}
+
+**目标**: ${workspaceMdData.goal}
+`;
+
+  // 添加规则
+  if (workspaceMdData.rules && workspaceMdData.rules.length > 0) {
+    context += `
+**规则** (必须遵守):
+${workspaceMdData.rules.map(r => `- ${r}`).join('\n')}
+`;
+  }
+
+  // 添加聚焦节点信息（优先使用 graph.currentFocus 作为权威来源）
+  const focusNodeId = graph?.currentFocus || binding.focusedNodeId;
+  if (focusNodeId && focusedNodeInfo) {
+    const nodeStatus = graph?.nodes[focusNodeId]?.status || focusedNodeInfo.status;
+    context += `
+**当前聚焦节点**: ${focusedNodeInfo.title}
+- 节点 ID: ${focusNodeId}
+- 状态: ${nodeStatus}
+${focusedNodeInfo.requirement ? `- 需求: ${focusedNodeInfo.requirement}` : ''}
+`;
+  }
+
+  context += `
+---
+**工作流程提醒**:
+- 开始任务前: 调用 node_transition(action='start')
+- 完成任务后: 调用 node_transition(action='complete', conclusion='...')
+- 重要事件时: 调用 log_append 记录
+- 遇到问题时: 调用 problem_update 记录
+</zero-workspace-context>
+`;
+
+  return context;
+}
+
+/**
+ * 生成 sessionId 注入内容（让 AI 知道自己的会话 ID）
+ * @param {string} sessionId - 会话 ID
+ * @param {string} platform - 平台标识 ('claude-code' | 'cursor')
+ * @returns {string} 注入内容
+ */
+function generateSessionIdContext(sessionId, platform = 'claude-code') {
+  const idName = platform === 'cursor' ? 'conversation_id' : 'sessionId';
+  const paramName = platform === 'cursor' ? 'sessionId' : 'sessionId';
+
+  return `<zero-session-info>
+当前会话 ID: ${sessionId}
+如需使用 ZeroWorkspace 管理任务，请调用: session_bind(${paramName}="${sessionId}", workspaceId="...")
+可用 session_status(${paramName}="${sessionId}") 查看可用工作区列表。
+</zero-session-info>`;
+}
+
+/**
+ * 生成绑定提醒内容（未绑定但检测到工作区关键词时）
+ * @param {string} sessionId - 会话 ID
+ * @param {string} platform - 平台标识 ('claude-code' | 'cursor')
+ * @returns {string} 提醒内容
+ */
+function generateBindingReminder(sessionId, platform = 'claude-code') {
+  const paramName = platform === 'cursor' ? 'sessionId' : 'sessionId';
+
+  return `<zero-workspace-reminder>
+检测到可能涉及工作区任务，但当前会话未绑定工作区。
+如需使用 ZeroWorkspace 功能，请先绑定：
+- 查看可用工作区: session_status(${paramName}="${sessionId}")
+- 绑定工作区: session_bind(${paramName}="${sessionId}", workspaceId="...")
+</zero-workspace-reminder>`;
+}
+
+/**
+ * 获取完整的工作区上下文（组合调用）
+ * @param {object} binding - 会话绑定信息
+ * @returns {string|null} 上下文内容或 null
+ */
+function getFullWorkspaceContext(binding) {
+  const { getWorkspaceConfig, getWorkspaceMdData, getNodeGraph, getNodeInfo } = require('./workspace.cjs');
+
+  const config = getWorkspaceConfig(binding.workspaceId);
+  if (!config) {
+    return null;
+  }
+
+  const workspaceMdData = getWorkspaceMdData(binding.workspaceId);
+  const graph = getNodeGraph(binding.workspaceId);
+  // 优先使用 graph.currentFocus 作为权威来源
+  const focusNodeId = graph?.currentFocus || binding.focusedNodeId;
+  const focusedNodeInfo = focusNodeId ? getNodeInfo(binding.workspaceId, focusNodeId) : null;
+
+  if (workspaceMdData) {
+    return generateWorkspaceContext(binding, config, workspaceMdData, graph, focusedNodeInfo);
+  }
+
+  return null;
+}
+
+module.exports = {
+  generateWorkspaceContext,
+  generateSessionIdContext,
+  generateBindingReminder,
+  getFullWorkspaceContext
+};

package/plugin/scripts/shared/index.cjs

@@ -0,0 +1,34 @@
+/**
+ * 共享模块统一导出
+ */
+
+const config = require('./config.cjs');
+const utils = require('./utils.cjs');
+const binding = require('./binding.cjs');
+const workspace = require('./workspace.cjs');
+const context = require('./context.cjs');
+const reminder = require('./reminder.cjs');
+const logger = require('./logger.cjs');
+
+module.exports = {
+  // 配置
+  ...config,
+
+  // 工具函数
+  ...utils,
+
+  // 绑定逻辑
+  ...binding,
+
+  // 工作区数据
+  ...workspace,
+
+  // 上下文生成
+  ...context,
+
+  // 智能提醒
+  ...reminder,
+
+  // 日志
+  ...logger
+};

package/plugin/scripts/shared/logger.cjs

@@ -0,0 +1,196 @@
+/**
+ * Zero-Workspace 日志模块
+ *
+ * 功能：
+ * - 仅在 dev 模式下记录日志
+ * - 按会话 ID 分文件记录
+ * - 自动轮转，最多保留 50 个日志文件
+ */
+
+const fs = require('node:fs');
+const path = require('node:path');
+const { IS_DEV, ZERO_HOME } = require('./config.cjs');
+
+// 日志目录
+const LOGS_DIR = path.join(ZERO_HOME, 'logs');
+
+// 最大日志文件数
+const MAX_LOG_FILES = 50;
+
+/**
+ * 确保日志目录存在
+ */
+function ensureLogsDir() {
+  if (!IS_DEV) return false;
+
+  try {
+    if (!fs.existsSync(LOGS_DIR)) {
+      fs.mkdirSync(LOGS_DIR, { recursive: true });
+    }
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+/**
+ * 获取日志文件路径
+ * @param {string} sessionId - 会话 ID
+ * @returns {string} 日志文件路径
+ */
+function getLogFilePath(sessionId) {
+  const fileName = sessionId ? `${sessionId}.log` : 'default.log';
+  return path.join(LOGS_DIR, fileName);
+}
+
+/**
+ * 格式化时间戳
+ * @returns {string} 格式化的时间戳
+ */
+function formatTimestamp() {
+  return new Date().toISOString();
+}
+
+/**
+ * 清理旧日志文件，保留最新的 MAX_LOG_FILES 个
+ */
+function rotateLogsIfNeeded() {
+  if (!IS_DEV) return;
+
+  try {
+    if (!fs.existsSync(LOGS_DIR)) return;
+
+    const files = fs.readdirSync(LOGS_DIR)
+      .filter(f => f.endsWith('.log'))
+      .map(f => ({
+        name: f,
+        path: path.join(LOGS_DIR, f),
+        mtime: fs.statSync(path.join(LOGS_DIR, f)).mtime.getTime()
+      }))
+      .sort((a, b) => b.mtime - a.mtime); // 按修改时间降序
+
+    // 删除超出限制的旧文件
+    if (files.length > MAX_LOG_FILES) {
+      const toDelete = files.slice(MAX_LOG_FILES);
+      for (const file of toDelete) {
+        try {
+          fs.unlinkSync(file.path);
+        } catch {
+          // 忽略删除失败
+        }
+      }
+    }
+  } catch {
+    // 忽略轮转错误
+  }
+}
+
+/**
+ * 写入日志
+ * @param {string} sessionId - 会话 ID（可选）
+ * @param {string} type - 日志类型 (hook/mcp)
+ * @param {string} event - 事件名称
+ * @param {object} data - 日志数据
+ */
+function log(sessionId, type, event, data = {}) {
+  if (!IS_DEV) return;
+
+  if (!ensureLogsDir()) return;
+
+  try {
+    const logPath = getLogFilePath(sessionId);
+    const timestamp = formatTimestamp();
+
+    const logLine = `[${timestamp}] [${type.toUpperCase()}] ${event}: ${JSON.stringify(data)}\n`;
+
+    fs.appendFileSync(logPath, logLine, 'utf-8');
+
+    // 每次写入后检查是否需要轮转
+    rotateLogsIfNeeded();
+  } catch {
+    // 静默失败，不影响主流程
+  }
+}
+
+/**
+ * 记录 Hook 调用
+ * @param {string} sessionId - 会话 ID
+ * @param {string} hookEvent - Hook 事件类型
+ * @param {object} details - 详细信息
+ */
+function logHook(sessionId, hookEvent, details = {}) {
+  log(sessionId, 'hook', hookEvent, details);
+}
+
+/**
+ * 记录 Hook 完整输出（包含发送给 AI 的内容）
+ * @param {string} sessionId - 会话 ID
+ * @param {string} hookEvent - Hook 事件类型
+ * @param {string} action - 动作类型 (output|throttled|silent)
+ * @param {object} details - 详细信息
+ * @param {string} [output] - 实际输出给 AI 的内容（截断到 500 字符）
+ */
+function logHookOutput(sessionId, hookEvent, action, details = {}, output = null) {
+  const logData = {
+    action,
+    ...details
+  };
+
+  if (output && typeof output === 'string') {
+    // 截断输出内容，避免日志过大
+    logData.output = output.length > 500 ? output.slice(0, 500) + '...[truncated]' : output;
+  }
+
+  log(sessionId, 'hook', `${hookEvent}:${action}`, logData);
+}
+
+/**
+ * 记录 MCP 工具调用开始
+ * @param {string} sessionId - 会话 ID
+ * @param {string} toolName - 工具名称
+ * @param {object} params - 调用参数
+ */
+function logMcpStart(sessionId, toolName, params = {}) {
+  log(sessionId, 'mcp', `${toolName}:start`, { params });
+}
+
+/**
+ * 记录 MCP 工具调用结束
+ * @param {string} sessionId - 会话 ID
+ * @param {string} toolName - 工具名称
+ * @param {boolean} success - 是否成功
+ * @param {any} result - 调用结果（截断）
+ * @param {number} durationMs - 耗时（毫秒）
+ */
+function logMcpEnd(sessionId, toolName, success, result, durationMs = 0) {
+  const resultStr = typeof result === 'object' ? JSON.stringify(result) : String(result);
+  const details = {
+    success,
+    result: resultStr.length > 500 ? resultStr.slice(0, 500) + '...' : resultStr,
+    durationMs
+  };
+  log(sessionId, 'mcp', `${toolName}:end`, details);
+}
+
+/**
+ * 记录错误
+ * @param {string} sessionId - 会话 ID
+ * @param {string} context - 错误上下文
+ * @param {Error|string} error - 错误信息
+ */
+function logError(sessionId, context, error) {
+  const errorMsg = error instanceof Error ? error.message : String(error);
+  log(sessionId, 'error', context, { error: errorMsg });
+}
+
+module.exports = {
+  LOGS_DIR,
+  MAX_LOG_FILES,
+  log,
+  logHook,
+  logHookOutput,
+  logMcpStart,
+  logMcpEnd,
+  logError,
+  rotateLogsIfNeeded
+};