whistle.mockbubu 1.0.0-dev.1

package/lib/uiServer/util.js

@@ -0,0 +1,99 @@
+const {
+  getApiList,
+  getProperty,
+} = require('../utils')
+const { RangeFilterMap, RuleFilterMap, LockedFilterMap } = require('../const')
+
+// 统一错误响应格式
+const createErrorResponse = (message, code = 500) => ({
+  code,
+  msg: message || '操作失败',
+})
+
+// 统一成功响应格式
+const createSuccessResponse = (data = null, message = '操作成功') => ({
+  code: 200,
+  msg: message,
+  ...(data && { data }),
+})
+
+// 执行筛选
+const execFilter = (result, filter) => {
+  const { key, value, method } = filter
+  const list = result.filter((item) => {
+    const type = Object.prototype.toString.call(value)
+
+    switch (method) {
+    case 'indexOf':
+      if (type === '[object String]') {
+        return ~item[key].indexOf(value)
+      }
+      break
+    case 'equal':
+      return item[key] === value
+    case 'unequal':
+      return item[key] !== value
+    }
+
+    return true
+  })
+
+  return list
+}
+
+// 处理筛选条件
+const handleFilterList = (list, filterOptions) => {
+  if (!Array.isArray(list) || !filterOptions) return []
+
+  const filters = []
+  const { name, rule, range, ruleValue, locked } = filterOptions
+  if (name) {
+    filters.push({
+      key: 'name',
+      value: name.trim(),
+      method: 'indexOf',
+    })
+  }
+
+  if (rule) {
+    filters.push({
+      key: 'rule',
+      value: rule.trim(),
+      method: 'equal',
+    })
+  }
+
+  // 范围筛选
+  RangeFilterMap[range] && filters.push(RangeFilterMap[range])
+  // 规则值筛选
+  RuleFilterMap[ruleValue] && filters.push(RuleFilterMap[ruleValue])
+  // 锁定筛选
+  LockedFilterMap[locked] && filters.push(LockedFilterMap[locked])
+
+  // 执行筛选
+  const filteredList = filters.reduce((result, filter) => {
+    return execFilter(result, filter)
+  }, list)
+
+  return filteredList
+}
+
+// 获取文件列表（包含文件属性数据）
+const getFullDataList = (localStorage) => {
+  // 获取接口列表
+  const list = getApiList(localStorage) || []
+
+  // 合并属性数据
+  list.forEach((item) => {
+    const props = getProperty(localStorage, item.name) || {}
+    delete item.data
+    Object.assign(item, props)
+  })
+  return list
+}
+
+exports.createErrorResponse = createErrorResponse
+exports.createSuccessResponse = createSuccessResponse
+exports.execFilter = execFilter
+exports.handleFilterList = handleFilterList
+exports.getFullDataList = getFullDataList

package/lib/utils.js

@@ -0,0 +1,287 @@
+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 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
+  }
+  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) => {
+  if (!filename) {
+    throw new Error('文件名不能为空')
+  }
+  // storage.writeFile(filename, JSON.stringify(body))
+
+  const sessionCache = storage.readFile(filename)
+  if (sessionCache) {
+    const session = JSON.parse(sessionCache)
+    session.res.body = JSON.stringify(body)
+    storage.writeFile(filename, JSON.stringify(session))
+  }
+}
+
+const readFile = (storage, filename) => {
+  // return storage.readFile(filename)
+
+  const sessionCache = storage.readFile(filename)
+  if (sessionCache) {
+    const resCache = JSON.parse(sessionCache)?.res
+    console.log(resCache)
+    return resCache.body
+  }
+}
+
+const removeFile = (storage, name) => {
+  storage.removeFile(name)
+  removeProperty(storage, name)
+}
+// 获取mock文件名
+const getFilename = (originalReq) => {
+  const { url, ruleValue, pattern } = originalReq
+  const u = new URL(url)
+  let filename = url
+
+  switch (ruleValue) {
+  case RuleValueMap.href:
+    filename = u.href
+    break
+  case RuleValueMap.pathname:
+    filename = u.origin + u.pathname
+    break
+  case RuleValueMap.pattern:
+    filename = pattern
+    break
+  default:
+    filename = u.origin + u.pathname
+  }
+
+  return filename
+}
+
+const getApiListUpdated = (storage) => {
+  return !!storage.getProperty('api-list-updated')
+}
+const setApiListUpdated = (storage, updated) => {
+  return storage.setProperty('api-list-updated', updated)
+}
+const getRule = (originalReq) => {
+  const { pattern, ruleValue } = originalReq
+
+  return `${pattern} mockbubu://${ruleValue}`
+}
+
+const isJsonReq = (headers) => {
+  return (
+    (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)
+
+  addNewVersion(storage, filename, { versionName: newVersion, file })
+  deleteVersion({ storage, filename, versionName })
+}
+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)
+    }
+  }
+}
+
+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
+exports.updateFile = updateFile
+exports.readFile = readFile
+exports.removeFile = removeFile
+exports.getFilename = getFilename
+exports.getApiListUpdated = getApiListUpdated
+exports.setApiListUpdated = setApiListUpdated
+exports.getRule = getRule
+exports.isJsonReq = isJsonReq
+
+exports.withTryCatch = withTryCatch
+
+exports.handleBuffer2String = withTryCatch(({ body, 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
+  }
+
+  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
+  }
+  return content
+})

package/package.json

@@ -0,0 +1,44 @@
+{
+  "name": "whistle.mockbubu",
+  "version": "1.0.0-dev.1",
+  "description": "mock response data",
+  "scripts": {
+    "lint": "eslint . --ext .js",
+    "lint:fix": "eslint . --ext .js --fix",
+    "lint:check": "eslint . --ext .js --max-warnings 0"
+  },
+  "dependencies": {
+    "koa": "^2.7.0",
+    "koa-bodyparser": "^4.2.1",
+    "koa-onerror": "^4.1.0",
+    "koa-router": "^7.4.0",
+    "koa-static": "^5.0.0",
+    "koa2-cors": "^2.0.6",
+    "qs": "^6.10.3",
+    "string_decoder": "^1.3.0"
+  },
+  "files": [
+    "lib/*",
+    "public/*",
+    "rules.txt",
+    ".editorconfig",
+    ".gitignore",
+    "index.js",
+    "package.json",
+    "README.md",
+    "step1.png",
+    "step2.png"
+  ],
+  "keywords": [
+    "proxy",
+    "mock",
+    "whistle"
+  ],
+  "devDependencies": {
+    "eslint": "^8.0.0",
+    "eslint-config-standard": "^17.0.0",
+    "eslint-plugin-import": "^2.27.0",
+    "eslint-plugin-node": "^11.1.0",
+    "eslint-plugin-promise": "^6.1.0"
+  }
+}