// ==UserScript==
// @name         抖音直播间人数显示
// @namespace    https://www.phpnbw.com/
// @version      1.0
// @description  显示抖音直播间实时观看人数
// @author       phpnbw
// @match        *://live.douyin.com/*
// @grant        GM_xmlhttpRequest
// @downloadURL none
// ==/UserScript==

(function() {
    'use strict';

    function getWebcastId() {
        const url = window.location.href;

        // 如果是主页面，直接返回null
        if (url === 'https://live.douyin.com/' || url === 'https://live.douyin.com') {
            console.log('当前是直播主页面，不显示观看人数');
            return null;
        }

        // 尝试匹配两种格式的URL
        // 格式1: https://live.douyin.com/613217711064
        // 格式2: https://live.douyin.com/613217711064?from_tab_name=main
        const match = url.match(/\/(\d+)(?:\?|$)/);

        return match ? match[1] : null;
    }

    async function fetchViewerCount() {
        const webcastId = getWebcastId();
        if (!webcastId) {
            throw new Error('未找到直播间ID');
        }

        return new Promise((resolve, reject) => {
            GM_xmlhttpRequest({
                method: 'GET',
                url: `https://dyapi.phpnbw.com/get_live_room_num?webcast_id=${webcastId}`,
                timeout: 5000,
                onload: function(response) {
                    try {
                        const data = JSON.parse(response.responseText);
                        // 打印获取到的数据
                        console.log('API Response:', data);
                        const viewerCount = data?.data?.data?.data?.[0]?.room_view_stats?.display_value || '未知';
                        console.log('观看人数:', viewerCount); // 打印观看人数
                        resolve(viewerCount);
                    } catch (error) {
                        console.error('解析数据失败:', error);
                        reject(error);
                    }
                },
                onerror: function(error) {
                    console.error('请求失败:', error);
                    reject(error);
                },
                ontimeout: function() {
                    reject(new Error('请求超时'));
                }
            });
        });
    }

    function showViewerCount(count) {
        // 移除旧的显示元素（如果存在）
        const oldDisplay = document.getElementById('viewer-count-display');
        if (oldDisplay) {
            oldDisplay.remove();
        }

        // 创建显示元素
        const display = document.createElement('div');
        display.id = 'viewer-count-display';

        display.style.cssText = `
            display: inline-block;
            background-color: rgba(0, 0, 0, 0.7);
            color: red;
            padding: 5px 10px;
            border-radius: 5px;
            font-size: 16px;
            z-index: 9999;
        `;
        display.textContent = `观看人数: ${count}`;

        // 使用 XPath 找到目标元素
        const xpath = "/html/body/div[2]/div[2]/div/main/div[2]/div/div/div/pace-island/div/div/div[1]/div[1]";
        const targetElement = document.evaluate(xpath, document, null, XPathResult.FIRST_ORDERED_NODE_TYPE, null).singleNodeValue;

        console.log('目标元素是否存在:', !!targetElement);

        if (targetElement) {
            console.log('找到目标元素，尝试添加显示');
            targetElement.style.position = 'relative';
            display.style.marginLeft = '10px';
            targetElement.appendChild(display);
        } else {
            console.log('未找到目标元素，显示在右上角');
            display.style.position = 'fixed';
            display.style.top = '75px';
            display.style.right = '50px';
            document.body.appendChild(display);
        }
    }

    async function updateViewerCount() {
        try {
            const count = await fetchViewerCount();
            showViewerCount(count);
        } catch (error) {
            console.error('获取观看人数失败:', error);
            showViewerCount('获取失败');
        }
    }

    function init() {
        const url = window.location.href;

        // 如果是主页面，直接返回，不执行任何操作
        if (url === 'https://live.douyin.com/' || url === 'https://live.douyin.com') {
            console.log('当前是直播主页面，不显示观看人数');
            return;
        }

        // 只有在不是主页面的情况下才执行更新操作
        setInterval(updateViewerCount, 5000);
        updateViewerCount();
    }

    // 页面加载完成后执行
    if (document.readyState === 'loading') {
        document.addEventListener('DOMContentLoaded', init);
    } else {
        init();
    }
})();