import ReconnectingWebSocket from 'reconnecting-websocket'

/**
 * ConnectWebSocket 类
 * TODO: 优化messageHandler: EventEmitter。暂时传入回调函数
 */

class ConnectWebSocket {
	_url
	_socket
	_hasInit
	_messageHandler

	constructor(url) {
		this._url = url
		this._socket = null
		this._hasInit = false
		this._messageHandler = null
	}

	initSocket() {
		if (this._hasInit) {
			return
		}
		if (!this._url) {
			return
		}
		// 会自动重连，无需处理重连逻辑
		this._socket = new ReconnectingWebSocket(this._url, [], {
			maxReconnectionDelay: 20000, // 断开后最大的重连时间： 20s，每多一次重连，会增加 1.3 倍，5 * 1.3 * 1.3 * 1.3...
			minReconnectionDelay: 5000, // 断开后最短的重连时间： 5s
			maxRetries: 5,
		})
		this._hasInit = true

		this._socket.addEventListener('open', this._onOpen.bind(this))
		this._socket.addEventListener('close', this._onClose.bind(this))
		this._socket.addEventListener('error', this._onError.bind(this))
		this._socket.addEventListener('message', this._onMessage.bind(this))
	}

	_onOpen() {
		console.log('ws连接成功')
	}

	_onClose() {
		console.log('ws连接已断开')
	}

	_onError() {
		console.log('ws连接 error')
	}

	registerMessageHandler(messageHandler) {
		this._messageHandler = messageHandler
	}

	_onMessage(msg) {
		// const data = JSON.parse(msg.data);
		this._messageHandler && this._messageHandler(msg.data)
	}

	sendMessage = message => {
		// 目前只供AI聊天使用
		this._socket?.send(message)
		// this._socket?.send(JSON.stringify(message.data));
	}

	close() {
		this._socket?.close()
		this._messageHandler = null
		this._socket = null
	}
}

export default ConnectWebSocket