bakery-ia/services/notification/app/api/notifications.py

# ================================================================
# services/notification/app/api/notifications.py - COMPLETE IMPLEMENTATION
# ================================================================
"""
Complete notification API routes with full CRUD operations
"""

from fastapi import APIRouter, Depends, HTTPException, status, Query, Path, BackgroundTasks
from typing import List, Optional, Dict, Any
import structlog
from datetime import datetime
from sqlalchemy.ext.asyncio import AsyncSession

import uuid
from sqlalchemy import select, delete, func

from app.schemas.notifications import (
    NotificationCreate,
    NotificationResponse,
    NotificationHistory,
    NotificationStats,
    NotificationPreferences,
    PreferencesUpdate,
    BulkNotificationCreate,
    TemplateCreate,
    TemplateResponse,
    DeliveryWebhook,
    ReadReceiptWebhook,
    NotificationType,
    NotificationStatus
)
from app.services.notification_service import NotificationService
from app.services.messaging import (
    handle_email_delivery_webhook,
    handle_whatsapp_delivery_webhook,
    process_scheduled_notifications
)

# Import unified authentication from shared library
from shared.auth.decorators import (
    get_current_user_dep,
    get_current_tenant_id_dep,
    require_role
)

from app.core.database import get_db

router = APIRouter()
logger = structlog.get_logger()

# ================================================================
# NOTIFICATION ENDPOINTS
# ================================================================

