# BotLibrary/library/logger.py
# Создание логгеров и подключение функций логирования в файлы

import sys
from loguru import logger

from BotLibrary.configs import ProjectPath
from BotLibrary.analitics.message_to_file import write_message_to_file
from BotLibrary.analitics.user_data_to_file import write_user_info_to_file

# Настройка экспорта модулей и логирования
__all__ = ("logger", "setup_logger", "cmd_logginger", "error_cmd_logginger",
           "logginger", "common_msg_logginger", )


# Класс для параметров логгера
class LogsSet:
    # Максимальный размер лог-файла
    max_size = "500 MB"

    # Шаблон логов для информации
    info_text = (
        "<green>{time:YYYY-MM-DD HH:mm:ss}</green> <red>|</red> "
        "<blue>PRIMO-{extra[log_type]}</blue> <red>|</red> "
        "<red>{extra[user]} |</red> <level>{message}</level>"
    )

    # Шаблон логов для ошибок
    error_text = (
        "<level>{time:YYYY-MM-DD HH:mm:ss} | "
        "<bold>ERROR-{extra[log_type]}</bold> | "
        "{extra[user]} | {message}</level>"
    )

    # Шаблон логов для отладки
    debug_text = (
        "<cyan>{time:YYYY-MM-DD HH:mm:ss}</cyan> <red>|</red> "
        "<magenta>DEBUG-{extra[log_type]}</magenta> <red>|</red> "
        "<yellow>{extra[user]} |</yellow> <level>{message}</level>"
    )

    # Шаблон логов для предупреждений
    warning_text = (
        "<yellow>{time:YYYY-MM-DD HH:mm:ss}</yellow> <red>|</red> "
        "<orange>WARNING-{extra[log_type]}</orange> <red>|</red> "
        "<red>{extra[user]} |</red> <level>{message}</level>"
    )


# Создание обычного логгера + логгер в файл
async def setup_logger():
    logger.remove()  # Удаляем все логгеры

    # Пустой логгер для записи отступов в файл уровня TRACE
    logger.add(ProjectPath.log_file,
               rotation=LogsSet.max_size,
               format="\n\n\n",
               backtrace=True,
               diagnose=True,
               level="TRACE",
               filter=lambda record: record["level"].name == "TRACE")
    logger.trace("")

    # Логгер для вывода в консоль и файл с уровнем INFO
    logger.add(sys.stderr,
               colorize=True,
               format=LogsSet.info_text,
               level="INFO",
               filter=lambda record: record["level"].name == "INFO")
    logger.add(ProjectPath.log_file,
               rotation=LogsSet.max_size,
               format=LogsSet.info_text,
               backtrace=True,
               diagnose=True,
               level="INFO",
               filter=lambda record: record["level"].name == "INFO")

    # Логгер для вывода в консоль и файл с уровнем ERROR
    logger.add(sys.stderr,
               colorize=True,
               format=LogsSet.error_text,
               level="ERROR",
               filter=lambda record: record["level"].name == "ERROR")
    logger.add(ProjectPath.log_error_file,
               rotation=LogsSet.max_size,
               format=LogsSet.error_text,
               backtrace=True,
               diagnose=True,
               level="ERROR",
               filter=lambda record: record["level"].name == "ERROR")


# Запись сообщения в файл и информации о пользователи
async def logginger(message):
    write_user_info_to_file(message.from_user)
    await write_message_to_file(message)


# Создание функции логирования на обычные сообщения
async def common_msg_logginger(message, name, message_type, log_type):
    # Проверка на наличие текста и его типа
    if message.text is None:
        logger.bind(log_type=log_type, user=f"@{message.from_user.username}").info(
            f"Получено сообщение из ({name}) : {message_type}")
    else:
        logger.bind(log_type=log_type, user=f"@{message.from_user.username}").info(
            f"Получено сообщение из ({name}) : {message.text}")


# Специальный логгер для команд. Вывод в консоль, файл и запись информации о пользователи
async def cmd_logginger(message, log_type, text):
    await logginger(message)
    logger.bind(log_type=log_type, user=f"@{message.from_user.username}").info(text)


# Специальный логгер для ошибок с командами. Вывод в консоль, файл и запись информации о пользователи
async def error_cmd_logginger(message, log_type, e):
    text_error = f"Ошибка при использовании команды /{log_type.lower()}: {str(e)}\n"
    await logginger(message)
    logger.bind(log_type=log_type, user=f"@{message.from_user.username}").error(text_error)