whistle.mockbubu 2.0.0 → 2.1.0

package/lib/uiServer/router/files/file-save-router.js

@@ -0,0 +1,146 @@
+/**
+ * 文件名: file-save-router.js
+ * 功能: 首次保存路由（前端缓存 → 文件系统）
+ * 依赖: storage-adapter, group-manager
+ *
+ * 职责:
+ * - 接收前端发送的完整会话数据
+ * - 写入 V3 文件系统 (file.json + source version)
+ * - 创建/更新文件配置 (Layer 3 - properties)
+ *
+ * 路由:
+ * - POST /cgi-bin/mockbubu/file-save - 首次保存文件
+ */
+
+const { LOG_PREFIX } = require('../../../constants')
+const { createSuccessResponse, createErrorResponse } = require('../../utils/util')
+const { wrapRouteHandler } = require('../../utils/router-helpers')
+
+module.exports = (router) => {
+  /**
+   * 首次保存文件（前端缓存 → 文件系统）
+   *
+   * 功能:
+   * - 接收前端传递的完整会话数据
+   * - 写入 V3 文件存储（file.json + versions/source.json）
+   * - 创建文件配置（Layer 3 - properties）
+   *
+   * 请求体:
+   * ```javascript
+   * {
+   *   url: 'https://api.example.com/users',
+   *   method: 'GET',
+   *   status: 200,
+   *   session: {...}  // 完整会话数据
+   * }
+   * ```
+   *
+   * 响应:
+   * ```javascript
+   * {
+   *   code: 0,
+   *   data: { success: true },
+   *   msg: 'success'
+   * }
+   * ```
+   */
+  router.post('/cgi-bin/mockbubu/file-save', wrapRouteHandler(async (ctx) => {
+    console.log(`${LOG_PREFIX.FILE_SAVE} ========== 路由被调用 ==========`)
+    console.log(`${LOG_PREFIX.FILE_SAVE} 请求体:`, JSON.stringify(ctx.request.body, null, 2))
+
+    const { storageAdapter, groupManager } = ctx
+    const { url, method, status, session, config } = ctx.request.body
+
+    // 参数校验
+    if (!url || !method || !status || !session) {
+      console.error(`${LOG_PREFIX.FILE_SAVE} 参数不完整`, {
+        hasUrl: !!url,
+        hasMethod: !!method,
+        hasStatus: !!status,
+        hasSession: !!session,
+      })
+      ctx.body = createErrorResponse('参数不完整')
+      return
+    }
+
+    console.log(`${LOG_PREFIX.FILE_SAVE} 开始保存文件 | URL: ${url}`, config ? `| 配置: mock=${config.mock}, locked=${config.locked}` : '')
+
+    try {
+      const currentGroupId = await groupManager.getCurrentGroupId()
+      const filename = url // 使用 URL 作为 filename
+      const v3Storage = storageAdapter.v3Storage
+
+      // 检查文件是否已存在（通过索引文件）
+      const indexData = await v3Storage.getIndex(currentGroupId)
+      const existingFileEntry = indexData.files.find(f => f.url === filename)
+
+      console.log(`${LOG_PREFIX.FILE_SAVE} 检查文件是否存在: ${filename}, 结果:`, existingFileEntry ? '已存在' : '不存在')
+
+      // 如果文件已存在（在索引中找到了）
+      if (existingFileEntry) {
+        console.log(`${LOG_PREFIX.FILE_SAVE} 文件已存在，跳过保存: ${filename}`)
+        ctx.body = createSuccessResponse({ success: true, skipped: true })
+        return
+      }
+
+      // 生成新的文件ID
+      const newFileId = v3Storage.generateFileId()
+      console.log(`${LOG_PREFIX.FILE_SAVE} 生成文件ID: ${newFileId}`)
+
+      // 使用 V3 Storage 的 createFile 方法创建文件
+      console.log(`${LOG_PREFIX.FILE_SAVE} 准备调用 v3Storage.createFile, groupId=${currentGroupId}, fileId=${newFileId}`)
+      const fileData = {
+        url: filename,
+        method,
+        status,
+        headers: session.res?.headers || {},
+        session, // 完整 session 数据
+        config: {
+          mock: config?.mock || false,
+          locked: config?.locked || false,
+          mockVersion: 'source', // 新文件默认使用 source 版本
+          mockTime: config?.mock ? Date.now() : null,
+        },
+      }
+      console.log(`${LOG_PREFIX.FILE_SAVE} fileData keys:`, Object.keys(fileData))
+
+      await v3Storage.createFile(currentGroupId, newFileId, fileData)
+
+      console.log(`${LOG_PREFIX.FILE_SAVE} ✅ V3 Storage createFile 执行完成: ${filename}`)
+
+      // 注意：session 数据已保存在 file.json 中，无需创建 versions/source.json
+      // 所有读取操作都从 file.json 的 session 字段读取
+
+      // ⚠️ 关键修复:再次调用 setGroupFileConfig 确保 config 正确写入
+      const configToSet = {
+        mock: config?.mock !== undefined ? config.mock : false,
+        locked: config?.locked !== undefined ? config.locked : false,
+        mockVersion: config?.mock ? 'source' : null,
+        mockTime: config?.mock ? Date.now() : null,
+      }
+      console.log(`${LOG_PREFIX.FILE_SAVE} 准备更新配置:`, configToSet)
+      await groupManager.setGroupFileConfig(currentGroupId, filename, configToSet)
+      console.log(`${LOG_PREFIX.FILE_SAVE} ✅ 配置更新完成`)
+
+      // 返回保存后的完整配置（供前端更新使用）
+      ctx.body = createSuccessResponse({
+        success: true,
+        file: {
+          id: newFileId,
+          url: filename,
+          method,
+          status,
+          date: Date.now(),
+          ...configToSet,
+          rule: session.req?.rule || '',
+          pattern: session.req?.pattern || '',
+          ruleValue: session.req?.ruleValue || 'pathname',
+          name: filename,
+        },
+      })
+    } catch (error) {
+      console.error(`${LOG_PREFIX.FILE_SAVE} 保存失败:`, error)
+      ctx.body = createErrorResponse(`保存失败: ${error.message}`)
+    }
+  }, '首次保存文件'))
+}

