PagerMaid_Plugins_Pyro/silly_girl/main.py

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
from datetime import datetime, timedelta
from pagermaid.services import bot
from pyrogram.types import ChatPermissions

import json


class SillyGirl:
    address = ""
    token = ""
    self_user_id = ""
    init = False
    working = False

    def init_connect_info(self, address):
        sillyGirl.self_user_id = bot.me.id
        self.init = True
        if address:
            sqlite["silly_girl_address"] = address
        else:
            address = sqlite.get("silly_girl_address")
        if "@" in address:
            s1 = address.split("//", 1)
            s2 = s1[1].split("@", 1)
            sillyGirl.token = s2[0]
            address = f"{s1[0]}//{s2[1]}"
        sillyGirl.address = address

    async def polls(self):
        while True:
            await self.poll([])

    async def poll(self, data):
        try:
            init = ""
            if sillyGirl.init == False:
                init = "&init=true"
                sillyGirl.init = True
            req_data = await client.post(
                f"{self.address}/pgm?token={self.token}{init}", json=data
            )

        except Exception as e:
            print(e)
            print(e, "???====")
            await sleep(0.1)
            return
        if req_data.status_code != 200:
            await sleep(0.1)
            return
        try:
            replies = json.loads(req_data.text)
            results = []
            for reply in replies:
                if reply["command"]:
                    try:
                        id = reply["id"]
                        cid = reply["chat_id"]
                        uid = reply["sender_id"]
                        if reply["command"] == "ban":
                            if id != 0:
                                await bot.restrict_chat_member(
                                    cid,
                                    uid,
                                    ChatPermissions(),
                                    datetime.now() + timedelta(seconds=id),
                                )
                            else:
                                await bot.restrict_chat_member(
                                    cid, uid, ChatPermissions()
                                )
                        elif reply["command"] == "unban":
                            await bot.restrict_chat_member(
                                cid,
                                uid,
                                ChatPermissions(),
                                datetime.now() + timedelta(seconds=60),
                            )
                        elif reply["command"] == "kick":
                            if id != 0:
                                await bot.ban_chat_member(
                                    cid, uid, datetime.now() + timedelta(seconds=60)
                                )
                            else:
                                await bot.ban_chat_member(cid, uid)
                    except Exception as e:
                        print(e, "----")
                    continue
                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["videos"] and len(reply["videos"]) != 0:
                    message = await bot.send_video(
                        reply["chat_id"],
                        reply["videos"][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 sillyGirl.poll(results)
        except Exception as e:
            print(e, "???")
            await sleep(0.1)
            return


sillyGirl = SillyGirl()


@Hook.on_startup()
async def connect_sillyGirl():
    sillyGirl.init_connect_info("")
    bot.loop.create_task(sillyGirl.polls())
    bot.loop.create_task(sillyGirl.polls())


@listener(
    is_plugin=True,
    outgoing=True,
    ignore_edited=True,
    command="sillyGirl",
    description="连接到傻妞服务器",
    parameters="[auth]",
)
async def Connect(message: Message):
    try:
        await edit_delete(message, "连接中，建议重启...")
        sillyGirl.init_connect_info(message.arguments)
    except Exception as e:
        print(e, "+++")
        print(e)


@listener(outgoing=True, ignore_edited=True, incoming=True)
async def handle_receive(message: Message):
    try:
        reply_to = message.id
        reply_to_sender_id = 0
        chat_id = message.chat.id
        sender_id = 0
        user_name = ""
        chat_name = message.chat.title or ""
        if message.from_user:
            user_name = message.from_user.first_name
            if message.from_user.last_name:
                user_name += f" {message.from_user.last_name}"
            sender_id = message.from_user.id
            if reply := message.reply_to_message:
                reply_to = reply.id
                if reply.from_user:
                    reply_to_sender_id = reply.from_user.id
            if message.reply_to_message_id:
                reply_to = message.reply_to_message_id
            if message.reply_to_message:
                reply_to = message.reply_to_message.id
                if message.reply_to_message.from_user:
                    reply_to_sender_id = message.reply_to_message.from_user.id
        if sillyGirl.init != True:
            sillyGirl.init_connect_info("")
        # if reply_to_sender_id==0 or sillyGirl.self_user_id == reply_to_sender_id:
        #     reply_to = 0
        await sillyGirl.poll(
            [
                {
                    "id": message.id,
                    "chat_id": 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
                    in [ChatType.SUPERGROUP, ChatType.CHANNEL],
                    "user_name": user_name,
                    "chat_name": chat_name,
                }
            ]
        )

    except Exception as e:
        print(e)
        print(e, "---")
    return