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, strengthData } from "../assets/dataMap";
import { Ref, ref } from "vue";

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

interface SettingsType {
    version: number;
    strengthData: typeof strengthData;
    waveData: typeof waveData;
    guardLevel: number;
    fansMedal: boolean;
}
interface TestSettingsType {
    falloff: number[];
}

let settings: Ref<SettingsType> = ref({
    version: 1,
    waveData: waveData,
    strengthData: strengthData,
    guardLevel: 0,
    fansMedal: false
})

let testSettings: Ref<TestSettingsType> = ref({
    falloff: [0, 0],
})

if (window.localStorage.getItem("settings")) {
    settings.value = JSON.parse(window.localStorage.getItem("settings") || '{}');
    console.log(settings.value)
}
if (window.localStorage.getItem("test")) {
    testSettings.value = JSON.parse(window.localStorage.getItem("test") || '{}');
    console.log(testSettings.value)
}

let waveCounter = ref(0)

/**
 * 转换大航海等级
 * @description 由于B站返回的大航海等级是（1-总督，2-提督，3-舰长）所以要转换一下方便处理
 * @param guardLevel
 */
const transformGuardLevel = (guardLevel: number) => {
    if (guardLevel == 0) {
        return 0
    } else if (guardLevel == 1) {
        return 3
    } else if (guardLevel == 2) {
        return 2
    } else if (guardLevel == 3) {
        return 1
    } else {
        return 0
    }
}

/**
 * 创建socket长连接
 * @param authBody
 * @param wssLinks
 */
function createSocket(authBody: string, wssLinks: string[]) {
    const opt = {
        ...getWebSocketConfig(authBody, wssLinks),
        // 收到消息,
        onReceivedMessage: (res) => {
            // 从本地存储中获取设置
            settings = window.localStorage.getItem("settings") ? ref(JSON.parse(window.localStorage.getItem("settings") || '{}')) : null

            // 粉丝勋章
            let execute_1 = settings.value.fansMedal ? !!res.data.fans_medal_wearing_status : true

            // 大航海
            let execute_2 = transformGuardLevel(res.data.guard_level) >= settings.value.guardLevel

            if (res.cmd == "LIVE_OPEN_PLATFORM_SEND_GIFT" && execute_1 && execute_2) {
                if (settings && res.data.gift_id.toString() === settings.value.strengthData[0]) {
                    // 加强度
                    try {
                        addOrIncrease(2, 1, res.data.gift_num)
                        addOrIncrease(2, 2, res.data.gift_num)
                        notyf.success("收到" + res.data.gift_name + "，强度+" + res.data.gift_num)
                    }
                    catch (e) {
                        console.log(e)
                        notyf.error("强度操作失败！")
                    }
                } else if (settings && res.data.gift_id.toString() === settings.value.strengthData[1]) {
                    // 减强度
                    try {
                        addOrIncrease(1, 1, res.data.gift_num)
                        addOrIncrease(1, 2, res.data.gift_num)
                        notyf.success("收到" + res.data.gift_name + "，强度-" + res.data.gift_num)
                    }
                    catch (e) {
                        console.log(e)
                        notyf.error("强度操作失败！")
                    }
                } else if(settings && settings.value.waveData[res.data.gift_id]) {
                    // 其他礼物，发送波形数据
                    try {
                        sendWaveData(5 * res.data.gift_num, 5 * res.data.gift_num, settings.value.waveData[res.data.gift_id], settings.value.waveData[res.data.gift_id])
                        notyf.success("收到礼物" + res.data.gift_name + "*"+res.data.gift_num)
                        waveCounter.value++
                        if (waveCounter.value % testSettings.value.falloff[0] == 0 && testSettings.value.falloff[0] > 0) {
                            notyf.success("触发强度衰减")
                            addOrIncrease(1, 1, testSettings.value.falloff[1])
                            addOrIncrease(1, 2, testSettings.value.falloff[1])
                        }
                    }
                    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, waveCounter }