package com.vincent.rsf.server.ai.service.diagnosis;

import com.vincent.rsf.server.system.entity.AiDiagnosisRecord;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

@Service
public class AiDiagnosisReportService {

    /**
     * 基于模型结论和工具摘要回填诊断报告结构化字段与 Markdown 内容。
     */
    public void fillReport(AiDiagnosisRecord record, String conclusion, String toolSummary) {
        if (record == null) {
            return;
        }
        Map<String, String> sections = extractSections(conclusion);
        String executiveSummary = firstNonBlank(
                sections.get("问题概述"),
                sections.get("执行摘要"),
                safeText(conclusion)
        );
        String evidenceSummary = firstNonBlank(
                sections.get("关键证据"),
                sections.get("证据摘要"),
                safeText(toolSummary)
        );
        String actionSummary = firstNonBlank(
                sections.get("建议动作"),
                sections.get("处置建议"),
                "请结合现场流程继续确认。"
        );
        String riskSummary = firstNonBlank(
                sections.get("风险评估"),
                sections.get("潜在风险"),
                "如未及时处理，可能继续影响当前业务运行。"
        );
        record.setReportTitle(buildTitle(record));
        record.setExecutiveSummary(executiveSummary);
        record.setEvidenceSummary(evidenceSummary);
        record.setActionSummary(actionSummary);
        record.setRiskSummary(riskSummary);
        record.setReportMarkdown(buildMarkdown(record, executiveSummary, evidenceSummary, actionSummary, riskSummary, conclusion));
    }

    /**
     * 尝试从原始结论文本中提取“问题概述/关键证据/建议动作/风险评估”等章节。
     */
    private Map<String, String> extractSections(String conclusion) {
        Map<String, String> sections = new LinkedHashMap<>();
        if (conclusion == null || conclusion.trim().isEmpty()) {
            return sections;
        }
        String[] lines = conclusion.replace("\r", "").split("\n");
        String currentTitle = "问题概述";
        StringBuilder currentBody = new StringBuilder();
        for (String rawLine : lines) {
            String line = rawLine == null ? "" : rawLine.trim();
            String title = normalizeTitle(line);
            if (title != null) {
                saveSection(sections, currentTitle, currentBody);
                currentTitle = title;
                currentBody = new StringBuilder();
                String remainder = line.substring(line.indexOf(title) + title.length()).trim();
                if (remainder.startsWith("：") || remainder.startsWith(":")) {
                    remainder = remainder.substring(1).trim();
                }
                if (!remainder.isEmpty()) {
                    currentBody.append(remainder);
                }
                continue;
            }
            if (currentBody.length() > 0) {
                currentBody.append("\n");
            }
            currentBody.append(rawLine == null ? "" : rawLine);
        }
        saveSection(sections, currentTitle, currentBody);
        return sections;
    }

    /**
     * 将当前章节内容写入 section map。
     */
    private void saveSection(Map<String, String> sections, String title, StringBuilder body) {
        String content = body == null ? "" : body.toString().trim();
        if (content.isEmpty()) {
            return;
        }
        sections.put(title, content);
    }

    /**
     * 判断一行文本是否可视为报告章节标题。
     */
    private String normalizeTitle(String line) {
        if (line == null) {
            return null;
        }
        String normalized = line.replace("#", "").replace("：", "").replace(":", "").trim();
        List<String> titles = new ArrayList<>();
        titles.add("问题概述");
        titles.add("执行摘要");
        titles.add("关键证据");
        titles.add("证据摘要");
        titles.add("可能原因");
        titles.add("建议动作");
        titles.add("处置建议");
        titles.add("风险评估");
        titles.add("潜在风险");
        for (String title : titles) {
            if (normalized.startsWith(title)) {
                return title;
            }
        }
        return null;
    }

    /**
     * 构造诊断报告标题。
     */
    private String buildTitle(AiDiagnosisRecord record) {
        String diagnosisNo = record.getDiagnosisNo() == null ? "-" : record.getDiagnosisNo();
        return "WMS诊断报告-" + diagnosisNo;
    }

    /**
     * 生成最终报告 Markdown。
     */
    private String buildMarkdown(AiDiagnosisRecord record, String executiveSummary, String evidenceSummary,
                                 String actionSummary, String riskSummary, String conclusion) {
        List<String> parts = new ArrayList<>();
        parts.add("# " + buildTitle(record));
        parts.add("");
        parts.add("## 问题概述");
        parts.add(safeText(executiveSummary));
        parts.add("");
        parts.add("## 关键证据");
        parts.add(safeText(evidenceSummary));
        parts.add("");
        parts.add("## 建议动作");
        parts.add(safeText(actionSummary));
        parts.add("");
        parts.add("## 风险评估");
        parts.add(safeText(riskSummary));
        if (conclusion != null && !conclusion.trim().isEmpty()) {
            parts.add("");
            parts.add("## 原始结论");
            parts.add(conclusion.trim());
        }
        return String.join("\n", parts);
    }

    /**
     * 返回第一个非空文本。
     */
    private String firstNonBlank(String... values) {
        if (values == null) {
            return "";
        }
        for (String value : values) {
            if (value != null && !value.trim().isEmpty()) {
                return value.trim();
            }
        }
        return "";
    }

    /**
     * 对展示文本做空值兜底。
     */
    private String safeText(String value) {
        return value == null || value.trim().isEmpty() ? "暂无内容" : value.trim();
    }
}