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

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.vincent.rsf.server.ai.model.AiDiagnosticToolResult;
import com.vincent.rsf.server.ai.model.AiPromptContext;
import com.vincent.rsf.server.manager.entity.DeviceSite;
import com.vincent.rsf.server.manager.mapper.DeviceSiteMapper;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

@Service
public class AiDeviceSiteSummaryService implements AiDiagnosticDataProvider {

    private static final String TOOL_CODE = "device_site_summary";
    private static final String TOOL_NAME = "设备站点摘要";

    @Resource
    private DeviceSiteMapper deviceSiteMapper;

    /**
     * 返回设备站点工具默认顺序。
     */
    @Override
    public int getOrder() {
        return 30;
    }

    /**
     * 返回设备站点工具编码。
     */
    @Override
    public String getToolCode() {
        return TOOL_CODE;
    }

    /**
     * 返回设备站点工具展示名。
     */
    @Override
    public String getToolName() {
        return TOOL_NAME;
    }

    /**
     * 返回设备站点工具默认说明。
     */
    @Override
    public String getDefaultToolPrompt() {
        return "结合设备站点摘要判断设备状态、巷道分布和最近更新站点。";
    }

    /**
     * 汇总站点状态、设备分布和最近更新站点，生成设备站点摘要。
     */
    @Override
    public AiDiagnosticToolResult buildDiagnosticData(AiPromptContext context) {
        String summary = buildDeviceSiteSummary(context);
        String severity = summary.contains("未查询到") ? "WARN" : "INFO";
        return new AiDiagnosticToolResult()
                .setToolCode(getToolCode())
                .setToolName(getToolName())
                .setSeverity(severity)
                .setSummaryText(summary);
    }

    /**
     * 基于 man_device_site 生成设备站点总览。
     */
    private String buildDeviceSiteSummary(AiPromptContext context) {
        List<DeviceSite> deviceSites = deviceSiteMapper.selectList(new LambdaQueryWrapper<DeviceSite>()
                .select(DeviceSite::getSite, DeviceSite::getName, DeviceSite::getType, DeviceSite::getDevice,
                        DeviceSite::getDeviceCode, DeviceSite::getDeviceSite, DeviceSite::getStatus,
                        DeviceSite::getChannel, DeviceSite::getUpdateTime)
                .eq(DeviceSite::getTenantId, context.getTenantId())
                .orderByDesc(DeviceSite::getUpdateTime)
                .last("limit 50"));

        if (deviceSites.isEmpty()) {
            return "当前未查询到 man_device_site 设备站点数据，请提示用户核对基础数据或租户配置。";
        }

        Map<String, Long> statusCounters = new LinkedHashMap<>();
        Map<String, Long> deviceCounters = new LinkedHashMap<>();
        Map<String, Long> channelCounters = new LinkedHashMap<>();
        for (DeviceSite deviceSite : deviceSites) {
            String status = resolveStatus(deviceSite.getStatus());
            String device = emptyToDefault(deviceSite.getDevice$(), "未配置设备类型");
            String channel = deviceSite.getChannel() == null ? "未配置巷道" : "巷道" + deviceSite.getChannel();
            statusCounters.put(status, statusCounters.getOrDefault(status, 0L) + 1);
            deviceCounters.put(device, deviceCounters.getOrDefault(device, 0L) + 1);
            channelCounters.put(channel, channelCounters.getOrDefault(channel, 0L) + 1);
        }

        StringBuilder summary = new StringBuilder();
        summary.append("以下是基于 man_device_site 的实时设备站点摘要，请结合任务和库存数据综合判断：");
        summary.append("\n设备站点总览：共 ").append(deviceSites.size()).append(" 条有效站点记录。");
        summary.append("\n站点状态分布：").append(formatCounter(statusCounters)).append("。");
        summary.append("\n设备类型分布：").append(formatCounter(deviceCounters)).append("。");
        summary.append("\n巷道分布：").append(formatCounter(channelCounters)).append("。");
        summary.append("\n最近更新站点 TOP5：").append(formatLatestSites(deviceSites.subList(0, Math.min(deviceSites.size(), 5)))).append("。");
        return summary.toString();
    }

    /**
     * 格式化计数类摘要。
     */
    private String formatCounter(Map<String, Long> counter) {
        List<String> parts = new ArrayList<>();
        for (Map.Entry<String, Long> item : counter.entrySet()) {
            parts.add(item.getKey() + " " + item.getValue() + " 条");
        }
        return String.join("，", parts);
    }

    /**
     * 格式化最近更新站点列表。
     */
    private String formatLatestSites(List<DeviceSite> deviceSites) {
        List<String> parts = new ArrayList<>();
        for (DeviceSite deviceSite : deviceSites) {
            parts.add(emptyToDefault(deviceSite.getName(), emptyToDefault(deviceSite.getSite(), "未命名站点"))
                    + "（"
                    + resolveStatus(deviceSite.getStatus())
                    + "，设备 "
                    + emptyToDefault(deviceSite.getDevice$(), "-")
                    + "，接驳位 "
                    + emptyToDefault(deviceSite.getDeviceCode(), "-")
                    + "，巷道 "
                    + (deviceSite.getChannel() == null ? "-" : deviceSite.getChannel())
                    + "）");
        }
        return String.join("；", parts);
    }

    /**
     * 将站点状态码转成可读文本。
     */
    private String resolveStatus(Integer status) {
        if (status == null) {
            return "未知状态";
        }
        return status == 1 ? "正常" : status == 0 ? "冻结" : "状态" + status;
    }

    /**
     * 统一处理空值展示。
     */
    private String emptyToDefault(String value, String defaultValue) {
        return value == null || value.trim().isEmpty() ? defaultValue : value;
    }
}