# app/core/lifespan.py
"""Summary description ....

This module provides classes and functions for:
- ....
- ....
- ....
"""

import logging
from contextlib import asynccontextmanager

from fastapi import FastAPI
from sqlalchemy import text

from middleware.auth import AuthManager
from models.database_models import Base

from .database import SessionLocal, engine
from .logging import setup_logging

logger = logging.getLogger(__name__)
auth_manager = AuthManager()


@asynccontextmanager
async def lifespan(app: FastAPI):
    """Application lifespan events."""
    # Startup
    app_logger = setup_logging()  # Configure logging first
    app_logger.info("Starting up ecommerce API")

    # Create tables
    Base.metadata.create_all(bind=engine)

    # Create indexes
    create_indexes()

    # Create default admin user
    db = SessionLocal()
    try:
        auth_manager.create_default_admin_user(db)
    except Exception as e:
        logger.error(f"Failed to create default admin user: {e}")
    finally:
        db.close()

    yield

    # Shutdown
    app_logger.info("Shutting down ecommerce API")


def create_indexes():
    """Create database indexes."""
    with engine.connect() as conn:
        try:
            # User indexes
            conn.execute(
                text("CREATE INDEX IF NOT EXISTS idx_user_email ON users(email)")
            )
            conn.execute(
                text("CREATE INDEX IF NOT EXISTS idx_user_username ON users(username)")
            )

            # Product indexes
            conn.execute(
                text("CREATE INDEX IF NOT EXISTS idx_product_gtin ON products(gtin)")
            )
            conn.execute(
                text(
                    "CREATE INDEX IF NOT EXISTS idx_product_marketplace ON products(marketplace)"
                )
            )

            # Stock indexes
            conn.execute(
                text(
                    "CREATE INDEX IF NOT EXISTS idx_stock_gtin_location ON stock(gtin, location)"
                )
            )

            conn.commit()
            logger.info("Database indexes created successfully")
        except Exception as e:
            logger.warning(f"Index creation warning: {e}")