json-api-mocker 2.2.2 → 2.3.1

package/dist/server.d.ts

@@ -1,21 +1,22 @@
-import type { Config } from './types';
-export declare class MockServer {
-    private app;
-    private configPath;
-    private config;
-    private serverConfig;
-    private logs;
-    private readonly MAX_LOGS;
-    private upload;
-    private wss;
-    private clients;
-    constructor(serverConfig?: Config, configPath?: string);
-    private loadConfig;
-    private saveConfig;
-    private setupConfigRoutes;
-    private setupMockRoutes;
-    private sendResponse;
-    private setupUploadRoute;
-    start(port?: number): void;
-    private broadcastLog;
-}
+import { Express } from 'express';
+import { Config } from './types';
+export declare class MockServer {
+    private app;
+    private server;
+    private wss;
+    private config;
+    private configPath;
+    constructor(config: Config, configPath?: string);
+    getApp(): Express;
+    private setupMiddleware;
+    private logRequest;
+    private generateMockData;
+    private handleRequest;
+    private setupRoutes;
+    private createRoute;
+    private findRouteConfig;
+    private generateMockResponse;
+    private setupWebSocket;
+    start(): void;
+    close(): void;
+}

package/dist/server.js

@@ -1,325 +1,205 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.MockServer = void 0;
-const express_1 = __importDefault(require("express"));
-const cors_1 = __importDefault(require("cors"));
-const fs_1 = require("fs");
-const path_1 = require("path");
-const uuid_1 = require("uuid");
-const mockjs_1 = __importDefault(require("mockjs"));
-const multer_1 = __importDefault(require("multer"));
-const path_2 = __importDefault(require("path"));
-const ws_1 = require("ws");
-class MockServer {
-    constructor(serverConfig = {}, configPath = 'data.json') {
-        this.logs = []; // 存储请求日志
-        this.MAX_LOGS = 1000; // 最大日志数量
-        this.clients = new Set();
-        this.app = (0, express_1.default)();
-        this.serverConfig = serverConfig;
-        this.configPath = (0, path_1.join)(process.cwd(), configPath);
-        this.config = this.loadConfig();
-        // 添加跨域支持
-        this.app.use((0, cors_1.default)());
-        this.app.use(express_1.default.json());
-        // 注意：先设置配置管理接口，再设置mock接口
-        this.setupConfigRoutes();
-        this.setupMockRoutes();
-        // 配置文件上传
-        const storage = multer_1.default.diskStorage({
-            destination: (req, file, cb) => {
-                const uploadDir = path_2.default.join(process.cwd(), 'uploads');
-                cb(null, uploadDir);
-            },
-            filename: (req, file, cb) => {
-                // 生成唯一文件名
-                const uniqueSuffix = Date.now() + '-' + Math.round(Math.random() * 1E9);
-                cb(null, file.fieldname + '-' + uniqueSuffix + path_2.default.extname(file.originalname));
-            }
-        });
-        this.upload = (0, multer_1.default)({ storage });
-        // 添加文件上传路由
-        this.setupUploadRoute();
-        // 使用固定的 WebSocket 端口
-        this.wss = new ws_1.WebSocketServer({ port: serverConfig.wsPort || 88866 });
-        this.wss.on('connection', (ws) => {
-            this.clients.add(ws);
-            ws.send(JSON.stringify({
-                type: 'init',
-                data: this.logs
-            }));
-            ws.on('close', () => {
-                this.clients.delete(ws);
-            });
-        });
-    }
-    loadConfig() {
-        try {
-            const data = (0, fs_1.readFileSync)(this.configPath, 'utf-8');
-            const fullConfig = JSON.parse(data);
-            // 确保返回的是数组
-            const routes = fullConfig.routes || [];
-            // 验每个配置项
-            return routes.filter(api => {
-                const isValid = api &&
-                    api.id &&
-                    api.route &&
-                    typeof api.route.path === 'string' &&
-                    api.route.methods &&
-                    typeof api.route.methods === 'object';
-                if (!isValid) {
-                    console.warn('Filtered out invalid API config:', api);
-                }
-                return isValid;
-            });
-        }
-        catch (error) {
-            console.error('Error loading config:', error);
-            return [];
-        }
-    }
-    saveConfig() {
-        try {
-            const data = (0, fs_1.readFileSync)(this.configPath, 'utf-8');
-            const fullConfig = JSON.parse(data);
-            // 处理每个 API 的响应数据
-            const processedRoutes = this.config.map(api => ({
-                ...api,
-                route: {
-                    ...api.route,
-                    methods: Object.entries(api.route.methods).reduce((acc, [method, config]) => ({
-                        ...acc,
-                        [method]: {
-                            ...config,
-                            // 解析并重新格式化 JSON，移除多余的转义
-                            response: typeof config.response === 'string'
-                                ? JSON.stringify(JSON.parse(config.response))
-                                : JSON.stringify(config.response),
-                            headers: config.headers || {}
-                        }
-                    }), {})
-                }
-            }));
-            fullConfig.routes = processedRoutes;
-            (0, fs_1.writeFileSync)(this.configPath, JSON.stringify(fullConfig, null, 2));
-        }
-        catch (error) {
-            console.error('Error saving config:', error);
-        }
-    }
-    setupConfigRoutes() {
-        const baseProxy = this.serverConfig.baseProxy || '';
-        // 获取所有API配置
-        this.app.get('/api/_config', (req, res) => {
-            console.log('Sending config:', this.config);
-            res.json(this.config);
-        });
-        // 创建新API配置
-        this.app.post('/api/_config', (req, res) => {
-            const newApi = {
-                id: (0, uuid_1.v4)(),
-                route: req.body.route
-            };
-            this.config.push(newApi);
-            this.saveConfig();
-            res.json(newApi);
-        });
-        // 更新API配置
-        this.app.put('/api/_config/:id', (req, res) => {
-            const index = this.config.findIndex(api => api.id === req.params.id);
-            if (index === -1) {
-                res.status(404).json({ error: 'API not found' });
-                return;
-            }
-            this.config[index] = {
-                id: req.params.id,
-                route: req.body.route
-            };
-            this.saveConfig();
-            res.json(this.config[index]);
-        });
-        // 删除API配置
-        this.app.delete('/api/_config/:id', (req, res) => {
-            const index = this.config.findIndex(api => api.id === req.params.id);
-            if (index === -1) {
-                res.status(404).json({ error: 'API not found' });
-                return;
-            }
-            this.config.splice(index, 1);
-            this.saveConfig();
-            res.status(204).send();
-        });
-        // 添加日志接口
-        this.app.get('/api/_logs', (req, res) => {
-            const { page = 1, size = 20 } = req.query;
-            const start = (Number(page) - 1) * Number(size);
-            const end = start + Number(size);
-            const total = this.logs.length;
-            res.json({
-                total,
-                list: this.logs.slice(start, end)
-            });
-        });
-        // 清除日志
-        this.app.delete('/api/_logs', (req, res) => {
-            this.logs = [];
-            res.status(204).send();
-        });
-    }
-    setupMockRoutes() {
-        const baseProxy = this.serverConfig.baseProxy || '';
-        this.app.all('*', (req, res, next) => {
-            // 如果是配置接口，跳过日志记录
-            if (req.path === '/api/_config' || req.path.startsWith('/api/_config/') ||
-                req.path === '/api/_logs' || req.path.startsWith('/api/_logs/')) {
-                return next();
-            }
-            const startTime = Date.now();
-            const originalSend = res.send;
-            // 修复 this 指向问题
-            const self = this;
-            // 记录请求日志
-            res.send = function (body) {
-                const endTime = Date.now();
-                const duration = endTime - startTime;
-                const log = {
-                    id: (0, uuid_1.v4)(),
-                    path: req.path,
-                    method: req.method,
-                    timestamp: new Date().toISOString(),
-                    status: res.statusCode,
-                    duration,
-                    params: {
-                        query: req.query,
-                        body: req.body,
-                        params: req.params
-                    },
-                    requestBody: req.body,
-                    responseBody: body
-                };
-                // 使用正确的 this 引用
-                self.logs.unshift(log);
-                if (self.logs.length > self.MAX_LOGS) {
-                    self.logs.pop();
-                }
-                // 广播新日志
-                self.broadcastLog(log);
-                return originalSend.call(this, body);
-            };
-            next();
-        });
-        // 处理所有mock请求，但排除配置接口
-        this.app.all('*', (req, res, next) => {
-            try {
-                // 如果是配置接口，跳过这个中间件
-                if (req.path === '/api/_config' || req.path.startsWith('/api/_config/')) {
-                    return next();
-                }
-                console.log('Received request for path:', req.path);
-                console.log('Current config:', this.config);
-                if (!Array.isArray(this.config)) {
-                    console.error('Config is not an array:', this.config);
-                    res.status(500).json({ error: 'Internal server error' });
-                    return;
-                }
-                const api = this.config.find(api => {
-                    if (!api || !api.route || typeof api.route.path !== 'string') {
-                        console.error('Invalid API config:', api);
-                        return false;
-                    }
-                    // 移除 baseProxy 前缀再比较
-                    const requestPath = req.path.replace(baseProxy, '');
-                    return api.route.path === requestPath;
-                });
-                if (!api) {
-                    res.status(404).json({ error: 'API not found' });
-                    return;
-                }
-                if (!api.route || !api.route.methods) {
-                    res.status(500).json({ error: 'Invalid API configuration' });
-                    return;
-                }
-                const method = req.method.toLowerCase();
-                const methodConfig = api.route.methods[method];
-                if (!methodConfig) {
-                    res.status(405).json({ error: 'Method not allowed' });
-                    return;
-                }
-                if (methodConfig.delay) {
-                    setTimeout(() => {
-                        this.sendResponse(res, methodConfig);
-                    }, methodConfig.delay);
-                }
-                else {
-                    this.sendResponse(res, methodConfig);
-                }
-            }
-            catch (error) {
-                console.error('Error handling request:', error);
-                res.status(500).json({ error: 'Internal server error' });
-            }
-        });
-    }
-    sendResponse(res, config) {
-        try {
-            if (config.headers) {
-                Object.entries(config.headers).forEach(([key, value]) => {
-                    res.setHeader(key, value);
-                });
-            }
-            // 解析存储的响应数据
-            let responseData = typeof config.response === 'string'
-                ? JSON.parse(config.response.replace(/\\/g, '')) // 移除多余的反斜杠
-                : config.response;
-            // 使用 Mock.js 处理数据
-            const mockedData = mockjs_1.default.mock(responseData);
-            res.status(config.status || 200).json(mockedData);
-        }
-        catch (error) {
-            console.error('Error processing response:', error);
-            res.status(500).json({ error: 'Internal server error' });
-        }
-    }
-    setupUploadRoute() {
-        // 处理文件上传
-        this.app.post('/api/upload', this.upload.single('file'), (req, res) => {
-            if (!req.file) {
-                return res.status(400).json({ error: 'No file uploaded' });
-            }
-            // 返回文件URL
-            const fileUrl = `/uploads/${req.file.filename}`;
-            res.json({
-                success: true,
-                data: {
-                    url: fileUrl,
-                    filename: req.file.originalname,
-                    size: req.file.size
-                }
-            });
-        });
-        // 提供静态文件访问
-        this.app.use('/uploads', express_1.default.static(path_2.default.join(process.cwd(), 'uploads')));
-    }
-    start(port) {
-        const serverPort = port || this.serverConfig.port || 3000;
-        this.app.listen(serverPort, () => {
-            console.log(`Mock server is running on http://localhost:${serverPort}`);
-        });
-    }
-    // 广播日志给所有客户端
-    broadcastLog(log) {
-        const message = JSON.stringify({
-            type: 'log',
-            data: log
-        });
-        this.clients.forEach(client => {
-            if (client.readyState === ws_1.WebSocket.OPEN) {
-                client.send(message);
-            }
-        });
-    }
-}
-exports.MockServer = MockServer;
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MockServer = void 0;
+const mockjs_1 = __importDefault(require("mockjs"));
+const express_1 = __importDefault(require("express"));
+const cors_1 = __importDefault(require("cors"));
+const ws_1 = require("ws");
+const http_1 = __importDefault(require("http"));
+class MockServer {
+    constructor(config, configPath = 'data.json') {
+        this.app = (0, express_1.default)();
+        this.wss = null;
+        this.logRequest = (req, res, next) => {
+            const startTime = Date.now();
+            const requestId = Math.random().toString(36).substring(7);
+            console.log(`[${new Date().toISOString()}] Request ${requestId}:`);
+            console.log(`  Method: ${req.method}`);
+            console.log(`  URL: ${req.url}`);
+            console.log(`  Query Params: ${JSON.stringify(req.query)}`);
+            console.log(`  Body: ${JSON.stringify(req.body)}`);
+            res.on('finish', () => {
+                const duration = Date.now() - startTime;
+                console.log(`[${new Date().toISOString()}] Response ${requestId}:`);
+                console.log(`  Status: ${res.statusCode}`);
+                console.log(`  Duration: ${duration}ms`);
+                console.log('----------------------------------------');
+            });
+            next();
+        };
+        this.config = config;
+        this.configPath = configPath;
+        this.server = http_1.default.createServer(this.app);
+        this.setupMiddleware();
+        this.setupRoutes();
+        if (config.websocket?.enabled) {
+            this.setupWebSocket();
+        }
+    }
+    getApp() {
+        return this.app;
+    }
+    setupMiddleware() {
+        this.app.use((0, cors_1.default)());
+        this.app.use(express_1.default.json());
+        this.app.use('/uploads', express_1.default.static('uploads'));
+        this.app.use(this.logRequest);
+    }
+    generateMockData(config) {
+        try {
+            if (config.mock?.enabled && config.mock.template) {
+                const { total, template } = config.mock;
+                return mockjs_1.default.mock({
+                    [`data|${total}`]: [template]
+                }).data;
+            }
+            return config.response;
+        }
+        catch (error) {
+            console.error('Error generating mock data:', error);
+            return config.response;
+        }
+    }
+    handleRequest(config) {
+        return (req, res) => {
+            try {
+                let responseData = this.generateMockData(config);
+                if (config.pagination?.enabled && Array.isArray(responseData)) {
+                    const page = parseInt(req.query.page) || 1;
+                    const pageSize = parseInt(req.query.pageSize) || config.pagination.pageSize;
+                    const startIndex = (page - 1) * pageSize;
+                    const endIndex = startIndex + pageSize;
+                    const paginatedData = responseData.slice(startIndex, endIndex);
+                    res.header('X-Total-Count', responseData.length.toString());
+                    responseData = paginatedData;
+                }
+                res.json(responseData);
+            }
+            catch (error) {
+                console.error('Error handling request:', error);
+                res.status(500).json({ error: 'Internal server error' });
+            }
+        };
+    }
+    setupRoutes() {
+        this.config.routes.forEach((route) => {
+            Object.entries(route.methods).forEach(([method, methodConfig]) => {
+                this.createRoute(route.path, method, methodConfig);
+            });
+        });
+    }
+    createRoute(path, method, config) {
+        const fullPath = `${this.config.server.baseProxy}${path}`;
+        console.log(`创建路由: ${method.toUpperCase()} ${fullPath}`);
+        switch (method.toLowerCase()) {
+            case 'get':
+                this.app.get(fullPath, this.handleRequest(config));
+                break;
+            case 'post':
+                if (path === '/upload/avatar') {
+                    // 对于文件上传路由，使用特殊处理
+                    this.app.post(fullPath, (req, res) => {
+                        const mockResponse = this.generateMockData(config);
+                        res.json(mockResponse);
+                    });
+                }
+                else {
+                    this.app.post(fullPath, this.handleRequest(config));
+                }
+                break;
+            case 'put':
+                this.app.put(`${fullPath}/:id`, this.handleRequest(config));
+                break;
+            case 'delete':
+                this.app.delete(`${fullPath}/:id`, this.handleRequest(config));
+                break;
+        }
+    }
+    findRouteConfig(path, method) {
+        const route = this.config.routes.find(r => r.path === path);
+        return route?.methods[method] || null;
+    }
+    generateMockResponse(config) {
+        if (config.mock?.enabled && config.mock.template) {
+            return mockjs_1.default.mock(config.mock.template);
+        }
+        return config.response;
+    }
+    setupWebSocket() {
+        if (!this.config.websocket)
+            return;
+        this.wss = new ws_1.Server({
+            server: this.server,
+            path: this.config.websocket.path
+        });
+        this.wss.on('connection', (ws) => {
+            console.log('WebSocket client connected');
+            // 处理客户端消息
+            ws.on('message', (message) => {
+                try {
+                    const data = JSON.parse(message.toString());
+                    const eventConfig = this.config.websocket?.events?.[data.event];
+                    if (eventConfig?.mock.enabled) {
+                        const response = mockjs_1.default.mock(eventConfig.mock.template);
+                        ws.send(JSON.stringify({
+                            event: data.event,
+                            data: response
+                        }));
+                    }
+                }
+                catch (error) {
+                    console.error('Error handling WebSocket message:', error);
+                }
+            });
+            // 设置自动发送数据的定时器
+            if (this.config.websocket && this.config.websocket.events) {
+                Object.entries(this.config.websocket.events).forEach(([event, config]) => {
+                    if (config.mock.interval) {
+                        setInterval(() => {
+                            const response = mockjs_1.default.mock(config.mock.template);
+                            ws.send(JSON.stringify({
+                                event,
+                                data: response
+                            }));
+                        }, config.mock.interval);
+                    }
+                });
+            }
+            ws.on('close', () => {
+                // 移除日志，避免测试完成后的日志输出
+                // console.log('WebSocket client disconnected');
+            });
+        });
+    }
+    start() {
+        this.server.listen(this.config.server.port, () => {
+            console.log(`Mock 服务器已启动:`);
+            console.log(`- HTTP 地址: http://localhost:${this.config.server.port}`);
+            if (this.config.websocket?.enabled) {
+                console.log(`- WebSocket 地址: ws://localhost:${this.config.server.port}${this.config.websocket.path}`);
+            }
+            console.log(`- 基础路径: ${this.config.server.baseProxy}`);
+            console.log('可用的接口:');
+            this.config.routes.forEach(route => {
+                Object.keys(route.methods).forEach(method => {
+                    console.log(`  ${method.toUpperCase()} http://localhost:${this.config.server.port}${this.config.server.baseProxy}${route.path}`);
+                });
+            });
+        });
+    }
+    close() {
+        // 关闭所有 WebSocket 连接
+        if (this.wss) {
+            this.wss.clients.forEach(client => {
+                client.close();
+            });
+            this.wss.close();
+        }
+        // 关闭 HTTP 服务器
+        this.server.close();
+    }
+}
+exports.MockServer = MockServer;