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

package/lib/utils.js

@@ -1,124 +1,98 @@
+/**
+ * Whistle.mockbubu 工具函数集
+ * V3 架构：简化版本，移除废弃的版本管理函数
+ */
+
 const zlib = require('zlib')
 const { RuleValueMap } = require('./const')
-// const URL = require('url')
-const setProperty = (storage, property, val) => {
-  if (storage.hasProperty(property)) {
-    const oldVal = storage.getProperty(property)
-
-    if (typeof oldVal === 'object') {
-      storage.setProperty(property, Object.assign(oldVal, val))
-    } else {
-      storage.setProperty(property, val)
-    }
-  } else {
-    storage.setProperty(property, val)
-  }
-}
 
-const removePropertyAttr = (storage, property, attrName) => {
-  if (storage.hasProperty(property)) {
-    const oldVal = storage.getProperty(property)
+// ========================================
+// 文件操作函数
+// ========================================
 
-    if (typeof oldVal === 'object') {
-      delete oldVal[attrName]
-      storage.setProperty(property, oldVal)
-    }
+// ✅ 按组获取文件列表
+const getGroupFileList = async (storage, groupId) => {
+  if (!groupId) {
+    console.warn('[mockbubu] getGroupFileList: groupId 为空，返回空数组')
+    return []
   }
-}
-const getPropertyAttr = (storage, property, attrName) => {
-  const props = storage.getProperty(property)
 
-  return props[attrName]
-}
-const getProperty = (storage, property) => {
-  return storage.getProperty(property)
-}
-const removeProperty = (storage, property) => {
-  return storage.removeProperty(property)
-}
-
-const writeFile = ({ storage, filename, body, properties, encoding }) => {
-  switch (encoding) {
-  case 'gzip':
-    body = zlib.gunzipSync(body)
-    break
-  case 'deflate':
-    body = zlib.inflateSync(body)
-    break
-  case 'br':
-    body = zlib.brotliDecompressSync(body)
-    break
-  default:
-    break
-  }
-  setProperty(storage, filename, properties || {})
-
-  const type = Object.prototype.toString.call(body)
-  let content
-  switch (type) {
-  case '[object String]':
-    content = body
-    break
-  case '[object Object]':
-    content = JSON.stringify(body)
-    break
-  case '[object Uint8Array]':
-    content = body.toString()
-    break
-  default:
-    content = body
+  try {
+    // 获取所有文件列表
+    // StorageAdapter 返回的格式已经是: groupId/filename (不需要解码)
+    const allFiles = await storage.getFileList() || []
+    const groupPrefix = `${groupId}/`
+    const result = []
+
+    // 过滤出属于该组的文件
+    allFiles.forEach((item) => {
+      try {
+        // StorageAdapter 返回的文件名格式: groupId/filename
+        // 例如: default/https://m.shein.com/us/bff-api/...
+        // 不需要任何解码，直接使用
+        const name = item.name
+
+        // 检查是否以 "groupId/" 开头
+        if (name.startsWith(groupPrefix)) {
+          // 提取纯文件名（去除 groupId/ 前缀）
+          const filename = name.substring(groupPrefix.length)
+
+          result.push({
+            name: filename, // 返回纯文件名，不包含 groupId 前缀
+            size: item.size || 0,
+          })
+        }
+      } catch (err) {
+        // 处理失败时跳过该文件
+        console.warn(`[mockbubu] 处理文件失败: ${item.name}`, err.message)
+      }
+    })
+
+    return result
+  } catch (error) {
+    console.error('[mockbubu] getGroupFileList 执行失败:', error.message)
+    console.error('[mockbubu] 错误堆栈:', error.stack)
+    return []
   }
-  storage.writeFile(filename, content)
-  setApiListUpdated(storage, true)
-  console.log('%c [ content ]-58', 'font-size:13px; background:pink; color:#bf2c9f;', content)
-  // switch (type) {
-  // case '[object String]':
-  //   storage.writeFile(filename, body)
-  //   break
-  // case '[object Object]':
-  //   storage.writeFile(filename, JSON.stringify(body))
-  //   break
-  // case '[object Uint8Array]':
-  //   storage.writeFile(filename, body.toString())
-  //   break
-  // default:
-  //   storage.writeFile(filename, body) // TODO:
-  // }
-}
-
-const getApiList = (storage) => {
-  return storage.getFileList()
 }
 
-const updateFile = (storage, filename, body) => {
+const updateFile = (storage, filename, body, groupId) => {
   if (!filename) {
     throw new Error('文件名不能为空')
   }
-  // storage.writeFile(filename, JSON.stringify(body))
 
-  const sessionCache = storage.readFile(filename)
+  const filePath = groupId ? `${groupId}/${filename}` : filename
+  const sessionCache = storage.readFile(filePath)
   if (sessionCache) {
     const session = JSON.parse(sessionCache)
     session.res.body = JSON.stringify(body)
-    storage.writeFile(filename, JSON.stringify(session))
+    storage.writeFile(filePath, JSON.stringify(session))
   }
 }
 
-const readFile = (storage, filename) => {
-  // return storage.readFile(filename)
-
-  const sessionCache = storage.readFile(filename)
+const readFile = async (storage, filename, groupId) => {
+  const filePath = groupId ? `${groupId}/${filename}` : filename
+  const sessionCache = await storage.readFile(filePath)
   if (sessionCache) {
     const resCache = JSON.parse(sessionCache)?.res
-    console.log(resCache)
     return resCache.body
   }
 }
 
-const removeFile = (storage, name) => {
-  storage.removeFile(name)
-  removeProperty(storage, name)
+// 读取完整的 session 数据（包含 req 和 res）
+const readSession = async (storage, filename, groupId) => {
+  const filePath = groupId ? `${groupId}/${filename}` : filename
+  const sessionCache = await storage.readFile(filePath)
+  if (sessionCache) {
+    return JSON.parse(sessionCache)
+  }
+  return null
 }
+
+// ========================================
+// URL 和请求处理函数
+// ========================================
+
 // 获取mock文件名
 const getFilename = (originalReq) => {
   const { url, ruleValue, pattern } = originalReq
@@ -142,11 +116,12 @@ const getFilename = (originalReq) => {
   return filename
 }
 
-const getApiListUpdated = (storage) => {
-  return !!storage.getProperty('api-list-updated')
+const getApiListUpdated = async (storage) => {
+  const value = await storage.getProperty('api-list-updated')
+  return !!value
 }
-const setApiListUpdated = (storage, updated) => {
-  return storage.setProperty('api-list-updated', updated)
+const setApiListUpdated = async (storage, updated) => {
+  return await storage.setProperty('api-list-updated', updated)
 }
 const getRule = (originalReq) => {
   const { pattern, ruleValue } = originalReq
@@ -155,102 +130,63 @@ const getRule = (originalReq) => {
 }
 
 const isJsonReq = (headers) => {
-  return (
+  const result = (
     (headers.accept && headers.accept.includes('application/json')) ||
     headers['sec-fetch-dest'] === 'empty' ||
     headers['Sec-Fetch-Dest'] === 'empty'
   )
-}
 
-function getVersionName(name) {
-  return 'version.' + name
-}
-
-/**
- * mock版本管理
- */
-
-// 新增版本
-const addNewVersion = ({ storage, filename, versionName, content }) => {
-  const name = getVersionName(versionName)
-
-  setProperty(storage, filename, { [name]: content })
-}
-// 更新版本文件内容
-const updateVersionContent = ({ storage, filename, versionName, content }) => {
-  const name = getVersionName(versionName)
-
-  setProperty(storage, filename, { [name]: content })
-}
-// 获取版本文件内容
-const getVersionContent = ({ storage, filename, versionName }) => {
-  const name = getVersionName(versionName)
-
-  return getPropertyAttr(storage, filename, name)
-}
-// 删除版本
-const deleteVersion = ({ storage, filename, versionName }) => {
-  const name = getVersionName(versionName)
-
-  removePropertyAttr(storage, filename, name)
-}
-// 更新版本名称
-const updateVersionName = ({ storage, filename, versionName, newVersion }) => {
-  const file = getVersionContent(versionName)
+  // 调试:所有 m.shein.com 的请求都输出日志
+  if (headers.host && headers.host === 'm.shein.com') {
+    const fullUrl = headers['x-whistle-full-url'] || 'unknown'
+    console.log(`[isJsonReq] m.shein.com 请求: ${decodeURIComponent(fullUrl)}`)
+    console.log('  - Accept:', headers.accept)
+    console.log('  - sec-fetch-dest:', headers['sec-fetch-dest'])
+    console.log('  - Sec-Fetch-Dest:', headers['Sec-Fetch-Dest'])
+    console.log('  - 判断结果:', result)
+  }
 
-  addNewVersion(storage, filename, { versionName: newVersion, file })
-  deleteVersion({ storage, filename, versionName })
+  return result
 }
-const getVersions = ({ storage, filename }) => {
-  const props = getProperty(storage, filename)
-  const list = []
-
-  Object.keys(props).forEach((key) => {
-    const matchs = key.match(/^version\.{1}(.+)/)
-
-    if (matchs && matchs[1]) {
-      list.push({
-        filename: matchs[1],
-        content: props[key],
-      })
-    }
-  })
 
-  return list
-}
+// ========================================
+// 工具函数
+// ========================================
 
 const withTryCatch = (fn) => {
   return async (...args) => {
     try {
       return await fn(...args)
     } catch (e) {
-      console.error('全局捕获异常:', e)
+      // 仅在开发环境输出错误日志
+      if (process.env.NODE_ENV === 'development') {
+        console.error('[mockbubu] 全局捕获异常:', e)
+      }
+      throw e
     }
   }
 }
 
-exports.addNewVersion = addNewVersion
-exports.updateVersionContent = updateVersionContent
-exports.deleteVersion = deleteVersion
-exports.updateVersionName = updateVersionName
-exports.getVersions = getVersions
-exports.setProperty = setProperty
-exports.getProperty = getProperty
-exports.getPropertyAttr = getPropertyAttr
-exports.getVersionContent = getVersionContent
-exports.removeProperty = removeProperty
-exports.removePropertyAttr = removePropertyAttr
-exports.writeFile = writeFile
-exports.getApiList = getApiList
+// ========================================
+// 导出 - V3 架构使用的函数
+// ========================================
+
+// 文件操作
+exports.getGroupFileList = getGroupFileList
 exports.updateFile = updateFile
 exports.readFile = readFile
-exports.removeFile = removeFile
+exports.readSession = readSession
+
+// URL 和请求处理
 exports.getFilename = getFilename
-exports.getApiListUpdated = getApiListUpdated
-exports.setApiListUpdated = setApiListUpdated
 exports.getRule = getRule
 exports.isJsonReq = isJsonReq
 
+// API 列表更新标志
+exports.getApiListUpdated = getApiListUpdated
+exports.setApiListUpdated = setApiListUpdated
+
+// 工具函数
 exports.withTryCatch = withTryCatch
 
 exports.handleBuffer2String = withTryCatch(({ body, encoding }) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "whistle.mockbubu",
-  "version": "1.0.0-dev.4",
+  "version": "2.0.0-beta.0",
   "description": "mock response data",
   "scripts": {
     "lint": "eslint . --ext .js",