package cc.mrbird.febs.firewall.task;

import cc.mrbird.febs.common.utils.RedisUtils;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

/**
 * Nginx Reload 后台任务
 * <p>
 * 每 10 秒轮询 Redis 队列，有任务时执行 nginx reload。
 * <p>
 * 设计目的：
 * 1. 避免 SpringBoot 直接调用系统命令
 * 2. 支持批量合并 reload（多次修改只 reload 一次）
 * 3. 防止误操作频繁 reload
 *
 * @author auto-generated
 * @date 2026-07-31
 */
@Slf4j
@Component
@RequiredArgsConstructor
public class NginxReloadTask {

    private final RedisUtils redisUtils;

    /** Redis reload 任务队列 key */
    private static final String NGINX_RELOAD_QUEUE_KEY = "firewall:nginx_reload_queue";
    /** reload 脚本路径 */
    private static final String RELOAD_SCRIPT = "/usr/local/bin/nginx_reload.sh";
    /** Nginx 命令（备用） */
    private static final String NGINX_CMD = "nginx -s reload";

    /**
     * 每 10 秒检查并消费 reload 队列
     */
    @Scheduled(fixedDelay = 10_000)
    public void checkAndReload() {
        // 批量消费队列中所有待 reload 任务
        int count = 0;
        while (true) {
            Object item = redisUtils.lGetIndex(NGINX_RELOAD_QUEUE_KEY, 0);
            if (item == null) {
                break;
            }
            count++;
            redisUtils.lRemove(NGINX_RELOAD_QUEUE_KEY, 1, item);
            log.info("消费 Nginx reload 任务: siteId={}", item);
        }

        if (count > 0) {
            log.info("开始执行 Nginx reload (合并 {} 个任务)", count);
            executeReload();
        }
    }

    /**
     * 执行 Nginx reload
     * 优先使用 sudo 脚本，失败时回退到直接命令
     */
    private void executeReload() {
        // 方案1：通过 sudo 脚本执行（推荐，生产环境）
        try {
            Process process = Runtime.getRuntime().exec(new String[]{"sudo", RELOAD_SCRIPT});
            process.waitFor();
            if (process.exitValue() == 0) {
                log.info("Nginx reload 成功 (via sudo script)");
                return;
            }
            log.warn("Nginx reload 脚本执行返回非0: exitCode={}", process.exitValue());
        } catch (Exception e) {
            log.warn("Nginx reload 脚本执行失败，尝试直接命令: {}", e.getMessage());
        }

        // 方案2：直接执行 nginx reload（开发/测试环境）
        try {
            Process process = Runtime.getRuntime().exec(NGINX_CMD);
            process.waitFor();
            if (process.exitValue() == 0) {
                log.info("Nginx reload 成功 (via direct command)");
            } else {
                log.error("Nginx reload 失败: exitCode={}", process.exitValue());
            }
        } catch (Exception e) {
            log.error("Nginx reload 执行异常", e);
        }
    }
}