background.js

/**
 * Created by rumi.zhao on 2017/10/20.
 * 参考： https://developer.chrome.com/docs/extensions/mv3/background_pages/
 * 自从manifest-v3版本以后，background.common 的作用是一个后台的 service worker，它只在需要的时候启动，比如其中监听的事件被触发时
 * 其核心设计逻辑是在后台运行，与网页分开，专注与处理网页外的工作，比如推送通知、丰富的离线支持、后台同步和“添加到主屏幕”，需要注意下面两个问题
 *
 * 首先，service worker 在不使用时终止，需要时重新启动（比如收到事件通知）。
 * 相比于v2版本的background.common，不应再使用局部变量来保持值，service worker 的生命周期很短暂，使用 chrome.storage.local 代替
 * 在service worker 中使用 setTimeout或setInterval方法执行延迟或定期操作可能会失败，因为调度程序会在 Service Worker 终止时取消计时器，使用 chrome.alarms 代替
 *
 * 其次，服务工作者无权访问页面 DOM。
 * Service Worker 无法访问DOMParser API 或创建 iframe 来解析和遍历文档，可以通过 chrome.tabs.create() 创建新选项卡或使用库
 * 创建画布使用离线画布： const canvas = new OffscreenCanvas(width, height);
 *
 * 注意 manifest.json 中配置，在Chrome 93以前， service worker 必须在根目录
 */

importScripts('common/function.js');
importScripts('common/cookie.js');

// 插件安装成功后的回调
chrome.runtime.onInstalled.addListener(() => {
    console.log('on install event.');

    // 可以在这儿绑定右键菜单
    bindContextMenu();
});

/**
 * 监听创建标签
 * chrome.bookmarks.onCreated.addListener(() => {});
 */

/**
 * 监听内容脚本发送的消息
 * 以type区分类型
 * type: download - 下载任务
 */
chrome.runtime.onMessage.addListener(async function(message, sender, sendResponse) {
    console.log(sender.tab ? "from a content script:" + sender.tab.url : "from the extension");
    if (!message.type) {
        console.log("please set the message type. message: {}", message);
        return;
    }

    if (message.type === _u_constant.messageType.downloadTask) {
        await processDownloadUrl(message);
    }
});

chrome.contextMenus.onClicked.addListener((info, tab) => {
    console.log('content menu');
    console.log(info, tab);
});

/**
 * 监听网络请求，拦截一些关心的url数据
 */
chrome.webRequest.onHeadersReceived.addListener(async function (detail) {
    // 请求地址中包含 .m3u8
    if (detail.url.match(/(\.m3u8|\.mp4\?)/)) {

    }
}, {urls: ["<all_urls>"]}, ["responseHeaders"]);

/**
 * 监听请求，获取请求头
 */
chrome.webRequest.onSendHeaders.addListener(async function (detail){
    if (detail.url.match(/(\.m3u8|\.mp4\?)/)) {
        console.log('catch video request on send headers.')
        let m3u8Videos = await _u_api.getStorage(_u_constant.storageKey.m3u8Videos);

        // 兼容处理和初始化
        if (!m3u8Videos || typeof m3u8Videos !== 'object') {
            m3u8Videos = {};
        }

        // 检查当前tab的视频地址，初始化
        if (!m3u8Videos.hasOwnProperty(detail.tabId) || !m3u8Videos[detail.tabId]) {
            m3u8Videos[detail.tabId] = [];
        }

        if (detail.tabId <= 0) {
            console.log('tabId is not valid.');
            return false;
        }

        // 加入当前tab页面下载地址
        const m3u8Video = {
            type: _u_constant.messageType.m3u8UrlAttach,
            m3u8Url: detail.url,
            tabId: detail.tabId,
            initiator: detail.initiator,
            requestId: detail.requestId,
            referer: detail.requestHeaders
        }
        for (const header of detail.requestHeaders) {
            if (header.name === 'Referer') {
                m3u8Video['referer'] = header.value
            }
        }
        console.log('receive video: ', m3u8Video);

        // 检查是否已经有该链接，没有则加入
        const existUrls = m3u8Videos[detail.tabId].map(item => item.m3u8Url);
        if (existUrls.indexOf(m3u8Video.m3u8Url) === -1) {
            m3u8Videos[detail.tabId].push(m3u8Video);
        }
        await _u_api.setStorage(_u_constant.storageKey.m3u8Videos, m3u8Videos);
    }
}, {urls: ["<all_urls>"]}, ["requestHeaders", "extraHeaders"]);

/**
 * 将tabId保存，方便 content_script 获取
 */
chrome.tabs.onUpdated.addListener(async (tabId, changeInfo, tab) => {
    await _u_api.setStorage(_u_constant.storageKey.activeTabId, tabId);
});

function bindContextMenu() {
    chrome.contextMenus.create({
        id: "download_image",
        title: "下载",
        contexts: ["all"]
    });

    chrome.contextMenus.create({
        id: 'open',
        title: 'opem',
        contexts: ['all'],
    });
}

// 处理下载任务
async function processDownloadUrl(message) {
    if (!message || !message.hasOwnProperty('url')) {
        console.log('There is not any need download url.');
        return;
    }

    let cacheDownloadUrls = await _u_api.getStorage(_u_constant.storageKey.downloadUrls) || [];
    if (cacheDownloadUrls.indexOf(message.url) >= 0 && !message.force) {
        console.log('The file has been download: ' + message.url);
        return;
    }

    // 支持相对路径，windows的”下载“文件夹下
    let filename = getFilename(message.url);
    if (message.hasOwnProperty('filename') && message.filename) {
        filename = message.filename;
    }

    filename = filename.replaceAll(/[\/\\:*?"<>|]/g, '-');  // Windows特殊字符替换
    if (message.hasOwnProperty('path')) {
        if (typeof message.path === 'string') {
            filename = message.path + '\\' + filename;
        } else if (Array.isArray(message.path)) {
            filename = message.path.join('\\') + '\\' + filename;
        } else {
            console.log('unknown path: ', message.path);
        }
    }
    console.log('begin download: ' + message.url + ', filename: ' + filename);

    // 通过地址下载文件
    chrome.downloads.download({url: message.url, filename: filename});
    cacheDownloadUrls.push(message.url);

    if (cacheDownloadUrls.length >= 100) {
        // 只记录最近的地址，避免太多，先进先出
        cacheDownloadUrls.shift();
    }

    await _u_api.setStorage(_u_constant.storageKey.downloadUrls, cacheDownloadUrls);
}