BLive_Coyote/src/socket/index.ts

import DanmakuWebSocket from "../assets/danmaku-websocket.min.js"
import { Notyf } from 'notyf'
import { closeCoyoteSocket, addOrIncrease, sendWaveData } from "./coyote"
import { waveData } from "../assets/dataMap";

let ws: DanmakuWebSocket
const notyf = new Notyf({ duration: 4000 })

/**
 * 创建socket长连接
 * @param authBody
 * @param wssLinks
 */
function createSocket(authBody: string, wssLinks: string[]) {
    const opt = {
        ...getWebSocketConfig(authBody, wssLinks),
        // 收到消息,
        onReceivedMessage: (res) => {
            console.log("收到"+ res.cmd +"消息：")
            //console.log(res.data.uname + "（大航海" +res.data.guard_level + "级）：" + res.data.msg)

            // if (res.data.msg == "#UPA1") {
            //     try {
            //         addOrIncrease(2, 1, 1)
            //         notyf.success("A通道强度增加成功")
            //     }
            //     catch (e) {
            //         console.log(e)
            //         notyf.error("A通道强度增加失败")
            //     }
            // }

            if (res.cmd == "LIVE_OPEN_PLATFORM_SEND_GIFT") {
                if (res.data.gift_id == 31036) {
                    // 小花花：减强度1
                    try {
                        addOrIncrease(1, 1, 1)
                        addOrIncrease(1, 2, 1)
                        notyf.success("收到花花，强度-1")
                    }
                    catch (e) {
                        console.log(e)
                        notyf.error("强度操作失败！")
                    }
                } else if (res.data.gift_id == 31039) {
                    // 牛哇牛哇：加强度1
                    try {
                        addOrIncrease(2, 1, 1)
                        addOrIncrease(2, 2, 1)
                        notyf.success("收到牛牛，强度+1")
                    }
                    catch (e) {
                        console.log(e)
                        notyf.error("强度操作失败！")
                    }
                } else {
                    // 其他礼物，发送波形数据
                    try {
                        sendWaveData(5, 5, waveData[res.data.gift_id], waveData[res.data.gift_id])
                        notyf.success("收到礼物" + res.data.gift_name)
                    }
                    catch (e) {
                        console.log(e)
                        notyf.error("发送波形数据失败！")
                    }
                }
            }

            console.log(res)
        },
        // 收到心跳处理回调
        onHeartBeatReply: (data) => console.log("收到心跳处理回调:", data),
        onError: (data) => console.log("error", data),
        onListConnectError: () => {
            console.log("list connect error")
            destroySocket()
        },
    }

    if (!ws) {
        ws = new DanmakuWebSocket(opt)
    }

    return ws
}

/**
 * 获取websocket配置信息
 * @param authBody
 * @param wssLinks
 */
function getWebSocketConfig(authBody: string, wssLinks: string[]) {
    const url = wssLinks[0]
    const urlList = wssLinks
    const auth_body = JSON.parse(authBody)
    return {
        url,
        urlList,
        customAuthParam: [
            {
                key: "key",
                value: auth_body.key,
                type: "string",
            },
            {
                key: "group",
                value: auth_body.group,
                type: "string",
            },
        ],
        rid: auth_body.roomid,
        protover: auth_body.protoover,
        uid: auth_body.uid,
    }
}

/**
 * 销毁websocket
 */
function destroySocket() {
    console.log("destroy1")
    ws && ws.destroy()
    ws = undefined
    console.log("destroy2")
}

/**
 * 获取websocket实例
 */
function getWsClient() {
    return ws
}

export { createSocket, destroySocket, getWebSocketConfig, getWsClient }