package com.xcong.excoin.modules.gateApi;

import com.xcong.excoin.modules.gateApi.wsHandler.handler.AutoOrdersChannelHandler;
import com.xcong.excoin.modules.gateApi.wsHandler.handler.CandlestickChannelHandler;
import com.xcong.excoin.modules.gateApi.wsHandler.handler.PositionClosesChannelHandler;
import com.xcong.excoin.modules.gateApi.wsHandler.handler.PositionsChannelHandler;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import java.math.BigDecimal;

/**
 * Gate 模块 Spring 容器入口 — 组件组装 + 生命周期管理。
 *
 * <h3>组装顺序（{@code @PostConstruct}）</h3>
 * <ol>
 *   <li>{@link GateConfig} — 构建配置（API 密钥、合约、策略参数）</li>
 *   <li>{@link GateGridTradeService} — init()：获取用户 ID → 切双向持仓 → 清旧条件单 → 平仓 → 设杠杆</li>
 *   <li>{@link GateKlineWebSocketClient} — 注册 6 个频道处理器 → init()：建立 WS 连接并订阅</li>
 *   <li>{@code gridTradeService.startGrid()} — 状态重置，等待首根 K 线</li>
 * </ol>
 *
 * <h3>6 个频道处理器</h3>
 * <ol>
 *   <li>CandlestickChannelHandler — 公开频道，K线 → onKline()</li>
 *   <li>PositionsChannelHandler — 私有频道，仓位 → onPositionUpdate()</li>
 *   <li>PositionClosesChannelHandler — 私有频道，平仓 → onPositionClose()</li>
 *   <li>OrdersChannelHandler — 私有频道，订单成交 → onOrderUpdate()</li>
 *   <li>UserTradesChannelHandler — 私有频道，用户成交 → onUserTrade()</li>
 *   <li>AutoOrdersChannelHandler — 私有频道，条件单状态 → onAutoOrder()</li>
 * </ol>
 *
 * <h3>销毁顺序（{@code @PreDestroy}）</h3>
 * <ol>
 *   <li>gridTradeService.stopGrid()：取消所有条件单 → 关闭交易线程池</li>
 *   <li>wsClient.destroy()：取消订阅 → 断开 WS → 关闭线程池</li>
 * </ol>
 *
 * @author Administrator
 */
@Slf4j
@Component
public class GateWebSocketClientManager {

    // ==================== 修改配置只需改这里 ===================
    static final String DEFAULT_API_KEY = "08d6108fc29378fe64d6f0e89745b3fb";
    static final String DEFAULT_API_SECRET = "4d16e40117477a2344f6fd5872c5b075dc9664a599e65422d3e6791ff64379f1";
    // 不可调参数
    static final String DEFAULT_CONTRACT = "ETH_USDT";
    static final String DEFAULT_LEVERAGE = "100";
    static final String DEFAULT_MARGIN_MODE = "CROSS";
    static final String DEFAULT_POSITION_MODE = "dual";
    static final int DEFAULT_RESTART_GRID_SPAN = 2;
    static final int DEFAULT_PRICE_SCALE = 2;
    static final String DEFAULT_CONTRACT_MULTIPLIER = "0.01";
    static final String DEFAULT_PNL_MODE = "LAST_PRICE";
    static final int DEFAULT_REOPEN_MAX_RETRIES = 3;
    // ==========================================================

    @Autowired
    private GateConfigPersistenceService persistenceService;

    @Autowired
    private GateLogBuffer logBuffer;

    /** WebSocket 连接管理器 */
    private GateKlineWebSocketClient wsClient;
    /** 网格交易策略服务 */
    private GateGridTradeService gridTradeService;
    /** 统一配置 */
    private GateConfig config;

