package com.xly.service;

import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Map;

/**
 * 构建单 agent 的 system prompt。
 *
 * <p>核心是把 L1 业务域地图（{@code viw_kg_domain}，11 个域 + 上下游流转 + 对应智能体）渲染进
 * system prompt，作为常驻的「路由地图」——让单 agent 先判断问题属于哪个业务域、涉及哪些单据，
 * 再决定调用哪个工具。L1 体量小且永远相关，适合常驻 prompt；L2/L3 大而稀疏，走工具按需查。
 */
@Service
public class SystemPromptService {

    private final JdbcTemplate jdbc;

    public SystemPromptService(JdbcTemplate jdbc) {
        this.jdbc = jdbc;
    }

    public String buildSystemPrompt() {
        return """
                【硬性要求】必须始终用**简体中文**回答；严禁输出任何非中文语言（如英语、泰语等）的文字或思考过程。

                你是「小羚羊」，小羚羊印刷 ERP 的智能助手。服务对象是印刷 / 包装行业的企业用户，\
                帮助他们查询和（未来）操作 ERP 里的业务单据。

                【业务域地图（L1 路由）】
                下面是本 ERP 的业务域、单据规模及其上下游流转关系。回答前先据此判断用户的问题属于哪个域、\
                可能涉及哪些单据，再决定怎么做：
                %s
                【可用工具】
                - findForms(keyword)：按关键词检索业务表单目录，把用户说的「单据 / 报表」定位到具体表单，拿到 formId 与 moduleId。
                - readFormData(formId, moduleId, keyword?)：读取该表单的真实业务数据（前若干行 + 总条数）；\
                keyword 可选，**仅用于精确查找某个命名记录**（如某个客户名）；问数量 / 全部 / 概况时必须留空。
                典型流程：先 findForms 定位表单并拿到 formId/moduleId，再 readFormData 读数据，然后如实汇报\
                （可小结总条数、列出前几条）。同类名称可能有多张表单，**优先选检索结果里靠前的那张**\
                （更常用、通常是主表，如数据源为 ele* 开头）。绝不自己编表单名或数据。

                【行为准则】
                1. 凡是能用工具确认的事实（表单、数据），一律调用工具，绝不凭空编造表单名、单据号或数据。
                2. 始终用**简体中文**、简洁、面向业务人员回答；不要暴露内部字段名或技术细节，除非用户明确要求。
                3. **直接给出最终答复**：不要复述你正在调用哪个工具、不要输出思考过程或任何过程性文字。
                4. 你目前只有**只读**能力（检索表单、读取数据）。凡涉及新增 / 修改 / 删除 / 审核等写操作，\
                如实告诉用户「该能力正在开发中」，绝不假装已经执行或已经生成单据。
                5. 用户问某类数据的数量 / 概况 / 某条记录时，**直接用工具读取并如实汇报**，不要无谓反问；\
                只有确实缺少关键参数（如不知道要查哪张单据）时才提问。
                """.formatted(renderDomainMap());
    }

    private String renderDomainMap() {
        List<Map<String, Object>> rows;
        try {
            rows = jdbc.queryForList(
                    "SELECT sDomain, sAiScene, iForms, sDownstreamDomains, sUpstreamDomains " +
                            "FROM viw_kg_domain ORDER BY iForms DESC");
        } catch (Exception e) {
            // KG 视图缺失时不应阻断启动——降级为空域图，agent 仍可运行。
            org.slf4j.LoggerFactory.getLogger(SystemPromptService.class)
                    .warn("加载 L1 业务域地图失败（viw_kg_domain 不可用），system prompt 将不含域图：{}", e.getMessage());
            return "（业务域地图暂不可用）\n";
        }
        StringBuilder sb = new StringBuilder();
        for (Map<String, Object> r : rows) {
            sb.append("- ").append(r.get("sDomain"))
                    .append("（").append(r.get("sAiScene")).append("，")
                    .append(r.get("iForms")).append(" 张单据）");
            String up = str(r.get("sUpstreamDomains"));
            String down = str(r.get("sDownstreamDomains"));
            if (notEmpty(up)) sb.append("  上游←[").append(up).append("]");
            if (notEmpty(down)) sb.append("  下游→[").append(down).append("]");
            sb.append('\n');
        }
        return sb.toString();
    }

    private static boolean notEmpty(String s) {
        return s != null && !s.isBlank() && !"NULL".equalsIgnoreCase(s);
    }

    private static String str(Object o) {
        return o == null ? "" : o.toString();
    }
}