private static final String SYSTEM_PROMPT = """
    你是一个专业的 Draw.io 图表生成助手。根据用户的描述，生成 Draw.io mxGraph XML 格式的图表。

    **重要：你必须严格按照以下 XML 模板格式生成，不要添加任何 JavaScript 代码或其他格式！**

    XML 模板格式：
    ```xml
    <mxGraphModel dx="800" dy="600" grid="1" gridSize="10" guides="1" tooltips="1" connect="1" arrows="1" fold="1" page="1" pageScale="1" pageWidth="827" pageHeight="1169">
      <root>
        <mxCell id="0"/>
        <mxCell id="1" parent="0"/>
        <!-- 矩形节点示例 -->
        <mxCell id="node1" value="节点文字" style="rounded=1;whiteSpace=wrap;html=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
          <mxGeometry x="100" y="100" width="120" height="60" as="geometry"/>
        </mxCell>
        <!-- 连接线示例 -->
        <mxCell id="edge1" value="" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;" edge="1" parent="1" source="node1" target="node2">
          <mxGeometry relative="1" as="geometry"/>
        </mxCell>
      </root>
    </mxGraphModel>

public interface IDrawioMcpGateway {

    /** 调用 MCP 工具 */
    JSONObject callTool(String toolName, Map<String, Object> params);

    /** 添加矩形 */
    JSONObject addRectangle(int x, int y, int width, int height, String text, String style);

    /** 添加连接线 */
    JSONObject addEdge(String sourceId, String targetId, String text, String style);

    /** 获取选中的单元格 */
    JSONObject getSelectedCell();

    /** 删除单元格 */
    JSONObject deleteCell(String cellId);

    /** 获取形状分类 */
    JSONObject getShapeCategories();

    /** 添加指定形状 */
    JSONObject addCellOfShape(String shapeName, int x, int y, int width, int height, String text);

    /** 获取分页模型 */
    JSONObject listPagedModel(int page, int pageSize);

    /** 是否已连接 */
    boolean isConnected();

    /** 是否启用 */
    boolean isEnabled();
}

@Override
public JSONObject callTool(String toolName, Map<String, Object> params) {
    if (!enabled) {
        return errorResponse("Draw.io MCP 未启用");
    }

    if (!connected) {
        connect();
        if (!connected) {
            return errorResponse("无法连接到 MCP 服务器");
        }
    }

    String requestId = String.valueOf(requestIdCounter.incrementAndGet());
    CompletableFuture<JSONObject> future = new CompletableFuture<>();
    pendingRequests.put(requestId, future);

    try {
        JSONObject request = new JSONObject();
        request.put("jsonrpc", "2.0");
        request.put("id", requestId);
        request.put("method", "tools/call");
        JSONObject toolParams = new JSONObject();
        toolParams.put("name", toolName);
        toolParams.put("arguments", params);
        request.put("params", toolParams);

        String requestJson = request.toJSONString();
        log.debug("[DrawioMcpClient] 发送请求: {}", requestJson);
        webSocket.sendText(requestJson, true);

        // 等待响应，设置超时
        return future.get(timeout, TimeUnit.MILLISECONDS);
    } catch (TimeoutException e) {
        pendingRequests.remove(requestId);
        log.error("[DrawioMcpClient] 请求超时: {}", toolName);
        return errorResponse("请求超时");
    } catch (Exception e) {
        pendingRequests.remove(requestId);
        log.error("[DrawioMcpClient] 调用工具失败: {}", toolName, e);
        return errorResponse("调用失败: " + e.getMessage());
    }
}

// 添加矩形
@Override
public JSONObject addRectangle(int x, int y, int width, int height, String text, String style) {
    return callTool("add-rectangle", Map.of(
            "x", x,
            "y", y,
            "width", width,
            "height", height,
            "text", text,
            "style", style != null ? style : "whiteSpace=wrap;html=1;fillColor=#dae8fc;strokeColor=#6c8ebf;"
    ));
}

// 添加连接线
@Override
public JSONObject addEdge(String sourceId, String targetId, String text, String style) {
    Map<String, Object> params = new HashMap<>();
    params.put("source_id", sourceId);
    params.put("target_id", targetId);
    if (text != null) params.put("text", text);
    if (style != null) params.put("style", style);
    return callTool("add-edge", params);
}

/**
 * Draw.io XML 解析与修复工具类
 */
public class DrawioXmlParser {
    
    private static final Pattern XML_PATTERN = 
        Pattern.compile("<mxGraphModel[\\s\\S]*?</mxGraphModel>");
    
    /**
     * 从 AI 响应中提取 XML
     * AI 可能返回：1. 纯 XML  2. Markdown 代码块包裹  3. 带说明文字
     */
    public static String extractXml(String aiResponse) {
        if (aiResponse == null || aiResponse.isEmpty()) {
            return null;
        }
        
        Matcher matcher = XML_PATTERN.matcher(aiResponse);
        if (matcher.find()) {
            return matcher.group();
        }
        return null;
    }
    
    /**
     * 修复 XML 中的常见问题
     */
    public static String fixXml(String xml) {
        if (xml == null) return null;
        
        // 1. 修复未转义的 & 符号（最常见问题）
        // 注意：要避免重复转义已经是 &amp; 的情况
        xml = xml.replaceAll("&(?!(amp|lt|gt|quot|apos);)", "&amp;");
        
        // 2. 修复中文引号
        xml = xml.replace(""", "\"").replace(""", "\"");
        xml = xml.replace("'", "'").replace("'", "'");
        
        // 3. 移除可能的 BOM 头
        xml = xml.replace("\uFEFF", "");
        
        // 4. 修复换行符
        xml = xml.replace("\r\n", "\n").replace("\r", "\n");
        
        // 5. 移除 XML 声明（Draw.io 不需要）
        xml = xml.replaceFirst("<\\?xml[^>]*\\?>\\s*", "");
        
        return xml.trim();
    }
    
    /**
     * 验证 XML 是否合法
     */
    public static boolean validateXml(String xml) {
        try {
            DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
            // 禁用外部实体，防止 XXE 攻击
            factory.setFeature("http://apache.org/xml/features/disallow-doctype-decl", true);
            DocumentBuilder builder = factory.newDocumentBuilder();
            builder.parse(new InputSource(new StringReader(xml)));
            return true;
        } catch (Exception e) {
            log.warn("XML 验证失败: {}", e.getMessage());
            return false;
        }
    }
}

/**
 * 处理 AI 响应，提取并修复 XML
 */
private String processAiResponse(String aiResponse) {
    // Step 1: 提取 XML
    String xml = DrawioXmlParser.extractXml(aiResponse);
    if (xml == null) {
        log.error("无法从 AI 响应中提取 XML");
        throw new DrawioException("AI 未生成有效的图表数据");
    }
    
    // Step 2: 修复常见问题
    xml = DrawioXmlParser.fixXml(xml);
    
    // Step 3: 验证 XML 格式
    if (!DrawioXmlParser.validateXml(xml)) {
        log.warn("XML 格式验证失败，尝试进一步修复...");
        // 可以添加更激进的修复策略
        xml = attemptAggressiveFix(xml);
    }
    
    // Step 4: 添加标记，方便前端识别
    return "<!-- DRAWIO_START -->\n" + xml + "\n<!-- DRAWIO_END -->";
}

// 使用正则避免重复转义
xml = xml.replaceAll("&(?!(amp|lt|gt|quot|apos);)", "&");

@Scheduled(fixedDelay = 30000)  // 每 30 秒检查一次
public void checkConnection() {
    if (enabled && !connected) {
        log.info("检测到 MCP 连接断开，尝试重连...");
        connect();
    }
}

private void connect() {
    int retryCount = 0;
    while (retryCount < MAX_RETRY && !connected) {
        try {
            doConnect();
        } catch (Exception e) {
            retryCount++;
            log.warn("MCP 连接失败，第 {} 次重试", retryCount);
            Thread.sleep(1000 * retryCount);  // 指数退避
        }
    }
}

// Spring Boot 配置
@Bean
public WebMvcConfigurer webMvcConfigurer() {
    return new WebMvcConfigurer() {
        @Override
        public void configureAsyncSupport(AsyncSupportConfigurer configurer) {
            configurer.setDefaultTimeout(300000);  // 5 分钟
        }
    };
}

@Cacheable(value = "drawio", key = "#request.hashCode()")
public String generateDiagram(DrawioRequest request) {
    // 调用 AI 生成
}

// 移除可能的脚本标签
xml = xml.replaceAll("<script[^>]*>.*?</script>", "");
// 移除事件处理器
xml = xml.replaceAll("\\s+on\\w+\\s*=", " data-removed=");

DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
factory.setFeature("http://apache.org/xml/features/disallow-doctype-decl", true);
factory.setFeature("http://xml.org/sax/features/external-general-entities", false);
factory.setFeature("http://xml.org/sax/features/external-parameter-entities", false);

@RateLimiter(name = "drawio", fallbackMethod = "rateLimitFallback")
public void generateDiagram(DrawioRequest request) {
    // ...
}

public void rateLimitFallback(DrawioRequest request, RequestNotPermitted ex) {
    throw new TooManyRequestsException("请求过于频繁，请稍后再试");
}