whistle.mock-plugins 1.0.4 → 1.0.5

package/app/cgi-bin/interfaces.js

@@ -176,6 +176,7 @@ module.exports = function(req, res) {
           proxyType: interfaceData.proxyType,
           responseContent: interfaceData.responseContent,
           targetUrl: interfaceData.targetUrl,
+          customHeaders: interfaceData.customHeaders || {},
           filePath: interfaceData.filePath,
           httpStatus: interfaceData.httpStatus || 200,
           responseDelay: interfaceData.responseDelay || 0,
@@ -200,6 +201,7 @@ module.exports = function(req, res) {
           proxyType: interfaceData.proxyType || 'response',
           responseContent: interfaceData.responseContent || '',
           targetUrl: interfaceData.targetUrl || '',
+          customHeaders: interfaceData.customHeaders || {},
           filePath: interfaceData.filePath || '',
           httpStatus: interfaceData.httpStatus || 200,
           responseDelay: interfaceData.responseDelay || 0,
@@ -316,6 +318,7 @@ module.exports = function(req, res) {
         proxyType: interfaceData.proxyType || interfacesData.interfaces[index].proxyType,
         responseContent: interfaceData.responseContent !== undefined ? interfaceData.responseContent : interfacesData.interfaces[index].responseContent,
         targetUrl: interfaceData.targetUrl !== undefined ? interfaceData.targetUrl : interfacesData.interfaces[index].targetUrl,
+        customHeaders: interfaceData.customHeaders !== undefined ? interfaceData.customHeaders : interfacesData.interfaces[index].customHeaders || {},
         filePath: interfaceData.filePath !== undefined ? interfaceData.filePath : interfacesData.interfaces[index].filePath,
         httpStatus: interfaceData.httpStatus || interfacesData.interfaces[index].httpStatus,
         responseDelay: interfaceData.responseDelay !== undefined ? interfaceData.responseDelay : interfacesData.interfaces[index].responseDelay,

package/app/cgi-bin/test-interface.js

@@ -1,6 +1,22 @@
 const fs = require('fs-extra');
 const path = require('path');
 const Mock = require('mockjs');
+const url = require('url');
+
+// 生成随机数工具函数（与rules-server.js中保持一致）
+const generateRandomValue = (pattern) => {
+  // 如果不是以@开头的模式，直接返回原值
+  if (!pattern || !pattern.startsWith('@')) {
+    return pattern;
+  }
+  
+  const formatPattern = pattern.substring(1); // 去掉@前缀
+  // 为每个x生成一个随机字符（字母或数字）
+  return formatPattern.replace(/x/g, () => {
+    const chars = 'abcdefghijklmnopqrstuvwxyz0123456789';
+    return chars.charAt(Math.floor(Math.random() * chars.length));
+  });
+};
 
 module.exports = async function(req, res) {
   const dataDir = this.dataDir;
@@ -208,11 +224,26 @@ module.exports = async function(req, res) {
             });
           }
           
+          // 处理自定义请求头，包括随机值
+          const redirectHeaders = targetInterface.customHeaders || {};
+          const processedHeaders = {};
+          
+          // 处理随机值，转换为实际值
+          Object.entries(redirectHeaders).forEach(([key, value]) => {
+            processedHeaders[key] = generateRandomValue(value);
+          });
+          
+          const formattedRedirectHeaders = Object.entries(processedHeaders)
+            .map(([key, value]) => `${key}: ${value}`)
+            .join('\n');
+          
           responseData = {
             statusCode: 302,
             contentType: 'application/json; charset=utf-8',
             proxyType: 'redirect',
             targetUrl: redirectTarget,
+            customHeaders: processedHeaders,
+            formattedHeaders: formattedRedirectHeaders,
             mockInfo: {
               matchedPattern: targetInterface.urlPattern,
               requestMethod: targetInterface.httpMethod,
@@ -254,11 +285,26 @@ module.exports = async function(req, res) {
             });
           }
           
+          // 处理自定义请求头
+          const urlRedirectHeaders = targetInterface.customHeaders || {};
+          const processedUrlRedirectHeaders = {};
+          
+          // 处理随机值，转换为实际值
+          Object.entries(urlRedirectHeaders).forEach(([key, value]) => {
+            processedUrlRedirectHeaders[key] = generateRandomValue(value);
+          });
+          
+          const formattedUrlRedirectHeaders = Object.entries(processedUrlRedirectHeaders)
+            .map(([key, value]) => `${key}: ${value}`)
+            .join('\n');
+          
           responseData = {
             statusCode: 302,
             contentType: 'application/json; charset=utf-8',
             proxyType: 'url_redirect',
             targetUrl: baseUrl,
+            customHeaders: processedUrlRedirectHeaders,
+            formattedHeaders: formattedUrlRedirectHeaders,
             mockInfo: {
               matchedPattern: targetInterface.urlPattern,
               requestMethod: targetInterface.httpMethod,