import { EventDispatcher } from './dispatcher';

export class WebSocketClient extends EventDispatcher {
    // #socket链接
    url = '';
    // #socket实例
    socket = null;
    // #重连次数
    reconnectAttempts = 0;
    // #最大重连数
    maxReconnectAttempts = 20;
    // #重连间隔
    reconnectInterval = 60000; // 10 seconds
    // #发送心跳数据间隔
    heartbeatInterval = 1000 * 30;
    // #计时器id
    heartbeatTimer = undefined;
    // #彻底终止ws
    stopWs = false;
    // *构造函数
    constructor(url) {
        super();
        this.url = url;
    }
    // >生命周期钩子
    onopen(callBack) {
        this.addEventListener('open', callBack);
    }
    onmessage(callBack) {
        this.addEventListener('message', callBack);
    }
    onclose(callBack) {
        this.addEventListener('close', callBack);
    }
    onerror(callBack) {
        this.addEventListener('error', callBack);
    }
    // >消息发送
    send(message) {
        if (this.socket && this.socket.readyState === WebSocket.OPEN) {
            this.socket.send({
				data:message,				
			});
        } else {
            console.error('[WebSocket] 未连接');
        }
    }

    // !初始化连接
    connect() {
        if (this.reconnectAttempts === 0) {
            this.log('WebSocket', `初始化连接中...          ${this.url}`);
        }
        if (this.socket && this.socket.readyState === WebSocket.OPEN) {
            return;
        }
        this.socket = uni.connectSocket({
			url: this.url,
			complete: ()=> {}
		});
		this.socket.onOpen((event)=>{			
			this.stopWs = false;
			// 重置重连尝试成功连接
			this.reconnectAttempts = 0;
			// 在连接成功时停止当前的心跳检测并重新启动
			this.startHeartbeat();			
			this.log('WebSocket', `连接成功,等待服务端数据推送[onopen]...     ${this.url}`);
			this.dispatchEvent('open', event);
		})
        // !websocket连接成功
        // this.socket.onOpen = event => {			
        //     this.stopWs = false;
        //     // 重置重连尝试成功连接
        //     this.reconnectAttempts = 0;
        //     // 在连接成功时停止当前的心跳检测并重新启动
        //     this.startHeartbeat();			
        //     this.log('WebSocket', `连接成功,等待服务端数据推送[onopen]...     ${this.url}`);
        //     this.dispatchEvent('open', event);
        // };
		this.socket.onMessage((event)=>{
			this.dispatchEvent('message', event);
			this.startHeartbeat();
		})
   //      this.socket.onMessage = event => {
			// console.log(event)
           // this.dispatchEvent('message', event);
           // this.startHeartbeat();
   //      };
		this.socket.onClose((event)=>{
			if (this.reconnectAttempts === 0) {
			    this.log('WebSocket', `连接断开[onclose]...    ${this.url}`);
			}
			if (!this.stopWs) {
			    this.handleReconnect();
			}
			this.dispatchEvent('close', event);
		})
        // this.socket.onClose = event => {
        //     if (this.reconnectAttempts === 0) {
        //         this.log('WebSocket', `连接断开[onclose]...    ${this.url}`);
        //     }
        //     if (!this.stopWs) {
        //         this.handleReconnect();
        //     }
        //     this.dispatchEvent('close', event);
        // };
		this.socket.onError((event)=>{
			if (this.reconnectAttempts === 0) {
			    this.log('WebSocket', `连接异常[onerror]...    ${this.url}`);
			}
			if (!this.stopWs) {
			    this.handleReconnect();
			}
			this.closeHeartbeat();
			this.dispatchEvent('error', event);
		})
		
		// uni.onSocketError((event)=>{
		// 	console.log("333")
		// 	if (this.reconnectAttempts === 0) {
		// 	    this.log('WebSocket', `连接异常[onerror]...    ${this.url}`);
		// 	}
		// 	if (!this.stopWs) {
		// 	    this.handleReconnect();
		// 	}
		// 	this.closeHeartbeat();
		// 	this.dispatchEvent('error', event);
		// })
        // this.socket.onError = event => {
        //     if (this.reconnectAttempts === 0) {
        //         this.log('WebSocket', `连接异常[onerror]...    ${this.url}`);
        //     }
        //     this.closeHeartbeat();
        //     this.dispatchEvent('error', event);
        // };
    }

    // > 断网重连逻辑
    handleReconnect() {
        if (this.reconnectAttempts < this.maxReconnectAttempts) {
            this.reconnectAttempts++;
            this.log('WebSocket', `尝试重连... (${this.reconnectAttempts}/${this.maxReconnectAttempts})       ${this.url}`);
            setTimeout(() => {
                this.connect();
            }, this.reconnectInterval);
        } else {
			plus.runtime.restart();
            // this.closeHeartbeat();
            // this.log('WebSocket', `最大重连失败，终止重连: ${this.url}`);
        }
    }

    // >关闭连接
    close() {
        if (this.socket) {
            this.stopWs = true;
            this.socket.close();
            this.socket = null;
            this.removeEventListener('open');
            this.removeEventListener('message');
            this.removeEventListener('close');
            this.removeEventListener('error');
        }
        this.closeHeartbeat();
    }

    // >开始心跳检测 -> 定时发送心跳消息
    startHeartbeat() {		
        if (this.stopWs) return;
        if (this.heartbeatTimer) {
            this.closeHeartbeat();
        }
        this.heartbeatTimer = setInterval(() => {
            if (this.socket) {
                this.socket.send({
                	data: JSON.stringify({ type: 'heartBeat', data: {} }),
                	
                });
                this.log('WebSocket', '送心跳数据...');
            } else {
                console.error('[WebSocket] 未连接');
            }
        }, this.heartbeatInterval);
    }

    // >关闭心跳
    closeHeartbeat() {
        clearInterval(this.heartbeatTimer);
        this.heartbeatTimer = undefined;
    }
}