import os
import logging
import json
import sys
from datetime import datetime, timezone
from context import get_request_id, get_current_user  # Importamos las funciones
from logging.handlers import RotatingFileHandler


# ============================================================================
class JsonFormatter(logging.Formatter):
    """Formateador personalizado para exportar logs en formato JSON puro."""
    EXTRA_FIELDS = ("http_method", "endpoint", "status_code")

    def format(self, record):
        user = get_current_user()
        log_record = {
            "timestamp": datetime.fromtimestamp(record.created, tz=timezone.utc).isoformat(),
            "level": record.levelname,
            "request_id": get_request_id(),  # ⚡️ Inyección automática del ID de rastreo
            "webUser": user["webUser"],
            "emp_id": user["emp_id"],
            "full_name": user["full_name"],
            "message": record.getMessage(),
            "module": record.module,
            "function": record.funcName,
            "line": record.lineno
        }
        for field in self.EXTRA_FIELDS:
            if hasattr(record, field):
                log_record[field] = getattr(record, field)
        # Si el log incluye excepciones, las añade al JSON
        if record.exc_info:
            log_record["exception"] = self.formatException(record.exc_info)
        return json.dumps(log_record, ensure_ascii=False)


# ============================================================================
def setup_logging(environment: str = "production"):
    """Configura el logger global de la aplicación."""
    root_logger = logging.getLogger()
    
    # Evita duplicar logs si la función se ejecuta más de una vez
    if root_logger.hasHandlers():
        root_logger.handlers.clear()

    # Definir el nivel mínimo de captura (INFO por defecto)
    root_logger.setLevel(logging.INFO)

    # Handler para la consola (Standard Output)
    console_handler = logging.StreamHandler(sys.stdout)

    if environment == "production":
        # En producción usamos JSON estructurado
        console_handler.setFormatter(JsonFormatter())
    else:
        # En desarrollo usamos texto limpio y con colores opcionales
        readable_formatter = logging.Formatter(
            fmt="%(asctime)s [%(levelname)s] %(module)s.%(funcName)s:%(lineno)d - %(message)s",
            datefmt="%Y-%m-%d %H:%M:%S"
        )
        console_handler.setFormatter(readable_formatter)

    root_logger.addHandler(console_handler)

    # 2. CREAR CARPETA Y ARCHIVO FÍSICO EN DISCO
    log_dir = "logs"
    if not os.path.exists(log_dir):
        os.makedirs(log_dir) # Crea la carpeta 'logs/' si no existe

    # Usamos 'RotatingFileHandler' para que el archivo no crezca infinitamente.
    # Cuando llegue a 5MB, lo cierra, lo renombra a backend.log.1 y crea uno nuevo.
    file_handler = RotatingFileHandler(
        filename=os.path.join(log_dir, "backend.log"),
        maxBytes=5 * 1024 * 1024,  # 5 Megabytes
        backupCount=3,             # Mantener máximo 3 archivos viejos de respaldo
        encoding="utf-8"
    )
    file_handler.setFormatter(JsonFormatter())
    root_logger.addHandler(file_handler)


    # Silenciar logs ruidosos de librerías externas (opcional)
    logging.getLogger("uvicorn.access").setLevel(logging.WARNING)
# ============================================================================