whistle.mock-plugins/lib/ui-server.js

Whistle 插件，用于快速创建 API 模拟数据

const express = require('express');
const path = require('path');
const bodyParser = require('body-parser');
const app = express();
const fs = require('fs-extra');
const storage = require('./storage');

// 引入工具函数
const util = require('../app/util');

// 设置app路径
const APP_ROOT = util.formatPath(path.join(__dirname, '../app'));
const APP_CGI_ROOT = util.formatPath(path.join(APP_ROOT, 'cgi-bin'));
// 加载CGI模块
const CGI_MODULES = util.loadModulesSync(APP_CGI_ROOT);

// 数据存储目录
const DATA_DIR = storage.DATA_DIR;

// 确保数据目录存在
fs.ensureDirSync(DATA_DIR);

// 处理请求和响应错误
app.use(function(req, res, next) {
  req.on('error', util.noop);
  res.on('error', util.noop);
  next();
});

// 解析请求体
app.use(bodyParser.urlencoded({ extended: true, limit: '100mb'}));
app.use(bodyParser.json({ limit: '100mb' }));

// 处理CGI请求
app.all('/cgi-bin/*', function(req, res) {
  if (req.headers.origin) {
    res.setHeader('Access-Control-Allow-Origin', req.headers.origin);
    res.setHeader('Access-Control-Allow-Credentials', true);
  }

  const controller = CGI_MODULES[APP_ROOT + req.url.replace(/\?.*$/, '')];
  if (!controller) {
    res.status(404).end('Not found');
    return;
  }

  try {
    // 提供数据目录给CGI处理程序
    controller.call({ dataDir: DATA_DIR }, req, res);
  } catch(e) {
    res.status(500).end(e.stack);
  }
});

// 静态文件服务
app.use(express.static(path.join(__dirname, '../app/public')));

// 单页应用支持：所有非API请求返回index.html
app.get('*', (req, res) => {
  if (!req.path.startsWith('/cgi-bin/')) {
    res.sendFile(path.join(__dirname, '../app/public/index.html'));
  }
});

// 导出启动UI服务器的函数
module.exports = function startUIServer(server, options) {
  server.on('request', app);
};