import time
import psutil
from datetime import datetime

from ATRI.service import Service
from ATRI.log import logger as log
from ATRI.rule import is_in_service
from ATRI.exceptions import GetStatusError


_status_msg = """
> Status Overview

[CPU: {cpu}%]
[Memory: {mem}%]
[Disk usage: {disk}%]

[Net sent: {inteSENT}MB]
[Net recv: {inteRECV}MB]

[Runtime: {up_time}]
{msg}
""".strip()


class IsSurvive(Service):
    def __init__(self):
        Service.__init__(self, "状态", "检查自身状态", rule=is_in_service("状态"))

    @staticmethod
    def ping() -> str:
        return "I'm fine."

    @staticmethod
    def get_status():
        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

            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 = "咱感觉有些头晕..."
            is_ok = False
            if mem > 90:
                msg = "咱感觉有点头晕并且有点累..."
                is_ok = False
        elif mem > 90:
            msg = "咱感觉有点累..."
            is_ok = False
        elif disk > 90:
            msg = "咱感觉身体要被塞满了..."
            is_ok = False
        else:
            log.info("资源占用正常")
            is_ok = True

        msg0 = _status_msg.format(
            cpu=cpu,
            mem=mem,
            disk=disk,
            inteSENT=inteSENT,
            inteRECV=inteRECV,
            up_time=up_time,
            msg=msg,
        )

        return msg0, is_ok