import json

from aws_lambda_powertools.event_handler.api_gateway import Router
from elasticsearch import Elasticsearch
from layercake.dynamodb import (
    DynamoDBCollection,
    DynamoDBPersistenceLayer,
    KeyPair,
    SortKey,
    TransactKey,
)
from pydantic import UUID4, BaseModel

import elastic
from boto3clients import dynamodb_client
from middlewares.audit_log_middleware import AuditLogMiddleware
from middlewares.authentication_middleware import User
from rules.enrollment import set_status_as_canceled
from settings import ELASTIC_CONN, ENROLLMENT_TABLE, USER_TABLE

router = Router()
elastic_client = Elasticsearch(**ELASTIC_CONN)
enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
enrollment_collect = DynamoDBCollection(enrollment_layer)
user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
user_collect = DynamoDBCollection(user_layer)


@router.get('/', compress=True, tags=['Enrollment'])
def get_enrollments():
    event = router.current_event
    query = event.get_query_string_value('query', '{}')
    page_size = event.get_query_string_value('page_size', '25')

    return elastic.search(  # type: ignore
        index=ENROLLMENT_TABLE,
        page_size=int(page_size),
        query=json.loads(query),
        elastic_client=elastic_client,
    )


@router.get('/<id>', compress=True, tags=['Enrollment'])
def get_enrollment(id: str):
    return enrollment_collect.get_items(
        TransactKey(id)
        + SortKey('0')
        + SortKey('started_date')
        + SortKey('finished_date')
        + SortKey('failed_date')
        + SortKey('canceled_date')
        + SortKey('archived_date')
        + SortKey('cancel_policy')
        + SortKey('parent_vacancy', path_spec='vacancy')
        + SortKey('lock', path_spec='hash')
        + SortKey('author')
        + SortKey('tenant')
        + SortKey('cert')
    )


class Cancel(BaseModel):
    id: UUID4 | str
    lock_hash: str
    course: dict = {}
    vacancy: dict = {}


@router.patch(
    '/<id>/cancel',
    compress=True,
    tags=['Enrollment'],
    middlewares=[
        AuditLogMiddleware('ENROLLMENT_CANCEL', user_collect, ('id', 'course'))
    ],
)
def cancel(id: str, payload: Cancel):
    user: User = router.context['user']

    set_status_as_canceled(
        id,
        lock_hash=payload.lock_hash,
        author=user.model_dump(),  # type: ignore
        course=payload.course,  # type: ignore
        vacancy_key=KeyPair.parse_obj(payload.vacancy),
        persistence_layer=enrollment_layer,
    )

    return payload


@router.post('/', compress=True, tags=['Enrollment'])
def enroll():
    return {}