from decimal import Decimal
from http import HTTPStatus

from aws_lambda_powertools.event_handler.api_gateway import Router
from aws_lambda_powertools.event_handler.exceptions import (
    BadRequestError,
)
from layercake.dateutils import now
from layercake.dynamodb import (
    DynamoDBPersistenceLayer,
    KeyPair,
    PartitionKey,
)
from pydantic import BaseModel
from pydantic.types import UUID4

from api_gateway import JSONResponse
from boto3clients import dynamodb_client
from config import COURSE_TABLE

router = Router()
course_layer = DynamoDBPersistenceLayer(COURSE_TABLE, dynamodb_client)


@router.get('/<id>/custompricing', compress=True)
def get_custom_pricing(id: str):
    result = course_layer.collection.query(
        PartitionKey(f'CUSTOM_PRICING#ORG#{id}'),
        limit=100,
    )

    return result


class CustomPricing(BaseModel):
    course_id: UUID4
    unit_price: Decimal


@router.post('/<id>/custompricing', compress=True)
def post_custom_pricing(id: str, payload: CustomPricing):
    now_ = now()

    with course_layer.transact_writer() as transact:
        transact.put(
            item={
                'id': f'CUSTOM_PRICING#ORG#{id}',
                'sk': f'COURSE#{payload.course_id}',
                'unit_price': payload.unit_price,
                'created_at': now_,
            },
            cond_expr='attribute_not_exists(sk)',
            exc_cls=CoursConflictError,
        )
        transact.condition(
            key=KeyPair(str(payload.course_id), '0'),
            cond_expr='attribute_exists(sk)',
            exc_cls=CourseNotFoundError,
        )

    return JSONResponse(status_code=HTTPStatus.CREATED)


class Delete(BaseModel):
    course_id: UUID4


@router.delete('/<id>/custompricing', compress=True)
def delete_custom_pricing(id: str, payload: Delete):
    if course_layer.delete_item(
        KeyPair(
            f'CUSTOM_PRICING#ORG#{id}',
            f'COURSE#{payload.course_id}',
        )
    ):
        return JSONResponse(status_code=HTTPStatus.OK)


class CoursConflictError(BadRequestError): ...


class CourseNotFoundError(BadRequestError): ...