app/api/maintenance.py

from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select, delete, func
from typing import Dict, Any, List
from datetime import datetime, timedelta
from ..core.dependencies import get_current_active_user
from ..db.database import get_db
from ..db.models import User, Order, Notification, Event
from ..utils.logger import logger

router = APIRouter()

@router.post("/sessions/cleanup")
async def cleanup_sessions(
    current_user: User = Depends(get_current_active_user),
    db: AsyncSession = Depends(get_db)
) -> Dict[str, int]:
    """Manually trigger session cleanup"""
    if "admin" not in current_user.roles:
        raise HTTPException(
            status_code=403,
            detail="Only administrators can perform maintenance operations"
        )
    
    cutoff_date = datetime.utcnow() - timedelta(days=7)
    stmt = delete(Event).where(Event.created_at < cutoff_date)
    result = await db.execute(stmt)
    await db.commit()
    
    return {"deleted_sessions": result.rowcount}

@router.post("/data/archive")
async def archive_data(
    current_user: User = Depends(get_current_active_user),
    db: AsyncSession = Depends(get_db)
) -> Dict[str, int]:
    """Manually trigger data archiving"""
    if "admin" not in current_user.roles:
        raise HTTPException(
            status_code=403,
            detail="Only administrators can perform maintenance operations"
        )
    
    archive_date = datetime.utcnow() - timedelta(days=365)
    archived = {}
    
    # Archive old orders
    orders_stmt = delete(Order).where(
        Order.created_at < archive_date,
        Order.status.in_(["delivered", "cancelled"])
    )
    orders_result = await db.execute(orders_stmt)
    archived["orders"] = orders_result.rowcount
    
    # Archive old notifications
    notif_stmt = delete(Notification).where(
        Notification.created_at < archive_date,
        Notification.read == True
    )
    notif_result = await db.execute(notif_stmt)
    archived["notifications"] = notif_result.rowcount
    
    await db.commit()
    return archived

@router.get("/health")
async def check_health(
    current_user: User = Depends(get_current_active_user),
    db: AsyncSession = Depends(get_db)
) -> Dict[str, Any]:
    """Check system health metrics"""
    if "admin" not in current_user.roles:
        raise HTTPException(
            status_code=403,
            detail="Only administrators can view system health"
        )
    
    try:
        # Check database connection
        await db.execute(select(1))
        
        # Get database statistics
        total_users = await db.scalar(select(func.count()).select_from(User))
        total_orders = await db.scalar(select(func.count()).select_from(Order))
        total_notifications = await db.scalar(select(func.count()).select_from(Notification))
        
        return {
            "status": "healthy",
            "timestamp": datetime.utcnow(),
            "database": {
                "connected": True,
                "total_users": total_users,
                "total_orders": total_orders,
                "total_notifications": total_notifications
            }
        }
    except Exception as e:
        logger.error(f"Health check error: {str(e)}")
        return {
            "status": "unhealthy",
            "error": str(e),
            "timestamp": datetime.utcnow()
        }

@router.post("/database/maintenance")
async def perform_db_maintenance(
    current_user: User = Depends(get_current_active_user),
    db: AsyncSession = Depends(get_db)
) -> Dict[str, Any]:
    """Manually trigger database maintenance"""
    if "admin" not in current_user.roles:
        raise HTTPException(
            status_code=403,
            detail="Only administrators can perform maintenance operations"
        )
    
    try:
        # Cleanup expired sessions
        await cleanup_sessions(current_user, db)
        
        # Run VACUUM ANALYZE (requires raw SQL)
        await db.execute("VACUUM ANALYZE;")
        
        return {
            "status": "success",
            "message": "Database maintenance completed successfully"
        }
    except Exception as e:
        logger.error(f"Database maintenance error: {str(e)}")
        raise HTTPException(
            status_code=500,
            detail=f"Database maintenance failed: {str(e)}"
        )