package com.vincent.rsf.server.ai.tool;

import com.vincent.rsf.framework.exception.CoolException;
import org.springframework.util.StringUtils;

import java.util.ArrayList;
import java.util.List;

public final class BuiltinToolGovernanceSupport {

    private BuiltinToolGovernanceSupport() {
    }

    /**
     * 把工具入参里的 limit 统一收敛到安全范围内。
     * 所有内置只读工具都通过该方法限制返回规模，避免模型一次查询过多数据。
     */
    public static int normalizeLimit(Integer limit, int defaultValue, int maxValue) {
        if (limit == null) {
            return defaultValue;
        }
        if (limit < 1 || limit > maxValue) {
            throw new CoolException("limit 必须在 1 到 " + maxValue + " 之间");
        }
        return limit;
    }

    /**
     * 要求多个过滤条件里至少有一个有效值。
     * 这是防止 AI 工具被模型当成“全表扫描接口”使用的第一道保护。
     */
    public static void requireAnyFilter(String message, String... values) {
        if (values == null || values.length == 0) {
            throw new CoolException(message);
        }
        for (String value : values) {
            if (StringUtils.hasText(value)) {
                return;
            }
        }
        throw new CoolException(message);
    }

    /**
     * 清洗单个文本型查询参数，并限制最大长度。
     * 这里只做轻量治理，不做模糊兜底或自动纠错，非法输入直接拒绝。
     */
    public static String sanitizeQueryText(String value, String fieldLabel, int maxLength) {
        if (!StringUtils.hasText(value)) {
            return null;
        }
        String normalized = value.trim();
        if (normalized.length() > maxLength) {
            throw new CoolException(fieldLabel + "长度不能超过 " + maxLength);
        }
        return normalized;
    }

    /**
     * 清洗字符串数组型参数，常用于站点类型、状态列表等批量过滤条件。
     * 返回结果会自动剔除空值，但如果最终为空仍然视为非法请求。
     */
    public static List<String> sanitizeStringList(List<String> values, String fieldLabel, int maxSize, int maxItemLength) {
        if (values == null || values.isEmpty()) {
            throw new CoolException(fieldLabel + "不能为空");
        }
        if (values.size() > maxSize) {
            throw new CoolException(fieldLabel + "数量不能超过 " + maxSize);
        }
        List<String> result = new ArrayList<>();
        for (String value : values) {
            String normalized = sanitizeQueryText(value, fieldLabel + "项", maxItemLength);
            if (!StringUtils.hasText(normalized)) {
                continue;
            }
            result.add(normalized);
        }
        if (result.isEmpty()) {
            throw new CoolException(fieldLabel + "不能为空");
        }
        return result;
    }
}