from loguru import logger
import logging
import sys
import os

def chat_log_filter(record):
    return "chat_msg" in record["extra"]

def not_chat_log_filter(record):
    return "chat_msg" not in record["extra"]

def formatter_with_clip(record):
    # Note this function returns the string to be formatted, not the actual message to be logged
    # record["extra"]["serialized"] = "555555"
    max_len = 12
    record['function_x'] = record['function'].center(max_len)
    if len(record['function_x']) > max_len:
        record['function_x'] = ".." + record['function_x'][-(max_len-2):]
    record['line_x'] = str(record['line']).ljust(3)
    return '<green>{time:HH:mm}</green> | <cyan>{function_x}</cyan>:<cyan>{line_x}</cyan> | <level>{message}</level>\n'

def setup_logging(PATH_LOGGING):
    
    admin_log_path = os.path.join(PATH_LOGGING, "admin")
    os.makedirs(admin_log_path, exist_ok=True)
    sensitive_log_path = os.path.join(admin_log_path, "chat_secrets.log")
    regular_log_path = os.path.join(admin_log_path, "console_log.log")
    logger.remove()
    logger.configure(
        levels=[dict(name="WARNING", color="<g>")],
    )

    logger.add(
        sys.stderr, 
        format=formatter_with_clip,
        # format='<green>{time:HH:mm}</green> | <cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>',
        filter=(lambda record: not chat_log_filter(record)),
        colorize=True,
        enqueue=True
    )

    logger.add(
        sensitive_log_path, 
        format='<green>{time:MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>',
        rotation="10 MB",
        filter=chat_log_filter,
        enqueue=True, 
    )

    logger.add(
        regular_log_path, 
        format='<green>{time:MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>',
        rotation="10 MB",
        filter=not_chat_log_filter,
        enqueue=True, 
    )

    logging.getLogger("httpx").setLevel(logging.WARNING)

    logger.warning(f"所有对话记录将自动保存在本地目录{sensitive_log_path}, 请注意自我隐私保护哦！")


# logger.bind(chat_msg=True).info("This message is logged to the file!")
# logger.debug(f"debug message")
# logger.info(f"info message")
# logger.success(f"success message")
# logger.error(f"error message")
# logger.add("special.log", filter=lambda record: "special" in record["extra"])
# logger.debug("This message is not logged to the file")