    @PostConstruct
    public void init() {
        logBuffer.info("[管理器] 加载配置，策略待手动启动...");

        try {
            // 优先读取持久化配置，不存在则用默认值创建文件
            GateConfigDTO saved = persistenceService.load(DEFAULT_API_KEY);
            if (saved != null) {
                config = persistenceService.buildFromDTO(saved, DEFAULT_API_KEY);
                logBuffer.info("[管理器] 已加载持久化配置, apiKey=" + mask(DEFAULT_API_KEY));
            } else {
                // 构建默认配置 DTO（只有 7 项可调参数）
                GateConfigDTO defaults = GateConfigDTO.builder()
                        .apiKey(DEFAULT_API_KEY)
                        .gridRate(new BigDecimal("0.005"))
                        .expectedProfit(new BigDecimal("0.15"))
                        .maxLoss(new BigDecimal("1.5"))
                        .baseQuantity("2")
                        .quantity("2")
                        .maxPositionSize(4)
                        .stopLossCount(0)
                        .build();
                // 确保配置文件存在
                persistenceService.ensureExists(DEFAULT_API_KEY, defaults);
                config = persistenceService.buildFromDTO(defaults, DEFAULT_API_KEY);
                logBuffer.info("[管理器] 已创建默认配置文件");
            }

            logBuffer.info("[管理器] 配置就绪，请通过控制面板点击「重启策略」启动");
        } catch (Exception e) {
            logBuffer.error("[管理器] 配置加载失败: " + e.getMessage());
        }
    }

    /**
     * 销毁：停止策略 → 关闭交易线程池 → 取消 WS 订阅 → 断开连接 → 关闭 WS 线程池。
     */
    @PreDestroy
    public void destroy() {
        log.info("[管理器] 开始销毁...");
        if (gridTradeService != null) {
            gridTradeService.stopGrid();
        }
        if (wsClient != null) {
            wsClient.destroy();
        }
        log.info("[管理器] 销毁完成");
    }

    /**
     * @return WebSocket 连接管理器实例
     */
    public GateKlineWebSocketClient getKlineWebSocketClient() { return wsClient; }
    /**
     * @return 网格交易策略服务实例
     */
    public GateGridTradeService getGridTradeService() { return gridTradeService; }
    /**
     * @return 当前生效的配置
     */
    public GateConfig getConfig() { return config; }

    /**
     * 使用新配置重启策略：停止旧策略 → 重建 WS → 重新 init + startGrid。
     */
    public void restartWithConfig(GateConfig newConfig) {
        logBuffer.info("[管理器] 开始策略重启...");
        // 1. 停止旧策略 + 断开旧 WS
        if (gridTradeService != null) {
            gridTradeService.stopGrid();
        }
        if (wsClient != null) {
            wsClient.destroy();
        }
        // 2. 使用新配置重建
        this.config = newConfig;
        this.gridTradeService = new GateGridTradeService(config);
        gridTradeService.init();
        // 3. 重建 WS 客户端并重新注册 Handler
        this.wsClient = new GateKlineWebSocketClient(config.getWsUrl());
        wsClient.addChannelHandler(new CandlestickChannelHandler(config.getContract(), gridTradeService));
        wsClient.addChannelHandler(new PositionsChannelHandler(
                config.getApiKey(), config.getApiSecret(), config.getContract(), gridTradeService));
        wsClient.addChannelHandler(new PositionClosesChannelHandler(
                config.getApiKey(), config.getApiSecret(), config.getContract(), gridTradeService));
        wsClient.addChannelHandler(new AutoOrdersChannelHandler(
                config.getApiKey(), config.getApiSecret(), config.getContract(), gridTradeService));
        gridTradeService.setWsClient(wsClient);
        wsClient.init();
        logBuffer.info("[管理器] WS已重连, Handler 已重新绑定");
        // 4. 启动策略
        gridTradeService.startGrid();
        logBuffer.info("[管理器] 策略已使用新配置重启完成");
    }

    /**
     * 停止策略：取消所有条件单 → 停止网格 → 断开 WS。
     */
    public void stopStrategy() {
        logBuffer.info("[管理器] 停止策略...");
        if (gridTradeService != null) {
            gridTradeService.stopGrid();
        }
        if (wsClient != null) {
            wsClient.destroy();
        }
        logBuffer.info("[管理器] 策略已停止");
    }

    private static String mask(String key) {
        if (key == null || key.length() <= 8) return "***";
        return key.substring(0, 4) + "****" + key.substring(key.length() - 4);
    }
}