from http import HTTPStatus
from os import rename
from typing import Annotated

from aws_lambda_powertools import Logger
from aws_lambda_powertools.event_handler.api_gateway import Router
from aws_lambda_powertools.event_handler.exceptions import NotFoundError
from aws_lambda_powertools.event_handler.openapi.params import Body
from glom import T, glom
from layercake.dateutils import now
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, SortKey, TransactKey

from api_gateway import JSONResponse
from boto3clients import dynamodb_client
from config import ENROLLMENT_TABLE

from .cancel import router as cancel
from .dedup_window import router as dedup_window
from .download_cert import router as download_cert
from .enroll import router as enroll

__all__ = ['cancel', 'dedup_window', 'download_cert', 'enroll']

logger = Logger(__name__)
router = Router()
dyn = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)


@router.get('/<enrollment_id>')
def get_enrollment(enrollment_id: str):
    return dyn.collection.get_items(
        TransactKey(enrollment_id)
        + SortKey('0')
        + SortKey('ORG')
        + SortKey('LOCK')
        + SortKey('SCORM_COMMIT#LAST', rename_key='last_commit'),
    )


@router.get('/<enrollment_id>/scorm')
def get_scorm(enrollment_id: str):
    enrollment = dyn.collection.get_items(
        TransactKey(enrollment_id) + SortKey('0') + SortKey('SCORM_COMMIT#LAST')
    )

    if not enrollment:
        raise NotFoundError('Enrollment not found')

    spec = {'id': 'course.id', 'scormset': 'course.scormset'}
    course_id, scormset_id = glom(enrollment, spec).values()
    scormset = dyn.collection.get_item(
        KeyPair(course_id, f'SCORMSET#{scormset_id}'),
        raise_on_error=False,
        default={},
    )

    return enrollment | {'scormset': scormset}


@router.post('/<enrollment_id>')
def post_enrollment(
    enrollment_id: str,
    cmi: Annotated[dict, Body(embed=True)],
):
    now_ = now()

    with dyn.transact_writer() as transact:
        transact.condition(
            key=KeyPair(enrollment_id, '0'),
            cond_expr='attribute_exists(sk)',
            exc_cls=NotFoundError,
        )
        transact.put(
            item={
                'id': enrollment_id,
                'sk': 'SCORM_COMMIT#LAST',
                'cmi': cmi,
                'created_at': now_,
            }
        )
        transact.put(
            item={
                'id': f'SCORM_COMMIT#{enrollment_id}',
                'sk': now_.isoformat(),
                'cmi': cmi,
                'created_at': now_,
            }
        )

    return JSONResponse(HTTPStatus.NO_CONTENT)