path: root/ATRI/plugins/essential/__init__.py

import json
import shutil
import asyncio
from pathlib import Path
from datetime import datetime
from random import choice, randint

from nonebot.adapters.onebot.v11 import (
    Bot,
    MessageEvent,
    FriendRequestEvent,
    GroupRequestEvent,
    GroupIncreaseNoticeEvent,
    GroupDecreaseNoticeEvent,
    GroupAdminNoticeEvent,
    GroupBanNoticeEvent,
    GroupRecallNoticeEvent,
    FriendRecallNoticeEvent,
)

from ATRI import conf
from ATRI.log import log
from ATRI.service import Service
from ATRI.utils.apscheduler import scheduler
from ATRI.utils import FileDealer, MessageChecker
from ATRI.permission import MASTER
from ATRI.message import MessageBuilder

from .models import RequestInfo
from .data_source import recall_msg_dealer


__ESSENTIAL_DIR = Path(".") / "data" / "plugins" / "essential"
__TEMP_DIR = Path(".") / "data" / "temp"
__ESSENTIAL_DIR.mkdir(parents=True, exist_ok=True)
__TEMP_DIR.mkdir(parents=True, exist_ok=True)

# __REJECT_RECALL_LIST = list()


plugin = Service("基础部件").document("对基础请求进行处理")


friend_add_event = plugin.on_request("好友申请", "好友申请检测")


@friend_add_event.handle()
async def _(event: FriendRequestEvent):
    path = __ESSENTIAL_DIR / "friend_add.json"
    file = FileDealer(path)
    if not path.is_file():
        await file.write(json.dumps(dict()))
        data = dict()

    apply_code = event.flag
    user_id = event.get_user_id()
    apply_comment = event.comment
    now_time = str(datetime.now().timestamp())

    data = file.json()
    data[apply_code] = RequestInfo(
        user_id=user_id,
        comment=apply_comment,
        time=now_time,
    )
    await file.write(json.dumps(data))

    result = (
        MessageBuilder("咱收到一条好友请求!")
        .text(f"请求人: {user_id}")
        .text(f"申请信息: {apply_comment}")
        .text(f"申请码: {apply_code}")
        .text("Tip: 好友申请列表")
    )
    await plugin.send_to_master(result)


group_invite_request = plugin.on_request("应邀入群", "应邀入群检测")


@group_invite_request.handle()
async def _(event: GroupRequestEvent):
    path = __ESSENTIAL_DIR / "group_invite.json"
    file = FileDealer(path)
    if not path.is_file():
        await file.write(json.dumps(dict()))
        data = dict()

    apply_code = event.flag
    user_id = event.get_user_id()
    apply_comment = event.comment
    now_time = str(datetime.now().timestamp())

    data = file.json()
    data[apply_code] = RequestInfo(
        user_id=user_id,
        comment=apply_comment,
        time=now_time,
    )
    await file.write(json.dumps(data))

    result = (
        MessageBuilder("咱收到一条应邀入群请求!")
        .text(f"申请人: {user_id}")
        .text(f"申请信息: {apply_comment}")
        .text(f"申请码: {apply_code}")
        .text("Tip: 应邀入群列表")
    )
    await plugin.send_to_master(result)


group_member_event = plugin.on_notice("群成员变动", "群成员变动检测")


@group_member_event.handle()
async def _(event: GroupIncreaseNoticeEvent):
    await asyncio.sleep(randint(1, 6))
    await group_member_event.finish(
        MessageBuilder("好欸! 事新人!").text(f"在下 {choice(list(conf.BotConfig.nickname))}")
    )


@group_member_event.handle()
async def _(event: GroupDecreaseNoticeEvent):
    await asyncio.sleep(randint(1, 6))
    await group_member_event.finish("呜——有人跑了...")


group_admin_event = plugin.on_notice("群管理变动", "群管理变动检测")


