import logging

from django.db import transaction
from django.utils import timezone
from procrastinate.contrib.django import app
from api.notifications import send_sms
from billing.models import Topup
from django.utils.timezone import localtime

logger = logging.getLogger(__name__)


@app.periodic(
    cron="*/1 * * * * *", periodic_id="notify_expired_topups", queue="heavy_tasks"
)  # every 1 minute
@app.task
def update_expired_topups(timestamp: int):
    expired_topups_qs = Topup.objects.filter(
        expires_at__lte=timezone.now(),
        expiry_notification_sent=False,
        paid=False,
    ).select_related("user")
    if not expired_topups_qs.exists():
        logger.info("No expired topups found.")
        return {"total_expired_topups": 0}

    with transaction.atomic():
        count = expired_topups_qs.count()
        logger.info(f"Found {count} topups to expire.")

        for topup in expired_topups_qs:
            if topup.user and topup.user.mobile and not topup.expiry_notification_sent:
                send_sms_task.defer(
                    mobile=topup.user.mobile,
                    amount=topup.amount,
                    topup_id=str(topup.id),
                    created_at=localtime(topup.created_at).isoformat(),
                    user=f"{topup.user.first_name + ' ' + topup.user.last_name}"
                    if topup.user.last_name and topup.user.first_name
                    else "User",
                )
                topup.expiry_notification_sent = True
                topup.save()
            else:
                # Mark as notified even if we can't send SMS (no mobile number)
                topup.expiry_notification_sent = True
                topup.save()
                return

    return {
        "total_expired_topups": count,
    }


@app.task
def send_sms_task(
    user: str, mobile: str, amount: float, topup_id: str, created_at: str
):
    message = (
        f"Dear {user}, \n\nYour topup of {amount} MVR [created at {created_at}] has expired. "
        "Please make a new topup to update your wallet. \n\n- SAR Link"
    )
    send_sms(mobile, message)
    logger.info(f"SMS sent to {mobile} for expired topup of {amount} MVR.")

    # Mark the topup as notified after successful SMS sending
    try:
        topup = Topup.objects.get(id=topup_id)
        topup.expiry_notification_sent = True
        topup.save()
        logger.info(f"Marked topup {topup_id} as notified.")
    except Topup.DoesNotExist:
        logger.error(f"Topup {topup_id} not found when trying to mark as notified.")