yapi/common/mock-extra.js

var mockjs = require('mockjs');
var strRegex = /\${([a-zA-Z0-9_\.]+)\}/g
var varSplit = ".";
var mockSplit = "|";

function mock(mockJSON, context) {
  context = context || {};
  var filtersMap = {
    regexp: handleRegexp
  }
  return mockjs.mock(parse(mockJSON))

  function parse(p, c) {
    c = c || {};
    for (var i in p) {
      if (!p.hasOwnProperty(i)) {
        continue;
      }
      if (typeof p[i] === 'object') {
        c[i] = (p[i].constructor === Array) ? [] : {};
        parse(p[i], c[i]);
      } else {
        p[i] = handleStr(p[i]);
        var filters = i.split(mockSplit), newFilters = [].concat(filters);
        if (filters.length > 1) {
          for (var f = 1, l = filters.length, index; f < l; f++) {
            if (filters[f] in filtersMap) {
              if ((index = newFilters.indexOf(filters[f])) !== -1) {
                newFilters.splice(index, 1);
              }
              c[newFilters.join(mockSplit)] = filtersMap[filters[f]].call(p, p[i])
            }
          }
        } else {
          c[i] = p[i]
        }
      }
    }
    return c;
  }

  function handleRegexp(item) {
    return new RegExp(item)
  }

  function handleStr(str) {
    if(typeof str !== 'string' ||str.indexOf("{") === -1 || str.indexOf("}") === -1 || str.indexOf("$") === -1){
      return str;
    }
    str = str.replace(strRegex, function (matchs, name) {
      var names = name.split(varSplit);
      var data = context;
      names.forEach(function(n){
        if(data === false) return false;
        if(n in data){
          data = data[n];
        }else{
          data = false;
        }
      })
      if(data === false){
        return matchs;
      }
      return data;
    })
    return str;
  }
}

module.exports = mock;