"""Payment Group Payment Report logic."""

from urllib.parse import urlparse

from payment.config import Config
from payment.connectors.s3 import create_presigned_url, get_s3_client
from payment.logic.exceptions import EntityDoesNotExist
from payment.models import PaymentGroupPayment, ReportPaymentGroupPayment


def create_or_update_report(
    payment_group_payment_id: int, report_export_url: str, report_type: str
) -> tuple[ReportPaymentGroupPayment, bool]:
    """Create or update a report_payment_group_payment record.

    Args:
        payment_group_payment_id (int): id of the parent payment_group_payment
        report_export_url (str): url of where the report is saved on S3
        report_type (str): one of 'approval' or 'summary'

    Returns:
        [ReportPaymentGroupPayment, bool]: the created or updated report instance and created flag
    """
    payment_group_payment = PaymentGroupPayment.get_by_id_or_error(
        payment_group_payment_id
    )

    existing_report = next(
        (
            report
            for report in payment_group_payment.reports
            if report.report_type == report_type
        ),
        None,
    )

    if existing_report:
        existing_report.update_attributes(report_export_url=report_export_url)
        ReportPaymentGroupPayment.commit_changes()
        return existing_report, False

    return ReportPaymentGroupPayment.create(
        payment_group_payment_id=payment_group_payment_id,
        report_export_url=report_export_url,
        report_type=report_type,
    ), True


def get_report(report_payment_group_payment_id: int) -> ReportPaymentGroupPayment:
    """Get a report_payment_group_payment by id.

    Args:
        report_payment_group_payment_id (int): id of the report_payment_group_payment

    Returns:
        ReportPaymentGroupPayment: the report instance
    """
    report = ReportPaymentGroupPayment.get_by_id(report_payment_group_payment_id)

    if not report:
        raise EntityDoesNotExist(
            ReportPaymentGroupPayment, report_payment_group_payment_id
        )

    return report


def delete_report(report_payment_group_payment_id: int) -> None:
    """Delete a report_payment_group_payment by id.

    Args:
        report_payment_group_payment_id (int): id of the report_payment_group_payment
    """

    report = ReportPaymentGroupPayment.get_by_id(report_payment_group_payment_id)

    if not report:
        raise EntityDoesNotExist(
            ReportPaymentGroupPayment, report_payment_group_payment_id
        )

    ReportPaymentGroupPayment.delete_by_id(report_payment_group_payment_id)


def get_by_payment_group_payment(
    payment_group_payment_id: int,
) -> list[ReportPaymentGroupPayment]:
    """Get list of reports by payment_group_payment.

    Args:
        payment_group_payment_id (int): id of the parent payment_group_payment

    Returns:
        list[ReportPaymentGroupPayment]: list of report instances
    """
    payment_group_payment = PaymentGroupPayment.get_by_id_or_error(
        payment_group_payment_id
    )
    return payment_group_payment.reports


def get_download(report_payment_group_payment_id: int) -> dict:
    """Get presigned download URL for a report_payment_group_payment.

    Args:
        report_payment_group_payment_id (int): id of the report_payment_group_payment

    Returns:
        dict: download info with presigned URL
    """
    report = ReportPaymentGroupPayment.get_by_id_or_error(
        report_payment_group_payment_id
    )
    bucket = Config.S3_PAYMENTS_BUCKET_NAME
    parsed_url = urlparse(report.report_export_url)
    key = parsed_url.path[1:]
    client = get_s3_client()
    presigned_url = create_presigned_url(bucket, client, key)
    return {'download_url': presigned_url}