/**仅供测试**/
// const WebSocket = require('ws');
// const WebSocketServer = WebSocket.Server;
// const zlib = require('zlib');
const url = require('url');
const fs = require('fs');
const http = require('http');
const { get_sign, getDmHtml } = require('../js/_lib.douyin_sign.cjs');
eval(fs.readFileSync('./js/_lib.douyin_pb.cjs', 'utf8'));

var rule = {
    类型: '影视',
    title: '抖音直播',
    desc: '抖音直播纯js版本',
    host: 'https://live.douyin.com',
    url: '',
    searchUrl: '/search',
    class_name: '射击游戏&竞技游戏&单机游戏&棋牌游戏&休闲益智&角色扮演&策略卡牌&娱乐天地&科技文化',
    class_url: '1$1&2$1&3$1&4$1&5$1&6$1&7$1&10000$3&10001$3',
    filter_def: {},
    filter: ' ',
    searchable: 1,
    quickSearch: 0,
    filterable: 1,
    headers: {},
    timeout: 5000,
    hikerClassListCol: 'movie_2',
    play_parse: true,
    class_parse: async () => {
        return []
    },
    预处理: async function () {
        let ck = (await axios({ url: rule.host })).headers['set-cookie'];
        const regex = /ttwid=([^;]+)/;
        const match = ck[0].match(regex);
        if (match) {
            rule.headers.cookie = match[0];
        }
    },
    一级: async function (tid, pg, filter, extend) {
        let { MY_CATE, MY_FL, MY_PAGE, input } = this;
        let page = 15 * (MY_PAGE - 1);
        let select_partition = MY_FL.sort || MY_CATE;
        const partition = select_partition.split('$')[0];
        const type = select_partition.split('$')[1];
        let url = `https://live.douyin.com/webcast/web/partition/detail/room/?aid=6383&count=15&partition=${partition}&partition_type=${type}&offset=${page}`;
        let html = await request(url, {
            headers: rule.headers
        });
        let d = [];
        let list = JSON.parse(html).data.data;
        list.forEach(it => {
            d.push({
                title: it.room.title,
                desc: it.room.owner.nickname + '(🔥' + it.room.stats.user_count_str + ')',
                img: it.room.cover.url_list[0],
                url: `https://live.douyin.com/webcast/room/web/enter/?aid=6383&live_id=1&device_platform=web&enter_from=web_live&browser_language=zh-CN&browser_platform=Linux+aarch64&browser_name=Chrome&browser_version=58.0.3029.110&web_rid=${it.web_rid}&room_id_str=${it.room.id_str}##${it.room.id_str}`
            })
        })
        return setResult(d);
    },
    二级: async function (ids) {
        let { input } = this;
        let url = input.split('##')[0];
        let room_id = input.split('##')[1];
        let html = await request(url, {
            headers: rule.headers
        });
        const resolutionName = {
            "FULL_HD1": "蓝光",
            "HD1": "超清",
            "ORIGION": "原画",
            "SD1": "标清",
            "SD2": "高清"
        };
        let info = JSON.parse(html).data.data[0];
        const offwss = '#关闭弹幕$off';
        const flv = Object.entries(info.stream_url.flv_pull_url).map(([key, value]) => `${resolutionName[key]}$${value}@@${room_id}`).join('#') + offwss;
        const hls = Object.entries(info.stream_url.hls_pull_url_map).map(([key, value]) => `${resolutionName[key]}$${value}@@${room_id}`).join('#') + offwss;
        let vod = {
            vod_id: input,
            vod_name: info.title,
            vod_pic: info.cover.url_list[0],
            vod_director: info.owner.nickname,
            vod_content: ''
        };
        let playFroms = ['线路_flv', '线路_hls'];
        let playUrls = [flv, hls];
        vod.vod_play_from = playFroms.join('$$$');
        vod.vod_play_url = playUrls.join('$$$');
        return vod;
    },
    搜索: async function (wd, quick, pg) {
        let { input } = this;
        let page = 10 * (pg - 1);
        rule.headers.referer = `${rule.host}/`;
        let url = `https://www.douyin.com/aweme/v1/web/live/search/?device_platform=webapp&aid=6383&channel=channel_pc_web&search_channel=aweme_live&keyword=${wd}&offset=${page}&count=10&os_version=10`;
        let html = await request(url, {
            headers: rule.headers
        });
        let d = [];
        let list = JSON.parse(html).data;
        list.forEach(it => {
            let rawdata = JSON.parse(it.lives.rawdata);
            let web_rid = rawdata.owner.web_rid;
            let id_str = rawdata.id_str;
            let video_feed_tag = rawdata.video_feed_tag;
            let user_count = rawdata.user_count;
            d.push({
                title: rawdata.owner.nickname,
                desc: video_feed_tag + '(' + user_count + ')',
                content: rawdata.title,
                img: rawdata.owner.avatar_large.url_list[0],
                url: `https://live.douyin.com/webcast/room/web/enter/?aid=6383&live_id=1&device_platform=web&enter_from=web_live&browser_language=zh-CN&browser_platform=Linux+aarch64&browser_name=Chrome&browser_version=58.0.3029.110&web_rid=${web_rid}&room_id_str=${id_str}##${id_str}`
            })
        })
        return setResult(d);
    },
    lazy: async function (flag, id, flags) {
        let { input, hostUrl, hostname,getProxyUrl } = this;
        // log('hostUrl:', hostUrl);
        // log('hostname:', hostname);
        if (input === 'off') {
            if (currentWs) {
                console.log("有直播间正在运行弹幕WebSocket服务，即将关闭.");
                currentWs.close();
                // 清除全局变量中的连接，表示没有WebSocket服务正在运行
                currentWs = null;
                startTime = null;
            } else {
                console.log("暂无运行的弹幕WebSocket服务.");
            }
            return;
        } else {
            let url = input.split('@@')[0];
            let room_id = input.split('@@')[1];
            let ttwid = rule.headers.cookie;
            connectWebSocket(hostname, room_id, ttwid);
            // let danmu = `web://${hostUrl}:4201/danmu.html`;
            // return {parse: 0, url: url, danmaku: danmu};
            return { parse: 0, url: url, danmaku: 'web://' + getProxyUrl() + '&url=danmu.html' };
        }
    },
    proxy_rule: async function () {
        let { input, hostname } = this;
        // log('hostname:', hostname);
        if (input) {
            input = decodeURIComponent(input);
            log(`${rule.title}代理播放:${input}`);
            if (input.includes('danmu.html')) {
                const danmuHTML = getDmHtml(hostname);
                return [200, 'text/html', danmuHTML];
            }
        }
    }
};

