app.py

#!/usr/bin/env python3

"""
CDK Application for managing containers in AWS
"""

import os

from aws_cdk import (
    # Aspects,
    App,
    Environment,
    Tags,
)
# import cdk_nag

from ContainerManager.base_stack import BaseStackMain, BaseStackDomain
from ContainerManager.leaf_stack.main import ContainerManagerStack
from ContainerManager.leaf_stack.start_system import LeafStackStartSystem
from ContainerManager.utils import load_base_config, load_leaf_config


# https://docs.aws.amazon.com/cdk/api/v2/docs/aws-cdk-lib.App.html
app = App()
application_id = app.node.get_context("_application_id")
APPLICATION_ID_TAG_NAME = "ApplicationId"
### TODO: Finish going through all the cdk_nag checks:
# Aspects.of(app).add(cdk_nag.AwsSolutionsChecks(verbose=True))
Tags.of(app).add(APPLICATION_ID_TAG_NAME, application_id)

### Fact-check the maturity, and save it for leaf stacks:
# (Makefile defaults to prod if not set. We want to fail-fast
# here, so throw if it doesn't exist)
maturity = app.node.get_context("maturity")
supported_maturities = ["devel", "prod"]
assert maturity in supported_maturities, f"ERROR: Unknown maturity. Must be in {supported_maturities}"


# Lets you reference self.account and self.region in your CDK code
# if you need to:
main_env = Environment(
    account=os.getenv('CDK_DEFAULT_ACCOUNT'),
    region=os.getenv('CDK_DEFAULT_REGION'),
)
us_east_1_env = Environment(
    account=main_env.account,
    region="us-east-1",
)

##################
### Base Stack ###
##################
base_config = load_base_config("./base-stack-config.yaml")
### Create the Base Stack VPC for ALL leaf stacks:
base_stack_main = BaseStackMain(
    app,
    f"{app.node.get_context('_base_stack_prefix')}-Vpc",
    description="The base VPC for all other ContainerManage stacks to use.",
    # cross_region_references=True,
    env=main_env,
    config=base_config,
    application_id_tag_name=APPLICATION_ID_TAG_NAME,
    application_id_tag_value=application_id,
)
### Create the Base Stack Domain for ALL leaf stacks:
base_stack_domain = BaseStackDomain(
    app,
    f"{app.node.get_context('_base_stack_prefix')}-Domain",
    description="The base HostedZone for all other ContainerManage stacks to use.",
    # cross_region_references=True,
    env=us_east_1_env,
    config=base_config,
    main_stack_region=main_env.region,
)

##################
### Leaf Stack ###
##################
### Create the application for ONE Container:
file_path = app.node.try_get_context("config-file")
if file_path:
    leaf_config = load_leaf_config(file_path, maturity=maturity)
    # You can override container_id if you need to:
    container_id = app.node.try_get_context("container-id")
    if not container_id:
        container_id = os.path.basename(os.path.splitext(file_path)[0])
    container_id = container_id.lower()
    # For stack names, turn "minecraft.java.example" into "MinecraftJavaExample":
    container_id_alpha = "".join(e for e in container_id.title() if e.isalpha()) # pylint: disable=invalid-name

    stack_tags = {
        "ContainerId": container_id,
        "StackId": f"{application_id}-{container_id_alpha}",
    }

    leaf_stack_manager = ContainerManagerStack(
        app,
        f"{application_id}-{container_id_alpha}-Stack",
        description="For automatically managing and spinning down the container.",
        # cross_region_references=True,
        env=main_env,
        base_stack_main=base_stack_main,
        base_stack_domain=base_stack_domain,
        application_id=application_id,
        container_id=container_id,
        config=leaf_config,
    )
    for key, val in stack_tags.items():
        Tags.of(leaf_stack_manager).add(key, val)

    leaf_stack_start_system = LeafStackStartSystem(
        app,
        f"{application_id}-{container_id_alpha}-StartSystemStack",
        description="Everything for spinning up the container when someone connects.",
        # cross_region_references=True,
        env=us_east_1_env,
        base_stack_domain=base_stack_domain,
        leaf_stack_manager=leaf_stack_manager,
        container_id=container_id,
    )
    for key, val in stack_tags.items():
        Tags.of(leaf_stack_start_system).add(key, val)

app.synth()