whistle.mockbubu 1.0.0-dev.4 → 2.0.0

package/lib/server.js

@@ -1,17 +1,22 @@
 const {
-  getProperty,
   setProperty,
   getFilename,
   getRule,
   isJsonReq,
   setApiListUpdated,
-  getVersionContent,
   handleBuffer2String,
   withTryCatch,
 } = require('./utils')
 const qs = require('qs')
+const GroupManager = require('./group-manager')
 
 module.exports = (server, { storage }) => {
+  // 初始化组管理器
+  const groupManager = new GroupManager(storage)
+
+  // 首次启动时迁移现有数据到默认组
+  groupManager.migrateExistingData()
+
   server.on('request', (req, res) => {
     const { originalReq, method } = req
     const { headers, ruleValue, url, pattern } = originalReq
@@ -19,9 +24,13 @@ module.exports = (server, { storage }) => {
     const rule = getRule(originalReq)
 
     // 非json请求直接透传
-    if (!isJsonReq(headers)) return req.passThrough()
+    if (!isJsonReq(headers)) {
+      console.log(`[mockbubu] ⏭️  非 JSON 请求已跳过 | File: ${filename}`)
+      console.log(`[mockbubu] 📋 Headers: Accept=${headers.accept}, Sec-Fetch-Dest=${headers['sec-fetch-dest'] || headers['Sec-Fetch-Dest']}`)
+      return req.passThrough()
+    }
 
-    // 获取请求信息
+    // 获取请求信息 - 这些写入到全局 properties
     let str = ''
     req.on('data', (chunk) => {
       str += chunk
@@ -34,30 +43,133 @@ module.exports = (server, { storage }) => {
     })
 
     try {
-      const { mock, mockVersion } = getProperty(storage, filename) || {}
-      const sessionCache = storage.readFile(filename)
+      // 获取当前激活的组ID
+      const currentGroupId = groupManager.getCurrentGroupId()
+      // 读取该组的配置
+      const groupConfig = groupManager.getGroupFileConfig(currentGroupId, filename)
+      const { mock, mockVersion } = groupConfig
 
-      if (mock) {
-        if (sessionCache) {
-          const resCache = JSON.parse(sessionCache)?.res
-          const { statusCode, statusMessage, headers } = resCache
-
-          headers['from-res-cache'] = 'true'
-          delete headers['content-encoding']
-          delete headers['content-length']
-          res.writeHead(statusCode, statusMessage, headers)
-          if (mockVersion) {
-            const mockVersionContent = getVersionContent({ storage, filename, versionName: mockVersion })
+      // 添加调试日志 - 使用一行显示关键信息
+      console.log(`[mockbubu] 🔍 REQUEST | File: ${filename} | Group: ${currentGroupId} | Mock: ${mock} | Version: ${mockVersion || 'source'}`)
+      console.log('[mockbubu] 📝 Full Config:', JSON.stringify(groupConfig, null, 2))
+
+      // 从组目录读取缓存文件
+      const filePath = `${currentGroupId}/${filename}`
+      const sessionCache = storage.readFile(filePath)
+      console.log(`[mockbubu] 💾 Cache Status: ${sessionCache ? 'EXISTS' : 'NO CACHE'} | Path: ${filePath}`)
+
+      if (mock && sessionCache) {
+        // Mock: true 且有缓存，返回模拟数据
+        const resCache = JSON.parse(sessionCache)?.res
+        const { statusCode, statusMessage, headers } = resCache
+
+        headers['from-res-cache'] = 'true'
+        delete headers['content-encoding']
+        delete headers['content-length']
+        res.writeHead(statusCode, statusMessage, headers)
+
+        if (mockVersion) {
+          // 从当前组配置读取版本内容
+          console.log(`[mockbubu] ⚡ 使用版本数据 | Version: ${mockVersion}`)
+          const mockVersionContent = groupManager.getGroupVersionContent(currentGroupId, filename, mockVersion)
+          if (mockVersionContent) {
+            console.log(`[mockbubu] ✅ 版本数据存在,返回版本: ${mockVersion}`)
             res.end(JSON.stringify(mockVersionContent))
           } else {
+            // 版本不存在时降级到原始数据
+            console.log(`[mockbubu] ⚠️  版本 ${mockVersion} 不存在,降级到原始数据`)
             res.end(resCache.body)
           }
+        } else {
+          console.log('[mockbubu] ⚡ 使用原始数据 (无版本)')
+          res.end(resCache.body)
         }
+      } else if (mock && !sessionCache) {
+        // Mock: true 但无缓存，需要先捕获数据
+        console.log(`[mockbubu] ⚠️  Mock enabled but no cache for ${filename}, need to capture first`)
+        // 不要 return，继续往下执行捕获逻辑
+        // 无缓存，捕获最新的接口数据
+        console.log(`[mockbubu] 🎣 开始捕获（无缓存）: ${filename}`)
+        const client = req.request((svrRes) => {
+          const encoding = svrRes.headers['content-encoding']
+          let body
+
+          svrRes.on('data', (data) => {
+            body = body ? Buffer.concat([body, data]) : data
+          })
+          svrRes.on('end', withTryCatch(async () => {
+            if (!body) return
+
+            const content = await handleBuffer2String({ body, encoding })
+            // 获取完整的抓包数据，要等待响应完成
+            req.getSession(async (session) => {
+              // 如果设置了 enable://hide 会获取到空数据
+              if (!session) {
+                return
+              }
+
+              // 完全隔离架构：将元数据和配置写入当前组
+              // 检查组配置是否真实存在（而不是默认返回值）
+              const configKey = groupManager.getGroupConfigKey(currentGroupId, filename)
+              const isFirstCapture = !storage.hasProperty(configKey)
+
+              if (isFirstCapture) {
+                // 首次捕获：创建完整配置（保持当前 mock 状态）
+                groupManager.setGroupFileConfig(currentGroupId, filename, {
+                  method,
+                  rule,
+                  status: session.res.statusCode,
+                  pattern,
+                  ruleValue: ruleValue || 'pathname',
+                  url,
+                  date: Date.now(),
+                  mock, // 使用当前的 mock 状态（true）
+                  locked: false,
+                  mockVersion: null,
+                  mockTime: null,
+                })
+                console.log(`[mockbubu] ✅ 首次捕获文件到组 ${currentGroupId}: ${filename} | Mock: ${mock}`)
+              } else {
+                // 已存在配置，仅更新元数据（不覆盖 mock、locked 等用户配置）
+                const existingGroupConfig = groupManager.getGroupFileConfig(currentGroupId, filename)
+                groupManager.setGroupFileConfig(currentGroupId, filename, {
+                  ...existingGroupConfig,
+                  method,
+                  rule,
+                  status: session.res.statusCode,
+                  pattern,
+                  ruleValue: ruleValue || 'pathname',
+                  url,
+                  date: Date.now(),
+                })
+                console.log(`[mockbubu] 🔄 更新文件元数据到组 ${currentGroupId}: ${filename}`)
+              }
+
+              // 将文件写入组目录
+              const tempSession = JSON.parse(JSON.stringify(session))
+              tempSession.res.body = content
+              const groupFilePath = `${currentGroupId}/${filename}`
+              storage.writeFile(groupFilePath, JSON.stringify(tempSession))
+              console.log(`[mockbubu] 💾 写入文件: ${groupFilePath}`)
+
+              setApiListUpdated(storage, true)
+            })
+          }))
+
+          // 将响应透传给客户端
+          svrRes.pipe(res)
+        })
+
+        req.pipe(client)
       } else {
-        // 已有缓存则直接透传
-        if (sessionCache) return req.passThrough()
+        // mock: false，已有缓存则直接透传，无缓存则捕获
+        if (sessionCache) {
+          console.log(`[mockbubu] 📤 直接透传（已有缓存）: ${filename}`)
+          return req.passThrough()
+        }
 
-        // 命中插件规则，没有勾选mock的缓存最新的接口数据
+        // 无缓存，捕获最新的接口数据
+        console.log(`[mockbubu] 🎣 开始捕获（无缓存）: ${filename}`)
         const client = req.request((svrRes) => {
           const encoding = svrRes.headers['content-encoding']
           let body
@@ -75,22 +187,50 @@ module.exports = (server, { storage }) => {
               if (!session) {
                 return
               }
-              setProperty(storage, filename, {
-                method,
-                rule,
-                status: session.res.statusCode,
-                pattern,
-                ruleValue: ruleValue || 'pathname',
-                url,
-                mock: false,
-                locked: false,
-                date: Date.now(),
-                mockTime: null,
-              })
 
+              // 完全隔离架构：将元数据和配置写入当前组
+              // 检查组配置是否真实存在（而不是默认返回值）
+              const configKey = groupManager.getGroupConfigKey(currentGroupId, filename)
+              const isFirstCapture = !storage.hasProperty(configKey)
+
+              if (isFirstCapture) {
+                // 首次捕获：创建完整配置
+                groupManager.setGroupFileConfig(currentGroupId, filename, {
+                  method,
+                  rule,
+                  status: session.res.statusCode,
+                  pattern,
+                  ruleValue: ruleValue || 'pathname',
+                  url,
+                  date: Date.now(),
+                  mock: false,
+                  locked: false,
+                  mockVersion: null,
+                  mockTime: null,
+                })
+                console.log(`[mockbubu] ✅ 首次捕获文件到组 ${currentGroupId}: ${filename}`)
+              } else {
+                // 已存在配置，仅更新元数据（不覆盖 mock、locked 等用户配置）
+                const existingGroupConfig = groupManager.getGroupFileConfig(currentGroupId, filename)
+                groupManager.setGroupFileConfig(currentGroupId, filename, {
+                  ...existingGroupConfig,
+                  method,
+                  rule,
+                  status: session.res.statusCode,
+                  pattern,
+                  ruleValue: ruleValue || 'pathname',
+                  url,
+                  date: Date.now(),
+                })
+                console.log(`[mockbubu] 🔄 更新文件元数据到组 ${currentGroupId}: ${filename}`)
+              }
+
+              // 将文件写入组目录
               const tempSession = JSON.parse(JSON.stringify(session))
               tempSession.res.body = content
-              storage.writeFile(filename, JSON.stringify(tempSession))
+              const groupFilePath = `${currentGroupId}/${filename}`
+              storage.writeFile(groupFilePath, JSON.stringify(tempSession))
+              console.log(`[mockbubu] 💾 写入文件: ${groupFilePath}`)
 
               setApiListUpdated(storage, true)
             })

package/lib/uiServer/index.js

@@ -6,10 +6,41 @@ const path = require('path')
 const router = require('koa-router')()
 const setupRouter = require('./router')
 const cors = require('koa2-cors')
+const GroupManager = require('../group-manager')
 const MAX_AGE = 1000 * 60 * 5
 
 module.exports = (server, options) => {
   const app = new Koa()
+  const { storage } = options
+
+  console.log('[mockbubu] 插件开始初始化...')
+
+  // 初始化组管理器
+  let groupManager
+  try {
+    console.log('[mockbubu] 创建 GroupManager...')
+    groupManager = new GroupManager(storage)
+    console.log('[mockbubu] GroupManager 创建成功')
+
+    // 首次启动时迁移现有数据到默认组(已添加大文件保护)
+    console.log('[mockbubu] 开始数据迁移...')
+    groupManager.migrateExistingData()
+    console.log('[mockbubu] 数据迁移完成')
+  } catch (error) {
+    console.error('[mockbubu] 初始化失败:', error)
+    console.error('[mockbubu] 错误堆栈:', error.stack)
+    // 如果初始化失败,创建一个简单的 mock 对象
+    groupManager = {
+      getCurrentGroupId: () => 'default',
+      getCurrentGroup: () => ({ id: 'default', name: '默认组' }),
+      getGroups: () => [{ id: 'default', name: '默认组' }],
+    }
+  }
+
+  // 将 groupManager 挂载到 app.context,让所有中间件都能访问
+  app.context.groupManager = groupManager
+
+  console.log('[mockbubu] 插件初始化完成')
 
   app.proxy = true
   app.silent = true
@@ -17,16 +48,27 @@ module.exports = (server, options) => {
   setupRouter(router, options)
   app.use(
     cors({
-      origin: '*',
+      origin: (ctx) => {
+        const origin = ctx.get('Origin')
+        // 只允许 localhost 和 127.0.0.1 的请求
+        if (!origin || /^https?:\/\/(localhost|127\.0\.0\.1)(:\d+)?$/i.test(origin)) {
+          return origin || '*'
+        }
+        // Whistle UI 的域名
+        if (/^https?:\/\/local\.whistlejs\.com(:\d+)?$/i.test(origin)) {
+          return origin
+        }
+        return false
+      },
       maxAge: 10,
       credentials: true,
       allowMethods: ['GET', 'POST', 'PUT', 'DELETE', 'OPTIONS'],
     }),
   )
   app.use(bodyParser({
-    jsonLimit: '10mb',
-    formLimit: '10mb',
-    textLimit: '10mb',
+    jsonLimit: '100mb',
+    formLimit: '100mb',
+    textLimit: '100mb',
   }))
   app.use(router.routes())
   app.use(router.allowedMethods())

package/lib/uiServer/router/group-router.js

@@ -0,0 +1,218 @@
+const {
+  createErrorResponse,
+  createSuccessResponse,
+} = require('../util')
+const {
+  validate,
+} = require('../validator')
+
+/**
+ * 组管理路由
+ */
+module.exports = (router) => {
+  // 获取组列表
+  router.post('/cgi-bin/mockbubu/groups/list', (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const groups = groupManager.getGroups()
+      const currentGroupId = groupManager.getCurrentGroupId()
+
+      ctx.body = createSuccessResponse({
+        groups,
+        currentGroupId,
+      })
+    } catch (error) {
+      ctx.body = createErrorResponse(error.message)
+    }
+  })
+
+  // 获取当前组
+  router.post('/cgi-bin/mockbubu/groups/current', (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const currentGroup = groupManager.getCurrentGroup()
+
+      ctx.body = createSuccessResponse({
+        currentGroup,
+        currentGroupId: groupManager.getCurrentGroupId(),
+      })
+    } catch (error) {
+      ctx.body = createErrorResponse(error.message)
+    }
+  })
+
+  // 创建组
+  router.post('/cgi-bin/mockbubu/groups/create', validate({
+    name: (val) => {
+      if (!val || !val.trim()) {
+        throw new Error('组名不能为空')
+      }
+      if (val.trim().length > 50) {
+        throw new Error('组名长度不能超过50个字符')
+      }
+      return null
+    },
+    description: (val) => {
+      if (val && val.length > 200) {
+        throw new Error('描述长度不能超过200个字符')
+      }
+      return null
+    },
+    copyFromGroupId: (val) => {
+      // 可选参数,如果提供则必须是字符串
+      if (val !== undefined && val !== null && typeof val !== 'string') {
+        throw new Error('copyFromGroupId 必须是字符串')
+      }
+      return null
+    },
+  }), (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const { name, description, copyFromGroupId } = ctx.request.body
+
+      const newGroup = groupManager.createGroup({
+        name: name.trim(),
+        description: description ? description.trim() : '',
+        copyFromGroupId: copyFromGroupId || null,
+      })
+
+      ctx.body = createSuccessResponse(newGroup, '创建成功')
+    } catch (error) {
+      ctx.body = createErrorResponse(error.message)
+    }
+  })
+
+  // 更新组
+  router.post('/cgi-bin/mockbubu/groups/update', validate({
+    groupId: (val) => {
+      if (!val || typeof val !== 'string') {
+        throw new Error('groupId 必须是字符串')
+      }
+      return null
+    },
+    name: (val) => {
+      if (val !== undefined && val !== null) {
+        if (!val.trim()) {
+          throw new Error('组名不能为空')
+        }
+        if (val.trim().length > 50) {
+          throw new Error('组名长度不能超过50个字符')
+        }
+      }
+      return null
+    },
+    description: (val) => {
+      if (val !== undefined && val !== null && val.length > 200) {
+        throw new Error('描述长度不能超过200个字符')
+      }
+      return null
+    },
+  }), (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const { groupId, name, description } = ctx.request.body
+
+      const updatedGroup = groupManager.updateGroup(groupId, {
+        name: name ? name.trim() : undefined,
+        description: description !== undefined ? description.trim() : undefined,
+      })
+
+      ctx.body = createSuccessResponse(updatedGroup, '更新成功')
+    } catch (error) {
+      ctx.body = createErrorResponse(error.message)
+    }
+  })
+
+  // 删除组
+  router.post('/cgi-bin/mockbubu/groups/delete', validate({
+    groupId: (val) => {
+      if (!val || typeof val !== 'string') {
+        throw new Error('groupId 必须是字符串')
+      }
+      return null
+    },
+  }), (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const { groupId } = ctx.request.body
+
+      groupManager.deleteGroup(groupId)
+
+      ctx.body = createSuccessResponse(null, '删除成功')
+    } catch (error) {
+      ctx.body = createErrorResponse(error.message)
+    }
+  })
+
+  // 切换当前组
+  router.post('/cgi-bin/mockbubu/groups/switch', validate({
+    groupId: (val) => {
+      if (!val || typeof val !== 'string') {
+        throw new Error('groupId 必须是字符串')
+      }
+      return null
+    },
+  }), (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const { groupId } = ctx.request.body
+
+      const currentGroup = groupManager.setCurrentGroup(groupId)
+
+      ctx.body = createSuccessResponse({
+        currentGroup,
+        currentGroupId: groupId,
+      }, '切换成功')
+    } catch (error) {
+      ctx.body = createErrorResponse(error.message)
+    }
+  })
+
+  // 复制组
+  router.post('/cgi-bin/mockbubu/groups/copy', validate({
+    sourceGroupId: (val) => {
+      if (!val || typeof val !== 'string') {
+        throw new Error('sourceGroupId 必须是字符串')
+      }
+      return null
+    },
+    name: (val) => {
+      if (!val || !val.trim()) {
+        throw new Error('组名不能为空')
+      }
+      if (val.trim().length > 50) {
+        throw new Error('组名长度不能超过50个字符')
+      }
+      return null
+    },
+    description: (val) => {
+      if (val && val.length > 200) {
+        throw new Error('描述长度不能超过200个字符')
+      }
+      return null
+    },
+  }), (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const { sourceGroupId, name, description } = ctx.request.body
+
+      // 检查源组是否存在
+      const groups = groupManager.getGroups()
+      const sourceGroup = groups.find(g => g.id === sourceGroupId)
+      if (!sourceGroup) {
+        throw new Error(`源组不存在: ${sourceGroupId}`)
+      }
+
+      // 创建新组并复制配置
+      const newGroup = groupManager.createGroup({
+        name: name.trim(),
+        description: description ? description.trim() : '',
+        copyFromGroupId: sourceGroupId,
+      })
+
+      ctx.body = createSuccessResponse(newGroup, '复制成功')
+    } catch (error) {
+      ctx.body = createErrorResponse(error.message)
+    }
+  })
+}