import mysql.connector
import os
from dotenv import load_dotenv

# Load environment variables from backend/.env
load_dotenv('backend/.env')

def check_db():
    try:
        conn = mysql.connector.connect(
            host=os.getenv('DB_HOST'),
            port=os.getenv('DB_PORT'),
            user=os.getenv('DB_USER'),
            password=os.getenv('DB_PASSWORD'),
            database=os.getenv('DB_NAME')
        )
        cursor = conn.cursor()

        print("\n--- Aiven Cloud DB Status ---")
        
        # Check Sources
        cursor.execute("SELECT COUNT(*) FROM sources")
        source_count = cursor.fetchone()[0]
        print(f"Total Sources: {source_count}")

        # Check Chunks
        cursor.execute("SELECT COUNT(*) FROM chunks")
        chunk_count = cursor.fetchone()[0]
        print(f"Total Chunks stored: {chunk_count}")

        if source_count > 0:
            print("\nRecent Sources:")
            cursor.execute("SELECT title, type, chunk_count FROM sources ORDER BY created_at DESC LIMIT 5")
            for title, stype, count in cursor.fetchall():
                print(f" - [{stype.upper()}] {title} ({count} chunks)")

        cursor.close()
        conn.close()
    except Exception as e:
        print(f"Error connecting to cloud DB: {e}")

if __name__ == "__main__":
    check_db()