✨ adapter Bilibili Live

Author: RF-Tar-Railt

.github/workflows/test.yml

@@ -14,7 +14,7 @@ jobs:
   test:
     strategy:
       matrix:
-        py_ver: ['3.9', '3.10', '3.11', '3.12']
+        py_ver: ['3.10', '3.11', '3.12', '3.13']
     runs-on:  ubuntu-latest
     steps:
       - uses: actions/checkout@v5

README.md

@@ -214,6 +214,7 @@ dev = [
     "nonebot-adapter-milky>=1.0.0",
     "nonebot-adapter-efchat>=0.1.9.post2",
     "pytest-asyncio==0.26.0",
+    "nonebot-adapter-bilibili-live>=0.2.4",
 ]
 test = [
     "nonebug>=0.4.3",

pdm.lock

@@ -0,0 +1,17 @@
+from nonebot_plugin_alconna.uniseg.constraint import SupportAdapter
+from nonebot_plugin_alconna.uniseg.loader import BaseLoader
+
+
+class Loader(BaseLoader):
+    def get_adapter(self) -> SupportAdapter:
+        return SupportAdapter.bililive
+
+    def get_builder(self):
+        from .builder import BiliLiveMessageBuilder
+
+        return BiliLiveMessageBuilder()
+
+    def get_exporter(self):
+        from .exporter import BiliLiveMessageExporter
+
+        return BiliLiveMessageExporter()

pyproject.toml

@@ -0,0 +1,23 @@
+from nonebot.adapters.bilibili_live.message import TextSegment, AtSegment, EmoticonSegment
+
+from nonebot_plugin_alconna.uniseg.builder import MessageBuilder, build
+from nonebot_plugin_alconna.uniseg.constraint import SupportAdapter
+from nonebot_plugin_alconna.uniseg.segment import At, Text, Emoji
+
+
+class BiliLiveMessageBuilder(MessageBuilder):
+    @classmethod
+    def get_adapter(cls) -> SupportAdapter:
+        return SupportAdapter.bililive
+
+    @build("text")
+    def text(self, seg: TextSegment):
+        return Text(seg.data["text"])
+
+    @build("at")
+    def at(self, seg: AtSegment):
+        return At("user", str(seg.user_id), seg.data.get("name"))
+
+    @build("emoticon")
+    def emoticon(self, seg: EmoticonSegment):
+        return Emoji(seg.data["emoji"], seg.data.get("descript"))

src/nonebot_plugin_alconna/uniseg/adapters/bililive/__init__.py

@@ -0,0 +1,61 @@
+from typing import Union
+
+from nonebot.adapters import Bot, Event
+from nonebot.adapters.bilibili_live.event import MessageEvent, DanmakuEvent, SuperChatEvent
+from nonebot.adapters.bilibili_live.message import Message, MessageSegment
+from nonebot.adapters.bilibili_live.bot import WebBot
+
+from nonebot_plugin_alconna.uniseg.constraint import SupportScope
+from nonebot_plugin_alconna.uniseg.exporter import MessageExporter, SupportAdapter, Target, export
+from nonebot_plugin_alconna.uniseg.segment import At, Emoji, Text, Reply
+
+
+class BiliLiveMessageExporter(MessageExporter[Message]):
+    def get_message_type(self):
+        return Message
+
+    @classmethod
+    def get_adapter(cls) -> SupportAdapter:
+        return SupportAdapter.bililive
+
+    def get_target(self, event: Event, bot: Union[Bot, None] = None) -> Target:
+
+        if isinstance(event, MessageEvent):
+            return Target(
+                str(event.room_id),
+                adapter=self.get_adapter(),
+                self_id=bot.self_id if bot else None,
+                scope=SupportScope.bililive,
+            )
+        raise NotImplementedError
+
+    def get_message_id(self, event: Event) -> str:
+        if isinstance(event, DanmakuEvent):
+            return event.msg_id or f"{event.room_id}:{hash(event.content)}"
+        if isinstance(event, SuperChatEvent):
+            return str(event.msg_id or event.id)
+        raise NotImplementedError
+
+    @export
+    async def text(self, seg: Text, bot: Union[Bot, None]) -> "MessageSegment":
+        return MessageSegment.text(seg.text)
+
+    @export
+    async def at(self, seg: At, bot: Union[Bot, None]) -> "MessageSegment":
+        return MessageSegment.at(seg.target, seg.display)
+
+    @export
+    async def emoji(self, seg: Emoji, bot: Union[Bot, None]) -> "MessageSegment":
+        return MessageSegment.emoticon(seg.id)
+
+    @export
+    async def reply(self, seg: Reply, bot: Union[Bot, None]) -> "MessageSegment":
+        return MessageSegment.at(seg.id)
+
+    async def send_to(self, target: Union[Target, Event], bot: Bot, message: Message, **kwargs):
+        if isinstance(target, Target):
+            assert isinstance(bot, WebBot), "BiliLive currently only supports WebBot for sending messages."
+            reply_mid = message["at"][-1].data.get("uid", 0) if message["at"] else 0
+            msg = "".join(str(seg) for seg in message)
+            return await bot.send_danmaku(room_id=int(target.id), msg=msg, reply_mid=reply_mid, **kwargs)
+        return await bot.send(target, message=message, **kwargs)  # type: ignore

src/nonebot_plugin_alconna/uniseg/adapters/bililive/builder.py

@@ -11,6 +11,7 @@
 class SupportAdapter(str, Enum):
     """支持的适配器"""
 
+    bililive = "bilibili Live"
     console = "Console"
     ding = "Ding"
     discord = "Discord"
@@ -69,6 +70,8 @@ class SupportScope(str, Enum):
     """Tailchat平台"""
     efchat = "EFChat"
     """EFChat平台"""
+    bililive = "BiliLive"
+    """Bilibili直播平台"""
 
     onebot12_other = "Onebot12"
     """ob12 的其他平台"""
@@ -114,6 +117,7 @@ def ensure_satori(platform: str):
 class SupportAdapterModule(str, Enum):
     """支持的适配器的模块路径"""
 
+    bililive = "nonebot.adapters.bilibili_live"
     console = "nonebot.adapters.console"
     ding = "nonebot.adapters.ding"
     discord = "nonebot.adapters.discord"

src/nonebot_plugin_alconna/uniseg/adapters/bililive/exporter.py

@@ -508,3 +508,10 @@ async def select_efchat(target: "Target", bot: Bot):
     if target.channel:
         return False
     return bot.adapter.get_name() == SupportAdapter.efchat
+
+
+@_register(SupportScope.bililive)
+async def select_bililive(target: "Target", bot: Bot):
+    if target.channel or target.private:
+        return False
+    return bot.adapter.get_name() == SupportAdapter.bililive