// ==UserScript==
// @name         【伟哥自用】CB站 一键截图
// @description  Chaturbate 按 S 键一键截图当前直播画面，保存为 PNG（主播名@时间戳.png）；视频未出画面/截图失败时有明确提示
// @version      0.3.0
// @author       朱焱伟（移植自 KazurinのB站一键截图）
// @license      MIT
// @namespace    CB@https://bbs.tampermonkey.net.cn
// @match        https://chaturbate.com/*
// @match        https://*.chaturbate.com/*
// @icon         https://www.google.com/s2/favicons?sz=64&domain=chaturbate.com
// @grant        none
// @run-at       document-idle
// @noframes
// ==/UserScript==

/*
 * Chaturbate 伟哥のCB站一键截图（按 S 截图前，建议手动切到 1080p 最高清晰度）
 * 移植自: KazurinのB站一键截图 https://greasyfork.org/zh-CN/scripts/414896-bilibili-一键截图/code
 *
 * 2026-10 修复记录（v0.2.0，站点换播放器导致旧版失效）：
 * 站点把播放器从 video.js（#vjs_video_3_html5_api）换成了自研播放器
 * （styled-components 的 Player__Video 组件，页面直出 <video>，无固定 id）。
 * 旧版两个死点造成「按 S 没反应」：
 *   1. video 元素专用选择器失效，且新播放器自身的 keydown 处理若在冒泡期
 *      stopPropagation，挂在 document 冒泡期的旧监听收不到按键
 *      → 按键监听改到 document 捕获期（比一切页面处理器先拿到 S），video 查找
 *        改为通用策略：顶层文档 + 递归 shadow DOM，多个 video 时优先正在出画面的
 *   2. 找不到视频 / 视频没出画面 / 画布被跨域污染时只 console.log，
 *      用户视角＝毫无反应 → 全部改为右上角 toast 明确反馈（含失败原因）
 * 其他加固：e.key 判键（keyCode 已废弃，仅作老内核兜底）、输入焦点判断补
 * SELECT/contentEditable、长按不重复触发、blob URL 延迟 revoke（防个别浏览器
 * 下载刚发起就被掐断）。
 *
 * v0.3.0：
 *   - 全屏可见：toast 原来挂在 body 上，真全屏时只有全屏元素子树会渲染、
 *     body 上的提示看不见 → 每次显示按 document.fullscreenElement 动态挂到
 *     全屏子树内，并监听 fullscreenchange 在退出全屏时搬回 body
 *     （webkitFullscreenElement / webkitfullscreenchange 兼容 Safari）
 *   - toast 增加文件大小（blob.size，KB/MB 格式化）
 */

