sillyGirl 增加傻妞插件 (Sourcery refactored) (#71)

Co-authored-by: cdle <798731886@qq.com>
Co-authored-by: Sourcery AI <>
Co-authored-by: omg-xtao <100690902+omg-xtao@users.noreply.github.com>

list.json

@@ -719,6 +719,16 @@
             "supported": true,
             "des-short": "websocket 使用 ws 管理机器人",
             "des": "websocket 使用 ws 管理机器人"
+        },
+        {
+            "name": "silly_girl",
+            "version": "1.0",
+            "section": "chat",
+            "maintainer": "cdle",
+            "size": "8 kb",
+            "supported": true,
+            "des-short": "傻妞插件",
+            "des": "傻妞插件，用户可以通过此插件召唤傻妞，Repo：https://github.com/cdle/sillyGirl"
         }
     ]
 }

silly_girl/main.py

@@ -0,0 +1,145 @@
+from asyncio import sleep
+from pagermaid.listener import listener
+from pagermaid.enums import Message
+from pagermaid.single_utils import sqlite
+from pagermaid.utils import client, edit_delete
+from pagermaid import bot
+from pyrogram.enums.chat_type import ChatType
+from pagermaid.hook import Hook
+
+
+class SillyGirl:
+    address = ""
+    token = ""
+    self_user_id = ""
+    init = False
+
+    def init_connect_info(self, address):
+        self.init = True
+        self.self_user_id = bot.me.id
+        if address:
+            sqlite["silly_girl_address"] = address
+        else:
+            address = sqlite.get("silly_girl_address", "")
+        try:
+            if '@' in address:
+            s1 = address.split("//", 1)
+            s2 = s1[1].split("@", 1)
+            sillyGirl.token = s2[0]
+            self.address = f"{s1[0]}//{s2[1]}"
+        except:
+            self.address = ""
+
+    async def polls(self):
+        while True:
+            if self.address:
+                await self.poll([])
+            else:
+                return
+
+    async def poll(self, data):
+        try:
+            if not self.address:
+                return
+            init = ''
+            if not self.init:
+                init = "&init=true"
+                self.init = True
+            req_data = await client.post(
+                f"{self.address}/pgm?token={self.token}{init}", json=data
+            )
+        except Exception as e:
+            print(e)
+            await sleep(0.1)
+            return
+        if req_data.status_code != 200:
+            await sleep(0.1)
+            return
+        try:
+            replies = req_data.json()
+            results = []
+            for reply in replies:
+                if reply["delete"]:
+                    try:
+                        await bot.edit_message(reply["chat_id"], reply["id"], "打错字了，呱呱～")
+                    except Exception as e:
+                        pass
+                    try:
+                        await bot.delete_messages(reply["chat_id"], [reply["id"]])
+                    except Exception as e:
+                        pass
+                if reply["id"] != 0:
+                    try:
+                        await bot.edit_message(reply["chat_id"], reply["id"], reply["text"])
+                        continue
+                    except Exception as e:
+                        continue
+                message: Message
+                if reply["images"] and len(reply["images"]) != 0:
+                    message = await bot.send_photo(
+                        reply["chat_id"],
+                        reply["images"][0],
+                        caption=reply["text"],
+                        reply_to_message_id=reply["reply_to"],
+                    )
+                elif reply["text"] != '':
+                    message = await bot.send_message(reply["chat_id"], reply["text"], reply_to_message_id=reply["reply_to"])
+                if message:
+                    results.append({
+                        'id': message.id,
+                        'uuid': reply["uuid"],
+                    })
+            if len(results):
+                await self.poll(results)
+        except Exception as e:
+            print(e)
+            await sleep(0.1)
+
+
+sillyGirl = SillyGirl()
+
+
+@Hook.on_startup()
+async def connect_sillyGirl():
+    sillyGirl.init_connect_info("")
+    bot.loop.create_task(sillyGirl.polls())
+
+
+@listener(command="sillyGirl", description="连接到傻妞服务器", parameters="<auth>")
+async def sillyGirl_connect(message: Message):
+    try:
+        await edit_delete(message, "连接中...")
+        sillyGirl.init_connect_info(message.arguments)
+    except Exception as e:
+        print(e)
+
+
+@listener(outgoing=True, ignore_edited=True, incoming=True)
+async def handle_receive(message: Message):
+    try:                   
+        reply_to = message.id
+        reply = message.reply_to_message
+        reply_to_sender_id = 0
+        sender_id = 0
+        if message.from_user:
+            sender_id = message.from_user.id
+            if reply:
+                reply_to = reply.id
+                reply_to_sender_id = reply.from_user.id
+        if not sillyGirl.init:
+            sillyGirl.init_connect_info("")
+        if sillyGirl.self_user_id == sender_id:
+            reply_to = 0
+        await sillyGirl.poll(
+        [{
+            'id': message.id,
+            'chat_id': message.chat.id,
+            'text': message.text,
+            'sender_id': sender_id,
+            'reply_to': reply_to,
+            'reply_to_sender_id': reply_to_sender_id,
+            'bot_id': sillyGirl.self_user_id,
+            'is_group': message.chat.type != ChatType.PRIVATE,
+        }])
+    except Exception as e:
+        print(e)