"""群聊学习 —— 触发器(Trigger)层。"""

from __future__ import annotations

import asyncio
import random
import time

from nonebot import get_adapter, logger, on_message, require
from nonebot.adapters.onebot.v11 import (
    GROUP,
    ActionFailed,
    Adapter,
    GroupMessageEvent,
    Message,
)
from nonebot.params import Arg
from nonebot.rule import Rule
from nonebot.typing import T_State

require("nonebot_plugin_orm")
require("nonebot_plugin_apscheduler")
from nonebot_plugin_apscheduler import scheduler

from nonebot_plugin_orm import get_session

from ..config import NICKNAME, config_manager
from ..models import ChatMessage
from ..services.learn import LearningChat


async def ChatRule(event: GroupMessageEvent, state: T_State) -> bool:
    if answers := await LearningChat(event).answer():
        state["answers"] = answers
        return True
    return False


learning_chat = on_message(
    priority=99,
    block=False,
    rule=Rule(ChatRule),
    permission=GROUP,
    state={
        "pm_name": "群聊学习",
        "pm_description": "(被动技能)bot会学习群友们的发言",
        "pm_usage": "群聊学习",
        "pm_priority": 1,
    },
)


@learning_chat.handle()
async def _(event: GroupMessageEvent, answers=Arg("answers")):
    for answer in answers:
        try:
            logger.info(
                "群聊学习", f'{NICKNAME}将向群<m>{event.group_id}</m>回复<m>"{answer}"</m>'
            )
            msg = await learning_chat.send(Message(answer))
            async with get_session() as session:
                session.add(ChatMessage(
                    group_id=event.group_id,
                    user_id=event.self_id,
                    message_id=msg["message_id"],
                    message=answer,
                    raw_message=answer,
                    time=int(time.time()),
                    plain_text=Message(answer).extract_plain_text(),
                ))
                await session.commit()
            await asyncio.sleep(random.random() + 0.5)
        except ActionFailed:
            logger.info(
                "群聊学习",
                f'{NICKNAME}向群<m>{event.group_id}</m>的回复<m>"{answer}"</m>发送<r>失败，可能处于风控中</r>',
            )


@scheduler.scheduled_job("interval", minutes=3, misfire_grace_time=5)
async def speak_up():
    if not config_manager.config.total_enable:
        return
    try:
        bots = get_adapter(Adapter).bots
        if len(bots) == 0:
            return
        bot = list(bots.values())[0]
    except ValueError:
        return
    if not (speak := await LearningChat.speak(int(bot.self_id))):
        return
    group_id, messages = speak
    for msg in messages:
        try:
            logger.info("群聊学习", f'{NICKNAME}向群<m>{group_id}</m>主动发言<m>"{msg}"</m>')
            send_result = await bot.send_group_msg(
                group_id=group_id, message=Message(msg)
            )
            async with get_session() as session:
                session.add(ChatMessage(
                    group_id=group_id,
                    user_id=int(bot.self_id),
                    message_id=send_result["message_id"],
                    message=msg,
                    raw_message=msg,
                    time=int(time.time()),
                    plain_text=Message(msg).extract_plain_text(),
                ))
                await session.commit()
            await asyncio.sleep(random.randint(2, 4))
        except ActionFailed:
            logger.info(
                "群聊学习",
                f'{NICKNAME}向群<m>{group_id}</m>主动发言<m>"{msg}"</m><r>发送失败，可能处于风控中</r>',
            )