package/lib/uiServer/router/files/version-router.js

@@ -0,0 +1,260 @@
+/**
+ * 文件名: version-router.js
+ * 功能: 版本管理路由（创建、删除、更新版本）
+ * 依赖: router-helpers.js, util.js, validator.js
+ */
+
+const { createSuccessResponse } = require('../../utils/util')
+const {
+  validateFilename,
+  validateVersionName,
+  validateMockData,
+  validate,
+} = require('../../utils/validator')
+const {
+  getFileIdByUrl,
+  wrapRouteHandler,
+  validateRequiredParams,
+} = require('../../utils/router-helpers')
+
+module.exports = (router) => {
+  // 新增版本
+  router.post('/cgi-bin/mockbubu/add-new-version', validate({
+    versionName: validateVersionName,
+    content: validateMockData,
+  }), wrapRouteHandler(async (ctx) => {
+    const { storageAdapter, groupManager } = ctx
+    const { versionName, content = {}, fileId, description = '' } = ctx.request.body || {}
+
+    validateRequiredParams({ fileId }, ['fileId'])
+
+    const currentGroupId = await groupManager.getCurrentGroupId()
+    const v3Storage = storageAdapter.v3Storage
+
+    const newVersion = await v3Storage.createVersion(currentGroupId, fileId, versionName, {
+      content,
+      description,
+    })
+
+    ctx.body = createSuccessResponse(newVersion, '版本创建成功')
+  }, '创建版本'))
+
+  // 删除版本
+  router.post('/cgi-bin/mockbubu/delete-version', wrapRouteHandler(async (ctx) => {
+    const { storageAdapter, groupManager } = ctx
+    const { versionId, fileId, url } = ctx.request.body || {}
+
+    validateRequiredParams({ versionId, fileId, url }, ['versionId', 'fileId', 'url'])
+
+    const currentGroupId = await groupManager.getCurrentGroupId()
+    const v3Storage = storageAdapter.v3Storage
+
+    // 获取文件配置(检查是否是当前mock版本)
+    const fileConfig = await groupManager.getGroupFileConfig(currentGroupId, url)
+    const version = await v3Storage.getVersionById(currentGroupId, fileId, versionId)
+
+    if (version && version.name === fileConfig.mockVersion) {
+      // 清除mock版本设置
+      fileConfig.mockVersion = null
+      await groupManager.setGroupFileConfig(currentGroupId, url, fileConfig)
+    }
+
+    await v3Storage.deleteVersion(currentGroupId, fileId, versionId)
+
+    ctx.body = createSuccessResponse(null, '版本删除成功')
+  }, '删除版本'))
+
+  // 更新版本内容
+  router.post('/cgi-bin/mockbubu/update-version-content', validate({
+    content: validateMockData,
+  }), wrapRouteHandler(async (ctx) => {
+    const { storageAdapter, groupManager } = ctx
+    const { versionId, content, fileId } = ctx.request.body || {}
+
+    validateRequiredParams({ versionId, fileId }, ['versionId', 'fileId'])
+
+    const currentGroupId = await groupManager.getCurrentGroupId()
+    const v3Storage = storageAdapter.v3Storage
+
+    await v3Storage.updateVersion(currentGroupId, fileId, versionId, { content })
+
+    ctx.body = createSuccessResponse(null, '版本内容更新成功')
+  }, '更新版本内容'))
+
+  // 获取版本列表
+  router.post('/cgi-bin/mockbubu/get-versions', wrapRouteHandler(async (ctx) => {
+    const { storageAdapter, groupManager } = ctx
+    const { fileId } = ctx.request.body || {}
+
+    validateRequiredParams({ fileId }, ['fileId'])
+
+    const currentGroupId = await groupManager.getCurrentGroupId()
+    const v3Storage = storageAdapter.v3Storage
+
+    // 获取文件数据（用于构建 source）
+    const fileData = await v3Storage.getFile(currentGroupId, fileId)
+    if (!fileData) {
+      throw new Error('文件不存在')
+    }
+
+    // 获取历史版本列表
+    const versions = await v3Storage.listVersions(currentGroupId, fileId)
+    console.log(`[version-router] get-versions: fileId=${fileId}, versions count=${versions.length}`)
+
+    // 构建统一数据结构
+    const result = []
+
+    // 1. 添加 source 数据
+    // source 数据存储在 file.json 的 session.res.body 字段中
+    // 这是捕获请求时保存的原始响应数据，用于版本对比和回退
+    let sourceContent = {}
+
+    if (fileData.session && fileData.session.res && fileData.session.res.body) {
+      try {
+        const bodyData = fileData.session.res.body
+
+        if (typeof bodyData === 'string') {
+          try {
+            sourceContent = JSON.parse(bodyData)
+          } catch (parseError) {
+            // 不是 JSON 格式（可能是 HTML/图片/文本），使用空对象
+            console.warn('[version-router] source 数据不是有效的 JSON，使用空对象')
+            sourceContent = {}
+          }
+        } else if (bodyData && typeof bodyData === 'object') {
+          sourceContent = bodyData
+        } else {
+          // 其他情况（如 undefined/null），使用空对象
+          sourceContent = {}
+        }
+      } catch (error) {
+        console.error('[version-router] source 数据解析失败，使用空对象:', error)
+        sourceContent = {}
+      }
+    } else {
+      console.warn('[version-router] file.json 中缺少 session.res.body 数据，使用空对象')
+      sourceContent = {}
+    }
+
+    result.push({
+      type: 'source',
+      name: 'source',
+      content: sourceContent,
+      updateTime: fileData.updateTime || fileData.createTime,
+    })
+
+    // 2. 添加历史版本数据
+    // 过滤掉名称为 "source" 的版本（避免与 source 数据冲突）
+    versions
+      .filter(version => version.name !== 'source')
+      .forEach((version) => {
+        let versionContent = {}
+        try {
+          // 如果 content 是字符串则解析为 JSON
+          versionContent = typeof version.content === 'string'
+            ? JSON.parse(version.content)
+            : (version.content || {})
+        } catch (error) {
+          console.error(`[version-router] 版本 ${version.name} 数据解析失败:`, error)
+          versionContent = {}
+        }
+
+        result.push({
+          type: 'history',
+          name: version.name,
+          content: versionContent,
+          updateTime: version.updateTime,
+          // 保留其他有用字段
+          id: version.id,
+          description: version.description,
+          createTime: version.createTime,
+        })
+      })
+
+    console.log(`[version-router] get-versions: 返回 ${result.length} 条数据`)
+    ctx.body = createSuccessResponse(result, '获取版本列表成功')
+  }, '获取版本列表'))
+
+  // 设置mock的版本
+  router.post('/cgi-bin/mockbubu/set-mock-version', wrapRouteHandler(async (ctx) => {
+    const { groupManager } = ctx
+    const { url, versionName } = ctx.request.body || {}
+
+    validateRequiredParams({ url }, ['url'])
+
+    const currentGroupId = await groupManager.getCurrentGroupId()
+    const fileConfig = await groupManager.getGroupFileConfig(currentGroupId, url)
+
+    // 更新 mockVersion (使用版本name,不是ID)
+    fileConfig.mockVersion = versionName || null
+    await groupManager.setGroupFileConfig(currentGroupId, url, fileConfig)
+
+    ctx.body = createSuccessResponse(null, 'Mock版本设置成功')
+  }, '设置Mock版本'))
+
+  // 更新版本元信息（名称和描述）
+  router.post('/cgi-bin/mockbubu/update-version-meta', validate({
+    newVersionName: validateVersionName,
+  }), wrapRouteHandler(async (ctx) => {
+    const { storageAdapter, groupManager } = ctx
+    const { url, fileId, versionId, newVersionName, description = '' } = ctx.request.body || {}
+
+    validateRequiredParams({ url, fileId, versionId }, ['url', 'fileId', 'versionId'])
+
+    const currentGroupId = await groupManager.getCurrentGroupId()
+    const v3Storage = storageAdapter.v3Storage
+
+    // 获取旧版本信息
+    const oldVersion = await v3Storage.getVersionById(currentGroupId, fileId, versionId)
+    if (!oldVersion) {
+      throw new Error('版本不存在')
+    }
+
+    // 更新版本元信息
+    await v3Storage.updateVersion(currentGroupId, fileId, versionId, {
+      name: newVersionName,
+      description,
+    })
+
+    // 如果更新的是当前mock版本的名称,需要同步更新 mockVersion
+    const fileConfig = await groupManager.getGroupFileConfig(currentGroupId, url)
+    if (oldVersion.name === fileConfig.mockVersion && oldVersion.name !== newVersionName) {
+      fileConfig.mockVersion = newVersionName
+      await groupManager.setGroupFileConfig(currentGroupId, url, fileConfig)
+    }
+
+    ctx.body = createSuccessResponse(null, '版本信息更新成功')
+  }, '更新版本元信息'))
+
+  // 兼容旧路由：更新版本名称（已废弃）
+  router.post('/cgi-bin/mockbubu/update-version-name', validate({
+    name: validateFilename,
+    versionName: validateVersionName,
+    newVersion: validateVersionName,
+  }), wrapRouteHandler(async (ctx) => {
+    const { groupManager } = ctx
+    const { versionName, name, newVersion } = ctx.request.body || {}
+
+    const { fileId, currentGroupId, v3Storage } = await getFileIdByUrl(ctx, name)
+
+    // 使用 V3 Storage 通过名称找到版本ID
+    const version = await v3Storage.getVersionByName(currentGroupId, fileId, versionName)
+    if (!version) {
+      throw new Error('版本不存在')
+    }
+
+    // 更新版本名称
+    await v3Storage.updateVersion(currentGroupId, fileId, version.id, {
+      name: newVersion,
+    })
+
+    // 如果更新的是当前mock版本,需要更新mockVersion属性
+    const fileConfig = await groupManager.getGroupFileConfig(currentGroupId, name)
+    if (versionName === fileConfig.mockVersion && versionName !== newVersion) {
+      fileConfig.mockVersion = newVersion
+      await groupManager.setGroupFileConfig(currentGroupId, name, fileConfig)
+    }
+
+    ctx.body = createSuccessResponse(null, '版本名称更新成功')
+  }, '更新版本名称'))
+}

