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 boto3clients import dynamodb_client, sesv2_client
from config import (
    EMAIL_SENDER,
    ENROLLMENT_TABLE,
)

from .emails import TEMPLATES, send_email

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:
    old_image = event.detail['old_image']
    event_name = old_image['sk'].removeprefix('SCHEDULE#').lower()
    template = TEMPLATES[event_name]

    # If email is missing, use enrollment email
    if 'email' not in old_image:
        r = dyn.get_item(KeyPair(pk=old_image['id'], sk='0'))
        old_image['name'] = r['user']['name']
        old_image['email'] = r['user']['email']
        old_image['course'] = r['course']['name']

    return send_email(
        to=(old_image['name'], old_image['email']),
        subject=template['subject'],
        message=template['message'],
        context={
            'course': old_image['course'],
        },
        sender=EMAIL_SENDER,
        sesv2_client=sesv2_client,
        event={
            'id': old_image['id'],
            'sk': old_image['sk'],
        },
        dynamodb_persistence_layer=dyn,
    )