package cc.mrbird.febs.firewall.nginx;

import cc.mrbird.febs.firewall.entity.FirewallSite;
import cc.mrbird.febs.firewall.mapper.FirewallCountryRuleMapper;
import cc.mrbird.febs.firewall.mapper.FirewallSiteMapper;
import cn.hutool.core.io.FileUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import java.io.File;
import java.nio.charset.StandardCharsets;
import java.util.List;

/**
 * Nginx 国家黑名单配置文件生成器
 * <p>
 * 默认放行所有国家，仅拦截配置中选中的国家（黑名单模式）。
 * <p>
 * 输出文件: /etc/nginx/conf.d/firewall_country_{siteId}.conf
 * <p>
 * 不直接执行 reload，仅负责文件生成。
 * reload 由 NginxReloadTask 消费 Redis 队列后执行，防止误操作。
 *
 * @author auto-generated
 * @date 2026-07-31
 */
@Slf4j
@Component
@RequiredArgsConstructor
public class NginxConfigGenerator {

    private final FirewallCountryRuleMapper mapper;
    private final FirewallSiteMapper siteMapper;

    /** Nginx 配置目录 */
    private static final String NGINX_CONF_DIR = "/etc/nginx/conf.d/";
    /** 配置文件前缀 */
    private static final String CONF_PREFIX = "firewall_country_";

    /**
     * 生成 Nginx 国家黑名单 map 配置
     * <pre>
     * map $geoip_country_code $allow_country {
     *     default 1;
     *     JP 0;
     *     US 0;
     * }
     * </pre>
     * default 1 表示默认放行，列表中国家设为 0 表示拦截。
     * Nginx 中判断: if ($allow_country_1 = 0) { return 403; }
     *
     * @param siteId 站点ID
     */
    public void generate(Long siteId) {
        // 查询站点配置路径
        FirewallSite site = siteMapper.selectById(siteId);
        if (site == null) {
            log.warn("防火墙站点不存在: siteId={}", siteId);
            return;
        }

        List<String> countries = mapper.selectEnableCountry(siteId);

        StringBuilder sb = new StringBuilder();
        sb.append("# Firewall Country Block Map (Blacklist)\n");
        sb.append("# Site: ").append(site.getSiteName()).append(" (").append(site.getDomain()).append(")\n");
        sb.append("# Generated at: ").append(new java.util.Date()).append("\n\n");
        sb.append("map $geoip_country_code $allow_country_").append(siteId).append(" {\n");
        sb.append("    default 1;\n");

        for (String c : countries) {
            sb.append("    ").append(c.toUpperCase()).append(" 0;\n");
        }

        sb.append("}\n");

        // 写入文件
        String confFileName = site.getNginxConf();
        if (confFileName == null || confFileName.isEmpty()) {
            confFileName = NGINX_CONF_DIR + CONF_PREFIX + siteId + ".conf";
        }

        File confFile = new File(confFileName);
        try {
            FileUtil.writeString(sb.toString(), confFile, StandardCharsets.UTF_8);
            log.info("Nginx 黑名单配置已生成: {}, 拦截国家: {}", confFile.getAbsolutePath(), countries);
        } catch (Exception e) {
            log.error("Nginx 配置文件写入失败: path={}", confFile.getAbsolutePath(), e);
            throw new RuntimeException("Nginx 配置文件写入失败: " + e.getMessage(), e);
        }
    }
}