"""
日志配置工具
"""
import logging
import sys
from pathlib import Path
from logging.handlers import RotatingFileHandler, TimedRotatingFileHandler
from datetime import datetime


class LoggerConfig:
    """日志配置类"""
    
    def __init__(self, log_dir: str = "logs", log_level: str = "INFO"):
        self.log_dir = Path(log_dir)
        self.log_level = getattr(logging, log_level.upper(), logging.INFO)
        self.ensure_log_directory()
    
    def ensure_log_directory(self):
        """确保日志目录存在"""
        if not self.log_dir.exists():
            self.log_dir.mkdir(parents=True, exist_ok=True)
    
    def setup_logger(
        self, 
        name: str = "app",
        log_file: str = None,
        max_bytes: int = 10 * 1024 * 1024,  # 10MB
        backup_count: int = 5,
        use_timed_rotation: bool = False
    ) -> logging.Logger:
        """配置日志记录器
        
        Args:
            name: 日志记录器名称
            log_file: 日志文件名（None则使用时间戳）
            max_bytes: 单个日志文件最大大小
            backup_count: 保留的备份文件数量
            use_timed_rotation: 是否使用按时间轮转
            
        Returns:
            配置好的Logger实例
        """
        logger = logging.getLogger(name)
        logger.setLevel(self.log_level)
        
        # 避免重复添加handler
        if logger.handlers:
            return logger
        
        # 创建formatter
        formatter = logging.Formatter(
            '%(asctime)s - %(name)s - %(levelname)s - %(message)s',
            datefmt='%Y-%m-%d %H:%M:%S'
        )
        
        # 控制台handler
        console_handler = logging.StreamHandler(sys.stdout)
        console_handler.setLevel(self.log_level)
        console_handler.setFormatter(formatter)
        logger.addHandler(console_handler)
        
        # 文件handler
        if log_file is None:
            timestamp = datetime.now().strftime("%Y%m%d")
            log_file = f"app_{timestamp}.log"
        
        log_path = self.log_dir / log_file
        
        if use_timed_rotation:
            # 按时间轮转（每天）
            file_handler = TimedRotatingFileHandler(
                log_path,
                when='midnight',
                interval=1,
                backupCount=backup_count,
                encoding='utf-8'
            )
        else:
            # 按大小轮转
            file_handler = RotatingFileHandler(
                log_path,
                maxBytes=max_bytes,
                backupCount=backup_count,
                encoding='utf-8'
            )
        
        file_handler.setLevel(self.log_level)
        file_handler.setFormatter(formatter)
        logger.addHandler(file_handler)
        
        return logger
    
    @staticmethod
    def get_logger(name: str = "app") -> logging.Logger:
        """获取logger实例"""
        return logging.getLogger(name)


# 全局日志配置
def setup_logging(log_dir: str = "logs", log_level: str = "INFO") -> logging.Logger:
    """设置全局日志"""
    logger_config = LoggerConfig(log_dir, log_level)
    return logger_config.setup_logger()