package com.iplatform.tcp.config;

import com.iplatform.base.config.TcpProperties;
import com.iplatform.core.config.LoadBalanceProperties;
import com.iplatform.tcp.EngineType;
import com.iplatform.tcp.support.TestConnectionCallback;
import com.iplatform.tcp.support.WebSocketPush;
import com.iplatform.tcp.util.ws.LoginAction;
import com.walker.infrastructure.ApplicationRuntimeException;
import com.walker.push.PushManager;
import com.walker.push.Pushable;
import com.walker.queue.QueueManager;
import com.walker.tcp.ConnectionCallback;
import com.walker.tcp.ConnectionManager;
import com.walker.tcp.ProtocolResolver;
import com.walker.tcp.netty.WebSocketEngine;
import com.walker.tcp.netty.WebSocketNettyHandler;
import com.walker.tcp.netty.WebSocketServerInitializer;
import com.walker.tcp.websocket.WebsocketHandler;
import com.walker.tcp.websocket.WebsocketProtocolResolver;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.lang.Nullable;

import java.util.ArrayList;
import java.util.List;

@Configuration
@ConditionalOnProperty(prefix = "iplatform.tcp", name = "enabled", havingValue = "true", matchIfMissing = false)
public class WebsocketConfig {

    @Bean
    public WebsocketProtocolResolver websocketProtocolResolver(){
        WebsocketProtocolResolver websocketProtocolResolver =  new WebsocketProtocolResolver();
        websocketProtocolResolver.setPermitNotRegisterConnect(false);
        return websocketProtocolResolver;
    }

    private List<ProtocolResolver<?>> acquireProtocolResolverList(){
        List<ProtocolResolver<?>> protocolResolverList = new ArrayList<>(2);
        protocolResolverList.add(websocketProtocolResolver());
        return protocolResolverList;
    }

    /**
     * 配置业务定义的处理方法实现。
     * @param connectionManager
     * @param queueManager
     * @return
     * @date 2023-04-17
     */
    @Bean
    public WebsocketHandler wsServerHandler(ConnectionManager connectionManager
            , QueueManager queueManager, TcpProperties tcpProperties){
        WebsocketHandler websocketHandler = new WebsocketHandler();
        websocketHandler.setEmptyMsgDisconnect(false);
        websocketHandler.setEngineId(EngineType.INDEX_TCP_WEBSOCKET);
        websocketHandler.setConnectionManager(connectionManager);
        websocketHandler.setQueueManager(queueManager);
        websocketHandler.setProtocolResolverList(this.acquireProtocolResolverList());
//        websocketHandler.setScanPackages("com.iplatform.tcp.util.ws");
        websocketHandler.setScanPackages(tcpProperties.getScanPackagesWs());
        // 2023-09-27
        websocketHandler.setConnectionHost(tcpProperties.getConnectionHost());
        return websocketHandler;
    }

    @Bean
    public WebSocketNettyHandler wsNettyLongHandler(ConnectionManager connectionManager
            , WebsocketHandler wsServerHandler, TcpProperties tcpProperties){
        WebSocketNettyHandler webSocketNettyHandler = new WebSocketNettyHandler();
        webSocketNettyHandler.setConnectionManager(connectionManager);
        webSocketNettyHandler.setTcpServerHandler(wsServerHandler);
        webSocketNettyHandler.setUri(tcpProperties.getWebsocketUri());
        return webSocketNettyHandler;
    }

    @Bean
    public WebSocketServerInitializer wsServerInitializer(WebSocketNettyHandler wsNettyLongHandler, TcpProperties tcpProperties){
        WebSocketServerInitializer webSocketServerInitializer = new WebSocketServerInitializer();
        webSocketServerInitializer.setHandler(wsNettyLongHandler);
        webSocketServerInitializer.setProtocolResolverList(this.acquireProtocolResolverList());
        webSocketServerInitializer.setShowLog(tcpProperties.isShowLog());
        return webSocketServerInitializer;
    }

    @Bean
    public WebSocketEngine webSocketEngine(ConnectionManager connectionManager
            , WebSocketServerInitializer webSocketServerInitializer
            , TcpProperties tcpProperties, @Nullable ConnectionCallback connectionCallback
            , LoadBalanceProperties loadBalanceProperties){
        WebSocketEngine webSocketEngine = new WebSocketEngine();
        webSocketEngine.setId(EngineType.INDEX_TCP_WEBSOCKET);
        webSocketEngine.setName("websocket测试引擎");
        webSocketEngine.setConnectionManager(connectionManager);
        webSocketEngine.setServerInitializer(webSocketServerInitializer);
        webSocketEngine.setOpenHeartBeat(tcpProperties.isOpenHeartBeat());
        webSocketEngine.setBossThreadNum(tcpProperties.getBossThreadNum());
        webSocketEngine.setWorkerThreadNum(tcpProperties.getWorkerThreadNum());
//        webSocketEngine.setHeartBeatTimeInterval(180000);
        webSocketEngine.setHeartBeatTimeInterval(tcpProperties.getHeartBeatSeconds() * 1000);
        webSocketEngine.setPort(tcpProperties.getPortWs());
        // 2023-08-25
        if(connectionCallback != null){
            webSocketEngine.setConnectionCallback(connectionCallback);
        } else {
            webSocketEngine.setConnectionCallback(new TestConnectionCallback());
        }
        // 延时启动，测试用。正式注释掉。2023-09-27
        webSocketEngine.setEngineStartDelaySeconds(30);
        try {
            webSocketEngine.start();
        } catch (Exception e) {
            throw new ApplicationRuntimeException("启动'WebSocketEngine'失败：" + e.getMessage(), e);
        }
        return webSocketEngine;
    }

    /**
     * Websocket登录认证动作配置
     * @return
     * @date 2023-04-17
     */
    @Bean
    public LoginAction wsLoginAction(){
        LoginAction loginAction = new LoginAction();
        return loginAction;
    }

    /**
     * 配置'WebSocketPush'推送实现，同时注册到管理器中。
     * @param webSocketEngine
     * @param pushManager
     * @return
     * @date 2023-04-21
     */
    @Bean
    public Pushable websocketPush(WebSocketEngine webSocketEngine, PushManager pushManager){
        WebSocketPush webSocketPush = new WebSocketPush();
        webSocketPush.setWebSocketEngine(webSocketEngine);
        webSocketPush.startup();
        pushManager.register(webSocketPush);
        return webSocketPush;
    }

}