import logging
from celery import shared_task

logger = logging.getLogger(__name__)


@shared_task(name="send_notification_task")
def send_notification_task(user_id: int, title: str, body: str, data: dict = None):
    """
    Celery background task to dispatch notifications.
    Can be connected to Firebase Cloud Messaging (FCM), APNS, or email.
    """
    try:
        from apps.account.models import User, Notification
        user = User.objects.get(id=user_id)
        
        # Save in-app notification record
        Notification.objects.create(
            user=user,
            title=title,
            message=body,
            notification_type=data.get('type') if data else 'general',
            action=data.get('action', 'navigate') if data else 'navigate',
            navigate_to=data.get('navigate_to', '') if data else '',
        )
        logger.info(f"Notification recorded for user #{user_id}: {title}")
        return True
    except Exception as e:
        logger.error(f"Error sending notification to user #{user_id}: {e}")
        return False


@shared_task(name="cleanup_expired_tokens_task")
def cleanup_expired_tokens_task():
    """
    Periodic task to clean up old or expired auth tokens/sessions.
    """
    logger.info("Executing cleanup_expired_tokens_task...")
    return True