/**
 * 使用方法
 * import WebSocket from '@/utils/websocket';
 * const options = {
 *  url: 'ws://127.0.0.1:8080',
 *  onmessage: (res) => {
 *      // 接收數(shù)據(jù)后回調(diào)
 *  },
 *  // 保活周期 10s
 *  timer: 10000,
 *  // 斷線重連
 *  reconnect: true,
 * };
 * const socket = new WebSocket(options);
 * 
 * 手動(dòng)關(guān)閉
 * socket.close();
 */


class Socket {
    /**
     * 構(gòu)造函數(shù)
     * @param {object} params 構(gòu)造函數(shù)參數(shù)
     */
    constructor(params) {
        window.WebSocket = window.WebSocket || window.MozWebSocket;
        if (!window.WebSocket) { // 檢測(cè)瀏覽器支持
            console.error('錯(cuò)誤: 瀏覽器不支持websocket');
            return;
        }
        this.websocket = null;
        this.params = params;
        this.socketInit(params);
    }

    /**
     * 初始化socket
     * @param {string}   url     WebSocket服務(wù)器將響應(yīng)的URL
     * @param {function} onopen  open事件的回調(diào)函數(shù)
     * @param {function} onclose close事件的回調(diào)函數(shù)
     * @param {function} onerror error事件的回調(diào)函數(shù)
     * @param {function} onclose close事件的回調(diào)函數(shù)
     * @param {boolean}  reconnect 是否啟用斷線重連 默認(rèn)為false
     * @param {number}   timer   心跳周期 若為空或0則不啟用
     */
    socketInit({
        url,
        onopen,
        onmessage,
        onerror,
        onclose,
        reconnect = false,
        timer,
    }) {
        if (url !== undefined) {
            // 若不以wss?開(kāi)頭
            if (!/^wss?:\/\//.test(url)) {
                const {
                    protocol
                } = window.location;
                // 則自動(dòng)補(bǔ)上協(xié)議頭
                url = protocol === 'http:' ? 'ws://' + url : 'wss://' + url;
            }
            try {
                this.websocket = new WebSocket(url);
                // 用于指定連接成功后的回調(diào)函數(shù)。
                this.websocket.onopen = (e) => {
                    if (timer > 0) {
                        this.heartCheck(timer);
                    }
                    if (typeof onopen === 'function') {
                        onopen(e);
                    }
                };
                // 用于指定當(dāng)從服務(wù)器接受到信息時(shí)的回調(diào)函數(shù)雳攘。
                this.websocket.onmessage = (e) => {
                    onmessage(e);
                };
                // 用于指定連接關(guān)閉后的回調(diào)函數(shù)。
                this.websocket.onclose = (e) => {
                    reconnect && this.reconnect();
                    if (typeof onclose === 'function') {
                        onclose(e);
                    }
                };
                // 用于指定連接失敗后的回調(diào)函數(shù)汁尺。
                this.websocket.onerror = (e) => {
                    console.log('連接異常', e);
                    reconnect && this.reconnect();
                    if (typeof onerror === 'function') {
                        onerror(e);
                    }
                };
            } catch (e) {
                reconnect && this.reconnect();
            }
        }
    }

    // 發(fā)送消息
    send(message) {
        // readyState 1 表示已經(jīng)鏈接并且可以通訊
        if (!this.websocket || this.websocket.readyState !== 1) {
            console.log('請(qǐng)確認(rèn)websocket是否已經(jīng)鏈接并且可以通訊');
            return;
        }
        this.websocket.send(JSON.stringify(message));
    }

    // 手動(dòng)關(guān)閉socket
    close() {
        this.heartInterval && clearInterval(this.heartInterval);
        this.reconnectTimeout && clearTimeout(this.reconnectTimeout);
        if (!this.websocket) {
            console.log('websocket 不可用');
            return;
        }
        this.websocket.close();
    }

    // 周期性發(fā)送ping 保活
    heartCheck(timer) {
        this.heartInterval = window.setInterval(() => {
            this.send({
                type: 'ping'
            });
        }, timer);
    }

    // 斷線重連
    reconnect() {
        if (this.lockReconnect) return;
        this.lockReconnect = true;
        this.reconnectTimeout && clearTimeout(this.reconnectTimeout);
        this.reconnectTimeout = window.setTimeout(() => {
            this.socketInit(this.params);
            this.lockReconnect = false;
        }, 5000);
    }
}
export default Socket;