package com.xcong.excoin.modules.gateApi;

import com.xcong.excoin.configurations.RabbitMqConfig;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.*;
import org.springframework.context.annotation.DependsOn;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.nio.charset.StandardCharsets;

/**
 * JAR 启动时动态创建专属命令队列 QUEUE_GATE_CMD_{apiKeyMd5}
 */
@Slf4j
@Component
@DependsOn("gateWebSocketClientManager")
public class CommandQueueInitializer {

    @Resource
    private AmqpAdmin amqpAdmin;

    @Resource
    private GateWebSocketClientManager manager;

    private volatile String queueName;

    @PostConstruct
    public void init() {
        try {
            // manager.config 此时已由 Manager 的 @PostConstruct 加载完成
            String apiKey = manager.getConfig().getApiKey();
            String apiKeyMd5 = md5(apiKey);
            String routingKey = "cmd." + apiKeyMd5;
            queueName = "QUEUE_GATE_CMD_" + apiKeyMd5;

            DirectExchange exchange = new DirectExchange(RabbitMqConfig.EXCHANGE_GATE);
            Queue queue = new Queue(queueName, true, false, true); // durable, non-exclusive, auto-delete
            Binding binding = BindingBuilder.bind(queue).to(exchange).with(routingKey);

            amqpAdmin.declareQueue(queue);
            amqpAdmin.declareBinding(binding);

            log.info("[Gate] 命令队列已注册, queue={}, routingKey={}", queueName, routingKey);
        } catch (Exception e) {
            log.error("[Gate] 命令队列注册失败, 策略启停指令将无法接收", e);
            queueName = null;
        }
    }

    /** 返回队列名，供 @RabbitListener 引用 */
    public String getQueueName() {
        return queueName;
    }

    private static String md5(String input) {
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");
            byte[] digest = md.digest(input.getBytes(StandardCharsets.UTF_8));
            StringBuilder sb = new StringBuilder();
            for (byte b : digest) sb.append(String.format("%02x", b));
            return sb.toString();
        } catch (NoSuchAlgorithmException e) {
            return Integer.toHexString(input.hashCode());
        }
    }
}