package com.xcong.excoin.modules.gateApi;

import com.xcong.excoin.modules.station.model.GateStatsEvent;
import com.xcong.excoin.modules.station.model.HeartbeatMsg;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.net.InetAddress;
import java.security.MessageDigest;

/**
 * JAR 侧 — 心跳定时发送（每 30s）
 */
@Slf4j
@Component
@EnableScheduling
public class HeartbeatScheduler {

    @Resource
    private StatsEventProducer statsProducer;

    @Resource
    private GateWebSocketClientManager manager;

    private String apiKeyMd5;
    private String hostPort;

    // initialDelay = 0：启动后立即发第一次心跳，避免 Station 要等 30s 才能发现实例
    @Scheduled(fixedRate = 30_000, initialDelay = 1_000)
    public void heartbeat() {
        try {
            if (apiKeyMd5 == null) {
                apiKeyMd5 = md5(manager.getConfig().getApiKey());
                hostPort = resolveHostPort();
            }

            // 策略未启动（gridTradeService == null）时也发送心跳，
            // 否则 Station 无法发现该 JAR 实例，也就无法对它下发 START 指令（死锁）。
            GateGridTradeService strategy = manager.getGridTradeService();
            String state = (strategy != null && strategy.getState() != null)
                    ? strategy.getState().name() : "STOPPED";
            int currentRound = strategy != null ? strategy.getCurrentRound() : 0;
            String cumulativePnl = (strategy != null && strategy.getCumulativePnl() != null)
                    ? strategy.getCumulativePnl().toPlainString() : "0";
            String principal = (strategy != null && strategy.getInitialPrincipal() != null)
                    ? strategy.getInitialPrincipal().toPlainString() : "0";

            HeartbeatMsg hb = HeartbeatMsg.builder()
                    .contract(manager.getConfig().getContract())
                    .state(state)
                    .leverage(manager.getConfig().getLeverage())
                    .currentRound(currentRound)
                    .cumulativePnl(cumulativePnl)
                    .principal(principal)
                    .hostPort(hostPort)
                    .build();

            GateStatsEvent event = statsProducer.newHeartbeat(apiKeyMd5, hb);
            statsProducer.sendHeartbeat(event);

            log.info("[Gate] 心跳已发送, state={}, apiKeyMd5={}", state, apiKeyMd5);
        } catch (Exception e) {
            log.warn("[Gate] 心跳发送失败", e);
        }
    }

    private String resolveHostPort() {
        try {
            String host = InetAddress.getLocalHost().getHostAddress();
            // port 从 Spring 环境变量获取，默认 8888
            String port = System.getProperty("server.port", "8888");
            return host + ":" + port;
        } catch (Exception e) {
            return "unknown";
        }
    }

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