amis2/scripts/mockApiPlugin.ts

import type {Plugin} from 'vite';
// @ts-ignore
import handler from '../mock/index';
import express, {query} from 'express';
import setPrototypeOf from 'setprototypeof';
import path from 'path';
import fs from 'fs';
const app = express();

function initExpress(req: any, res: any, next: any, callback: () => void) {
  if (app.enabled('x-powered-by')) res.setHeader('X-Powered-By', 'Express');
  (req as any).res = res;
  res.req = req;
  (req as any).next = next;

  setPrototypeOf(req, app.request);
  setPrototypeOf(res, app.response);

  res.locals = res.locals || Object.create(null);

  query({})(req as any, res, callback);
}

export default function mockApiPlugin(options: {} = {}): Plugin {
  return {
    name: 'mock-api',
    enforce: 'pre' as 'pre',
    apply: 'serve' as 'serve',
    configureServer(server) {
      // examples/index 页面，不处理会直接吐出 html 文件
      server.middlewares.use('/examples/index', async (req, res, next) => {
        if (req.originalUrl !== '/examples/index') {
          next();
          return;
        }

        let template = fs.readFileSync(
          path.resolve(__dirname, '../index.html'),
          'utf-8'
        );
        template = await server.transformIndexHtml('/index.html', template);
        res.statusCode = 200;
        res.end(template);
      });

      server.middlewares.use('/api', (req, res, next) => {
        initExpress(req, res, next, () => {
          handler(req, res);
        });
      });

      // 处理 app 模式
      server.middlewares.use('/examples/app/', async (req, res, next) => {
        if (req.originalUrl !== '/examples/app/') {
          next();
          return;
        }

        let template = fs.readFileSync(
          path.resolve(__dirname, '../examples/app/index-vite.html'),
          'utf-8'
        );
        template = await server.transformIndexHtml(
          '/examples/app/index-vite.html',
          template
        );
        res.statusCode = 200;
        res.end(template);
      });

      server.middlewares.use('/schema.json', (req, res, next) => {
        initExpress(req, res, next, () => {
          const filepath = path.resolve(
            __dirname,
            '../packages/amis/schema.json'
          );

          if (!fs.existsSync(filepath)) {
            (res as any).json({
              $schema: 'http://json-schema.org/draft-07/schema#',
              type: 'object',
              description:
                'amis/schema.json 还没有构建，请执行 `npm run build-schemas --workspace amis` 后看效果'
            });
            return;
          }

          (res as any).sendFile(filepath);
        });
      });
    }
  };
}