✨ support Milky

Author: RF-Tar-Railt

pdm.lock

@@ -72,6 +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.0",
 ]
 test = [
     "nonebug>=0.4.3",

pyproject.toml

@@ -38,6 +38,10 @@ def mfbuild(builder: MessageBuilder, seg: BaseMessageSegment):
             id=str(seg.data["id"]),
             summary=seg.data["name"],
         )
+    if builder.get_adapter() is SupportAdapter.milky:
+        url = seg.data["url"]
+        id_ = url.split("/")[-2]
+        return MarketFace(id=id_)
     return None
 
 

src/nonebot_plugin_alconna/builtins/uniseg/market_face.py

@@ -0,0 +1,22 @@
+from nonebot_plugin_alconna.uniseg.loader import BaseLoader
+from nonebot_plugin_alconna.uniseg.constraint import SupportAdapter
+
+
+class Loader(BaseLoader):
+    def get_adapter(self) -> SupportAdapter:
+        return SupportAdapter.milky
+
+    def get_builder(self):
+        from .builder import MilkyMessageBuilder
+
+        return MilkyMessageBuilder()
+
+    def get_exporter(self):
+        from .exporter import MilkyMessageExporter
+
+        return MilkyMessageExporter()
+
+    def get_fetcher(self):
+        from .target import MilkyTargetFetcher
+
+        return MilkyTargetFetcher()

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

@@ -0,0 +1,99 @@
+from typing import TYPE_CHECKING
+
+from nonebot.adapters import Bot, Event
+from nonebot.adapters.milky.message import XML
+from nonebot.adapters.milky.event import MessageEvent
+from nonebot.adapters.milky.message import Face as FaceSegment
+from nonebot.adapters.milky.message import Image as ImageSegment
+from nonebot.adapters.milky.message import Reply as ReplySegment
+from nonebot.adapters.milky.message import Video as VideoSegment
+from nonebot.adapters.milky.message import Record as RecordSegment
+from nonebot.adapters.milky.message import Forward as ForwardSegment
+from nonebot.adapters.milky.message import Mention as MentionSegment
+from nonebot.adapters.milky.message import LightAPP as LightAppSegment
+from nonebot.adapters.milky.message import MentionAll as MentionAllSegment
+
+from nonebot_plugin_alconna.uniseg.constraint import SupportAdapter
+from nonebot_plugin_alconna.uniseg.builder import MessageBuilder, build
+from nonebot_plugin_alconna.uniseg.segment import (
+    At,
+    AtAll,
+    Emoji,
+    Hyper,
+    Image,
+    Reply,
+    Video,
+    Voice,
+    Reference,
+    CustomNode,
+)
+
+
+class MilkyMessageBuilder(MessageBuilder):
+    @classmethod
+    def get_adapter(cls) -> SupportAdapter:
+        return SupportAdapter.milky
+
+    @build("mention")
+    def mention(self, seg: MentionSegment):
+        if seg.data["user_id"] == 0:
+            return AtAll()
+        return At("user", str(seg.data["user_id"]))
+
+    @build("mention_all")
+    def mention_all(self, seg: MentionAllSegment):
+        return AtAll()
+
+    @build("face")
+    def face(self, seg: FaceSegment):
+        return Emoji(str(seg.data["face_id"]))
+
+    @build("image")
+    def image(self, seg: ImageSegment):
+        return Image(id=seg.data["resource_id"])
+
+    @build("video")
+    def video(self, seg: VideoSegment):
+        return Video(id=seg.data["resource_id"])
+
+    @build("record")
+    def record(self, seg: RecordSegment):
+        return Voice(id=seg.data["resource_id"])
+
+    @build("reply")
+    def reply(self, seg: ReplySegment):
+        if "client_seq" in seg.data:
+            msg_id = f"{seg.data['message_seq']}#{seg.data['client_seq']}"
+        else:
+            msg_id = str(seg.data["message_seq"])
+        return Reply(msg_id, origin=seg)
+
+    @build("forward")
+    def forward(self, seg: ForwardSegment):
+        if "forward_id" in seg.data:
+            return Reference(seg.data["forward_id"])
+        return Reference(
+            nodes=[
+                CustomNode(uid=str(msg.user_id), name=msg.name, content=self.generate(msg.segments))
+                for msg in seg.data["messages"]
+            ]
+        )
+
+    @build("light_app")
+    def light_app(self, seg: LightAppSegment):
+        return Hyper("json", seg.data["json_payload"])
+
+    @build("xml")
+    def xml(self, seg: XML):
+        return Hyper("xml", seg.data["xml_payload"])
+
+    async def extract_reply(self, event: Event, bot: Bot):
+        if TYPE_CHECKING:
+            assert isinstance(event, MessageEvent)
+        if _reply := event.reply:
+            if _reply.client_seq:
+                msg_id = f"{_reply.message_seq}#{_reply.client_seq}"
+            else:
+                msg_id = str(_reply.message_seq)
+            return Reply(msg_id, _reply.message, _reply)
+        return None