HeXi/hexi/plugins/nonebot_plugin_group_tools/handlers/chat/greet.py

"""群管理工具 —— 聊天互动(问候/尬聊)。"""

from __future__ import annotations

from nonebot import get_driver, on_command, on_fullmatch, on_keyword, require
from nonebot.adapters.onebot.v11 import Bot, MessageEvent
from nonebot.rule import to_me
from nonebot_plugin_alconna import UniMessage

from ...constants import RES_DIR
from ...services.moderation import silence
from ...utils.media import image_to_base64

require("nonebot_plugin_alconna")

SUPERUSERS = get_driver().config.superusers

say_hello = on_command(
    "zai?",
    aliases={"在?", "在？", "在吗", "在么？", "在嘛", "在嘛？"},
    rule=to_me(),
    block=True,
)


@say_hello.handle()
async def say_hello_handle(ev: MessageEvent):
    await UniMessage.text("はい！私はいつも貴方の側にいますよ！").send()


say_sorry = on_fullmatch("沙雕机器人")


@say_sorry.handle()
async def say_sorry_handle(ev: MessageEvent):
    await UniMessage.text("ごめんなさい！嘤嘤嘤(〒︿〒)").send()


say_heweidao = on_keyword({"何意味"})


@say_heweidao.handle()
async def say_heweidao_handle(ev: MessageEvent):
    await UniMessage.text("何意味是什么意思喵~").send()


chat_waifu = on_fullmatch(("老婆", "waifu", "laopo"), rule=to_me())


@chat_waifu.handle()
async def chat_waifu_handle(ev: MessageEvent):
    is_superuser = str(ev.user_id) in SUPERUSERS
    img_name = "love.jpg" if is_superuser else "laopo.jpg"
    base64_str = image_to_base64(RES_DIR / "img" / img_name)
    if not base64_str:
        return
    if is_superuser:
        msg = UniMessage.text("mua~") + UniMessage.image(url=base64_str)
    else:
        msg = UniMessage.image(url=base64_str)
    await msg.send()


chat_laogong = on_fullmatch("老公", rule=to_me())


@chat_laogong.handle()
async def chat_laogong_handle(ev: MessageEvent):
    await UniMessage.text("你给我滚！").send(at_sender=True)


chat_mua = on_fullmatch("mua", rule=to_me())


@chat_mua.handle()
async def chat_mua_handle(ev: MessageEvent):
    await UniMessage.text("笨蛋~").send(at_sender=True)


ddhaole = on_fullmatch(("我有个朋友说他好了", "我朋友说他好了"))


@ddhaole.handle()
async def ddhaole_handle(bot: Bot, ev: MessageEvent):
    await UniMessage.text("那个朋友是不是你弟弟？").send()
    await silence(bot, ev, 30)


nihaole = on_fullmatch("我好了")


@nihaole.handle()
async def nihaole_handle(bot: Bot, ev: MessageEvent):
    await UniMessage.text("不许好，憋回去！").send()