🐛 milky media build

Author: RF-Tar-Railt

pyproject.toml

@@ -72,7 +72,7 @@ dev = [
     "pyyaml>=6.0.1",
     "fix-future-annotations>=0.5.0",
     # "nonebot-adapter-heybox>=0.1.1; python_version >= \"3.11\"",
-    "nonebot-adapter-milky>=0.1.2",
+    "nonebot-adapter-milky>=0.1.3",
 ]
 test = [
     "nonebug>=0.4.3",

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

@@ -1,3 +1,4 @@
+from base64 import b64decode
 from typing import TYPE_CHECKING
 
 from nonebot.adapters import Bot, Event
@@ -50,14 +51,41 @@ def face(self, seg: FaceSegment):
 
     @build("image")
     def image(self, seg: ImageSegment):
+        uri = seg.data.get("uri")
+        if isinstance(uri, str):
+            if uri.startswith("http"):
+                return Image(id=seg.data["resource_id"], url=uri)
+            if uri.startswith("file://"):
+                return Image(path=uri[7:])
+            if uri.startswith("base64://"):
+                b64 = uri[9:]
+                return Image(id=seg.data["resource_id"], raw=b64decode(b64))
         return Image(id=seg.data["resource_id"])
 
     @build("video")
     def video(self, seg: VideoSegment):
+        uri = seg.data.get("uri")
+        if isinstance(uri, str):
+            if uri.startswith("http"):
+                return Video(id=seg.data["resource_id"], url=uri)
+            if uri.startswith("file://"):
+                return Video(path=uri[7:])
+            if uri.startswith("base64://"):
+                b64 = uri[9:]
+                return Video(id=seg.data["resource_id"], raw=b64decode(b64))
         return Video(id=seg.data["resource_id"])
 
     @build("record")
     def record(self, seg: RecordSegment):
+        uri = seg.data.get("uri")
+        if isinstance(uri, str):
+            if uri.startswith("http"):
+                return Voice(id=seg.data["resource_id"], url=uri)
+            if uri.startswith("file://"):
+                return Voice(path=uri[7:])
+            if uri.startswith("base64://"):
+                b64 = uri[9:]
+                return Voice(id=seg.data["resource_id"], raw=b64decode(b64))
         return Voice(id=seg.data["resource_id"])
 
     @build("reply")

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

@@ -180,13 +180,19 @@ async def reference(self, seg: Reference, bot: Union[Bot, None]) -> "MessageSegm
         messages = []
         for node in seg.children:
             if isinstance(node, RefNode):
-                continue
-            content = self.get_message_type()()
-            if isinstance(node.content, str):
-                content.extend(self.get_message_type()(node.content))
+                if not node.context:
+                    continue
+                source = await bot.get_message(
+                    message_scene="group", peer_id=int(node.context), message_seq=int(node.id)
+                )
+                messages.append(MessageSegment.node(source.sender_id, source.sender.nickname, source.message))
             else:
-                content.extend(await self.export(node.content, bot, True))
-            messages.append(MessageSegment.node(user_id=int(node.uid), name=node.name, segments=content))
+                content = self.get_message_type()()
+                if isinstance(node.content, str):
+                    content.extend(self.get_message_type()(node.content))
+                else:
+                    content.extend(await self.export(node.content, bot, True))
+                messages.append(MessageSegment.node(user_id=int(node.uid), name=node.name, segments=content))
         if not messages:
             raise SerializeFailed(lang.require("nbp-uniseg", "invalid_segment").format(type="forward", seg=seg))
         return MessageSegment.forward(messages)