from aws_cdk import (
    aws_lambda as _lambda,
    aws_s3 as s3,
    aws_cloudfront as cloudfront,
    aws_cloudfront_origins as origins,
    RemovalPolicy,
    Stack,
    App,
)
from constructs import Construct


# Define parent stack to deploy child stacks in different region
class ParentStack(Stack):
    def __init__(self, scope: Construct, id: str, **kwargs):
        super().__init__(scope, id, **kwargs)
        us_east_1_stack = USEast1Stack(
            self, "USEast1", env={"region": "us-east-1"}, cross_region_references=True
        )
        eu_central_1_stack = MyStack(
            self,
            "EUCentral",
            us_lambda=us_east_1_stack.lambda_function,
            env={"region": "eu-central-1"},
            cross_region_references=True,
        )


class USEast1Stack(Stack):
    def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
        super().__init__(scope, construct_id, **kwargs)

        # Define the Lambda function
        us_lambda_function = _lambda.Function(
            self,
            "LambdaFunction",
            code=_lambda.Code.from_asset("lambda"),
            handler="index.handler",
            runtime=_lambda.Runtime.PYTHON_3_11,
            current_version_options=_lambda.VersionOptions(
                removal_policy=RemovalPolicy.DESTROY,  # retain old versions
                retry_attempts=1,
            ),
        )

        # Assign function to local variable of object
        self.lambda_function = us_lambda_function


class MyStack(Stack):
    def __init__(self, scope: Construct, id: str, us_lambda: _lambda, **kwargs) -> None:
        super().__init__(scope, id, **kwargs)

        # Create an S3 bucket
        mybucket = s3.Bucket(
            self,
            "YQBucket",
            bucket_name="yq-cdk-bucket2",  # specify a unique bucket name
        )

        # Create edge lambda object and cross region reference lambda in US-East stack
        edge_lambda1 = cloudfront.EdgeLambda(
            event_type=cloudfront.LambdaEdgeEventType.ORIGIN_REQUEST,
            function_version=us_lambda.current_version,
            # the properties below are optional
            include_body=False,
        )

        # Create cloudfront distribution
        cloudfront.Distribution(
            self,
            "distro",
            default_behavior=cloudfront.BehaviorOptions(
                origin=origins.S3Origin(mybucket),
                edge_lambdas=[edge_lambda1],
            ),
        )


app = App()
ParentStack(app, "ParentStack")
app.synth()