✨feat: auto download

Author: xifangczy

CHANGELOG.md

@@ -5,6 +5,8 @@
 [Changed] Heart Beat
 
 [Added] 手机端模拟，手机环境下有更多资源可以被下载。
+
+[Added] 自动下载
 ### 1.0.24
 [Added] 导入/导出配置
 

css/popup.css

@@ -39,7 +39,8 @@ body {
 .url {
   padding: 5px;
 }
-.ico,.icon {
+.ico,
+.icon {
   padding-left: 2px;
   padding-right: 2px;
   transition: all 0.1s;
@@ -62,7 +63,8 @@ body {
   float: right;
   font-weight: bold;
 }
-#Tips {
+#Tips,
+#TipsFixed {
   left: 0;
   right: 0;
   text-align: center;
@@ -76,6 +78,10 @@ body {
   padding: 0 10px;
   margin-bottom: 1px;
 }
+#TipsFixed {
+  position: fixed;
+  display: none;
+}
 #CloseBtn {
   background: #fff;
   border: solid 1px #c7c7c7;
@@ -97,21 +103,32 @@ body {
 #player video {
   max-height: 300px;
 }
-.button {
+.button,
+.button2 {
   background: #fff;
   border: solid 1px #c7c7c7;
   border-radius: 4px;
   color: rgb(26, 115, 232);
   cursor: pointer;
-  padding: 3px 5px 3px 5px;
-  margin: 2px 5px 4px 5px;
+  padding: 3px 3px 3px 3px;
+  margin: 2px 2px 4px 2px;
 }
 .button:hover {
   background: #f5f5f5;
 }
 .button:active {
   background: #ddd;
 }
+.button2 {
+  background: rgb(26 115 232 / 80%);
+  color: rgb(255 255 255);
+}
+.button2:hover {
+  background: rgb(26 115 232);
+}
+.button2:active {
+  background: rgb(26 115 232 / 80%);
+}
 .Tabs {
   display: flex;
 }

js/background.js

@@ -150,7 +150,16 @@ function findMedia(data, isRegex = false, filter = false) {
                 SetIcon({ tips: false });
             }
         }
-        chrome.runtime.sendMessage(info);
+        // 发送到popup 并检查自动下载
+        chrome.runtime.sendMessage(info, function () {
+            if (G.Options.AutoDownTabId.includes(G.tabId)) {
+                let downFileName = G.Options.TitleName ? info.title + '.' + info.ext : info.name;
+                chrome.downloads.download({
+                    url: data.url,
+                    filename: "CatCatch-" + G.tabId + "/" + downFileName
+                });
+            }
+        });
     });
 }
 
@@ -168,25 +177,32 @@ chrome.runtime.onMessage.addListener(function (Message, sender, sendResponse) {
         sendResponse("OK");
         return;
     }
-    if (Message.Message == "getRulesTabId") {
-        sendResponse(G.MobileTabId);
-        return;
-    }
-    if (Message.Message == "OnMobileUserAgent") {
-        OnMobileUserAgent(Message.tabId);
-        chrome.tabs.reload(Message.tabId);
-        sendResponse("OK");
-        return;
-    }
-    if (Message.Message == "OffMobileUserAgent") {
-        OffMobileUserAgent(Message.tabId);
-        chrome.tabs.reload(Message.tabId);
-        sendResponse("OK");
-        return;
-    }
-    if (Message.Message == "HeartBeat") {
-        console.log("HeartBeat OK");
-        sendResponse("HeartBeat OK");
+    switch (Message.Message) {
+        case "getRulesTabId":
+            sendResponse(G.Options.MobileTabId);
+            break;
+        case "OnMobileUserAgent":
+            OnMobileUserAgent(Message.tabId);
+            chrome.tabs.reload(Message.tabId);
+            sendResponse("OK");
+            break;
+        case "OffMobileUserAgent":
+            OffMobileUserAgent(Message.tabId);
+            chrome.tabs.reload(Message.tabId);
+            sendResponse("OK");
+            break;
+        case "getAutoDownTabId":
+            sendResponse(G.Options.AutoDownTabId);
+            break;
+        case "OnAutoDown":
+            OnAutoDown(Message.tabId); sendResponse("OK");
+            break;
+        case "OffAutoDown":
+            sendResponse(OffAutoDown(Message.tabId));
+            break;
+        case "HeartBeat":
+            sendResponse("HeartBeat OK"); console.log("HeartBeat OK");
+            break;
     }
     sendResponse("Error");
 });
