package com.monkeylessey.websocket;

import cn.hutool.core.net.url.UrlBuilder;
import cn.hutool.core.util.URLUtil;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.monkeylessey.framework.utils.TokenUtil;
import com.monkeylessey.sys.domain.vo.SysUserVO;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.http.DefaultHttpRequest;
import io.netty.handler.codec.http.FullHttpRequest;
import io.netty.handler.codec.http.HttpHeaders;
import io.netty.util.AttributeKey;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.util.Map;
import java.util.Objects;


/**
 * URL参数处理程序，这时候连接还是个http请求，没有升级成webSocket协议，此处SimpleChannelInboundHandler泛型使用FullHttpRequest
 *
 * @author Nanase Takeshi
 * @date 2022/5/7 15:07
 */
@Slf4j
@ChannelHandler.Sharable
public class NettyWebSocketParamHandler extends ChannelInboundHandlerAdapter {

    private String secret;

    public NettyWebSocketParamHandler(String secret) {
        this.secret = secret;
    }

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        if (msg instanceof DefaultHttpRequest) {
            // 获取请求头中的身份验证令牌
            DefaultHttpRequest request = (DefaultHttpRequest) msg;
            HttpHeaders headers = request.headers();
            if (headers.size() < 1) {
                ctx.channel().close();
                return;
            }
            String token = headers.get("Sec-WebSocket-Protocol");
            // token验证
//            this.handleToken(token, ctx);
//            ctx.pipeline().remove(this);
//            // 对事件进行传播，知道完成WebSocket连接。
//            ctx.fireChannelRead(msg);
        }
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        cause.printStackTrace();
        ctx.channel().close();
    }

    // 处理token
    private String handleToken(String token, ChannelHandlerContext ctx) {
        if (! StringUtils.hasText(token)) {
            NettyConnect.getChannelGroup().remove(ctx.channel());
            throw new RuntimeException("非法的访问凭证");
        }
        // 获取 userId 参数
        SysUserVO currentUserInfo = null;
        try {
            currentUserInfo = TokenUtil.getCurrentUserInfoStatic(token, secret);
        } catch (JsonProcessingException e) {
            NettyConnect.getChannelGroup().remove(ctx.channel());
            throw new RuntimeException("非法的访问凭证");
        }
        if (Objects.isNull(currentUserInfo)) {
            NettyConnect.getChannelGroup().remove(ctx.channel());
            throw new RuntimeException("用户不存在");
        }
        String userId = currentUserInfo.getId();
        AttributeKey<String> userIdKey = AttributeKey.valueOf("userId");
        ctx.channel().attr(userIdKey).set(userId);
        NettyConnect.getUserChannelMap().put(userId, ctx.channel());
        return userId;
    }

}