94 lines
2.6 KiB
TypeScript
94 lines
2.6 KiB
TypeScript
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);
|
|
});
|
|
});
|
|
}
|
|
};
|
|
}
|