from aws_lambda_powertools import Logger
from aws_lambda_powertools.utilities.data_classes import (
    EventBridgeEvent,
    event_source,
)
from aws_lambda_powertools.utilities.typing import LambdaContext
from layercake.dateutils import now
from layercake.dynamodb import (
    DynamoDBPersistenceLayer,
    KeyPair,
)
from layercake.extra_types import CreditCard

from boto3clients import dynamodb_client
from config import IUGU_ACCOUNT_ID, IUGU_API_TOKEN, IUGU_TEST_MODE, ORDER_TABLE
from iugu import Credentials, Iugu

logger = Logger(__name__)
dyn = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
iugu = Iugu(
    Credentials(
        IUGU_ACCOUNT_ID,
        IUGU_API_TOKEN,
        test_mode=IUGU_TEST_MODE,
    )
)


@event_source(data_class=EventBridgeEvent)
@logger.inject_lambda_context
def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
    new_image = event.detail['new_image']
    order_id = new_image['id']
    invoice_id = new_image['invoice_id']
    installments = new_image['installments']
    credit_card = CreditCard(**new_image['credit_card'])
    now_ = now()

    token = iugu.payment_token(credit_card)
    charge = iugu.charge(
        invoice_id=invoice_id,
        token=token['id'],
        installments=installments,
    )

    with dyn.transact_writer() as transact:
        transact.delete(key=KeyPair(order_id, 'TRANSACTION'))
        transact.update(
            key=KeyPair(order_id, 'TRANSACTION#STATS'),
            update_expr='SET #count = if_not_exists(#count, :zero) + :one, \
                             updated_at = :now',
            expr_attr_names={
                '#count': 'payment_attempts',
            },
            expr_attr_values={
                ':zero': 0,
                ':one': 1,
                ':now': now(),
            },
        )

        if charge['success'] is True:
            transact.update(
                key=KeyPair(order_id, '0'),
                # Post-migration (orders): rename `payment_date` to `paid_at`
                update_expr='SET #status = :status, \
                                 payment_date = :now, \
                                 updated_at = :now',
                expr_attr_names={
                    '#status': 'status',
                },
                expr_attr_values={
                    ':status': 'PAID',
                    ':now': now_,
                },
                cond_expr='attribute_exists(sk)',
            )
            transact.put(
                item={
                    'id': order_id,
                    'sk': f'TRANSACTION#ATTEMPT#{now_.isoformat()}',
                    'brand': credit_card.brand,
                    'last4': credit_card.last4,
                    'status': 'SUCCEEDED',
                    'transaction': charge,
                },
            )
        else:
            transact.put(
                item={
                    'id': order_id,
                    'sk': f'TRANSACTION#ATTEMPT#{now_.isoformat()}',
                    'brand': credit_card.brand,
                    'last4': credit_card.last4,
                    'status': 'FAILED',
                    'transaction': charge,
                },
            )

    return charge['success']