import logging
import sys
import uuid
from app.routers import fixstuff_auto
from fastapi import FastAPI, Request
from app.routers import main, delivery, confirm, fixstuff_customer
from fastapi.middleware.cors import CORSMiddleware
from starlette.middleware.base import BaseHTTPMiddleware
import os
from config import load_config
from sqlalchemy import create_engine, text
from sqlalchemy.orm import sessionmaker


ApplicationConfig = load_config()

# Configure logging
def setup_logging():
    """Configure structured logging for the application."""
    log_level = logging.DEBUG if ApplicationConfig.CURRENT_SETTINGS != 'PRODUCTION' else logging.INFO

    formatter = logging.Formatter(
        '%(asctime)s - %(name)s - %(levelname)s - %(message)s',
        datefmt='%Y-%m-%d %H:%M:%S'
    )

    root_logger = logging.getLogger()
    root_logger.setLevel(log_level)
    root_logger.handlers.clear()

    console_handler = logging.StreamHandler(sys.stdout)
    console_handler.setLevel(log_level)
    console_handler.setFormatter(formatter)
    root_logger.addHandler(console_handler)

    logging.getLogger('uvicorn.access').setLevel(logging.WARNING)

    return logging.getLogger('eamco_auto_api')

logger = setup_logging()

# Database setup with connection pooling
engine = create_engine(
    ApplicationConfig.SQLALCHEMY_DATABASE_URI,
    pool_pre_ping=True,
    pool_size=5,
    max_overflow=10,
    pool_recycle=3600,
)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

def check_db_connection():
    """
    Test database connectivity.
    """
    try:
        db = SessionLocal()
        db.execute(text("SELECT 1"))
        db.close()
        return True
    except Exception:
        return False

app = FastAPI()


# Request ID middleware for request tracking/correlation
class RequestIDMiddleware(BaseHTTPMiddleware):
    async def dispatch(self, request: Request, call_next):
        request_id = request.headers.get("X-Request-ID") or str(uuid.uuid4())[:8]
        request.state.request_id = request_id
        response = await call_next(request)
        response.headers["X-Request-ID"] = request_id
        return response

app.add_middleware(RequestIDMiddleware)

app.include_router(main.router)
app.include_router(delivery.router)
app.include_router(confirm.router)
app.include_router(fixstuff_auto.router)
app.include_router(fixstuff_customer.router)

app.add_middleware(
    CORSMiddleware,
    allow_origins=ApplicationConfig.origins,
    allow_credentials=True,
    allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
    allow_headers=["Authorization", "Content-Type", "Accept", "Origin", "X-Requested-With"],
)




@app.get("/")
def read_root():
    return {"Status": "Auto Service is online"}

@app.on_event("startup")
async def startup_event():
    """Application startup - log configuration and test DB connection."""
    logger.info("🚀 eamco_auto_api STARTING")
    mode = ApplicationConfig.CURRENT_SETTINGS.upper()
    if mode in ['DEVELOPMENT', 'DEV']:
        logger.info("🤖🤖🤖🤖🤖 Mode: Development 🤖🤖🤖🤖🤖")
    elif mode in ['PRODUCTION', 'PROD']:
        logger.info("💀💀💀💀💀💀💀💀💀💀 ⚠️ WARNING PRODUCTION 💀💀💀💀💀💀💀💀💀💀")
    # Sanitize DB URI to avoid logging credentials
    db_uri = ApplicationConfig.SQLALCHEMY_DATABASE_URI
    if '@' in db_uri:
        db_uri = db_uri.split('@')[-1]  # Only show host/db portion
    logger.info(f"DB: ...@{db_uri[:50]}")
    logger.info(f"CORS: {len(ApplicationConfig.origins)} origins configured")

    # Test database connection
    if check_db_connection():
        logger.info("DB Connection: ✅ OK")
    else:
        logger.info("DB Connection: ❌ FAILED")