qu-build/es/mock-server.js

Build tools for FE

import "core-js/modules/es6.array.sort";
import "core-js/modules/es6.regexp.split";
import "core-js/modules/es6.array.index-of";
import "core-js/modules/es6.regexp.replace";
import "core-js/modules/es6.regexp.match";
import "core-js/modules/es6.string.trim";
import "core-js/modules/es6.array.map";
import fs from 'fs';
import Mock, { Random } from 'mockjs';
import walkdir from 'node-walkdir';
var RE = /^\s*\/\*[*\s]+?([^\r\n]+)[\s\S]+?@url\s+([^\n]+)[\s\S]+?\*\//im;

function mock(_opts) {
  var routes = {};

  _opts.modules.map(function (_dir) {
    fs.exists(_dir, function (exists) {
      if (exists) {
        walkdir(_dir, /\.js(on)?$/i, function (filepath) {
          var content = String(fs.readFileSync(filepath, 'utf8')).trim() || '{}';
          var url = filepath;
          var describe = 'no description';
          var m = content.match(RE);

          if (m) {
            url = m[2].trim();
            describe = m[1].replace(/(^[\s*]+|[\s*]+$)/g, '');
          }

          if (url[0] !== '/') {
            url = "/".concat(url);
          }

          var pathname = url;

          if (pathname.indexOf('?') > -1) {
            pathname = pathname.split('?')[0];
          }

          if (mock.debug && routes[pathname]) {
            console.warn("[Mock Warn]: [".concat(filepath, ": ").concat(pathname, "] already exists and has been covered with new data."));
          }

          routes[pathname] = {
            url: url,
            filepath: filepath,
            describe: describe
          };

          if (/\.js$/.test(filepath)) {
            routes[pathname].data = require(filepath);
          } else {
            try {
              routes[pathname].data = new Function("return (".concat(content, ")"))();
            } catch (e) {
              delete routes[pathname];
              mock.debug && console.warn('[Mock Warn]:', e);
            }
          }
        });
      }
    });
  });

  return function handle(req, res, next) {
    res.set('Access-Control-Allow-Origin', '*');
    res.set('Access-Control-Allow-Headers', 'Origin, X-Requested-With, Content-Type, Accept');
    res.set('Access-Control-Allow-Methods', 'PUT,POST,GET,DELETE,OPTIONS');

    if (req.method === 'OPTIONS') {
      return res.send('');
    }

    var url = req.originalUrl.split('?')[0];

    if (url === '/api') {
      var host = "".concat(req.protocol, "://").concat(req.headers.host).concat(req.baseUrl);
      var list = Object.keys(routes).sort().map(function (path) {
        var route = routes[path];
        return {
          title: route.describe,
          url: host + route.url,
          file: route.filepath
        };
      });
      res.json(list);
    }

    var data = (routes[url] || 0).data;

    if (data) {
      if (typeof data === 'function') {
        data = data(req, res, Mock);
      }

      var _mockData = Mock.mock(data);

      if (req.query.callback) {
        res.end("".concat(req.query.callback, "(").concat(JSON.stringify(_mockData), ")"));
      } else {
        res.json(_mockData);
      }
    } else {
      next();
    }
  };
}

export default mock;