Separate GUI

app/js/downloader.js

@@ -0,0 +1,251 @@
+const path = require("path");
+const fs = require("fs");
+const crypto = require("crypto");
+const http = require("http");
+const https = require("https");
+const progress = require("progress-stream");
+const mime = require("mime");
+const sanitize = require("filenamify");
+
+class Downloader {
+ constructor() {
+ this.type = "";
+ this.id = "";
+ this.url = "";
+ this.aid = -1;
+ this.pid = 1;
+ this.cid = -1;
+ this.name = "";
+ this.links = [];
+ this.downloading = [];
+ }
+
+ getVideoUrl(videoUrl) {
+ this.url = "";
+ const mapping = {
+ "BV": "https://www.bilibili.com/video/",
+ "bv": "https://www.bilibili.com/video/",
+ "av": "https://www.bilibili.com/video/",
+ "ep": "https://www.bilibili.com/bangumi/play/",
+ "ss": "https://www.bilibili.com/bangumi/play/"
+ };
+ for (const [key, value] of Object.entries(mapping)) {
+ if (videoUrl.includes(key)) {
+ this.type = key;
+ this.id = key + videoUrl.split(key)[1];
+ this.url = value + this.id;
+ break;
+ }
+ }
+ }
+
+ getAid() {
+ const { type, url } = this;
+ if (!url) return;
+ fetch(url)
+ .then(response => response.text())
+ .then(result => {
+ let data = result.match(/__INITIAL_STATE__=(.*?);\(function\(\)/)[1];
+ data = JSON.parse(data);
+ console.log("INITIAL STATE", data);
+ if (type === "BV" || type === "bv" || type === "av") {
+ this.aid = data.videoData.aid;
+ this.pid = parseInt(url.split("p=")[1], 10) || 1;
+ this.cid = data.videoData.pages[this.pid - 1].cid;
+ }
+ else if (type === "ep") {
+ this.aid = data.epInfo.aid;
+ this.cid = data.epInfo.cid;
+ }
+ else if (type === "ss") {
+ this.aid = data.epList[0].aid;
+ this.cid = data.epList[0].cid;
+ }
+ this.getInfo();
+ })
+ .catch(error => showError("获取视频 aid 出错！"));
+ }
+
+ getInfo() {
+ const { id, aid, cid } = this;
+ if (!cid) {
+ showError("获取视频 cid 出错！");
+ return;
+ }
+ this.getData();
+ getDanmaku(); //获取cid后，获取下载链接和弹幕信息
+ document.getElementById("cid").textContent = cid;
+ document.getElementById("nav").classList.remove("d-none");
+ document.querySelector("#nav .nav-link").click();
+ fetch("https://api.bilibili.com/x/web-interface/view?aid=" + aid)
+ .then(response => response.json())
+ .then(({ data }) => {
+ console.log("VIDEO INFO", data);
+ document.querySelector("#tab-pane-1 tbody").innerHTML = "";
+ for (let [key, value] of Object.entries(data)) {
+ if (mime.getType(value) && mime.getType(value).includes("image")) { //解析图片地址
+ value = `<a href="${value}" download=""><img src="${value}"></a>`;
+ } else if (typeof value === "object") {
+ value = `<pre>${JSON.stringify(value, null, 2)}</pre>`;
+ }
+ document.querySelector("#tab-pane-1 tbody").insertAdjacentHTML("beforeend", `<tr>
+ <th class="text-capitalize">${key}</th>
+ <td>${value}</td>
+ </tr>`)
+ }
+ this.name = `${id}-${data.title}`;
+ document.getElementById("videoName").value = sanitize(this.name);
+ })
+ .catch(error => showError("获取视频信息出错！"));
+ }
+
+ getData(fallback) {
+ const { cid, type } = this;
+ let playUrl;
+ if (fallback) {
+ const params = `cid=${cid}&module=movie&player=1&quality=112&ts=1`;
+ const sign = crypto.createHash("md5").update(params + "9b288147e5474dd2aa67085f716c560d").digest("hex");
+ playUrl = `https://bangumi.bilibili.com/player/web_api/playurl?${params}&sign=${sign}`;
+ } else {
+ if (type === "BV" || type === "bv" || type === "av") {
+ const params = `appkey=iVGUTjsxvpLeuDCf&cid=${cid}&otype=json&qn=112&quality=112&type=`;
+ const sign = crypto.createHash("md5").update(params + "aHRmhWMLkdeMuILqORnYZocwMBpMEOdt").digest("hex");
+ playUrl = `https://interface.bilibili.com/v2/playurl?${params}&sign=${sign}`;
+ } else {
+ playUrl = `https://api.bilibili.com/pgc/player/web/playurl?qn=80&cid=${cid}`;
+ }
+ }
+ fetch(playUrl)
+ .then(response => response.text())
+ .then(result => {
+ const data = fallback ? this.parseData(result) : JSON.parse(result);
+ const target = data.durl || data.result.durl;
+ console.log("PLAY URL", data);
+ if (target) {
+ const quality = data.quality || data.result.quality,
+ qualityArray = {
+ 112: "高清 1080P+",
+ 80: "高清 1080P",
+ 74: "高清 720P60",
+ 64: "高清 720P",
+ 48: "高清 720P",
+ 32: "清晰 480P",
+ 16: "流畅 360P",
+ 15: "流畅 360P"
+ }; //需要修改，不是一一对应
+ document.getElementById("quality").textContent = qualityArray[quality] || "未知";
+ $("#success").show();
+ fallback ? $("#error").show() : $("#error").hide();
+
+ this.links = [];
+ document.querySelector("#success tbody").innerHTML = "";
+ target.forEach(part => {
+ this.links.push(part.url);
+ document.querySelector("#success tbody").insertAdjacentHTML("beforeend", `<tr>
+ <td>${part.order}</td>
+ <td>${part.length / 1e3}</td>
+ <td>${part.size / 1e6}</td>
+ <td>
+ <div class="form-check">
+ <input class="form-check-input" type="checkbox" checked="true">
+ </div>
+ </td>
+ </tr>`);
+ });
+ } else {
+ if (fallback) throw Error();
+ this.getData(true);
+ }
+ })
+ .catch(error => {
+ showError("获取 PlayUrl 或下载链接出错！由于B站限制，只能下载低清晰度视频。");
+ });
+ }
+
+ parseData(target) {
+ const data = $(target);
+ const result = {};
+ result.durl = [];
+ result.quality = data.find("quality").text();
+ data.find("durl").each((i, o) => {
+ const part = $(o);
+ result.durl.push({
+ url: part.find("url").text(),
+ order: part.find("order").text(),
+ length: part.find("length").text(),
+ size: part.find("size").text()
+ });
+ });
+ return result;
+ }
+
+ downloadAll() {
+ const { cid } = this;
+ let flag = true;
+ document.querySelectorAll("tbody input[type=checkbox]").forEach((element, part) => {
+ if (!element.checked || this.downloading.includes(this.links[part])) return;
+ document.getElementById("download").insertAdjacentHTML("beforeend", `<span>${cid}-${part}</span>
+ <span class="speed"></span>
+ <span class="eta"></span>
+ <span class="addon"></span>
+ <div class="progress mt-1 mb-3">
+ <div class="progress-bar progress-bar-striped progress-bar-animated" role="progressbar" style="width: 0%;">
+ 0%
+ </div>
+ </div>`);
+ this.downloading.push(this.links[part]);
+ ipcRenderer.send("length", this.downloading.filter(item => item !== "").length);
+ flag = false;
+ this.downloadLink(part);
+ });
+ if (flag) showWarning("没有新的视频被下载！");
+ }
+
+ downloadLink(part) {
+ const { name, cid, url } = this;
+ const downloadPath = document.getElementById("downloadPath").value;
+ const filename = document.getElementById("videoName").value || name || cid;
+ const file = path.join(downloadPath, `${sanitize(filename)}-${part}.flv`);
+ fs.stat(file, (error, state) => {
+ const options = {
+ url: this.links[part],
+ headers: {
+ "Range": `bytes=${state ? state.size : 0}-`, //断点续传
+ "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.212 Safari/537.36",
+ "Referer": url
+ }
+ };
+ const downloads = fs.createWriteStream(file, state ? { flags: "a" } : {}),
+ index = this.downloading.indexOf(options.url);
+ this.download(index, options, downloads);
+ if (state) document.querySelectorAll(".addon")[index].textContent = `从 ${Math.round(state.size / 1e6)}MB 处恢复的下载`;
+ //console.log(this.cid, file, options.url);
+ });
+ }
+
+ download(index, options, downloads) {
+ // https://www.npmjs.com/package/progress-stream
+ const proStream = progress({
+ time: 250 //单位ms
+ }).on("progress", progress => {
+ const { speed, eta, percentage } = progress; //显示进度条
+ document.querySelectorAll(".speed")[index].textContent = Math.round(speed / 1e3) + "KB/s";
+ document.querySelectorAll(".eta")[index].textContent = `eta:${eta}s`;
+ const bar = document.querySelectorAll(".progress-bar")[index];
+ bar.style.setProperty("width", percentage + "%")
+ bar.textContent = Math.round(percentage) + "%";
+ if (percentage === 100) {
+ bar.classList.replace("progress-bar-animated", "bg-success");
+ this.downloading[index] = "";
+ ipcRenderer.send("length", this.downloading.filter(item => item !== "").length);
+ }
+ });
+ //先pipe到proStream再pipe到文件的写入流中
+ (options.url.startsWith("https") ? https : http).get(options.url, options, res => {
+ proStream.setLength(res.headers["content-length"]);
+ res.pipe(proStream).pipe(downloads).on("error", error => {
+ console.error(error);
+ });
+ });
+ }
+}