From 406511496d184bcad812dcbddd7ee15b3328a855 Mon Sep 17 00:00:00 2001
From: StarHeartHunt <starheart233@gmail.com>
Date: Sat, 3 Jul 2021 13:58:26 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=8F=B7=EF=B8=8F=20add=20events?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../nonebot/adapters/feishu/__init__.py       |   1 +
 .../nonebot/adapters/feishu/event.py          | 138 ++++++++++++++++--
 2 files changed, 130 insertions(+), 9 deletions(-)

diff --git a/packages/nonebot-adapter-feishu/nonebot/adapters/feishu/__init__.py b/packages/nonebot-adapter-feishu/nonebot/adapters/feishu/__init__.py
index e0cfb5cb..1569ff02 100644
--- a/packages/nonebot-adapter-feishu/nonebot/adapters/feishu/__init__.py
+++ b/packages/nonebot-adapter-feishu/nonebot/adapters/feishu/__init__.py
@@ -1,3 +1,4 @@
 from .bot import Bot
+from .event import *
 from .message import Message, MessageSegment
 from .event import Event
diff --git a/packages/nonebot-adapter-feishu/nonebot/adapters/feishu/event.py b/packages/nonebot-adapter-feishu/nonebot/adapters/feishu/event.py
index ae55076e..5d7f6dfe 100644
--- a/packages/nonebot-adapter-feishu/nonebot/adapters/feishu/event.py
+++ b/packages/nonebot-adapter-feishu/nonebot/adapters/feishu/event.py
@@ -1,31 +1,151 @@
-from pydantic import BaseModel
-from nonebot.adapters import Event as BaseEvent
+from typing import List, Literal
+from pydantic import BaseModel, root_validator
 
-from .message import Message
+from nonebot.adapters import Event as BaseEvent
+from nonebot.typing import overrides
+
+from .message import Message, MessageSegment
 
 
 class Event(BaseEvent):
+    """
+    飞书协议事件。各事件字段参考 `飞书文档`_
 
+    .. _飞书事件列表文档:
+        https://open.feishu.cn/document/ukTMukTMukTM/uYDNxYjL2QTM24iN0EjN/event-list
+    """
+    app_id: int
+    event_type: str
+
+    @overrides(BaseEvent)
     def get_type(self) -> str:
-        raise NotImplementedError
+        return self.event_type
 
+    @overrides(BaseEvent)
     def get_event_name(self) -> str:
-        raise NotImplementedError
+        return self.event_type
 
+    @overrides(BaseEvent)
     def get_event_description(self) -> str:
         return str(self.dict())
 
+    @overrides(BaseEvent)
     def get_message(self) -> Message:
-        raise NotImplementedError
+        raise ValueError("Event has no message!")
 
+    @overrides(BaseEvent)
     def get_plaintext(self) -> str:
-        raise NotImplementedError
+        raise ValueError("Event has no plaintext!")
 
+    @overrides(BaseEvent)
     def get_user_id(self) -> str:
-        raise NotImplementedError
+        raise ValueError("Event has no user_id!")
 
+    @overrides(BaseEvent)
     def get_session_id(self) -> str:
-        raise NotImplementedError
+        raise ValueError("Event has no session_id!")
 
+    @overrides(BaseEvent)
     def is_tome(self) -> bool:
         return False
+
+
+class UserId(BaseModel):
+    union_id: str
+    user_id: str
+    open_id: str
+
+
+class Sender(BaseModel):
+    sender_id: UserId
+    sender_type: str
+    tenant_key: str
+
+
+class Mention(BaseModel):
+    key: str
+    id: UserId
+    name: str
+    tenant_key: str
+
+
+class MessageBody(BaseModel):
+    message_id: str
+    root_id: str
+    parent_id: str
+    create_time: str
+    chat_id: str
+    chat_type: str
+    message_type: str
+    content: Message
+    mentions: List[Mention]
+
+    plaintext: str
+
+    @root_validator(pre=True)
+    def gen_message(cls, values: dict):
+        content = []
+        for piece in values["content"]:
+            for segment in piece:
+                content.append(
+                    MessageSegment(segment["tag"], segment.pop('name', segment)))
+
+        values["content"] = Message(content)
+        return values
+
+    @root_validator
+    def gen_plaintext(cls, values: dict):
+        values["plaintext"] = str(values["content"])
+        return values
+
+
+class MessageEvent(Event):
+    sender: Sender
+    message: MessageBody
+
+    @overrides(Event)
+    def get_type(self) -> Literal["message", "notice", "meta_event"]:
+        return "message"
+
+    @overrides(Event)
+    def get_event_name(self) -> str:
+        return f"{self.get_type()}.{super().get_type()}"
+
+    @overrides(Event)
+    def get_event_description(self) -> str:
+        return (
+            f"Message[{super().get_type()}]"
+            f" {self.message.message_id} from {self.sender.sender_id.user_id}"
+            f" {self.message.content}")
+
+    @overrides(Event)
+    def get_message(self) -> Message:
+        return self.message.content
+
+    @overrides(Event)
+    def get_plaintext(self) -> str:
+        return self.message.plaintext
+
+    @overrides(Event)
+    def get_user_id(self) -> str:
+        return self.sender.sender_id.user_id
+
+    @overrides(Event)
+    def get_session_id(self) -> str:
+        return self.sender.sender_id.user_id
+
+
+class PrivateMessageEvent(MessageEvent):
+    ...
+
+
+class GroupMessageEvent(MessageEvent):
+    ...
+
+
+class NoticeEvent(Event):
+    ...
+
+
+class MetaEvent(Event):
+    ...