import sys
import loguru
from enum import Enum

class LogClassification(Enum):
    BASE = "base"
    MEMORY = "memory"
    EMOJI = "emoji"
    CHAT = "chat"
    PBUILDER = "promptbuilder"

class LogModule:
    logger = loguru.logger.opt()

    def __init__(self):
        pass
    def setup_logger(self, log_type: LogClassification):
        """配置日志格式
        
        Args:
            log_type: 日志类型，可选值：BASE(基础日志)、MEMORY(记忆系统日志)、EMOJI(表情包系统日志)
        """
        # 移除默认日志处理器
        self.logger.remove()

        # 基础日志格式
        base_format = "<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
        
        chat_format = "<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
        
        # 记忆系统日志格式
        memory_format = "<green>{time:HH:mm}</green> | <level>{level: <8}</level> | <light-magenta>海马体</light-magenta> | <level>{message}</level>"
        
        # 表情包系统日志格式
        emoji_format = "<green>{time:HH:mm}</green> | <level>{level: <8}</level> | <yellow>表情包</yellow> | <cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
        
        promptbuilder_format = "<green>{time:HH:mm}</green> | <level>{level: <8}</level> | <yellow>Prompt</yellow> | <cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
        
        
        # 根据日志类型选择日志格式和输出
        if log_type == LogClassification.CHAT:
            self.logger.add(
                sys.stderr,
                format=chat_format,
                # level="INFO"
            )
        elif log_type == LogClassification.PBUILDER:
            self.logger.add(
                sys.stderr,
                format=promptbuilder_format,
                # level="INFO"
            )
        elif log_type == LogClassification.MEMORY:
            
            # 同时输出到控制台和文件
            self.logger.add(
                sys.stderr,
                format=memory_format,
                # level="INFO"
            )
            self.logger.add(
                "logs/memory.log",
                format=memory_format,
                level="INFO",
                rotation="1 day",
                retention="7 days"
            )
        elif log_type == LogClassification.EMOJI:
            self.logger.add(
                sys.stderr,
                format=emoji_format,
                # level="INFO"
            )
            self.logger.add(
                "logs/emoji.log",
                format=emoji_format,
                level="INFO",
                rotation="1 day",
                retention="7 days"
            )
        else:  # BASE
            self.logger.add(
                sys.stderr,
                format=base_format,
                level="INFO"
            )
        
        return self.logger