function generateSignature(wss) {
    const parsedUrl = new url.URL(wss);
    const searchParams = new URLSearchParams(parsedUrl.search);
    const params = ["live_id", "aid", "version_code", "webcast_sdk_version",
        "room_id", "sub_room_id", "sub_channel_id", "did_rule",
        "user_unique_id", "device_platform", "device_type", "ac",
        "identity"
    ];
    const wssMaps = {};
    params.forEach(param => {
        wssMaps[param] = searchParams.get(param) || '';
    });
    const tplParams = params.map(i => `${i}=${wssMaps[i]}`);
    const param = tplParams.join(',');
    let md5_param = md5(param);
    let signature = get_sign(md5_param);
    return signature;
}

// 定义一个全局变量来存储WebSocket连接
let currentWs;
let startTime;

function connectWebSocket(hostname, room_id, ttwid) {
    // 检查是否已经有WebSocket服务正在运行
    if (currentWs) {
        // 如果有，先关闭这个WebSocket服务
        console.log("有正在运行的WebSocket服务.");
        currentWs.close();
        // 清除全局变量中的连接，表示没有WebSocket服务正在运行
        currentWs = null;
    }
    let wss = `wss://webcast5-ws-web-hl.douyin.com/webcast/im/push/v2/?app_name=douyin_web&version_code=180800&webcast_sdk_version=1.0.14-beta.0&update_version_code=1.0.14-beta.0&compress=gzip&device_platform=web&cookie_enabled=true&screen_width=1536&screen_height=864&browser_language=zh-CN&browser_platform=Win32&browser_name=Mozilla&browser_version=5.0%20(Windows%20NT%2010.0;%20Win64;%20x64)%20AppleWebKit/537.36%20(KHTML,%20like%20Gecko)%20Chrome/126.0.0.0%20Safari/537.36&browser_online=true&tz_name=Asia/Shanghai&cursor=d-1_u-1_fh-7392091211001140287_t-1721106114633_r-1&internal_ext=internal_src:dim|wss_push_room_id:${room_id}|wss_push_did:7319483754668557238|first_req_ms:1721106114541|fetch_time:1721106114633|seq:1|wss_info:0-1721106114633-0-0|wrds_v:7392094459690748497&host=https://live.douyin.com&aid=6383&live_id=1&did_rule=3&endpoint=live_pc&support_wrds=1&user_unique_id=7319483754668557238&im_path=/webcast/im/fetch/&identity=audience&need_persist_msg_count=15&insert_task_id=&live_reason=&room_id=${room_id}&heartbeatDuration=0`;
    const signature = generateSignature(wss);
    wss += `&signature=${signature}`;
    const headers = {
        'Cookie': ttwid,
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36'
    };
    startServer(hostname);//启动弹幕服务
    const ws = new WebSocket(wss, {
        headers: headers
    });
    // 存储新的WebSocket连接到全局变量
    currentWs = ws;

    ws.on('open', () => {
        console.log("WebSocket服务已启动.");
    });

    ws.on('message', (message) => {
        let pushframe = new proto.douyin.PushFrame.deserializeBinary(message);
        let payload = pushframe.getPayload_asU8();
        try {
            const buffer = zlib.gunzipSync(payload);
            // console.log('zlib解压成功');
            let response = new proto.douyin.Response.deserializeBinary(buffer);
            let res = response.toObject();
            let frame = pushframe.toObject();
            if (res.needack) {
                let ack = new proto.douyin.PushFrame();
                ack.setLogid(frame.logid);
                ack.setPayloadtype('ack');
                ack.setPayload(Buffer.from(res.internalext, 'utf-8'));
                let data = ack.serializeBinary();
                ws.send(data, {
                    binary: true,
                    mask: true
                }, (err) => {
                    if (err) {
                        // console.error('发送数据时出错:', err);
                        console.error('发送数据时出错:', err.message);
                    } else {
                        //console.log('数据发送成功');
                    }
                });

            }
            let messageslistList = response.getMessageslistList();
            try {
                for (let msg of messageslistList) {
                    let method = msg.getMethod();
                    if (method == 'WebcastChatMessage') {
                        let payload = msg.getPayload_asU8();
                        let chatmessage = new proto.douyin.ChatMessage.deserializeBinary(payload);
                        // let user_name = chatmessage.getUser().getNickname();
                        // let eventTime = chatmessage.getEventtime();
                        let content = chatmessage.getContent();
                        sendDanmuArray(hostname, [{ text: content }]);
                        //   console.log(`【${user_name}】:${content}`);
                    }
                }
            } catch (error) {
                // console.error('message处理错误:', error);
            }
        } catch (error) {
            console.error('解压失败:', error);
        }
    });

    ws.on('error', (error) => {
        console.error('WebSocket error:', error);
        ws.close();
    });

    ws.on('close', () => {
        console.log('WebSocket服务已关闭.');
    });
}


