from enum import Enum
from http import HTTPStatus
from typing import Annotated

from aws_lambda_powertools.event_handler.api_gateway import Router
from aws_lambda_powertools.event_handler.openapi.params import Body
from layercake.dateutils import now
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair

from api_gateway import JSONResponse
from boto3clients import dynamodb_client
from config import USER_TABLE
from exceptions import OrgNotFoundError

router = Router()
dyn = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)


class PaymentMethod(str, Enum):
    PIX = 'PIX'
    BANK_SLIP = 'BANK_SLIP'
    MANUAL = 'MANUAL'


@router.get('/<org_id>/subscription')
def subscription(org_id: str):
    return dyn.collection.get_item(
        KeyPair(
            pk=org_id,
            sk='METADATA#SUBSCRIPTION',
        ),
        raise_on_error=False,
        default={},
    )


@router.post('/<org_id>/subscription')
def add(
    org_id: str,
    name: Annotated[str, Body(embed=True)],
    billing_day: Annotated[int, Body(embed=True, ge=1)],
    payment_method: Annotated[PaymentMethod, Body(embed=True)],
):
    now_ = now()

    with dyn.transact_writer() as transact:
        transact.condition(
            key=KeyPair(org_id, '0'),
            cond_expr='attribute_exists(sk)',
            exc_cls=OrgNotFoundError,
        )
        transact.put(
            item={
                'id': 'SUBSCRIPTION',
                'sk': f'ORG#{org_id}',
                'name': name,
                'created_at': now_,
            },
            cond_expr='attribute_not_exists(sk)',
        )
        transact.put(
            item={
                'id': org_id,
                'sk': 'METADATA#SUBSCRIPTION',
                'billing_day': billing_day,
                'payment_method': payment_method.value,
                'created_at': now_,
            }
        )

    return JSONResponse(status_code=HTTPStatus.CREATED)