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

import com.vincent.rsf.server.ai.dto.AiPromptPreviewDto;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.util.ArrayList;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

@Component
public class AiPromptRenderSupport {

    private static final Pattern VARIABLE_PATTERN = Pattern.compile("\\{\\{?([a-zA-Z0-9_.-]+)}}?");

    /**
     * 同时渲染 System Prompt 和 User Prompt，并回传本次解析到的变量清单。
     * 该方法用于 Prompt 管理页的预览能力，帮助管理员在不真正调用模型的前提下验证模板结果。
     */
    public AiPromptPreviewDto render(String systemPrompt, String userPromptTemplate, String input, Map<String, Object> metadata) {
        String finalInput = input == null ? "" : input;
        return AiPromptPreviewDto.builder()
                .renderedSystemPrompt(renderTemplate(systemPrompt, finalInput, metadata))
                .renderedUserPrompt(renderUserPrompt(userPromptTemplate, finalInput, metadata))
                .resolvedVariables(resolveVariables(systemPrompt, userPromptTemplate, metadata))
                .build();
    }

    /**
     * 只渲染用户消息模板。
     * 如果模板没有消费任何变量，则保留模板原文并把用户输入附加到末尾，
     * 这样可以显式暴露“模板未生效”的问题，而不是静默吞掉输入。
     */
    public String renderUserPrompt(String userPromptTemplate, String input, Map<String, Object> metadata) {
        if (!StringUtils.hasText(userPromptTemplate)) {
            return input;
        }
        String rendered = replaceTemplateVariables(userPromptTemplate, input, metadata);
        if (Objects.equals(rendered, userPromptTemplate)) {
            return userPromptTemplate + "\n\n" + input;
        }
        return rendered;
    }

    private String renderTemplate(String template, String input, Map<String, Object> metadata) {
        /** 渲染任意模板片段；空模板保持原样返回。 */
        if (!StringUtils.hasText(template)) {
            return template;
        }
        return replaceTemplateVariables(template, input, metadata);
    }

    private String replaceTemplateVariables(String template, String input, Map<String, Object> metadata) {
        /**
         * 统一处理 `{{input}}`、`{input}` 以及 metadata 里的占位变量替换。
         * 这里使用朴素替换而不是脚本执行，目的是让模板行为稳定、可预期、易排查。
         */
        String rendered = template
                .replace("{{input}}", input)
                .replace("{input}", input);
        if (metadata == null || metadata.isEmpty()) {
            return rendered;
        }
        for (Map.Entry<String, Object> entry : metadata.entrySet()) {
            String value = entry.getValue() == null ? "" : String.valueOf(entry.getValue());
            rendered = rendered.replace("{{" + entry.getKey() + "}}", value);
            rendered = rendered.replace("{" + entry.getKey() + "}", value);
        }
        return rendered;
    }

    private List<String> resolveVariables(String systemPrompt, String userPromptTemplate, Map<String, Object> metadata) {
        /** 收集当前 Prompt 中显式出现过的变量名，用于前端展示。 */
        LinkedHashSet<String> variables = new LinkedHashSet<>();
        collectVariables(variables, systemPrompt);
        collectVariables(variables, userPromptTemplate);
        if (metadata != null && !metadata.isEmpty()) {
            variables.addAll(metadata.keySet());
        }
        return new ArrayList<>(variables);
    }

    private void collectVariables(LinkedHashSet<String> variables, String template) {
        /** 扫描模板文本中的占位变量并按出现顺序去重。 */
        if (!StringUtils.hasText(template)) {
            return;
        }
        Matcher matcher = VARIABLE_PATTERN.matcher(template);
        while (matcher.find()) {
            variables.add(matcher.group(1));
        }
    }
}