let DMwss;
let isServerStarted = false;

function startServer(hostname) {
    if (isServerStarted) {
        return;
    }
    // 原生的server
    // let server = http.createServer((req, res) => {
    //     if (req.url === '/danmu.html') {
    //         const danmuHTML = getDmHtml(hostname);
    //         res.writeHead(200, {'Content-Type': 'text/html'});
    //         res.end(danmuHTML);
    //     } else {
    //         res.writeHead(404);
    //         res.end('Not Found');
    //     }
    // });
    // fastify的server
    let server = fServer;
    DMwss = new WebSocketServer({ server });
    // 监听客户端连接事件
    DMwss.on('connection', (ws, request) => {
        console.log(`Client connected at ${request.url}`);
        // 检测心跳的定时器
        const interval = setInterval(() => {
            ws.ping(); // 发送 ping 消息
        }, 5000);

        // 接收客户端 pong 消息
        ws.on('pong', () => {
            // console.log('Heartbeat received');
        });

        // 接收消息
        ws.on('message', (message) => {
            console.log('Received message:', message);
            ws.send('ok');
        });

        // 连接关闭时清理
        ws.on('close', () => {
            console.log('Client disconnected');
            clearInterval(interval);
            if (DMwss.clients.size === 0) {
                console.log("关闭所有websocket");
                currentWs.close();
                currentWs = null;
                DMwss.close();
                isServerStarted = false;
            }
        });

        // 连接出错时清理
        ws.on('error', (err) => {
            console.error('WebSocket error:', err);
            clearInterval(interval);
        });
    });
    isServerStarted = true;
    console.log(`[DMwss] WebSocketServer is running on ${hostname}`);
    // 原生启动服务需要在此监听端口，直接用fServer则不需要
    // server.listen(4201, () => {
    //     console.log(`Server is running on ${hostUrl}:4201/`);
    //     isServerStarted = true;
    // });
}

function sendDanmuArray(hostname, danmuArray) {
    // if (!isServerStarted) {
    //     startServer(hostname);
    // }
    const danmuJSON = JSON.stringify(danmuArray);
    DMwss.clients.forEach((client) => {
        if (client.readyState === WebSocket.OPEN) {
            client.send(danmuJSON);
        }
    });
}