@@ -202,7 +218,7 @@ chrome.tabs.onActivated.addListener(function (activeInfo) {
         }
     });
 });
-//标签更新，清除该标签的记录
+// 标签更新 清除数据
 chrome.tabs.onUpdated.addListener(function (tabId, changeInfo) {
     if (changeInfo.status == "loading") {
         chrome.storage.local.get({ MediaData: {} }, function (items) {
@@ -212,13 +228,14 @@ chrome.tabs.onUpdated.addListener(function (tabId, changeInfo) {
         });
     }
 });
-//标签关闭，清除该标签的记录
+// 标签关闭 清除数据
 chrome.tabs.onRemoved.addListener(function (tabId) {
     chrome.storage.local.get({ MediaData: {} }, function (items) {
         delete items.MediaData["tabId" + tabId];
         chrome.storage.local.set({ MediaData: items.MediaData });
     });
     OffMobileUserAgent(tabId);
+    OffAutoDown(tabId);
 });
 
 //检查扩展名以及大小限制
@@ -312,8 +329,10 @@ function SetIcon(obj) {
         chrome.action.setTitle({ title: "抓到 " + obj.number + " 条鱼", tabId: obj.tabId });
     }
 }
+// 手机端模拟
 function OnMobileUserAgent(tabId) {
-    G.MobileTabId.push(tabId);
+    G.Options.MobileTabId.push(tabId);
+    chrome.storage.sync.set({ MobileTabId: G.Options.MobileTabId });
     chrome.declarativeNetRequest.updateSessionRules({
         removeRuleIds: [tabId],
         addRules: [{
@@ -327,18 +346,34 @@ function OnMobileUserAgent(tabId) {
                 }]
             },
             "condition": {
-                "tabIds": G.MobileTabId,
+                "tabIds": G.Options.MobileTabId,
                 "resourceTypes": ["main_frame", "sub_frame", "stylesheet", "script", "image", "font", "object", "xmlhttprequest", "ping", "csp_report", "media", "websocket", "webtransport", "webbundle", "other"]
             }
         }]
     });
 }
 function OffMobileUserAgent(tabId) {
-    const index = G.MobileTabId.indexOf(tabId);
+    const index = G.Options.MobileTabId.indexOf(tabId);
     if (index > -1) {
-        G.MobileTabId.splice(index, 1);
+        G.Options.MobileTabId.splice(index, 1);
     }
     chrome.declarativeNetRequest.updateSessionRules({
         removeRuleIds: [tabId]
     });
+    chrome.storage.sync.set({ MobileTabId: G.Options.MobileTabId });
+}
+// 自动下载
+function OnAutoDown(tabId) {
+    G.Options.AutoDownTabId.push(tabId);
+    chrome.storage.sync.set({ AutoDownTabId: G.Options.AutoDownTabId });
+}
+function OffAutoDown(tabId) {
+    if (G.Options.AutoDownTabId.length == 0) { return false; }
+    const index = G.Options.AutoDownTabId.indexOf(tabId);
+    if (index > -1) {
+        G.Options.AutoDownTabId.splice(index, 1);
+        chrome.storage.sync.set({ AutoDownTabId: G.Options.AutoDownTabId });
+        return true;
+    }
+    return false;
 }

js/init.js

