ErisPulse-KookAdapter 0.1.0__tar.gz

erispulse_kookadapter-0.1.0/ErisPulse_KookAdapter.egg-info/PKG-INFO

@@ -0,0 +1,110 @@
+Metadata-Version: 2.4
+Name: ErisPulse-KookAdapter
+Version: 0.1.0
+Summary: ErisPulse Kook Adapter - Kook平台适配器
+Author-email: ShanFish <shanfishdev@qq.com>
+License: Apache-2.0
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: ErisPulse
+Requires-Dist: websockets>=10.0
+Requires-Dist: aiohttp>=3.8.0
+Dynamic: license-file
+
+# ErisPulse-KookAdapter
+基于开源机器人框架[ErisPulse](https://github.com/ErisPulse/ErisPulse)的Kook(开黑啦)机器人适配器。
+
+项目目前已托管至PyPi，使用以下命令以完成安装：
+```bash
+pip install ErisPulse-KookAdapter
+```
+此时执行`ep list`，应该可以看到`ErisPulse-KookAdapter`在已安装的适配器列表中。
+
+## 使用
+```python
+from ErisPulse import sdk
+adapter = sdk.adapter.get("Kook")
+```
+然后就可以对adapter变量进行适配器方法操作了。
+
+适配器支持链式调用：
+```python
+# @用户
+await kook.Send.To("group", "频道ID").At("用户ID").Text("Hello")
+
+# @多个用户
+await kook.Send.To("group", "频道ID").At("用户1").At("用户2").Text("Hello")
+
+# @全体
+await kook.Send.To("group", "频道ID").AtAll().Text("公告")
+
+# 回复消息
+await kook.Send.To("group", "频道ID").Reply("消息ID").Text("回复内容")
+
+# 组合使用
+await kook.Send.To("group", "频道ID").At("用户ID").Reply("消息ID").Text("Hello")
+```
+
+另外，支持不同消息`SendDSL`调用：
+```python
+# 文本消息
+await kook.Send.To("group", "频道ID").Text("Hello World")
+
+# 图片消息
+await kook.Send.To("group", "频道ID").Image("http://example.com/image.jpg")
+
+# 视频消息
+await kook.Send.To("group", "频道ID").Video("http://example.com/video.mp4")
+
+# 文件消息
+await kook.Send.To("group", "频道ID").File("http://example.com/file.pdf")
+
+# 语音消息
+await kook.Send.To("group", "频道ID").Voice("http://example.com/voice.mp3")
+
+# KMarkdown 消息
+await kook.Send.To("group", "频道ID").Markdown("**粗体** *斜体*")
+
+# 卡片消息
+await kook.Send.To("group", "频道ID").Card({
+    "type": "card",
+    "theme": "primary",
+    "size": "lg",
+    "modules": [
+        {"type": "header", "text": {"type": "plain-text", "content": "标题"}},
+        {"type": "section", "text": {"type": "kmarkdown", "content": "内容"}}
+    ]
+})
+```
+
+### 私信消息
+```python
+# 发送私信给指定用户
+await kook.Send.To("user", "用户ID").Text("Hello")
+await kook.Send.To("user", "用户ID").Markdown("**私信内容**")
+await kook.Send.To("user", "用户ID").Card({...})
+```
+
+### 消息编辑与撤回
+```python
+# 发送消息并获取消息ID
+result = await kook.Send.To("group", "频道ID").Markdown("**原始内容**")
+msg_id = result["data"]["msg_id"]
+
+# 编辑消息（仅支持 KMarkdown 和 CardMessage）
+await kook.Send.To("group", "频道ID").Edit(msg_id, "**更新后的内容**")
+
+# 撤回消息
+await kook.Send.To("group", "频道ID").Recall(msg_id)
+
+# 上传本地文件 并获取文件URL
+result = await kook.Send.Upload("C:/path/to/image.jpg")
+file_url = result["data"]["url"]
+```
+
+
+## 参考文档
+- [Kook API 文档](https://developer.kookapp.cn/doc)  
+- [ErisPulse 文档](https://erisdev.com)  
+- [ErisPulse 主库](https://github.com/ErisPulse/ErisPulse)

erispulse_kookadapter-0.1.0/ErisPulse_KookAdapter.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+LICENSE
+README.md
+pyproject.toml
+ErisPulse_KookAdapter.egg-info/PKG-INFO
+ErisPulse_KookAdapter.egg-info/SOURCES.txt
+ErisPulse_KookAdapter.egg-info/dependency_links.txt
+ErisPulse_KookAdapter.egg-info/entry_points.txt
+ErisPulse_KookAdapter.egg-info/requires.txt
+ErisPulse_KookAdapter.egg-info/top_level.txt
+Kook/CallApi.py
+Kook/Converter.py
+Kook/Core.py
+Kook/__init__.py

erispulse_kookadapter-0.1.0/ErisPulse_KookAdapter.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

erispulse_kookadapter-0.1.0/ErisPulse_KookAdapter.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[erispulse.adapter]
+Kook = Kook:KookAdapter

erispulse_kookadapter-0.1.0/ErisPulse_KookAdapter.egg-info/requires.txt

@@ -0,0 +1,3 @@
+ErisPulse
+websockets>=10.0
+aiohttp>=3.8.0

erispulse_kookadapter-0.1.0/ErisPulse_KookAdapter.egg-info/top_level.txt

@@ -0,0 +1 @@
+Kook

erispulse_kookadapter-0.1.0/Kook/CallApi.py

@@ -0,0 +1,291 @@
+import aiohttp
+import uuid
+
+class CallApi:
+    def __init__(self, token: str):
+        self.token = token
+        self.session = aiohttp.ClientSession()
+    
+    async def close(self):
+        if self.session.closed:
+            return
+        await self.session.close()
+
+    async def start(self):
+        if not self.session.closed:
+            return
+        await self.session.start()
+
+    async def send_message(
+        self,
+        target_id: str,
+        type: int,
+        content: str,
+        quote_msg_id: str = None,
+        template_id: str = None,
+        **kwargs
+    ) -> dict:
+        """发送频道消息"""
+        if not self.token:
+            return {
+                "code": -1,
+                "msg": "token未刷新, 请刷新后重试",
+                "data": {}
+            }
+        nonce = str(uuid.uuid4())
+        payload = {
+            "nonce": nonce,
+            "target_id": target_id,
+            "type": type,
+            "content": content,
+        }
+        if quote_msg_id:
+            payload["quote"] = quote_msg_id
+        if template_id:
+            payload["template_id"] = template_id
+        
+        # 添加其他可选参数（如 mention, mention_all 等）
+        for key, value in kwargs.items():
+            if value is not None:
+                payload[key] = value
+        
+        async with self.session.post(
+            "https://www.kookapp.cn/api/v3/message/create",
+            json=payload,
+            headers={
+                "Authorization": f"Bot {self.token}",
+                "Content-Type": "application/json"
+            }
+        ) as resp:
+            data = await resp.json()
+            return data
+
+    async def send_direct_message(
+        self,
+        target_id: str,
+        type: int,
+        content: str,
+        quote_msg_id: str = None,
+        template_id: str = None,
+        **kwargs
+    ) -> dict:
+        """发送私信消息"""
+        if not self.token:
+            return {
+                "code": -1,
+                "msg": "token未刷新, 请刷新后重试",
+                "data": {}
+            }
+        nonce = str(uuid.uuid4())
+        payload = {
+            "nonce": nonce,
+            "target_id": target_id,
+            "type": type,
+            "content": content,
+        }
+        if quote_msg_id:
+            payload["quote"] = quote_msg_id
+        if template_id:
+            payload["template_id"] = template_id
+        
+        # 添加其他可选参数（如 reply_msg_id 等）
+        for key, value in kwargs.items():
+            if value is not None:
+                payload[key] = value
+        
+        async with self.session.post(
+            "https://www.kookapp.cn/api/v3/direct-message/create",
+            json=payload,
+            headers={
+                "Authorization": f"Bot {self.token}",
+                "Content-Type": "application/json"
+            }
+        ) as resp:
+            data = await resp.json()
+            return data
+
+    async def update_direct_message(
+        self,
+        msg_id: str,
+        content: str,
+        quote: str = None,
+        template_id: str = None,
+        **kwargs
+    ) -> dict:
+        """更新私信消息（仅支持 KMarkdown type=9 和 CardMessage type=10）"""
+        if not self.token:
+            return {
+                "code": -1,
+                "msg": "token未刷新, 请刷新后重试",
+                "data": {}
+            }
+        payload = {
+            "msg_id": msg_id,
+            "content": content,
+        }
+        if quote is not None:
+            payload["quote"] = quote
+        if template_id:
+            payload["template_id"] = template_id
+        
+        # 添加其他可选参数
+        for key, value in kwargs.items():
+            if value is not None:
+                payload[key] = value
+        
+        async with self.session.post(
+            "https://www.kookapp.cn/api/v3/direct-message/update",
+            json=payload,
+            headers={
+                "Authorization": f"Bot {self.token}",
+                "Content-Type": "application/json"
+            }
+        ) as resp:
+            data = await resp.json()
+            return data
+
+    async def delete_direct_message(
+        self,
+        msg_id: str,
+        **kwargs
+    ) -> dict:
+        """删除私信消息"""
+        if not self.token:
+            return {
+                "code": -1,
+                "msg": "token未刷新, 请刷新后重试",
+                "data": {}
+            }
+        payload = {
+            "msg_id": msg_id,
+        }
+        
+        # 添加其他可选参数
+        for key, value in kwargs.items():
+            if value is not None:
+                payload[key] = value
+        
+        async with self.session.post(
+            "https://www.kookapp.cn/api/v3/direct-message/delete",
+            json=payload,
+            headers={
+                "Authorization": f"Bot {self.token}",
+                "Content-Type": "application/json"
+            }
+        ) as resp:
+            data = await resp.json()
+            return data
+
+    async def update_channel_message(
+        self,
+        msg_id: str,
+        content: str,
+        quote: str = None,
+        temp_target_id: str = None,
+        template_id: str = None,
+        **kwargs
+    ) -> dict:
+        """更新频道消息（仅支持 KMarkdown type=9 和 CardMessage type=10）"""
+        if not self.token:
+            return {
+                "code": -1,
+                "msg": "token未刷新, 请刷新后重试",
+                "data": {}
+            }
+        payload = {
+            "msg_id": msg_id,
+            "content": content,
+        }
+        if quote is not None:
+            payload["quote"] = quote
+        if temp_target_id:
+            payload["temp_target_id"] = temp_target_id
+        if template_id:
+            payload["template_id"] = template_id
+        
+        # 添加其他可选参数
+        for key, value in kwargs.items():
+            if value is not None:
+                payload[key] = value
+        
+        async with self.session.post(
+            "https://www.kookapp.cn/api/v3/message/update",
+            json=payload,
+            headers={
+                "Authorization": f"Bot {self.token}",
+                "Content-Type": "application/json"
+            }
+        ) as resp:
+            data = await resp.json()
+            return data
+
+    async def delete_channel_message(
+        self,
+        msg_id: str,
+        **kwargs
+    ) -> dict:
+        """删除频道消息"""
+        if not self.token:
+            return {
+                "code": -1,
+                "msg": "token未刷新, 请刷新后重试",
+                "data": {}
+            }
+        payload = {
+            "msg_id": msg_id,
+        }
+        
+        # 添加其他可选参数
+        for key, value in kwargs.items():
+            if value is not None:
+                payload[key] = value
+        
+        async with self.session.post(
+            "https://www.kookapp.cn/api/v3/message/delete",
+            json=payload,
+            headers={
+                "Authorization": f"Bot {self.token}",
+                "Content-Type": "application/json"
+            }
+        ) as resp:
+            data = await resp.json()
+            return data
+
+    async def upload_file(self, file_path: str) -> dict:
+        if not self.token:
+            return {
+                "code": -1,
+                "msg": "token未刷新, 请刷新后重试",
+                "data": {}
+            }
+        
+        async with self.session.post(
+            "https://www.kookapp.cn/api/v3/asset/create",
+            headers={
+                "Authorization": f"Bot {self.token}",
+                "Content-Type": "multipart/form-data"
+            },
+            data={
+                "file": open(file_path, "rb")
+            }
+        ) as resp:
+            data = await resp.json()
+            return data
+        
+    async def get_ws_gateway(self, need_compress: bool = True) -> str:
+        if not self.token:
+            return ""
+        async with self.session.post(
+            "https://www.kookapp.cn/api/v3/gateway/index",
+            headers={
+                "Authorization": f"Bot {self.token}",
+                "Content-Type": "application/json"
+            },
+            json={
+                "need_compress": 1 if need_compress else 0
+            }
+        ) as resp:
+            data = await resp.json()
+            if data.get("code") == 0:
+                return data.get("data", {}).get("url", "")
+            return ""

erispulse_kookadapter-0.1.0/Kook/Converter.py

@@ -0,0 +1,175 @@
+"""
+Kook Event -> OneBot V12
+标准协议转换器
+"""
+
+import json
+import uuid
+import time
+
+
+class KookAdapterConverter:
+    def convert(self, data):
+        d = data.get("d", {})
+        extra = d.get("extra", {})
+        author = extra.get("author", {})
+        kook_type = d.get("type", 0)
+        channel_type = d.get("channel_type", "")
+        
+        onebot_data = {
+            "id": str(uuid.uuid4()).replace("-", ""),
+            "time": time.time(),
+            "type": self._get_message_type(data),
+            "detail_type": self._get_detail_type(data),
+            "platform": "Kook",
+            "self": {
+                "platform": "Kook",
+                "user_id": ""
+            },
+            "Kook_raw": data,
+            "Kook_raw_type": kook_type,
+        }
+        
+        if onebot_data["type"] == "message":
+            onebot_data.update({
+                "message_id": d.get("msg_id", ""),
+                "user_id": author.get("id", ""),
+                "message": self._convert_message_content(data),
+                "alt_message": d.get("content", ""),
+            })
+            
+            if channel_type == "PERSON":
+                onebot_data["detail_type"] = "private"
+            elif channel_type == "GROUP":
+                onebot_data["group_id"] = d.get("target_id", "")
+                
+            mentions = extra.get("mention", [])
+            if mentions:
+                onebot_data["mentions"] = mentions
+                
+        elif onebot_data["type"] == "notice":
+            onebot_data["user_id"] = author.get("id", "")
+            onebot_data["guild_id"] = extra.get("guild_id", "")
+            onebot_data.update(self._convert_notice_data(data))
+            
+        return onebot_data
+    
+    def _get_message_type(self, data):
+        kook_type = data.get("d", {}).get("type", 0)
+        return {
+            255: "notice"
+        }.get(kook_type, "message")
+    
+    def _get_detail_type(self, data):
+        d = data.get("d", {})
+        kook_type = d.get("type", 0)
+        if kook_type == 255:
+            return data.get("d", {}).get("extra", {}).get("type", "")
+        
+        return {
+            1: "text",
+            2: "image",
+            3: "video",
+            4: "file",
+            8: "record",
+            9: "kmarkdown",
+            10: "json",
+        }.get(kook_type, "unknown")
+    
+    def _convert_message_content(self, data):
+        d = data.get("d", {})
+        extra = d.get("extra", {})
+        kook_type = d.get("type", 0)
+        content = d.get("content", "")
+        
+        message_segments = []
+        
+        if kook_type == 1:
+            message_segments.append({"type": "text", "data": {"text": content}})
+        elif kook_type == 2:
+            attachments = extra.get("attachments", {})
+            url = attachments.get("url", content)
+            message_segments.append({"type": "image", "data": {"file": url, "url": url}})
+        elif kook_type == 3:
+            attachments = extra.get("attachments", {})
+            url = attachments.get("url", content)
+            message_segments.append({"type": "video", "data": {"file": url, "url": url}})
+        elif kook_type == 4:
+            attachments = extra.get("attachments", {})
+            url = attachments.get("url", content)
+            message_segments.append({"type": "file", "data": {"file": url, "url": url}})
+        elif kook_type == 8:
+            attachments = extra.get("attachments", {})
+            url = attachments.get("url", content)
+            message_segments.append({"type": "record", "data": {"file": url, "url": url}})
+        elif kook_type == 9:
+            kmarkdown = extra.get("kmarkdown", {})
+            raw_content = kmarkdown.get("raw_content", content)
+            message_segments.append({"type": "text", "data": {"text": raw_content}})
+        elif kook_type == 10:
+            message_segments.append({"type": "json", "data": {"data": content}})
+        else:
+            message_segments.append({"type": "text", "data": {"text": content}})
+        
+        mention = extra.get("mention", [])
+        if mention:
+            for user_id in mention:
+                message_segments.insert(0, {"type": "mention", "data": {"user_id": user_id}})
+        
+        if extra.get("mention_all", False):
+            message_segments.insert(0, {"type": "mention_all", "data": {}})
+        
+        return message_segments
+    
+    def _convert_notice_data(self, data):
+        d = data.get("d", {})
+        extra = d.get("extra", {})
+        event_type = extra.get("type", "")
+        body = extra.get("body", {})
+        
+        notice_data = {}
+        
+        # 映射 Kook 事件类型到标准 notice_type
+        if event_type == "added_reaction":
+            notice_data["notice_type"] = "reaction_added"
+            notice_data["sub_type"] = "added_reaction"
+            notice_data["channel_id"] = body.get("channel_id", "")
+            notice_data["message_id"] = body.get("msg_id", "")
+            notice_data["user_id"] = body.get("user_id", "")
+            notice_data["emoji"] = body.get("emoji", {})
+        elif event_type == "deleted_reaction":
+            notice_data["notice_type"] = "reaction_removed"
+            notice_data["sub_type"] = "deleted_reaction"
+            notice_data["channel_id"] = body.get("channel_id", "")
+            notice_data["message_id"] = body.get("msg_id", "")
+            notice_data["user_id"] = body.get("user_id", "")
+            notice_data["emoji"] = body.get("emoji", {})
+        elif event_type == "private_added_reaction":
+            notice_data["notice_type"] = "reaction_added"
+            notice_data["sub_type"] = "private_added_reaction"
+            notice_data["message_id"] = body.get("msg_id", "")
+            notice_data["user_id"] = body.get("user_id", "")
+            notice_data["emoji"] = body.get("emoji", {})
+        elif event_type == "private_deleted_reaction":
+            notice_data["notice_type"] = "reaction_removed"
+            notice_data["sub_type"] = "private_deleted_reaction"
+            notice_data["message_id"] = body.get("msg_id", "")
+            notice_data["user_id"] = body.get("user_id", "")
+            notice_data["emoji"] = body.get("emoji", {})
+        elif event_type == "updated_private_message":
+            notice_data["notice_type"] = "message_updated"
+            notice_data["sub_type"] = "updated_private_message"
+            notice_data["message_id"] = body.get("msg_id", "")
+            notice_data["user_id"] = body.get("author_id", "")
+            notice_data["content"] = body.get("content", "")
+        elif event_type == "deleted_private_message":
+            notice_data["notice_type"] = "message_deleted"
+            notice_data["sub_type"] = "deleted_private_message"
+            notice_data["message_id"] = body.get("msg_id", "")
+            notice_data["user_id"] = body.get("author_id", "")
+        else:
+            notice_data["notice_type"] = "system"
+            notice_data["sub_type"] = event_type if event_type else "system"
+            notice_data["raw_event"] = extra
+
+        return notice_data