ece.suwa3d.com/build/mock/getMockData.js

const path = require('path')
const glob = require('glob')
const { pathToRegexp } = require('path-to-regexp')
const signale = require('signale')
const { winPath } = require('./utils')
let _mockconfig = {}
const cwd = process.cwd()
let mockList = []
// 获取mock文件列表
function getMockFile() {
  let mockFiles = glob
    .sync('mock/**/*.[jt]s', {
      cwd,
      ignore: [],
    })
    .map(p => path.join(cwd, p))
  // windows下路径处理
  mockFiles = mockFiles.map(p => winPath(p))

  return mockFiles
}

// 获取mock文件数据
function getMockData(files) {
  let mockData = {}
  files.forEach((mockFile) => {
    try {
      const m = require(mockFile)
      mockData = Object.assign(mockData, m.default || m)
      return mockData
    }
    catch (e) {
      console.error(e)
      throw e
    }
  })
  return mockData
}

// mock数据格式化
function parseMockData(config) {
  mockList = Object.keys(config).reduce((list, key) => {
    const handler = config[key]
    const { method, path } = parseMockUrl(key)
    const keys = []
    const re = pathToRegexp(path, keys)
    list.push({
      method,
      path,
      re,
      keys,
      handler: mockHandle(handler),
    })
    return list
  }, [])
}
function getMatchMock(url) {
  return (
    _mockconfig.mockUrlList.findIndex((whiteUrl) => {
      if (Object.prototype.toString.call(whiteUrl) === '[object RegExp]')
        return whiteUrl.test(url)

      else
        return url.includes(whiteUrl)
    }) !== -1 && mockList.filter(item => url.includes(item.path)).sort((a, b) => b.path.length - a.path.length)[0]
  )
}
function mockHandle(handler) {
  if (Object.prototype.toString.call(handler) === '[object Function]')
    return handler

  return (_, res) => {
    return res.end(JSON.stringify(handler))
  }
}

function parseMockUrl(key) {
  let method = 'get'
  let path = key
  if (/\s+/.test(key)) {
    const splited = key.split(/\s+/)
    method = splited[0].toLowerCase()
    path = splited[1]
  }
  return {
    method,
    path,
  }
}

function initMock(config) {
  try {
    _mockconfig = config
    mockList = []
    const files = getMockFile()
    const mockData = getMockData(files)
    parseMockData(mockData)
  }
  catch (e) {
    signale.warn('init mock failed')
    signale.fatal(e)
  }
}

module.exports = {
  getMatchMock,
  initMock,
}