whistle.mockbubu 1.0.0-dev.4 → 2.0.0-beta.0

package/lib/uiServer/index.js

@@ -6,27 +6,98 @@ const path = require('path')
 const router = require('koa-router')()
 const setupRouter = require('./router')
 const cors = require('koa2-cors')
+const GroupManager = require('../group-manager')
+const StorageAdapter = require('../storage-adapter')
 const MAX_AGE = 1000 * 60 * 5
 
 module.exports = (server, options) => {
   const app = new Koa()
+  const { storage } = options
+
+  console.log('[mockbubu] 插件开始初始化...')
+
+  // 使用 StorageAdapter 包装 Whistle storage
+  // Whistle storage 对象直接有 baseDir 属性
+  const actualBaseDir = storage.baseDir || storage._options?.baseDir
+  console.log('[mockbubu] 🔍 UI Server storage.baseDir:', storage.baseDir)
+  console.log('[mockbubu] 🔍 UI Server 使用 baseDir:', actualBaseDir)
+  const storageAdapter = new StorageAdapter({ baseDir: actualBaseDir })
+
+  // 同步初始化：确保初始化完成后再处理请求
+  let isInitialized = false
+  let groupManager
+
+  ;(async () => {
+    try {
+      // 等待 storage 初始化完成
+      await storageAdapter.init()
+      console.log('[mockbubu] ✅ V3 Storage (UI Server) 已初始化')
+
+      // 创建组管理器
+      console.log('[mockbubu] 创建 GroupManager...')
+      groupManager = new GroupManager(storageAdapter)
+      console.log('[mockbubu] GroupManager 创建成功')
+
+      // 确保默认组存在
+      console.log('[mockbubu] 确保默认组存在...')
+      await groupManager.ensureDefaultGroup()
+      console.log('[mockbubu] 默认组已准备就绪')
+
+
+      isInitialized = true
+      console.log('[mockbubu] 插件初始化完成')
+    } catch (error) {
+      console.error('[mockbubu] 初始化失败:', error)
+      console.error('[mockbubu] 错误堆栈:', error.stack)
+      // 如果初始化失败,创建一个简单的 mock 对象
+      groupManager = {
+        getCurrentGroupId: async () => 'default',
+        getCurrentGroup: async () => ({ id: 'default', name: '默认组' }),
+        getGroups: async () => [{ id: 'default', name: '默认组' }],
+      }
+      isInitialized = true
+    }
+  })()
+
+  // 中间件：等待初始化完成
+  app.use(async (ctx, next) => {
+    // eslint-disable-next-line no-unmodified-loop-condition
+    while (!isInitialized) {
+      await new Promise(resolve => setTimeout(resolve, 100))
+    }
+    // 将 groupManager 和 storageAdapter 挂载到 ctx，让所有路由都能访问
+    ctx.groupManager = groupManager
+    ctx.storageAdapter = storageAdapter
+    await next()
+  })
 
   app.proxy = true
   app.silent = true
   onerror(app)
-  setupRouter(router, options)
+  setupRouter(router, { ...options, storage: storageAdapter })
   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', async (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const groups = await groupManager.getGroups()
+      const currentGroupId = await groupManager.getCurrentGroupId()
+
+      ctx.body = createSuccessResponse({
+        groups,
+        currentGroupId,
+      })
+    } catch (error) {
+      ctx.body = createErrorResponse(error.message)
+    }
+  })
+
+  // 获取当前组
+  router.post('/cgi-bin/mockbubu/groups/current', async (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const currentGroup = await groupManager.getCurrentGroup()
+
+      ctx.body = createSuccessResponse({
+        currentGroup,
+        currentGroupId: await 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
+    },
+  }), async (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const { name, description, copyFromGroupId } = ctx.request.body
+
+      const newGroup = await 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
+    },
+  }), async (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const { groupId, name, description } = ctx.request.body
+
+      const updatedGroup = await 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
+    },
+  }), async (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const { groupId } = ctx.request.body
+
+      await 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
+    },
+  }), async (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const { groupId } = ctx.request.body
+
+      const currentGroup = await 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
+    },
+  }), async (ctx) => {
+    try {
+      const { groupManager } = ctx
+      const { sourceGroupId, name, description } = ctx.request.body
+
+      // 检查源组是否存在
+      const groups = await groupManager.getGroups()
+      const sourceGroup = groups.find(g => g.id === sourceGroupId)
+      if (!sourceGroup) {
+        throw new Error(`源组不存在: ${sourceGroupId}`)
+      }
+
+      // 创建新组并复制配置
+      const newGroup = await groupManager.createGroup({
+        name: name.trim(),
+        description: description ? description.trim() : '',
+        copyFromGroupId: sourceGroupId,
+      })
+
+      ctx.body = createSuccessResponse(newGroup, '复制成功')
+    } catch (error) {
+      ctx.body = createErrorResponse(error.message)
+    }
+  })
+}