Providing cookie to Bilibili (#102)

Fix #102
zhxie · Nov 14, 2022 · 7528135 · 7528135
1 parent 416f151
commit 7528135
Show file tree

Hide file tree

Showing 5 changed files with 36 additions and 4 deletions.
diff --git a/server/models/mv/bilibili/entry.js b/server/models/mv/bilibili/entry.js
@@ -8,15 +8,18 @@ class Entry {
   upic;
   page;
   part;
+  cookie;
 
-  constructor(author, bvid, title, pic, upic, page, part) {
+  // Too many props. Should be put into one config prop.
+  constructor(author, bvid, title, pic, upic, page, part, cookie) {
     this.author = author;
     this.bvid = bvid;
     this.title_ = title;
     this.pic = pic;
     this.upic = upic;
     this.page = page;
     this.part = part;
+    this.cookie = cookie;
   }
 
   id = () => {
@@ -52,6 +55,9 @@ class Entry {
   };
 
   downloadOptions = () => {
+    if (this.cookie) {
+      return ["--add-header", `cookie:${this.cookie}`];
+    }
     return [];
   };
 

diff --git a/server/models/mv/bilibili/provider.js b/server/models/mv/bilibili/provider.js
@@ -19,7 +19,9 @@ class Provider {
     return NAME;
   };
 
-  configure = (_config) => {};
+  configure = (config) => {
+    this.cookie = config.cookie;
+  };
 
   search = async (title) => {
     if (!this.cookie) {
@@ -59,7 +61,8 @@ class Provider {
             "http:" + ele["pic"],
             ele["upic"],
             page["page"],
-            part
+            part,
+            this.cookie
           )
         );
       }
@@ -90,7 +93,8 @@ class Provider {
       data["pic"],
       owner["face"],
       page,
-      part
+      part,
+      this.cookie
     );
   };
 }

diff --git a/src/components/SettingsWindow.css b/src/components/SettingsWindow.css
@@ -14,3 +14,7 @@
 .settings-window-form > .ant-form-item {
   margin: 0 0 8px;
 }
+
+.settings-window-form .ant-form-item-label .icon-info {
+  margin-left: 4px;
+}
diff --git a/src/components/SettingsWindow.js b/src/components/SettingsWindow.js
@@ -8,9 +8,11 @@ import {
   Space,
   Switch,
   Tabs,
+  Tooltip,
 } from "antd";
 import React, { useCallback, useEffect, useState } from "react";
 import { useTranslation } from "react-i18next";
+import InfoCircleOutlined from "@ant-design/icons/InfoCircleOutlined";
 import "./SettingsWindow.css";
 import "./Window.css";
 
@@ -41,6 +43,13 @@ const SettingsWindow = (props) => {
     form.submit();
   }, [form]);
 
+  const BilibiliCookiesConfigLabel = (
+    <Tooltip title={t("bilibili_cookies_tooltip")}>
+      <span>{t("cookie")}</span>
+      <InfoCircleOutlined className="icon-info" />
+    </Tooltip>
+  );
+
   return (
     <Space
       className={`settings-window-space ${className}`}
@@ -132,6 +141,12 @@ const SettingsWindow = (props) => {
             >
               <Switch />
             </Form.Item>
+            <Form.Item
+              label={BilibiliCookiesConfigLabel}
+              name={["providers", "mv", "bilibili", "cookie"]}
+            >
+              <Input />
+            </Form.Item>
             <Divider plain orientation="left" style={{ marginTop: 0 }}>
               {t("ncm_mv")}
             </Divider>

diff --git a/src/locales/zh.js b/src/locales/zh.js
@@ -90,6 +90,9 @@ const zh = {
     ncm_lyrics: "网易云音乐 (歌词)",
     petit_lyrics: "プチリリ",
     save: "保存",
+    cookie: "Cookie",
+    bilibili_cookies_tooltip:
+      "哔哩哔哩需要提供 Cookie 才能获取 480P 以上的画质",
   },
 };