#!/usr/bin/env python3
"""
Run method rotation system database migration.
This script applies the rotation system database schema to the existing database.
"""

import sys
import os
import sqlite3
import logging
from pathlib import Path

# Add project root to path
project_root = Path(__file__).parent
sys.path.insert(0, str(project_root))

# Setup logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)


def run_migration():
    """Run the method rotation system migration"""
    try:
        # Database path
        db_path = project_root / "database" / "accounts.db"
        migration_path = project_root / "database" / "migrations" / "add_method_rotation_system.sql"
        
        if not db_path.exists():
            logger.error(f"Database not found at {db_path}")
            return False
            
        if not migration_path.exists():
            logger.error(f"Migration file not found at {migration_path}")
            return False
            
        # Read migration SQL
        with open(migration_path, 'r', encoding='utf-8') as f:
            migration_sql = f.read()
            
        # Connect to database
        logger.info(f"Connecting to database at {db_path}")
        conn = sqlite3.connect(str(db_path))
        
        try:
            # Check if migration has already been run
            cursor = conn.cursor()
            cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='method_strategies'")
            if cursor.fetchone():
                logger.info("Method rotation tables already exist, skipping migration")
                return True
            
            # Run migration
            logger.info("Running method rotation system migration...")
            conn.executescript(migration_sql)
            conn.commit()
            
            # Verify migration
            cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name LIKE '%method%' OR name LIKE '%rotation%'")
            tables = cursor.fetchall()
            
            expected_tables = [
                'method_strategies',
                'rotation_sessions', 
                'rotation_events',
                'method_performance_analytics',
                'method_cooldowns',
                'platform_method_states'
            ]
            
            created_tables = [table[0] for table in tables]
            missing_tables = [t for t in expected_tables if t not in created_tables]
            
            if missing_tables:
                logger.error(f"Migration incomplete, missing tables: {missing_tables}")
                return False
                
            logger.info(f"Migration successful! Created tables: {created_tables}")
            
            # Verify default data was inserted
            cursor.execute("SELECT COUNT(*) FROM method_strategies")
            strategy_count = cursor.fetchone()[0]
            
            cursor.execute("SELECT COUNT(*) FROM platform_method_states")
            state_count = cursor.fetchone()[0]
            
            logger.info(f"Default data inserted: {strategy_count} strategies, {state_count} platform states")
            
            return True
            
        finally:
            conn.close()
            
    except Exception as e:
        logger.error(f"Migration failed: {e}")
        return False


def check_migration_status():
    """Check if migration has been applied"""
    try:
        db_path = project_root / "database" / "accounts.db"
        
        if not db_path.exists():
            logger.warning(f"Database not found at {db_path}")
            return False
            
        conn = sqlite3.connect(str(db_path))
        cursor = conn.cursor()
        
        # Check for rotation tables
        cursor.execute("""
            SELECT name FROM sqlite_master 
            WHERE type='table' AND (
                name LIKE '%method%' OR 
                name LIKE '%rotation%'
            )
            ORDER BY name
        """)
        
        tables = [row[0] for row in cursor.fetchall()]
        conn.close()
        
        if tables:
            logger.info(f"Method rotation tables found: {tables}")
            return True
        else:
            logger.info("No method rotation tables found")
            return False
            
    except Exception as e:
        logger.error(f"Failed to check migration status: {e}")
        return False


def main():
    """Main function"""
    logger.info("Method Rotation System Database Migration")
    logger.info("=" * 50)
    
    # Check current status
    logger.info("Checking current migration status...")
    if check_migration_status():
        logger.info("Migration already applied")
        return
    
    # Run migration
    logger.info("Applying migration...")
    if run_migration():
        logger.info("Migration completed successfully!")
        logger.info("Method rotation system is now ready to use")
    else:
        logger.error("Migration failed!")
        sys.exit(1)


if __name__ == "__main__":
    main()