(function () {
    'use strict';

    let canvas, ctx, link, toastEl, toastTimer = 0;

    /* ====================== 查找视频 ====================== */

    // 顶层文档 + 递归穿透 open shadow DOM（防播放器再搬进 web component）
    function deepQueryAll(selector, root) {
        root = root || document;
        let found = Array.from(root.querySelectorAll(selector));
        for (const el of root.querySelectorAll('*')) {
            if (el.shadowRoot) found = found.concat(deepQueryAll(selector, el.shadowRoot));
        }
        return found;
    }

    // 多个 video 时的优先级：正在出画面的（videoWidth>0）→ 可见的 → 最后挂载的
    function findVideo() {
        const videos = deepQueryAll('video');
        if (!videos.length) return null;
        return videos.find((v) => v.videoWidth > 0 && v.videoHeight > 0)
            || videos.find((v) => v.offsetWidth > 0 && v.offsetHeight > 0)
            || videos[videos.length - 1];
    }

    /* ====================== toast 反馈 ====================== */

    // toast 的挂载点：真全屏时只有全屏元素子树会渲染，必须挂进去才看得见。
    // 全屏元素是 <video> 本身时子元素不渲染（replaced element），退而求其次挂其父级
    function getToastHost() {
        const fsEl = document.fullscreenElement || document.webkitFullscreenElement || null;
        if (fsEl && fsEl.tagName === 'VIDEO') return fsEl.parentElement || document.body;
        return fsEl || document.body;
    }

    function toast(msg, isError) {
        if (!toastEl) {
            toastEl = document.createElement('div');
            toastEl.style.cssText =
                'position:fixed;top:16px;right:16px;z-index:2147483647;' +
                'padding:10px 16px;border-radius:6px;font-size:14px;color:#fff;' +
                'background:rgba(0,0,0,.8);box-shadow:0 2px 8px rgba(0,0,0,.4);' +
                'pointer-events:none;opacity:0;transition:opacity .3s;';
        }
        const host = getToastHost();
        if (toastEl.parentNode !== host) host.appendChild(toastEl);
        toastEl.textContent = msg;
        toastEl.style.background = isError ? 'rgba(192,57,43,.9)' : 'rgba(0,0,0,.8)';
        toastEl.style.opacity = '1';
        clearTimeout(toastTimer);
        toastTimer = setTimeout(() => { toastEl.style.opacity = '0'; }, 2500);
    }

    // 退出全屏时把 toast 搬回 body，避免残留在已退出全屏的子树里
    // （若播放器容器带 transform/will-change，position:fixed 会被困在错误的包含块里）
    function relocateToast() {
        if (toastEl && toastEl.parentNode) {
            const host = getToastHost();
            if (toastEl.parentNode !== host) host.appendChild(toastEl);
        }
    }
    document.addEventListener('fullscreenchange', relocateToast);
    document.addEventListener('webkitfullscreenchange', relocateToast);

    function formatSize(bytes) {
        if (bytes >= 1024 * 1024) return (bytes / 1024 / 1024).toFixed(2) + ' MB';
        if (bytes >= 1024) return (bytes / 1024).toFixed(1) + ' KB';
        return bytes + ' B';
    }

    /* ====================== 截图与下载 ====================== */

    // 房间 URL 形如 /username/ → 取末段作为主播名
    function getPerformerName() {
        const segments = location.pathname.split('/');
        const last = segments.pop() || segments.pop(); // 容忍末尾斜杠
        return last || 'cb';
    }

    function startDownload(uri, filename) {
        if (!link) {
            link = document.createElement('a');
            link.style.display = 'none';
            document.body.appendChild(link);
        }
        link.href = uri;
        link.download = filename;
        link.click();
    }

    function onError(error) {
        console.error('[CB截图] 截图失败：', error);
        toast('截图失败（详见 F12 控制台）', true);
    }

    function downloadScreenshot(video) {
        if (!canvas) {
            canvas = document.createElement('canvas');
            ctx = canvas.getContext('2d');
        }
        if (!video.videoWidth || !video.videoHeight) {
            toast('视频还没出画面，稍等片刻再按 S', true);
            return;
        }
        canvas.width = video.videoWidth;
        canvas.height = video.videoHeight;
        try {
            ctx.drawImage(video, 0, 0, canvas.width, canvas.height);
        } catch (e) {
            onError(e);
            return;
        }

        const filename = getPerformerName() + '@' + Date.now() + '.png';
        if (typeof canvas.toBlob === 'function') {
            try {
                canvas.toBlob((blob) => {
                    // blob 为空一般是画面受跨域保护（画布被污染）
                    if (!blob) { onError(new Error('toBlob 返回空，画面可能受跨域保护')); return; }
                    const url = URL.createObjectURL(blob);
                    startDownload(url, filename);
                    // 立即 revoke 在个别浏览器会掐断刚发起的下载，延迟回收
                    setTimeout(() => URL.revokeObjectURL(url), 10000);
                    toast('已保存：' + filename + '（' + formatSize(blob.size) + '）');
                }, 'image/png');
            } catch (e) {
                onError(e); // 画布被跨域污染时 SecurityError 从这里抛出
            }
        } else {
            try {
                const dataUri = canvas.toDataURL('image/png');
                startDownload(dataUri, filename);
                // dataURL 是 base64，长度 × 3/4 ≈ 原始字节数
                toast('已保存：' + filename + '（约 ' + formatSize(Math.round(dataUri.length * 3 / 4)) + '）');
            } catch (e) {
                onError(e);
            }
        }
    }

    /* ====================== 快捷键 ====================== */

    function isTypingTarget(el) {
        if (!el) return false;
        const tag = el.tagName;
        return tag === 'INPUT' || tag === 'TEXTAREA' || tag === 'SELECT' || el.isContentEditable;
    }

    // 捕获期监听：比页面/播放器自己的 keydown 处理器先拿到按键，
    // 即使新播放器在冒泡期 stopPropagation 也不影响截图
    document.addEventListener('keydown', (e) => {
        if (e.repeat || e.isComposing || isTypingTarget(e.target)) return;
        const isS = e.key === 's' || e.key === 'S' || e.keyCode === 83; // keyCode 兜底老内核
        if (!isS) return;

        const video = findVideo();
        if (!video) {
            toast('当前页面没有视频', true);
            return;
        }
        e.preventDefault();
        downloadScreenshot(video);
    }, true);
})();