package com.xcong.excoin.websocket.handler;

import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSONObject;
import com.xcong.excoin.common.contants.AppContants;
import com.xcong.excoin.netty.handler.WebSocketServerHandler;
import com.xcong.excoin.rabbit.producer.FishHitProducer;
import com.xcong.excoin.utils.RedisUtils;
import com.xcong.excoin.websocket.fish.model.MsgModel;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.*;
import org.springframework.web.socket.handler.TextWebSocketHandler;

import javax.websocket.Session;
import java.math.BigDecimal;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @author wzy
 * @date 2021-11-26
 **/
@Slf4j
@Component
public class FishHitWebSocketHandler extends TextWebSocketHandler {

    @Autowired
    private RedisUtils redisUtils;
    @Autowired
    private FishHitProducer fishHitProducer;

    private final Map<Long, WebSocketSession> MEMBER_SESSIONS = new ConcurrentHashMap<>();
    private final Map<String, Long> SID_MID = new ConcurrentHashMap<>();


    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        Object token = session.getAttributes().get("token");
        if (token != null) {
            String redisStr = AppContants.APP_LOGIN_PREFIX + token;
            String memberStr = redisUtils.getString(redisStr);
            if (StrUtil.isBlank(memberStr)) {
                session.sendMessage(new TextMessage("{ code:-1, msg: \"Unauthorized\"}"));
                return;
            }

            JSONObject memberObject = JSONObject.parseObject(memberStr);
            Long memberId = memberObject.getLong("id");
            MEMBER_SESSIONS.put(memberId, session);
            SID_MID.put(session.getId(), memberId);
            session.sendMessage(new TextMessage("{ code:0, msg: \"login succes\" }"));
        }
    }

    /**
     *
     * @param content { type : "hit", fortId : 1, fishType : "fish1" }
     * @param session
     */
    @Override
    public void handleTextMessage(WebSocketSession session, TextMessage content) throws Exception {
        String message = content.getPayload();

        JSONObject reqParam = JSONObject.parseObject(message);
        String type = reqParam.getString("type");
        if (!("hit").equals(type)) {
            session.sendMessage(new TextMessage("{code:-1}"));
            return;
        }

        // 打中鱼奖励金币
        int fishGold = 0;
        String fishType = reqParam.getString("fishType");

        // 如果请求参数中包含鱼类型，且类型不为空，则说明打中
        if (reqParam.containsKey("fishType") && StrUtil.isNotBlank(fishType)) {
            Object fishTypeObj =  redisUtils.hget(AppContants.DICTIONARY_TYPE_FISH, fishType);
            if (fishTypeObj == null) {
                session.sendMessage(new TextMessage("{code : -1, msg : \"fish error\"}"));
                return;
            }

            fishGold = (int) fishTypeObj;
        }

        String fortId = reqParam.getString("fortId");
        if (!reqParam.containsKey("fortId") || fortId == null) {
            return;
        }

        Object fortObj = redisUtils.hget(AppContants.CANNON_TYPE, fortId);
        if (fortObj == null) {
            session.sendMessage(new TextMessage("{code : -1, msg : \"fort error\"}"));
            return;
        }

        // 消耗金币
        BigDecimal consume = (BigDecimal) fortObj;

        MsgModel msg = new MsgModel(consume, fishGold, SID_MID.get(session.getId()), fortId);
        // 发送打渔消息
        fishHitProducer.sendFishHitMsg(JSONObject.toJSONString(msg));
    }

    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        log.info("离开websocket");
        Long memberId = SID_MID.get(session.getId());

        if (memberId != null) {
            MEMBER_SESSIONS.remove(memberId);
        }
        SID_MID.remove(session.getId());
    }
}