@@ -10,8 +10,19 @@ chrome.tabs.query({ active: true, currentWindow: true }, function (tabs) {
 //设置参数
 G.Options = new Object();
 //所有设置变量
-G.OptionLists = ["Ext", "Debug", "TitleName", "OtherAutoClear", "Potplayer", "Type", "Regex", "ShowWebIco", "MobileUserAgent"];
-G.MobileTabId = [];
+G.OptionLists = [
+    "Ext",
+    "Debug",
+    "TitleName",
+    "OtherAutoClear",
+    "Potplayer",
+    "Type",
+    "Regex",
+    "ShowWebIco",
+    "MobileUserAgent",
+    "MobileTabId",
+    "AutoDownTabId"
+];
 
 // Init
 InitOptions();
@@ -56,8 +67,7 @@ function GetDefault(Obj) {
         { "type": "image/*", "size": 0, "state": false }
     );
     const defaultRegex = new Array(
-        { "type": "ig", "regex": "video\\.weibocdn\\.com.*\\.mp4", "state": false },
-        { "type": "ig", "regex": "btrace\\.video\.qq\\.com.*vurl=([^&]*)", "state": true }
+        { "type": "ig", "regex": ".*vurl=([^&]*)", "state": true }
     );
     switch (Obj) {
         case "Ext": return defaultExt;
@@ -70,6 +80,8 @@ function GetDefault(Obj) {
         case "ShowWebIco": return false;
         case "MobileUserAgent":
             return "Mozilla/5.0 (iPhone; CPU iPhone OS 13_2_3 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/13.0.3 Mobile/15E148 Safari/604.1";
+        case "MobileTabId": return [];
+        case "AutoDownTabId": return [];
     }
 }
 //初始变量

js/popup.js

@@ -43,10 +43,7 @@ function AddMedia(data) {
     }
 
     //添加下载文件名
-    let downFileName = data.name;
-    if (G.Options.TitleName) {
-        downFileName = data.title + '.' + data.ext;
-    }
+    let downFileName = G.Options.TitleName ? data.title + '.' + data.ext : data.name;
 
     // 文件大小单位转换
     if (data.size) {
@@ -192,7 +189,7 @@ $(function () {
     });
     //下载选中文件
     $('#DownFile').click(function () {
-        let FileNum = $('.TabShow :checked').size();
+        let FileNum = $('.TabShow :checked').length;
         if (FileNum >= 10 && !confirm("共 " + FileNum + "个文件，是否确认下载?")) {
             return;
         }
@@ -202,7 +199,7 @@ $(function () {
     });
     //复制选中文件
     $('#AllCopy').click(function () {
-        let count = $('.TabShow :checked').size();
+        let count = $('.TabShow :checked').length;
         if (count == 0) { return false };
         let url = '';
         $('.TabShow :checked').each(function () {
@@ -263,6 +260,27 @@ $(function () {
             $('#Clear').click();
         });
     });
+
+    // 自动下载
+    chrome.runtime.sendMessage({ Message: "OffAutoDown", tabId: G.tabId }, function(result){
+        if(result){
+            Tips("已关闭自动下载", 500);
+        }
+    });
+    $("#AutoDown").click(function () {
+        let switchStatus = $(this).data("switch");
+        if (switchStatus == "OnAutoDown") {
+            if (confirm("找到资源立刻尝试下载\n开启后，点击扩展图标会关闭自动下载\n是否确认开启？")) {
+                chrome.runtime.sendMessage({ Message: switchStatus, tabId: G.tabId });
+                $("#AutoDown").html("关闭下载");
+                $("#AutoDown").data("switch", "OffAutoDown");
+            }
+        } else {
+            chrome.runtime.sendMessage({ Message: switchStatus, tabId: G.tabId });
+            $("#AutoDown").html("自动下载");
+            $("#AutoDown").data("switch", "OnAutoDown");
+        }
+    });
 });
 
 //html5播放器允许格式
@@ -290,11 +308,6 @@ function UItoggle() {
     $("#otherQuantity").text("[" + length + "]");
 }
 
-function Tips(text) {
-    let Original = $('#Tips').html();
-    $('#Tips').css("position", "fixed");
-    $('#Tips').html(text).fadeIn(500).delay(200).fadeOut(500, function () {
-        $(this).css("position", "static");
-        $('#Tips').html(Original);
-    });
+function Tips(text, delay = 200) {
+    $('#TipsFixed').html(text).fadeIn(500).delay(delay).fadeOut(500);
 }

popup.html

@@ -19,6 +19,7 @@
       </div>
     </div>
     <div id="Tips">还没闻到味儿~</div>
+    <div id="TipsFixed">~</div>
     <div id="mediaList" class="mediaList TabShow hide"></div>
     <div id="otherMediaList" class="mediaList hide"></div>
     <div id="down">
@@ -27,8 +28,9 @@
       <div id="AllSelect" class="button">全选</div>
       <div id="ReSelect" class="button">反选</div>
       <div id="Clear" class="button">清空</div>
-      <div id="MobileUserAgent" class="button" data-switch="OnMobileUserAgent">模拟手机端</div>
-      <div id="Options" class="button">设置</div>
+      <div id="MobileUserAgent" class="button2" data-switch="OnMobileUserAgent">模拟手机端</div>
+      <div id="AutoDown" class="button2"  data-switch="OnAutoDown">自动下载</div>
+      <div id="Options" class="button2">设置</div>
     </div>
     <div id="player" class="hide">
       <video controls width="100%"></video>