import json
import boto3
from aws_lambda_powertools import Metrics, Logger, Tracer
import os

print('Loading function')
step_function_arn = os.environ["VIDEO_PROCESSING_STEP_FUNCTION_ARN"]

logger = Logger()
tracer = Tracer()
metrics = Metrics()

s3 = boto3.client('s3')
step_functions = boto3.client("stepfunctions")

@metrics.log_metrics(capture_cold_start_metric=True)
@logger.inject_lambda_context(log_event=True)
@tracer.capture_lambda_handler
def lambda_handler(event, context):
    logger.info("Received event: %s", event)
    print(f"Preparing to trigger: {step_function_arn}")
    responses = []
    for record in event["Records"]:
        logger.info(f"Processing Record: %s", record)
        # try:
        bucket = record["s3"]["bucket"]["name"]
        key = record["s3"]["object"]["key"]
        logger.info("Bucket: %s", bucket)
        logger.info("Key: %s", key)
        state = {
            "bucket": bucket,
            "key": key
        }
        response = step_functions.start_execution(
            stateMachineArn=step_function_arn,
            input=json.dumps(state),
            name=key
        )
        print(response)
        responses.append(response)

    return {
        "statusCode": 200,
        "body": json.dumps({
            "message": "State Machine Executions Started"
        }),
    }