完成基本下载功能

YanhuiJessica · YanhuiJessica · commit 71f48faad25b · 2022-03-22T14:32:05.000+08:00
diff --git a/icons/tronclass-dl-32.png b/icons/tronclass-dl-32.png
diff --git a/icons/tronclass-dl-48.png b/icons/tronclass-dl-48.png
diff --git a/js/background.js b/js/background.js
@@ -0,0 +1,20 @@
+function getQueryVariable(query, variable)
+{
+       var query = query.substring(1);
+       var vars = query.split("&");
+       for (let i in vars) {
+            let pair = vars[i].split("=");
+            if(pair[0] == variable){ return pair[1]; }
+       }
+       return false;
+}
+
+chrome.runtime.onMessage.addListener(request => {
+    if (request.from === 'content') {
+        let filename = getQueryVariable(decodeURI(request.url), 'name');
+        chrome.downloads.download({
+            url: request.url,
+            filename: filename,
+        });
+    }
+});
diff --git a/js/content.js b/js/content.js
@@ -0,0 +1,42 @@
+const courseId = window.location.pathname.split('/')[2];
+
+let xhr = new XMLHttpRequest(), xhrDl = new XMLHttpRequest();
+var uploads = [];
+
+xhr.onload = function() {
+    const data = JSON.parse(this.responseText);
+    var coursewares = data['activities'];
+    for (let c of coursewares) {
+        if (c['uploads'].length > 0) {
+            for (let u of c['uploads']) {
+                uploads.push(u);
+            }
+        }
+    }
+};
+xhr.onerror = function() {
+    console.log('An error occurred fetching the JSON');
+};
+
+xhr.open('GET', `http://courses.cuc.edu.cn/api/courses/${courseId}/activities`, true);
+xhr.send();
+
+xhrDl.onload = function() {
+    const data = JSON.parse(this.responseText);
+    var url = data['url'];
+    chrome.runtime.sendMessage({from: 'content', url: url});
+};
+
+function startDownload(reference_id) {
+    xhrDl.open('GET', `http://courses.cuc.edu.cn/api/uploads/reference/document/${reference_id}/url?preview=true`, true);
+    xhrDl.send();
+}
+
+chrome.runtime.onMessage.addListener(function(request, sender, sendResponse) {
+    if (request.from === 'popup' && request.subject === 'getUploads') {
+        sendResponse({uploads: uploads});
+    }
+    if (request.from === 'popup' && request.subject === 'startDownloads') {
+        startDownload(request.reference_id);
+    }
+});
diff --git a/manifest.json b/manifest.json
@@ -0,0 +1,39 @@
+{
+
+  "manifest_version": 2,
+  "name": "CUC-Tronclass-Downloader",
+  "version": "0.0.1",
+
+  "description": "CUC 畅课课件下载（支持所有格式） Download coursewares for CUC Tronclass (all types available)",
+  "homepage_url": "https://github.com/YanhuiJessica/CUC-Tronclass-Downloader",
+
+  "icons": {
+    "32": "icons/tronclass-dl-32.png",
+    "48": "icons/tronclass-dl-48.png"
+  },
+
+  "permissions": [
+    "activeTab", "downloads",
+    "*://courses.cuc.edu.cn/*",
+    "*://cdn.staticfile.org/font-awesome/*"
+  ],
+
+  "browser_action": {
+    "default_icon": {
+      "32": "icons/tronclass-dl-32.png"
+    },
+    "default_title": "Tronclass-dl",
+    "default_popup": "popup/index.html"
+  },
+
+  "content_scripts": [
+    {
+      "matches": ["*://courses.cuc.edu.cn/course/*"],
+      "js": ["js/content.js"]
+    }
+  ],
+
+  "background": {
+    "scripts": ["js/background.js"]
+  }
+}
diff --git a/popup/index.html b/popup/index.html
@@ -0,0 +1,27 @@
+<!DOCTYPE html>
+
+<html>
+    <head>
+        <meta charset="utf-8">
+        <script src="popup.js" type="text/javascript"></script>
+        <link rel="stylesheet" href="https://cdn.staticfile.org/font-awesome/4.7.0/css/font-awesome.min.css">
+    </head>
+    <body>
+        <div id="download">
+            <table>
+                <thead>
+                    <tr>
+                        <th>
+                            <!-- <input type="checkbox" id="select-all"> -->
+                        </th>
+                        <th>课件名</th>
+                        <th style="cursor: pointer;"><i id="dlbtn" class="fa fa-download"></i></th>
+                    </tr>
+                </thead>
+                <tbody id="uploads"></tbody>
+            </table>
+        </div>
+
+        <script src="popup.js" type="text/javascript"></script>
+    </body>
+</html>
diff --git a/popup/popup.js b/popup/popup.js
@@ -0,0 +1,45 @@
+function onError(error) {
+    console.log(`Error: ${error}`);
+}
+
+var uploads;
+window.addEventListener('load', function() {
+    chrome.tabs.query({active: true, currentWindow: true}, function(tabs) {
+        chrome.tabs.sendMessage(tabs[0].id, {from: 'popup', subject: 'getUploads'},
+        function(response) {
+            uploads = response.uploads;
+            var tb = document.getElementById('uploads');
+            if (tb.innerHTML.length == 0) {
+                for (let u in uploads) {
+                    var tr = document.createElement('tr');
+                    tr.innerHTML = `
+                    <td><input type="radio" name="courseware" value="${u}" /></td>
+                    <td>${uploads[u].name}</td>
+                    <td></td>
+                    `;
+                    tb.appendChild(tr);
+                }
+            }
+        });
+    });
+});
+
+document.addEventListener("click", (e) => {
+    if (e.target.id == "dlbtn") {
+        var checkboxes = document.getElementsByName('courseware');
+        var ids = [];
+        for (var i = 0; i < checkboxes.length; i++) {
+            if (checkboxes[i].checked) {
+                ids.push(checkboxes[i].value);
+            }
+        }
+        chrome.tabs.query({active: true, currentWindow: true}, tabs => {
+            for (let i of ids) {
+                chrome.tabs.sendMessage(
+                    tabs[0].id,
+                    {from: 'popup', subject: 'startDownloads', reference_id: uploads[i].reference_id}
+                ).then().catch(onError);
+            }
+        });
+    }
+})
