"""
Database initialization script for SCM microservice.
Creates all collections, indexes, and validation rules.
"""
import asyncio
from motor.motor_asyncio import AsyncIOMotorClient
from app.core.logging import get_logger
from app.core.config import settings
from app.db_init.merchant_settings_init import initialize_merchant_settings_database

logger = get_logger(__name__)


async def initialize_database():
    """
    Initialize the complete SCM database structure.
    
    This script creates:
    - Collections with schema validation
    - Performance indexes
    - Default data (if needed)
    """
    client = None
    try:
        logger.info("Starting database initialization")
        
        # Connect to MongoDB
        logger.info(f"Connecting to MongoDB: {settings.MONGODB_URI}")
        client = AsyncIOMotorClient(settings.MONGODB_URI)
        db = client[settings.MONGODB_DB_NAME]
        
        # Test connection
        await client.admin.command('ping')
        logger.info(f"Connected to database: {settings.MONGODB_DB_NAME}")
        
        # Initialize merchant settings
        await initialize_merchant_settings_database(db)
        
        # Add other collection initializations here as needed
        # await initialize_merchants_database(db)
        # await initialize_employees_database(db)
        # await initialize_orders_database(db)
        
        logger.info("Database initialization completed successfully")
        
    except Exception as e:
        logger.error("Database initialization failed", exc_info=e)
        raise
    finally:
        if client:
            client.close()
            logger.info("Database connection closed")


async def verify_database_setup():
    """
    Verify that the database setup is correct.
    """
    client = None
    try:
        logger.info("Starting database verification")
        
        # Connect to MongoDB
        client = AsyncIOMotorClient(settings.MONGODB_URI)
        db = client[settings.MONGODB_DB_NAME]
        
        # Test connection
        await client.admin.command('ping')
        
        # List all collections
        collections = await db.list_collection_names()
        logger.info(f"Found collections: {collections}")
        
        # Check merchant_settings collection
        from app.merchants.constants import SCM_MERCHANT_SETTINGS_COLLECTION
        
        if SCM_MERCHANT_SETTINGS_COLLECTION in collections:
            collection = db[SCM_MERCHANT_SETTINGS_COLLECTION]
            
            # Check indexes
            indexes = await collection.index_information()
            logger.info(f"Merchant settings indexes: {list(indexes.keys())}")
            
            # Check collection stats
            stats = await db.command("collStats", SCM_MERCHANT_SETTINGS_COLLECTION)
            logger.info(f"Merchant settings collection stats:", extra={
                "count": stats.get("count", 0),
                "size": stats.get("size", 0),
                "indexes": stats.get("nindexes", 0)
            })
            
            logger.info("✅ Merchant settings collection is properly configured")
        else:
            logger.warning(f"❌ Collection {SCM_MERCHANT_SETTINGS_COLLECTION} not found")
        
        logger.info("Database verification completed")
        
    except Exception as e:
        logger.error("Database verification failed", exc_info=e)
        raise
    finally:
        if client:
            client.close()


if __name__ == "__main__":
    import sys
    
    # Check command line arguments
    if len(sys.argv) > 1:
        command = sys.argv[1]
        
        if command == "init":
            print("🚀 Initializing database...")
            asyncio.run(initialize_database())
            print("✅ Database initialization completed")
            
        elif command == "verify":
            print("🔍 Verifying database setup...")
            asyncio.run(verify_database_setup())
            print("✅ Database verification completed")
            
        else:
            print(f"Unknown command: {command}")
            print("Usage: python db_init.py [init|verify]")
            sys.exit(1)
    else:
        print("Usage:")
        print("  python db_init.py init    - Initialize database structure")
        print("  python db_init.py verify  - Verify database setup")
        sys.exit(1)