// ==UserScript==
// @name:en-US           CCTV-HLS
// @name                 CCTV视频解析
// @description:en-US    parse cctv video to hls url.
// @description          将CCTV视频解析成HLS地址.
// @namespace            https://greasyfork.org/users/135090
// @version              1.5.0
// @author               [ZWB](https://greasyfork.org/zh-CN/users/863179)
// @license              CC
// @grant                none
// @run-at               document-end
// @match                *://*.cctv.com/*/*/*/V*.shtml*
// @match                *://*.cctv.com/*/*/*/A*.shtml*
// @match                *://*.cctv.cn/*/*/*/V*.shtml*
// @match                *://*.cctv.cn/*/*/*/A*.shtml*
// @match                *://vdn.apps.cntv.cn/api/getHttpVideoInfo*
// @icon                 https://tv.cctv.cn/favicon.ico
// @downloadURL none
// ==/UserScript==

(async function () {
    if (location.hostname.indexOf("cctv.com") > 0 || location.hostname.indexOf("cctv.cn") > 0) {
        if (vodh5player?.playerList?.length > 1 ) {
            vodh5player?.playerList?.forEach((i,n)=>{
                let guid = i?.options_?.paras?.videoId;
                let base = "https://vdn.apps.cntv.cn";
                let pathname = "/api/getHttpVideoInfo.do";
                let apihref = base + pathname + `?client=flash&im=0&pid=${guid}`;
                let bts = n * 40 + 20;
                let btn = document.createElement("a");
                btn.href = apihref;
                btn.id = "btn" + n;
                btn.type = "button";
                btn.target = "_blank";
                btn.textContent = "点击跳转到下载页"+(n+1);
                btn.style = `
                position: fixed;
                z-index: 999;
                bottom: ${bts}px;
                right: 20px;
                background-color: #f86336;
                color: white;
                padding: 5px;
                border: none;
                cursor: pointer;
                font-size: 16px;
                `;
                document.body.appendChild(btn);
            });
            
        } else if (vodh5player?.playerList?.length == 1 ){
            let guid = vodh5player?.playerList[0]?.options_?.paras?.videoId;
            let base = "https://vdn.apps.cntv.cn";
            let pathname = "/api/getHttpVideoInfo.do";
            let apihref = base + pathname + `?client=flash&im=0&pid=${guid}`;
            let btn = document.createElement("a");
            btn.href = apihref;
            btn.id = "btn";
            btn.type = "button";
            btn.target = "_blank";
            btn.textContent = "点击跳转到下载页";
            btn.style = `
            position: fixed;
            z-index: 999;
            bottom: 20px;
            right: 20px;
            background-color: #f86336;
            color: white;
            padding: 5px;
            border: none;
            cursor: pointer;
            font-size: 16px;
            `;
            document.body.appendChild(btn);
        }
    }

    if (location.hostname.indexOf("vdn.apps.cntv.cn") > -1) {
        var data = JSON.parse(document?.body?.textContent);
        var hlsUrl = data?.hls_url?.replaceAll("main", "850");
        var title = data?.title?.replaceAll(" ","");
        if (data?.play_channel?.indexOf("4K") > 0) {
            hlsUrl = data?.hls_url.replaceAll("main", "4000");
        } else {
            const response = await fetch(hlsUrl);
            if (!response.ok) {
                if (response.status === 404) {
                    hlsUrl = data?.hls_url?.replaceAll("main", "450");
                }
                console.log(`HTTP状态: ${response.status}`);
            }
        }
        console.info(hlsUrl);
        await downloadM3U8Video(hlsUrl, title + '.ts', {
            onProgress: (current, total) => {
                var cotp = `${Math.round((current / total) * 100)}`;
                document.body.textContent = title + "---下载进程" + cotp + "%";
                console.info(`Progress: ${current}/${total} (${cotp}%)`);
            }
        });
    }

    async function downloadM3U8Video(m3u8Url, outputFilename = 'video.m2t', options = {}) {
        try {
            // 1. 获取并解析M3U8文件
            //void (alert('开始'));
            const response = await fetch(m3u8Url);
            if (!response.ok) throw new Error(`Failed to fetch M3U8: ${response.status}`);

            const m3u8Content = await response.text();
            const lines = m3u8Content.split('\n');
            const baseUrl = m3u8Url.substring(0, m3u8Url.lastIndexOf("/") + 1);
            const segments = [];

            // 解析TS分片URL
            for (const line of lines) {
                if (line && !line.startsWith('#') && (line.endsWith('.ts') || line.match(/\.ts\?/))) {
                    const segmentUrl = line.startsWith('http') ? line : new URL(line, baseUrl).href;
                    segments.push(segmentUrl);
                    // return;
                }
            }

            if (segments.length === 0) throw new Error('No TS segments found in the M3U8 file');
            console.log(`Found ${segments.length} TS segments`);

            // 2. 下载所有分片
            console.log('Downloading segments...');
            const blobs = [];
            const { onProgress } = options;

            for (let i = 0; i < segments.length; i++) {
                try {
                    const segmentResponse = await fetch(segments[i]);
                    if (!segmentResponse.ok) throw new Error(`Failed to fetch segment: ${segmentResponse.status}`);

                    const blob = await segmentResponse.blob();
                    blobs.push(blob);

                    // 调用进度回调
                    if (typeof onProgress === 'function') {
                        onProgress(i + 1, segments.length);
                    }
                } catch (error) {
                    console.error(`Error downloading segment ${segments[i]}:`, error);
                    throw error; // 可以选择继续或抛出错误
                }
            }

            // 3. 合并并下载
            console.log('Merging and downloading...');
            const mergedBlob = new Blob(blobs, { type: 'video/mp2t' });
            const url = URL.createObjectURL(mergedBlob);

            const a = document.createElement('a');
            a.href = url;
            a.download = outputFilename;
            document.body.appendChild(a);
            a.click();

            // 清理
            setTimeout(() => {
                document.body.removeChild(a);
                URL.revokeObjectURL(url);
            }, 100);

            console.log('Download completed!');
            return true;
        } catch (error) {
            console.error('Error downloading M3U8 video:', error);
            throw error;
        }
    }

})();