@group_admin_event.handle()
async def _(event: GroupAdminNoticeEvent):
    sub_type = event.sub_type
    if event.is_tome() and sub_type == "set":
        await plugin.send_to_master(f"好欸! 咱在群 {event.group_id} 成为了管理!!")
        return
    elif sub_type == "set":
        await group_admin_event.finish("新的py交易已达成")
    else:
        await group_admin_event.finish("有的人超能力到期了 (")


group_ban_event = plugin.on_notice("群禁言变动", "群禁言变动检测")


@group_ban_event.handle()
async def _(event: GroupBanNoticeEvent):
    if not event.is_tome():
        await group_ban_event.finish("群友喝下管理的红茶昏睡了过去")

    if event.duration:
        await plugin.send_to_master(
            MessageBuilder("那个...")
            .text(f"咱在群 {event.group_id} 被 {event.operator_id} 塞上了口球...")
            .text(f"时长...是 {event.duration} 秒")
        )
    else:
        await plugin.send_to_master(
            MessageBuilder("好欸!").text(
                f"咱在群 {event.group_id} 的口球被 {event.operator_id} 解除了!"
            )
        )


_acc_recall = True


recall_event = plugin.on_notice("撤回事件", "撤回事件检测")


@recall_event.handle()
async def _(bot: Bot, event: FriendRecallNoticeEvent):
    if event.is_tome() and not _acc_recall:
        return

    try:
        repo = await bot.get_msg(message_id=event.message_id)
    except Exception:
        return

    log.debug(f"Recall raw msg:\n{repo}")
    user = event.user_id
    repo = repo["message"]

    try:
        m = recall_msg_dealer(repo)
    except Exception:
        check = MessageChecker(repo).check_cq_code
        if not check:
            m = repo
        else:
            return

    await plugin.send_to_master(
        MessageBuilder("咱拿到了一条撤回信息!").text(f"{user}@[私聊]").text(f"撤回了\n{m}")
    )


@recall_event.handle()
async def _(bot: Bot, event: GroupRecallNoticeEvent):
    if event.is_tome() and not _acc_recall:
        return

    try:
        repo = await bot.get_msg(message_id=event.message_id)
    except Exception:
        return

    log.debug(f"Recall raw msg:\n{repo}")
    user = event.user_id
    group = event.group_id
    repo = repo["message"]

    try:
        m = recall_msg_dealer(repo)
    except Exception:
        check = MessageChecker(repo).check_cq_code
        if not check:
            m = repo
        else:
            return
    await plugin.send_to_master(
        MessageBuilder("咱拿到了一条撤回信息!").text(f"{user}@[群:{group}]").text(f"撤回了\n{m}")
    )


reject_recall = plugin.on_command("拒绝撤回", "拒绝撤回信息", permission=MASTER)


@reject_recall.handle()
async def _(event: MessageEvent):
    # global __REJECT_RECALL_LIST

    # user_id = event.get_user_id()
    # if not user_id in __REJECT_RECALL_LIST:
    #     __REJECT_RECALL_LIST.append(user_id)
    global _acc_recall

    _acc_recall = False
    await reject_recall.finish("完成~")


accept_recall = plugin.on_command("接受撤回", "拒绝撤回信息", permission=MASTER)


@accept_recall.handle()
async def _(event: MessageEvent):
    # global __REJECT_RECALL_LIST

    # user_id = event.get_user_id()
    # if user_id in __REJECT_RECALL_LIST:
    #     __REJECT_RECALL_LIST.remove(user_id)
    global _acc_recall

    _acc_recall = True
    await accept_recall.finish("完成~")


@scheduler.scheduled_job("interval", name="清除缓存", minutes=30, misfire_grace_time=5)  # type: ignore
async def _():
    try:
        shutil.rmtree(__TEMP_DIR)
        __TEMP_DIR.mkdir(parents=True, exist_ok=True)
    except Exception:
        log.warning("清除缓存失败, 请手动清除: data/temp")


from ATRI import driver

from .listener import init_listener


driver().on_startup(init_listener)