From 6eef3e3495d1f232d7d9bcfd75078e2d096dce68 Mon Sep 17 00:00:00 2001
From: hdk5 <hdk5@hdk5.xyz>
Date: Fri, 8 Nov 2024 23:17:48 +0200
Subject: [PATCH] [bilibili] initial support (#2824)

---
 gallery_dl/extractor/__init__.py |   1 +
 gallery_dl/extractor/bilibili.py | 102 +++++++++++++++++++++++++++++++
 2 files changed, 103 insertions(+)
 create mode 100644 gallery_dl/extractor/bilibili.py

diff --git a/gallery_dl/extractor/__init__.py b/gallery_dl/extractor/__init__.py
index f80b629a..90337021 100644
--- a/gallery_dl/extractor/__init__.py
+++ b/gallery_dl/extractor/__init__.py
@@ -30,6 +30,7 @@ modules = [
     "batoto",
     "bbc",
     "behance",
+    "bilibili",
     "blogger",
     "bluesky",
     "boosty",
diff --git a/gallery_dl/extractor/bilibili.py b/gallery_dl/extractor/bilibili.py
new file mode 100644
index 00000000..0be8284c
--- /dev/null
+++ b/gallery_dl/extractor/bilibili.py
@@ -0,0 +1,102 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2024 Mike Fährmann
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 2 as
+# published by the Free Software Foundation.
+
+"""Extractors for https://www.bilibili.com/"""
+
+from .common import Extractor, Message
+from .. import text, exception
+
+
+class BilibiliExtractor(Extractor):
+    """Base class for bilibili extractors"""
+    category = "bilibili"
+    root = "https://www.bilibili.com"
+
+    def __init__(self, match):
+        Extractor.__init__(self, match)
+        self.item = match.group(match.lastindex)
+
+    def _init(self):
+        self.api = BilibiliAPI(self)
+
+
+class BilibiliUserArticlesExtractor(BilibiliExtractor):
+    """Extractor for all articles of an user"""
+    subcategory = "user"
+    pattern = r"(?:https?://)?space\.bilibili\.com/(\d+)/article"
+    example = "https://space.bilibili.com/12345/article"
+
+    def items(self):
+        for article in self.api.user_articles(self.item):
+            article["_extractor"] = BilibiliArticleExtractor
+            url = "{}/opus/{}".format(self.root, article["opus_id"])
+            yield Message.Queue, url, article
+
+
+class BilibiliArticleExtractor(BilibiliExtractor):
+    """Extractor for images from an article"""
+    subcategory = "article"
+    pattern = (r"(?:https?://)?"
+               r"(?:t\.bilibili\.com|(?:www\.)?bilibili.com/opus)/(\d+)")
+    example = "https://www.bilibili.com/opus/12345"
+    directory_fmt = ("{category}", "{username}")
+    filename_fmt = "{id}_{num}.{extension}"
+    archive_fmt = "{id}_{num}"
+
+    def items(self):
+        article = self.api.article(self.item)
+        article["username"] = article["modules"]["module_author"]["name"]
+        article["id"] = article["id_str"]
+
+        dynamic_major = article["modules"]["module_dynamic"]["major"]
+        if dynamic_major["type"] == "MAJOR_TYPE_OPUS":
+            urls = [pic["url"] for pic in dynamic_major["opus"]["pics"]]
+        else:
+            urls = []
+            self.log.warning("%s: Unsupported article type '%s'",
+                             article["id"], dynamic_major["type"])
+
+        yield Message.Directory, article
+        for article["num"], url in enumerate(urls, 1):
+            yield Message.Url, url, text.nameext_from_url(url, article)
+
+
+class BilibiliAPI():
+    def __init__(self, extractor: BilibiliExtractor):
+        self.extractor = extractor
+
+    def _call(self, endpoint, params):
+        url = "https://api.bilibili.com/x/polymer/web-dynamic/v1" + endpoint
+        response = self.extractor.request(url, params=params).json()
+
+        if response["code"] != 0:
+            raise exception.StopExtraction("API request failed")
+
+        return response
+
+    def user_articles(self, user_id):
+        endpoint = "/opus/feed/space"
+        params = {"host_mid": user_id}
+
+        while True:
+            data = self._call(endpoint, params)
+
+            for item in data["data"]["items"]:
+                params["offset"] = item["opus_id"]
+                yield item
+
+            if not data["data"]["has_more"]:
+                break
+
+    def article(self, article_id):
+        endpoint = "/detail"
+        params = {
+            "id": article_id,
+            "features": "itemOpusStyle",
+        }
+        return self._call(endpoint, params)["data"]["item"]