feat: experimental htmlkit

Author: ProgramRipper

Dockerfile

@@ -1,4 +1,4 @@
-FROM python:alpine AS base
+FROM python:slim AS base
 WORKDIR /app
 
 

pdm.lock

@@ -10,6 +10,7 @@ dependencies = [
     "nonebot-adapter-milky~=0.5.0rc3",
     "nonebot-plugin-alconna~=0.59",
     "nonebot-plugin-apscheduler~=0.5",
+    "nonebot-plugin-htmlkit<1,>=0.1.0rc1",
     "nonebot-plugin-htmlrender~=0.6",
     "nonebot-plugin-orm[sqlite]~=0.8",
     "nonebot-plugin-uninfo @ https://github.com/RF-Tar-Railt/nonebot-plugin-uninfo/archive/5b87d916af09d6a878039d3445193efda2ed1af3.zip",
@@ -30,6 +31,7 @@ adapters = [
 plugins = [
     "nonebot_plugin_alconna",
     "nonebot_plugin_apscheduler",
+    "nonebot_plugin_htmlkit",
     "nonebot_plugin_htmlrender",
     "nonebot_plugin_orm",
     "nonebot_plugin_uninfo",

pyproject.toml

@@ -1,5 +1,6 @@
 from asyncio import gather
 from contextlib import asynccontextmanager
+from importlib.resources import as_file, files
 from queue import PriorityQueue
 from typing import Annotated, Any, AsyncGenerator
 
@@ -11,16 +12,18 @@
 from nonebot.plugin import PluginMetadata
 from nonebot_plugin_alconna import Alconna, Image, Subcommand, UniMessage, on_alconna
 from nonebot_plugin_apscheduler import scheduler
+from nonebot_plugin_htmlkit import template_to_pic
 from nonebot_plugin_htmlrender.browser import get_browser
 from nonebot_plugin_orm import async_scoped_session, get_session
 from nonebot_plugin_uninfo import MEMBER
 from nonebot_plugin_uninfo.orm import SceneModel, SceneOrm
 from playwright.async_api import BrowserContext, Page
 from sqlalchemy import exists, select
 
-from .....utils import run_task, send_message
+from .....utils import css_fetch_fn, img_fetch_fn, run_task, send_message
 from ... import plugin_config as bilibili_config
 from ...utils import UID_ARG, get_share_click, handle_error, raise_for_status
+from . import templates
 from .config import Config
 from .models import Dynamic, Dynamics, Subscription
 
@@ -163,7 +166,7 @@ async def render_screenshot(id_str: str) -> bytes:
         )
         await page.add_style_tag(
             content="""
-                * {
+                body {
                   font-family: "LXGW ZhenKai GB", "LXGW WenKai GB", sans-serif !important;
                 }
 
@@ -198,15 +201,40 @@ async def render_screenshot(id_str: str) -> bytes:
 async def broadcast(dynamics: list[Dynamic]):
     async with get_session() as session:
         for dynamic in dynamics:
-            screenshot, url, subs = await gather(
-                render_screenshot(dynamic["id_str"]),
-                get_share_click(dynamic["id_str"], "dynamic", "dt.dt-detail.0.0.pv"),
-                session.scalars(
-                    select(Subscription).where(
-                        Subscription.uid == dynamic["modules"]["module_author"]["mid"]
-                    )
-                ),
-            )
+            with as_file(files(templates)) as templates_path:
+                screenshot, url, subs = await gather(
+                    (
+                        template_to_pic(
+                            str(templates_path),
+                            "draw.html.j2",
+                            {"dynamic": dynamic},
+                            max_width=360 * 3,
+                            device_height=640 * 3,
+                            allow_refit=False,
+                            img_fetch_fn=img_fetch_fn,
+                            css_fetch_fn=css_fetch_fn,
+                            image_format="jpeg",
+                            jpeg_quality=80,
+                        )
+                        if dynamic["type"] == "DYNAMIC_TYPE_DRAW"
+                        and not (
+                            dynamic["modules"]["module_dynamic"]["additional"]
+                            or dynamic["modules"]["module_dynamic"]["major"]["opus"][
+                                "pics"
+                            ]
+                        )
+                        else render_screenshot(dynamic["id_str"])
+                    ),
+                    get_share_click(
+                        dynamic["id_str"], "dynamic", "dt.dt-detail.0.0.pv"
+                    ),
+                    session.scalars(
+                        select(Subscription).where(
+                            Subscription.uid
+                            == dynamic["modules"]["module_author"]["mid"]
+                        )
+                    ),
+                )
 
             msg = plugin_config.template.format(
                 name=dynamic["modules"]["module_author"]["name"],
@@ -316,10 +344,30 @@ async def _(id_str: str):
     except Exception:
         await handle_error("获取动态信息失败")
 
-    screenshot, url = await gather(
-        render_screenshot(id_str),
-        get_share_click(id_str, "dynamic", "dt.dt-detail.0.0.pv"),
-    )
+    with as_file(files(templates)) as templates_path:
+        screenshot, url = await gather(
+            (
+                template_to_pic(
+                    str(templates_path),
+                    "draw.html.j2",
+                    {"dynamic": dynamic},
+                    max_width=360 * 3,
+                    device_height=640 * 3,
+                    allow_refit=False,
+                    img_fetch_fn=img_fetch_fn,
+                    css_fetch_fn=css_fetch_fn,
+                    image_format="jpeg",
+                    jpeg_quality=80,
+                )
+                if dynamic["type"] == "DYNAMIC_TYPE_DRAW"
+                and not (
+                    dynamic["modules"]["module_dynamic"]["additional"]
+                    or dynamic["modules"]["module_dynamic"]["major"]["opus"]["pics"]
+                )
+                else render_screenshot(dynamic["id_str"])
+            ),
+            get_share_click(id_str, "dynamic", "dt.dt-detail.0.0.pv"),
+        )
     await plugin_config.template.format(
         name=dynamic["modules"]["module_author"]["name"],
         action=dynamic["modules"]["module_author"]["pub_action"]

src/plugins/bilibili/plugins/dynamic/__init__.py

@@ -1,4 +1,4 @@
-from typing import TypedDict
+from typing import Any, TypedDict
 
 from nonebot_plugin_orm import Model
 from nonebot_plugin_uninfo.orm import SceneModel
@@ -14,6 +14,7 @@ class ModuleAuthor(TypedDict):
 
 class Modules(TypedDict):
     module_author: ModuleAuthor
+    module_dynamic: dict[str, Any]
 
 
 class Dynamic(TypedDict):

src/plugins/bilibili/plugins/dynamic/models.py

@@ -0,0 +1,91 @@
+<head>
+    <link rel="stylesheet"
+        href="https://cdnjs.cloudflare.com/ajax/libs/modern-normalize/3.0.1/modern-normalize.min.css" />
+    <style type="text/css">
+        body {
+            font-family: "LXGW ZhenKai GB", "LXGW WenKai GB", sans-serif;
+        }
+
+        p {
+            margin: 0;
+            font-weight: 400;
+        }
+
+        .author {
+            padding: 3.2vmin;
+            display: flex;
+            align-items: center;
+            height: 19.2vmin;
+        }
+
+        .avatar {
+            width: 10.66667vmin;
+            height: 10.66667vmin;
+            margin-right: 3.2vmin;
+            flex-shrink: 0;
+            border-radius: 9999px;
+        }
+
+        .info {
+            flex: 1;
+        }
+
+        .name {
+            color: rgb(251, 114, 153);
+            font-size: 4vmin;
+        }
+
+        .time {
+            margin-top: 1px;
+            font-size: 3.2vmin;
+            line-height: 4.53333vmin;
+            color: #9499a0;
+        }
+
+        .content {
+            color: #18191c;
+            word-break: break-all;
+            letter-spacing: 0;
+            word-wrap: break-word;
+            font-size: 4.53333vmin;
+            line-height: 7.73333vmin;
+            white-space: pre-line;
+            padding: 0 3.2vmin;
+            margin-bottom: 3.2vmin;
+        }
+
+
+        .emoji {
+            padding: 0 1px;
+        }
+
+        .emoji img {
+            width: 12.26667vmin;
+            height: 12.26667vmin;
+            vertical-align: -.24em;
+        }
+    </style>
+</head>
+
+<body>
+    <div class="author">
+        <img class="avatar" src="{{ dynamic['modules']['module_author']['face'] }}" />
+        <div class="info">
+            <div class="name">
+                {{ dynamic["modules"]["module_author"]["name"] }}
+            </div>
+            <div class="time">
+                {{ dynamic["modules"]["module_author"]["pub_time"] }}
+            </div>
+        </div>
+    </div>
+    <p class="content">
+        {%- for node in dynamic["modules"]["module_dynamic"]["major"]["opus"]["summary"]["rich_text_nodes"] -%}
+        {%- if node["type"] == "RICH_TEXT_NODE_TYPE_TEXT" -%}
+        <span>{{ node["text"] }}</span>
+        {%- elif node["type"] == "RICH_TEXT_NODE_TYPE_EMOJI" -%}
+        <span class="emoji"><img src="{{ node['emoji']['icon_url'] }}" /></span>
+        {%- endif -%}
+        {%- endfor -%}
+    </p>
+</body>

src/plugins/bilibili/plugins/dynamic/templates/draw.html.j2

@@ -3,6 +3,7 @@
 from inspect import iscoroutine
 from typing import TYPE_CHECKING, Awaitable, TypeVar, cast
 
+from httpx import AsyncClient
 from nonebot_plugin_alconna import AtAll, SupportAdapter, UniMessage
 from nonebot_plugin_alconna.uniseg import Receipt
 from nonebot_plugin_uninfo.orm import SceneModel, get_bot_model
@@ -55,3 +56,19 @@ async def send_message(scene_model: SceneModel, msg: UniMessage) -> Receipt:
 
 def with_prefix(prefix: str) -> Callable[[str], str]:
     return lambda name: f"{prefix}_{name}"
+
+
+client = AsyncClient(
+    headers={
+        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+        "AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36 Edg/120.0.0.0"
+    }
+)
+
+
+async def img_fetch_fn(url: str) -> bytes:
+    return (await client.get(url)).content
+
+
+async def css_fetch_fn(url: str) -> str:
+    return (await client.get(url)).text