package com.xcong.excoin.modules.gateApi.wsHandler.handler;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.xcong.excoin.modules.gateApi.GateGridTradeService;
import com.xcong.excoin.modules.gateApi.wsHandler.AbstractPrivateChannelHandler;
import lombok.extern.slf4j.Slf4j;

import java.math.BigDecimal;

/**
 * 仓位频道处理器。
 * 私有频道，需认证。解析仓位推送后回调 {@link GateGridTradeService#onPositionUpdate}。
 *
 * @author Administrator
 */
@Slf4j
public class PositionsChannelHandler extends AbstractPrivateChannelHandler {

    private static final String CHANNEL_NAME = "futures.positions";

    public PositionsChannelHandler(String apiKey, String apiSecret,
                                    String contract,
                                    GateGridTradeService gridTradeService) {
        super(CHANNEL_NAME, apiKey, apiSecret, contract, gridTradeService);
    }

    @Override
    public boolean handleMessage(JSONObject response) {
        String channel = response.getString("channel");
        if (!CHANNEL_NAME.equals(channel)) {
            return false;
        }
        try {
            JSONArray resultArray = response.getJSONArray("result");
            if (resultArray == null || resultArray.isEmpty()) {
                return true;
            }
            for (int i = 0; i < resultArray.size(); i++) {
                JSONObject pos = resultArray.getJSONObject(i);
                if (!getContract().equals(pos.getString("contract"))) {
                    continue;
                }
                String mode = pos.getString("mode");
                BigDecimal size = new BigDecimal(pos.getString("size"));
                BigDecimal entryPrice = new BigDecimal(pos.getString("entry_price"));
                log.info("[{}] 推送: contract={}, mode={}, size={}, entry_price={}",
                        CHANNEL_NAME, getContract(), mode, size, entryPrice);
                if (getGridTradeService() != null) {
                    getGridTradeService().onPositionUpdate(getContract(), mode, size, entryPrice);
                }
            }
        } catch (Exception e) {
            log.error("[{}] 处理数据失败", CHANNEL_NAME, e);
        }
        return true;
    }
}