feat✨: 更新内置插件

2026-10-05 20:00:04 +08:00 · 2024-02-25 03:18:34 +08:00
parent 131200a28e
commit eb0572ea77
83 changed files with 7588 additions and 450 deletions
@@ -0,0 +1,117 @@
+import nonebot_plugin_alconna as alc
+from nonebot.adapters import Bot
+from nonebot.adapters.discord import Bot as DiscordBot
+from nonebot.adapters.dodo import Bot as DodoBot
+from nonebot.adapters.kaiheila import Bot as KaiheilaBot
+from nonebot.adapters.onebot.v11 import Bot as v11Bot
+from nonebot.adapters.onebot.v12 import Bot as v12Bot
+from nonebot_plugin_alconna import UniMsg
+from nonebot_plugin_saa import (
+    Image,
+    MessageFactory,
+    TargetDoDoChannel,
+    TargetQQGroup,
+    Text,
+)
+from nonebot_plugin_session import EventSession
+from pydantic import BaseModel
+
+from zhenxun.models.group_console import GroupConsole
+from zhenxun.services.log import logger
+
+
+class GroupChannel(BaseModel):
+
+    group_id: str
+    """群组id"""
+    channel_id: str | None = None
+    """频道id"""
+
+
+class BroadcastManage:
+
+    @classmethod
+    async def send(
+        cls, bot: Bot, message: UniMsg, session: EventSession
+    ) -> tuple[int, int]:
+        """发送广播消息
+
+        参数:
+            bot: Bot
+            message: 消息内容
+            session: Session
+
+        返回:
+            tuple[int, int]: 发送成功的群组数量, 发送失败的群组数量
+        """
+        message_list = []
+        for msg in message:
+            if isinstance(msg, alc.Image) and msg.url:
+                message_list.append(Image(msg.url))
+            elif isinstance(msg, alc.Text):
+                message_list.append(Text(msg.text))
+        if group_list := await cls.__get_group_list(bot):
+            error_count = 0
+            for group in group_list:
+                try:
+                    if not await GroupConsole.is_block_task(
+                        group.group_id, "broadcast", group.channel_id
+                    ):
+                        if isinstance(bot, (v11Bot, v12Bot)):
+                            target = TargetQQGroup(group_id=int(group.group_id))
+                        elif isinstance(bot, DodoBot):
+                            target = TargetDoDoChannel(channel_id=group.channel_id)  # type: ignore
+                        await MessageFactory(message_list).send_to(target, bot)
+                        logger.debug(
+                            "发送成功",
+                            "广播",
+                            session=session,
+                            target=f"{group.group_id}:{group.channel_id}",
+                        )
+                except Exception as e:
+                    error_count += 1
+                    logger.error(
+                        "发送失败",
+                        "广播",
+                        session=session,
+                        target=f"{group.group_id}:{group.channel_id}",
+                        e=e,
+                    )
+            return len(group_list) - error_count, error_count
+        return 0, 0
+
+    @classmethod
+    async def __get_group_list(cls, bot: Bot) -> list[GroupChannel]:
+        """获取群组id列表
+
+        参数:
+            bot: Bot
+
+        返回:
+            list[str]: 群组id列表
+        """
+        if isinstance(bot, (v11Bot, v12Bot)):
+            group_list = await bot.get_group_list()
+            return [GroupChannel(group_id=str(g["group_id"])) for g in group_list]
+        if isinstance(bot, DodoBot):
+            island_list = await bot.get_island_list()
+            source_id_list = [
+                g.island_source_id for g in island_list if g.island_source_id
+            ]
+            channel_id_list = []
+            for id in source_id_list:
+                channel_list = await bot.get_channel_list(island_source_id=id)
+                channel_id_list += [
+                    GroupChannel(group_id=id, channel_id=c.channel_id)
+                    for c in channel_list
+                ]
+            return channel_id_list
+        if isinstance(bot, KaiheilaBot):
+            pass
+            # group_list = await bot.guild_list()
+            # if group_list.guilds:
+            #     return [g.open_id for g in group_list.guilds if g.open_id]
+        if isinstance(bot, DiscordBot):
+            # TODO: discord获取群组列表
+            pass
+        return []