feat✨: 更新内置插件

zhenxun/builtin_plugins/superuser/broadcast/__init__.py

@@ -0,0 +1,61 @@
+from typing import Annotated
+
+from nonebot import on_command
+from nonebot.adapters import Bot
+from nonebot.params import Command
+from nonebot.permission import SUPERUSER
+from nonebot.plugin import PluginMetadata
+from nonebot_plugin_alconna import UniMsg
+from nonebot_plugin_saa import Text
+from nonebot_plugin_session import EventSession
+
+from zhenxun.configs.config import Config
+from zhenxun.configs.utils import PluginExtraData, RegisterConfig, Task
+from zhenxun.services.log import logger
+from zhenxun.utils.enum import PluginType
+
+from ._data_source import BroadcastManage
+
+__plugin_meta__ = PluginMetadata(
+    name="广播",
+    description="昭告天下！",
+    usage="""
+    广播 [消息] [图片]
+    示例：广播 你们好！
+    """.strip(),
+    extra=PluginExtraData(
+        author="HibiKier",
+        version="0.1",
+        plugin_type=PluginType.SUPERUSER,
+        configs=[
+            RegisterConfig(
+                module="_task",
+                key="DEFAULT_BROADCAST",
+                value=True,
+                help="被动 广播 进群默认开关状态",
+                default_value=True,
+                type=bool,
+            )
+        ],
+        tasks=[Task(module="broadcast", name="广播")],
+    ).dict(),
+)
+
+_matcher = on_command("广播", priority=1, permission=SUPERUSER, block=True)
+
+
+@_matcher.handle()
+async def _(
+    bot: Bot,
+    session: EventSession,
+    message: UniMsg,
+    command: Annotated[tuple[str, ...], Command()],
+):
+    message[0].text = message[0].text.replace(command[0], "").strip()
+    # await Text("正在发送..请等一下哦!").send()
+    count, error_count = await BroadcastManage.send(bot, message, session)
+    result = f"成功广播 {count} 个群组"
+    if error_count:
+        result += f"\n广播失败 {error_count} 个群组"
+    await Text(f"发送广播完成!\n{result}").send(reply=True)
+    logger.info(f"发送广播信息: {message}", "广播", session=session)

zhenxun/builtin_plugins/superuser/broadcast/_data_source.py

@@ -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 []