#!/usr/bin/env python3
"""
Script to delete ALL corrupted conversations where participants is a string
"""

import asyncio
import sys
from pathlib import Path

# Add parent directory to path
sys.path.insert(0, str(Path(__file__).parent.parent))

from app.database import connect_db, disconnect_db, get_db
import logging

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


async def fix_all_corrupted_conversations():
    """Delete ALL conversations where participants is stored as a string"""
    
    try:
        await connect_db()
        db = await get_db()
        
        # Method 1: Find by type (string instead of array)
        corrupted = await db.conversations.find({
            "participants": {"$type": "string"}
        }).to_list(length=None)
        
        logger.info(f"Found {len(corrupted)} corrupted by type check")
        
        # Method 2: Delete directly by the known problematic user ID
        known_corrupted = await db.conversations.find({
            "participants": "6901d49e03321114aa21e246"  # String, not in array
        }).to_list(length=None)
        
        logger.info(f"Found {len(known_corrupted)} with known corrupted participant")
        
        # Combine and dedupe
        all_ids = set()
        for conv in corrupted + known_corrupted:
            all_ids.add(conv["_id"])
        
        logger.info(f"Total unique corrupted: {len(all_ids)}")
        
        # Delete them
        deleted = 0
        for conv_id in all_ids:
            # Delete messages first
            await db.messages.delete_many({"conversation_id": str(conv_id)})
            # Delete conversation
            result = await db.conversations.delete_one({"_id": conv_id})
            if result.deleted_count > 0:
                deleted += 1
                logger.info(f"  ✅ Deleted corrupted conversation: {conv_id}")
        
        # Also check if there are any AIDA_BOT conversations with wrong structure
        aida_convs = await db.conversations.find({
            "$or": [
                {"participants": "AIDA_BOT"},
                {"participants": {"$elemMatch": {"$eq": "AIDA_BOT"}}}
            ]
        }).to_list(length=None)
        
        logger.info(f"Found {len(aida_convs)} AIDA conversations to verify")
        
        for conv in aida_convs:
            participants = conv.get("participants")
            # Check if it's not properly an array of 2 strings
            if not isinstance(participants, list) or len(participants) != 2:
                await db.messages.delete_many({"conversation_id": str(conv["_id"])})
                await db.conversations.delete_one({"_id": conv["_id"]})
                deleted += 1
                logger.info(f"  ✅ Deleted malformed AIDA conversation: {conv['_id']}")
        
        logger.info(f"\n{'='*60}")
        logger.info(f"✅ Cleanup Complete!")
        logger.info(f"Deleted {deleted} corrupted conversations")
        logger.info(f"{'='*60}\n")
        
    except Exception as e:
        logger.error(f"Cleanup failed: {e}")
        raise
    finally:
        await disconnect_db()


if __name__ == "__main__":
    logger.info("Fixing ALL corrupted conversations...")
    asyncio.run(fix_all_corrupted_conversations())