import time
import psutil
from datetime import datetime
from nonebot.adapters.cqhttp import Bot, MessageEvent

from ATRI.log import logger as log
from ATRI.service import Service as sv
from ATRI.rule import is_in_service
from ATRI.exceptions import GetStatusError
from ATRI.utils.apscheduler import scheduler
from ATRI.config import BotSelfConfig


__doc__ = """
测试机器人状态
权限组：所有人
用法：
  /ping
"""

ping = sv.on_command(cmd="/ping", docs="测试机器人", rule=is_in_service("ping"))


@ping.handle()
async def _ping(bot: Bot, event: MessageEvent) -> None:
    await ping.finish("I'm fine.")


__doc__ = """
检查机器人性能
权限组：所有人
用法：
  /status
"""

status = sv.on_command(cmd="/status", docs=__doc__, rule=is_in_service("status"))


@status.handle()
async def _status(bot: Bot, event: MessageEvent) -> None:
    try:
        cpu = psutil.cpu_percent(interval=1)
        mem = psutil.virtual_memory().percent
        disk = psutil.disk_usage("/").percent
        inteSENT = psutil.net_io_counters().bytes_sent / 1000000  # type: ignore
        inteRECV = psutil.net_io_counters().bytes_recv / 1000000  # type: ignore
        now = time.time()
        boot = psutil.boot_time()
        up_time = str(
            datetime.utcfromtimestamp(now).replace(microsecond=0)
            - datetime.utcfromtimestamp(boot).replace(microsecond=0)
        )
    except GetStatusError:
        raise GetStatusError("Failed to get status.")

    msg = "アトリは、高性能ですから！"

    if cpu > 90:  # type: ignore
        msg = "咱感觉有些头晕..."
        if mem > 90:
            msg = "咱感觉有点头晕并且有点累..."
    elif mem > 90:
        msg = "咱感觉有点累..."
    elif disk > 90:
        msg = "咱感觉身体要被塞满了..."

    msg0 = (
        "Self status:\n"
        f"* CPU: {cpu}%\n"
        f"* MEM: {mem}%\n"
        f"* DISK: {disk}%\n"
        f"* netSENT: {inteSENT}MB\n"
        f"* netRECV: {inteRECV}MB\n"
        f"* Runtime: {up_time}\n"
    ) + msg

    await status.finish(msg0)


@scheduler.scheduled_job("interval", minutes=5, misfire_grace_time=10)
async def _():
    log.info("开始自检")
    try:
        cpu = psutil.cpu_percent(interval=1)
        mem = psutil.virtual_memory().percent
        disk = psutil.disk_usage("/").percent
        inteSENT = psutil.net_io_counters().bytes_sent / 1000000  # type: ignore
        inteRECV = psutil.net_io_counters().bytes_recv / 1000000  # type: ignore
    except GetStatusError:
        raise GetStatusError("Failed to get status.")

    msg = str()
    if cpu > 90:  # type: ignore
        msg = "咱感觉有些头晕..."
        if mem > 90:
            msg = "咱感觉有点头晕并且有点累..."
    elif mem > 90:
        msg = "咱感觉有点累..."
    elif disk > 90:
        msg = "咱感觉身体要被塞满了..."
    else:
        log.info("运作正常")
        return

    msg0 = (
        "Self status:\n"
        f"* CPU: {cpu}%\n"
        f"* MEM: {mem}%\n"
        f"* DISK: {disk}%\n"
        f"* netSENT: {inteSENT}MB\n"
        f"* netRECV: {inteRECV}MB\n"
    ) + msg

    for sup in BotSelfConfig.superusers:
        await sv.NetworkPost.send_private_msg(user_id=sup, message=msg0)