@router.post("/send", response_model=NotificationResponse)
async def send_notification(
    notification: NotificationCreate,
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(get_current_user_dep),
):
    """Send a single notification"""
    try:
        logger.info("Sending notification",
                   tenant_id=tenant_id,
                   sender_id=current_user["user_id"],
                   type=notification.type.value)
        
        notification_service = NotificationService()
        
        # Ensure notification is scoped to tenant
        notification.tenant_id = tenant_id
        notification.sender_id = current_user["user_id"]
        
        # Check permissions for broadcast notifications
        if notification.broadcast and current_user.get("role") not in ["admin", "manager"]:
            raise HTTPException(
                status_code=403,
                detail="Only admins and managers can send broadcast notifications"
            )
        
        result = await notification_service.send_notification(notification)
        
        return result
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error("Failed to send notification", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.post("/send-bulk")
async def send_bulk_notifications(
    bulk_request: BulkNotificationCreate,
    background_tasks: BackgroundTasks,
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(get_current_user_dep),
):
    """Send bulk notifications"""
    try:
        # Check permissions
        if current_user.get("role") not in ["admin", "manager"]:
            raise HTTPException(
                status_code=403,
                detail="Only admins and managers can send bulk notifications"
            )
        
        logger.info("Sending bulk notifications",
                   tenant_id=tenant_id,
                   count=len(bulk_request.recipients),
                   type=bulk_request.type.value)
        
        notification_service = NotificationService()
        
        # Process bulk notifications in background
        background_tasks.add_task(
            notification_service.send_bulk_notifications,
            bulk_request
        )
        
        return {
            "message": "Bulk notification processing started",
            "total_recipients": len(bulk_request.recipients),
            "type": bulk_request.type.value
        }
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error("Failed to start bulk notifications", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.get("/history", response_model=NotificationHistory)
async def get_notification_history(
    page: int = Query(1, ge=1),
    per_page: int = Query(50, ge=1, le=100),
    type_filter: Optional[NotificationType] = Query(None),
    status_filter: Optional[NotificationStatus] = Query(None),
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(get_current_user_dep),
):
    """Get notification history for current user"""
    try:
        notification_service = NotificationService()
        
        history = await notification_service.get_notification_history(
            user_id=current_user["user_id"],
            tenant_id=tenant_id,
            page=page,
            per_page=per_page,
            type_filter=type_filter,
            status_filter=status_filter
        )
        
        return history
        
    except Exception as e:
        logger.error("Failed to get notification history", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.get("/stats", response_model=NotificationStats)
async def get_notification_stats(
    days: int = Query(30, ge=1, le=365),
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(require_role(["admin", "manager"])),
):
    """Get notification statistics for tenant (admin/manager only)"""
    try:
        notification_service = NotificationService()
        
        stats = await notification_service.get_notification_stats(
            tenant_id=tenant_id,
            days=days
        )
        
        return stats
        
    except Exception as e:
        logger.error("Failed to get notification stats", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.get("/{notification_id}", response_model=NotificationResponse)
async def get_notification(
    notification_id: str,
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(get_current_user_dep),
):
    """Get a specific notification by ID"""
    try:
        # This would require implementation in NotificationService
        # For now, return a placeholder response
        raise HTTPException(
            status_code=501,
            detail="Get single notification not yet implemented"
        )
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error("Failed to get notification", notification_id=notification_id, error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.patch("/{notification_id}/read")
async def mark_notification_read(
    notification_id: str,
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(get_current_user_dep),
):
    """Mark a notification as read"""
    try:
        # This would require implementation in NotificationService
        # For now, return a placeholder response
        return {"message": "Notification marked as read", "notification_id": notification_id}
        
    except Exception as e:
        logger.error("Failed to mark notification as read", notification_id=notification_id, error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

# ================================================================
# PREFERENCE ENDPOINTS
# ================================================================

@router.get("/preferences", response_model=NotificationPreferences)
async def get_notification_preferences(
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(get_current_user_dep),
):
    """Get user's notification preferences"""
    try:
        notification_service = NotificationService()
        
        preferences = await notification_service.get_user_preferences(
            user_id=current_user["user_id"],
            tenant_id=tenant_id
        )
        
        return NotificationPreferences(**preferences)
        
    except Exception as e:
        logger.error("Failed to get preferences", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.patch("/preferences", response_model=NotificationPreferences)
async def update_notification_preferences(
    updates: PreferencesUpdate,
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(get_current_user_dep),
):
    """Update user's notification preferences"""
    try:
        notification_service = NotificationService()
        
        # Convert Pydantic model to dict, excluding None values
        update_data = updates.dict(exclude_none=True)
        
        preferences = await notification_service.update_user_preferences(
            user_id=current_user["user_id"],
            tenant_id=tenant_id,
            updates=update_data
        )
        
        return NotificationPreferences(**preferences)
        
    except Exception as e:
        logger.error("Failed to update preferences", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

# ================================================================
# TEMPLATE ENDPOINTS
# ================================================================

@router.post("/templates", response_model=TemplateResponse)
async def create_notification_template(
    template: TemplateCreate,
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(require_role(["admin", "manager"])),
):
    """Create a new notification template (admin/manager only)"""
    try:
        # This would require implementation in NotificationService
        # For now, return a placeholder response
        raise HTTPException(
            status_code=501,
            detail="Template creation not yet implemented"
        )
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error("Failed to create template", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.get("/templates", response_model=List[TemplateResponse])
async def list_notification_templates(
    category: Optional[str] = Query(None),
    type_filter: Optional[NotificationType] = Query(None),
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(get_current_user_dep),
):
    """List notification templates"""
    try:
        # This would require implementation in NotificationService
        # For now, return a placeholder response
        return []
        
    except Exception as e:
        logger.error("Failed to list templates", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.get("/templates/{template_id}", response_model=TemplateResponse)
async def get_notification_template(
    template_id: str,
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(get_current_user_dep),
):
    """Get a specific notification template"""
    try:
        # This would require implementation in NotificationService
        # For now, return a placeholder response
        raise HTTPException(
            status_code=501,
            detail="Get template not yet implemented"
        )
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error("Failed to get template", template_id=template_id, error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.put("/templates/{template_id}", response_model=TemplateResponse)
async def update_notification_template(
    template_id: str,
    template: TemplateCreate,
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(require_role(["admin", "manager"])),
):
    """Update a notification template (admin/manager only)"""
    try:
        # This would require implementation in NotificationService
        # For now, return a placeholder response
        raise HTTPException(
            status_code=501,
            detail="Template update not yet implemented"
        )
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error("Failed to update template", template_id=template_id, error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.delete("/templates/{template_id}")
async def delete_notification_template(
    template_id: str,
    tenant_id: str = Depends(get_current_tenant_id_dep),
    current_user: Dict[str, Any] = Depends(require_role(["admin"])),
):
    """Delete a notification template (admin only)"""
    try:
        # This would require implementation in NotificationService
        # For now, return a placeholder response
        return {"message": "Template deleted successfully", "template_id": template_id}
        
    except Exception as e:
        logger.error("Failed to delete template", template_id=template_id, error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

# ================================================================
# WEBHOOK ENDPOINTS
# ================================================================

@router.post("/webhooks/email-delivery")
async def email_delivery_webhook(webhook: DeliveryWebhook):
    """Handle email delivery status webhooks from external providers"""
    try:
        logger.info("Received email delivery webhook", 
                   notification_id=webhook.notification_id,
                   status=webhook.status.value)
        
        await handle_email_delivery_webhook(webhook.dict())
        
        return {"status": "received"}
        
    except Exception as e:
        logger.error("Failed to process email delivery webhook", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.post("/webhooks/whatsapp-delivery")
async def whatsapp_delivery_webhook(webhook_data: Dict[str, Any]):
    """Handle WhatsApp delivery status webhooks from Twilio"""
    try:
        logger.info("Received WhatsApp delivery webhook", 
                   message_sid=webhook_data.get("MessageSid"),
                   status=webhook_data.get("MessageStatus"))
        
        await handle_whatsapp_delivery_webhook(webhook_data)
        
        return {"status": "received"}
        
    except Exception as e:
        logger.error("Failed to process WhatsApp delivery webhook", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.post("/webhooks/read-receipt")
async def read_receipt_webhook(webhook: ReadReceiptWebhook):
    """Handle read receipt webhooks"""
    try:
        logger.info("Received read receipt webhook", 
                   notification_id=webhook.notification_id)
        
        # This would require implementation to update notification read status
        # For now, just log the event
        
        return {"status": "received"}
        
    except Exception as e:
        logger.error("Failed to process read receipt webhook", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

# ================================================================
# ADMIN ENDPOINTS
# ================================================================

@router.post("/admin/process-scheduled")
async def process_scheduled_notifications_endpoint(
    background_tasks: BackgroundTasks,
    current_user: Dict[str, Any] = Depends(require_role(["admin"])),
):
    """Manually trigger processing of scheduled notifications (admin only)"""
    try:
        background_tasks.add_task(process_scheduled_notifications)
        
        return {"message": "Scheduled notification processing started"}
        
    except Exception as e:
        logger.error("Failed to start scheduled notification processing", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.get("/admin/queue-status")
async def get_notification_queue_status(
    current_user: Dict[str, Any] = Depends(require_role(["admin", "manager"])),
):
    """Get notification queue status (admin/manager only)"""
    try:
        # This would require implementation to check queue status
        # For now, return a placeholder response
        return {
            "pending_notifications": 0,
            "scheduled_notifications": 0,
            "failed_notifications": 0,
            "retry_queue_size": 0
        }
        
    except Exception as e:
        logger.error("Failed to get queue status", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.post("/admin/retry-failed")
async def retry_failed_notifications(
    background_tasks: BackgroundTasks,
    max_retries: int = Query(3, ge=1, le=10),
    current_user: Dict[str, Any] = Depends(require_role(["admin"])),
):
    """Retry failed notifications (admin only)"""
    try:
        # This would require implementation to retry failed notifications
        # For now, return a placeholder response
        return {"message": f"Retry process started for failed notifications (max_retries: {max_retries})"}
        
    except Exception as e:
        logger.error("Failed to start retry process", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

# ================================================================
# TESTING ENDPOINTS (Development only)
# ================================================================

@router.post("/test/send-email")
async def test_send_email(
    to_email: str = Query(...),
    subject: str = Query("Test Email"),
    current_user: Dict[str, Any] = Depends(require_role(["admin"])),
):
    """Send test email (admin only, development use)"""
    try:
        from app.services.email_service import EmailService
        
        email_service = EmailService()
        
        success = await email_service.send_email(
            to_email=to_email,
            subject=subject,
            text_content="This is a test email from the notification service.",
            html_content="<h1>Test Email</h1><p>This is a test email from the notification service.</p>"
        )
        
        return {"success": success, "message": "Test email sent" if success else "Test email failed"}
        
    except Exception as e:
        logger.error("Failed to send test email", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))

@router.post("/test/send-whatsapp")
async def test_send_whatsapp(
    to_phone: str = Query(...),
    message: str = Query("Test WhatsApp message"),
    current_user: Dict[str, Any] = Depends(require_role(["admin"])),
):
    """Send test WhatsApp message (admin only, development use)"""
    try:
        from app.services.whatsapp_service import WhatsAppService
        
        whatsapp_service = WhatsAppService()
        
        success = await whatsapp_service.send_message(
            to_phone=to_phone,
            message=message
        )
        
        return {"success": success, "message": "Test WhatsApp sent" if success else "Test WhatsApp failed"}
        
    except Exception as e:
        logger.error("Failed to send test WhatsApp", error=str(e))
        raise HTTPException(status_code=500, detail=str(e))
    
@router.post("/users/{user_id}/notifications/cancel-pending")
async def cancel_pending_user_notifications(
    user_id: str,
    current_user = Depends(get_current_user_dep),
    _admin_check = Depends(require_role(["admin"])),
    db: AsyncSession = Depends(get_db)
):
    """Cancel all pending notifications for a user (admin only)"""
    try:
        user_uuid = uuid.UUID(user_id)
    except ValueError:
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST,
            detail="Invalid user ID format"
        )
    
    try:
        from app.models.notifications import NotificationQueue, NotificationLog
        
        # Find pending notifications
        pending_notifications_query = select(NotificationQueue).where(
            NotificationQueue.user_id == user_uuid,
            NotificationQueue.status.in_(["pending", "queued", "scheduled"])
        )
        pending_notifications_result = await db.execute(pending_notifications_query)
        pending_notifications = pending_notifications_result.scalars().all()
        
        notifications_cancelled = 0
        cancelled_notification_ids = []
        errors = []
        
        for notification in pending_notifications:
            try:
                notification.status = "cancelled"
                notification.updated_at = datetime.utcnow()
                notification.cancelled_by = current_user.get("user_id")
                notifications_cancelled += 1
                cancelled_notification_ids.append(str(notification.id))
                
                logger.info("Cancelled pending notification", 
                          notification_id=str(notification.id),
                          user_id=user_id)
                
            except Exception as e:
                error_msg = f"Failed to cancel notification {notification.id}: {str(e)}"
                errors.append(error_msg)
                logger.error(error_msg)
        
        if notifications_cancelled > 0:
            await db.commit()
        
        return {
            "success": True,
            "user_id": user_id,
            "notifications_cancelled": notifications_cancelled,
            "cancelled_notification_ids": cancelled_notification_ids,
            "errors": errors,
            "cancelled_at": datetime.utcnow().isoformat()
        }
        
    except Exception as e:
        await db.rollback()
        logger.error("Failed to cancel pending user notifications", 
                    user_id=user_id, 
                    error=str(e))
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail="Failed to cancel pending notifications"
        )

@router.delete("/users/{user_id}/notification-data")
async def delete_user_notification_data(
    user_id: str,
    current_user = Depends(get_current_user_dep),
    _admin_check = Depends(require_role(["admin"])),
    db: AsyncSession = Depends(get_db)
):
    """Delete all notification data for a user (admin only)"""
    try:
        user_uuid = uuid.UUID(user_id)
    except ValueError:
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST,
            detail="Invalid user ID format"
        )
    
    try:
        from app.models.notifications import (
            NotificationPreference, 
            NotificationQueue, 
            NotificationLog,
            DeliveryAttempt
        )
        
        deletion_stats = {
            "user_id": user_id,
            "deleted_at": datetime.utcnow().isoformat(),
            "preferences_deleted": 0,
            "notifications_deleted": 0,
            "logs_deleted": 0,
            "delivery_attempts_deleted": 0,
            "errors": []
        }
        
        # Delete delivery attempts first (they reference notifications)
        try:
            delivery_attempts_query = select(DeliveryAttempt).join(
                NotificationQueue, DeliveryAttempt.notification_id == NotificationQueue.id
            ).where(NotificationQueue.user_id == user_uuid)
            delivery_attempts_result = await db.execute(delivery_attempts_query)
            delivery_attempts = delivery_attempts_result.scalars().all()
            
            for attempt in delivery_attempts:
                await db.delete(attempt)
            
            deletion_stats["delivery_attempts_deleted"] = len(delivery_attempts)
            
        except Exception as e:
            error_msg = f"Error deleting delivery attempts: {str(e)}"
            deletion_stats["errors"].append(error_msg)
            logger.error(error_msg)
        
        # Delete notification queue entries
        try:
            notifications_delete_query = delete(NotificationQueue).where(
                NotificationQueue.user_id == user_uuid
            )
            notifications_delete_result = await db.execute(notifications_delete_query)
            deletion_stats["notifications_deleted"] = notifications_delete_result.rowcount
            
        except Exception as e:
            error_msg = f"Error deleting notifications: {str(e)}"
            deletion_stats["errors"].append(error_msg)
            logger.error(error_msg)
        
        # Delete notification logs
        try:
            logs_delete_query = delete(NotificationLog).where(
                NotificationLog.user_id == user_uuid
            )
            logs_delete_result = await db.execute(logs_delete_query)
            deletion_stats["logs_deleted"] = logs_delete_result.rowcount
            
        except Exception as e:
            error_msg = f"Error deleting notification logs: {str(e)}"
            deletion_stats["errors"].append(error_msg)
            logger.error(error_msg)
        
        # Delete notification preferences
        try:
            preferences_delete_query = delete(NotificationPreference).where(
                NotificationPreference.user_id == user_uuid
            )
            preferences_delete_result = await db.execute(preferences_delete_query)
            deletion_stats["preferences_deleted"] = preferences_delete_result.rowcount
            
        except Exception as e:
            error_msg = f"Error deleting notification preferences: {str(e)}"
            deletion_stats["errors"].append(error_msg)
            logger.error(error_msg)
        
        await db.commit()
        
        logger.info("Deleted user notification data", 
                   user_id=user_id,
                   preferences=deletion_stats["preferences_deleted"],
                   notifications=deletion_stats["notifications_deleted"],
                   logs=deletion_stats["logs_deleted"])
        
        deletion_stats["success"] = len(deletion_stats["errors"]) == 0
        
        return deletion_stats
        
    except Exception as e:
        await db.rollback()
        logger.error("Failed to delete user notification data", 
                    user_id=user_id, 
                    error=str(e))
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail="Failed to delete user notification data"
        )

@router.post("/notifications/user-deletion")
async def send_user_deletion_notification(
    notification_data: dict,  # {"admin_email": str, "deleted_user_email": str, "deletion_summary": dict}
    current_user = Depends(get_current_user_dep),
    _admin_check = Depends(require_role(["admin"])),
    db: AsyncSession = Depends(get_db)
):
    """Send notification about user deletion to admins (admin only)"""
    try:
        admin_email = notification_data.get("admin_email")
        deleted_user_email = notification_data.get("deleted_user_email")
        deletion_summary = notification_data.get("deletion_summary", {})
        
        if not admin_email or not deleted_user_email:
            raise HTTPException(
                status_code=status.HTTP_400_BAD_REQUEST,
                detail="admin_email and deleted_user_email are required"
            )
        
    except Exception as e:
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST,
            detail=f"Invalid request data: {str(e)}"
        )
    
    try:
        from app.models.notifications import NotificationQueue
        from app.services.notification_service import NotificationService
        
        # Create notification for the admin about the user deletion
        notification_content = {
            "subject": f"Admin User Deletion Completed - {deleted_user_email}",
            "message": f"""
Admin User Deletion Summary

Deleted User: {deleted_user_email}
Deletion Performed By: {admin_email}
Deletion Date: {datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S UTC')}

Summary:
- Tenants Affected: {deletion_summary.get('total_tenants_affected', 0)}
- Models Deleted: {deletion_summary.get('total_models_deleted', 0)}
- Forecasts Deleted: {deletion_summary.get('total_forecasts_deleted', 0)}
- Notifications Deleted: {deletion_summary.get('total_notifications_deleted', 0)}
- Tenants Transferred: {deletion_summary.get('tenants_transferred', 0)}
- Tenants Deleted: {deletion_summary.get('tenants_deleted', 0)}

Status: {'Success' if deletion_summary.get('deletion_successful', False) else 'Completed with errors'}
Total Errors: {deletion_summary.get('total_errors', 0)}

This action was performed through the admin user deletion system and all associated data has been permanently removed.
            """.strip(),
            "notification_type": "user_deletion_admin",
            "priority": "high"
        }
        
        # Create notification queue entry
        notification = NotificationQueue(
            user_email=admin_email,
            notification_type="user_deletion_admin",
            subject=notification_content["subject"],
            message=notification_content["message"],
            priority="high",
            status="pending",
            created_at=datetime.utcnow(),
            metadata={
                "deleted_user_email": deleted_user_email,
                "deletion_summary": deletion_summary,
                "performed_by": current_user.get("user_id")
            }
        )
        
        db.add(notification)
        await db.commit()
        
        # Trigger immediate sending (assuming NotificationService exists)
        try:
            notification_service = NotificationService(db)
            await notification_service.process_pending_notification(notification.id)
        except Exception as e:
            logger.warning("Failed to immediately send notification, will be processed by background worker", 
                         error=str(e))
        
        logger.info("Created user deletion notification", 
                   admin_email=admin_email,
                   deleted_user=deleted_user_email,
                   notification_id=str(notification.id))
        
        return {
            "success": True,
            "message": "User deletion notification created successfully",
            "notification_id": str(notification.id),
            "recipient": admin_email,
            "created_at": datetime.utcnow().isoformat()
        }
        
    except Exception as e:
        await db.rollback()
        logger.error("Failed to send user deletion notification", 
                    admin_email=admin_email,
                    deleted_user=deleted_user_email, 
                    error=str(e))
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail="Failed to send user deletion notification"
        )

@router.get("/users/{user_id}/notification-data/count")
async def get_user_notification_data_count(
    user_id: str,
    current_user = Depends(get_current_user_dep),
    _admin_check = Depends(require_role(["admin"])),
    db: AsyncSession = Depends(get_db)
):
    """Get count of notification data for a user (admin only)"""
    try:
        user_uuid = uuid.UUID(user_id)
    except ValueError:
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST,
            detail="Invalid user ID format"
        )
    
    try:
        from app.models.notifications import (
            NotificationPreference, 
            NotificationQueue, 
            NotificationLog
        )
        
        # Count preferences
        preferences_count_query = select(func.count(NotificationPreference.id)).where(
            NotificationPreference.user_id == user_uuid
        )
        preferences_count_result = await db.execute(preferences_count_query)
        preferences_count = preferences_count_result.scalar()
        
        # Count notifications
        notifications_count_query = select(func.count(NotificationQueue.id)).where(
            NotificationQueue.user_id == user_uuid
        )
        notifications_count_result = await db.execute(notifications_count_query)
        notifications_count = notifications_count_result.scalar()
        
        # Count logs
        logs_count_query = select(func.count(NotificationLog.id)).where(
            NotificationLog.user_id == user_uuid
        )
        logs_count_result = await db.execute(logs_count_query)
        logs_count = logs_count_result.scalar()
        
        return {
            "user_id": user_id,
            "preferences_count": preferences_count,
            "notifications_count": notifications_count,
            "logs_count": logs_count,
            "total_notification_data": preferences_count + notifications_count + logs_count
        }
        
    except Exception as e:
        logger.error("Failed to get user notification data count", 
                    user_id=user_id, 
                    error=str(e))
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail="Failed to get notification data count"
        )