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.dynamodb import DynamoDBPersistenceLayer, KeyPair
from layercake.strutils import md5_hash

from boto3clients import dynamodb_client
from config import ENROLLMENT_TABLE

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


@event_source(data_class=EventBridgeEvent)
@logger.inject_lambda_context
def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
    new_image = event.detail['new_image']
    enrollment_id = new_image['id']
    user_id = new_image['user']['id']
    course_id = new_image['course']['id']
    lock_hash = md5_hash(f'{user_id}{course_id}')

    with dyn.transact_writer() as transact:
        transact.delete(
            key=KeyPair(enrollment_id, 'SCHEDULE#REMINDER_NO_ACCESS_AFTER_3_DAYS')
        )
        transact.delete(
            key=KeyPair(enrollment_id, 'SCHEDULE#REMINDER_NO_ACTIVITY_AFTER_7_DAYS')
        )
        transact.delete(
            key=KeyPair(enrollment_id, 'SCHEDULE#REMINDER_ACCESS_PERIOD_BEFORE_30_DAYS')
        )
        transact.delete(key=KeyPair(enrollment_id, 'CANCEL_POLICY'))
        # Remove locks related to this enrollment
        transact.delete(key=KeyPair(enrollment_id, 'LOCK'))
        transact.delete(key=KeyPair('LOCK', lock_hash))

    return True