package/lib/uiServer/router/global/plugin-control.js

@@ -0,0 +1,135 @@
+/**
+ * 插件控制路由
+ * 管理插件的工作模式
+ */
+
+const { PLUGIN_MODE_KEY, PLUGIN_MODE, DEFAULT_PLUGIN_MODE } = require('../../../config/const')
+const pluginModeManager = require('../../../core/plugin-mode-manager')
+
+module.exports = (router, options) => {
+  const storage = options.storage
+
+  /**
+   * 获取插件工作模式
+   */
+  router.post('/cgi-bin/mockbubu/get-plugin-mode', async (ctx) => {
+    try {
+      // 从模式管理器获取当前模式（内存缓存，与请求拦截器保持一致）
+      const mode = pluginModeManager.getMode()
+
+      ctx.body = {
+        success: true,
+        mode,
+      }
+    } catch (error) {
+      console.error('[mockbubu] ❌ 获取插件模式失败:', error)
+      ctx.body = {
+        success: false,
+        message: error.message,
+      }
+    }
+  })
+
+  /**
+   * 兼容旧接口：获取插件启用状态
+   * @deprecated 使用 get-plugin-mode 替代
+   */
+  router.post('/cgi-bin/mockbubu/get-plugin-enabled', async (ctx) => {
+    try {
+      const mode = await storage.getProperty(PLUGIN_MODE_KEY) || DEFAULT_PLUGIN_MODE
+      // 任何模式都算"启用"，因为都会处理 Mock
+      const enabled = true
+
+      ctx.body = {
+        success: true,
+        enabled,
+      }
+    } catch (error) {
+      console.error('[mockbubu] ❌ 获取插件状态失败:', error)
+      ctx.body = {
+        success: false,
+        message: error.message,
+      }
+    }
+  })
+
+  /**
+   * 切换插件工作模式
+   */
+  router.post('/cgi-bin/mockbubu/set-plugin-mode', async (ctx) => {
+    try {
+      const { mode } = ctx.request.body
+
+      // 验证模式参数
+      if (!Object.values(PLUGIN_MODE).includes(mode)) {
+        ctx.body = {
+          success: false,
+          message: `mode 参数必须是 ${Object.values(PLUGIN_MODE).join(' 或 ')}`,
+        }
+        return
+      }
+
+      // 使用模式管理器更新模式（会同时更新 storage 和内存缓存）
+      await pluginModeManager.setMode(mode)
+
+      // 通知 Whistle 更新规则（触发 rulesServer 钩子重新执行）
+      if (typeof storage.updateRules === 'function') {
+        storage.updateRules()
+      }
+
+      const modeNames = {
+        [PLUGIN_MODE.DEFAULT]: '默认模式',
+        [PLUGIN_MODE.MOCK_ONLY]: '仅Mock模式',
+      }
+      console.log(`[mockbubu] ✅ 已切换到: ${modeNames[mode]}`)
+
+      ctx.body = {
+        success: true,
+        mode,
+      }
+    } catch (error) {
+      console.error('[mockbubu] ❌ 切换插件模式失败:', error)
+      ctx.body = {
+        success: false,
+        message: error.message,
+      }
+    }
+  })
+
+  /**
+   * 兼容旧接口：切换插件启用/禁用状态
+   * @deprecated 使用 set-plugin-mode 替代
+   */
+  router.post('/cgi-bin/mockbubu/toggle-plugin-enabled', async (ctx) => {
+    try {
+      const { enabled } = ctx.request.body
+
+      // 启用 → capture 模式，禁用 → 不支持（返回错误）
+      if (!enabled) {
+        ctx.body = {
+          success: false,
+          message: '该功能已废弃，请使用模式切换功能',
+        }
+        return
+      }
+
+      // 启用 → 设置为 default 模式
+      await storage.setProperty(PLUGIN_MODE_KEY, PLUGIN_MODE.DEFAULT)
+
+      if (typeof storage.updateRules === 'function') {
+        storage.updateRules()
+      }
+
+      ctx.body = {
+        success: true,
+        enabled: true,
+      }
+    } catch (error) {
+      console.error('[mockbubu] ❌ 切换插件状态失败:', error)
+      ctx.body = {
+        success: false,
+        message: error.message,
+      }
+    }
+  })
+}