package com.iplatform.tcp.client.config;

import com.iplatform.tcp.client.LoginRequest;
import com.walker.infrastructure.utils.JsonUtils;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.drafts.Draft_6455;
import org.java_websocket.handshake.ServerHandshake;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.net.URI;
import java.util.HashMap;
import java.util.Map;

@Configuration
public class WebSocketClientConfig {

    protected final transient Logger logger = LoggerFactory.getLogger(this.getClass());

    private WebSocketClient webSocketClient = null;

    @Bean
    public ClientProperties clientProperties(){
        return new ClientProperties();
    }

    @Bean
    public WebSocketClient webSocketClient(ClientProperties properties) {
//        WebSocketClient webSocketClient = null;
        try {
            webSocketClient = new WebSocketClient(new URI(properties.getUrl()),new Draft_6455()) {
                @Override
                public void onOpen(ServerHandshake handshake) {
                    logger.info("ws 连接成功");
                    // 连接上就发送登录
//                    LoginRequest request = new LoginRequest();
//                    request.setName(properties.getName());
                    String data = null;
//                    try {
//                        data = JsonUtils.objectToJsonString(request);
//                    } catch (Exception e) {
//                        throw new RuntimeException(e);
//                    }

                    Map<String, String> map = new HashMap<>(4);
                    map.put("protocol", "login");
                    map.put("uid", properties.getName());
                    try {
                        data = JsonUtils.objectToJsonString(map);
                    } catch (Exception e) {
                        throw new RuntimeException(e);
                    }
                    webSocketClient.send(data.getBytes());
                    logger.info("客户端发送登录请求，name = {}", properties.getName());
                }

                @Override
                public void onMessage(String message) {
                    logger.info("ws 收到消息"+message);
                }

                @Override
                public void onClose(int code, String reason, boolean remote) {
                    logger.info("ws 退出");
                }

                @Override
                public void onError(Exception ex) {
                    ex.printStackTrace();
                    logger.info("连接错误"+ex.getMessage());
                }
            };
            webSocketClient.connect();
            return webSocketClient;

        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

}