Merge pull request #67 from Komalis/feat/add_auto_destroy

feat: add auto_destroy to clear stack automatically

Author: kbesrour-ma

easyecs/cloudformation/auto_destruction/harakiri.py

@@ -0,0 +1,14 @@
+import os
+
+import boto3
+
+
+def handler(event, context):
+    client = boto3.client("cloudformation")
+    stack_name = os.environ["StackName"]
+    response = client.delete_stack(
+        StackName=stack_name,
+    )
+    print(response)
+
+    return {"statusCode": response["ResponseMetadata"]["HTTPStatusCode"]}

easyecs/cloudformation/template/__init__.py

@@ -67,6 +67,12 @@ def create_template(
             targets=[service],
         )
 
+    if ecs_manifest.metadata.auto_destruction is not None:
+        assert isinstance(
+            ecs_manifest.metadata.auto_destruction, int
+        ), "auto_destruction must be an Integer (minutes)"
+        create_autodestroy(stack, ecs_manifest.metadata.auto_destruction)
+
     app.synth()
 
 
@@ -350,3 +356,39 @@ def create_ecs_service(
         enable_execute_command=True,
         min_healthy_percent=0,
     )
+
+
+def create_autodestroy(stack, deployment_timeout: int):
+    from pathlib import Path
+    from aws_cdk.aws_events import Rule, Schedule
+    from aws_cdk.aws_events_targets import LambdaFunction
+    from aws_cdk.aws_iam import PolicyStatement
+    from aws_cdk import aws_lambda, Duration
+    from aws_cdk.aws_logs import RetentionDays
+
+    lambdaDeleteStackPolicy = PolicyStatement(
+        actions=["cloudformation:DeleteStack", "lambda:RemovePermission"],
+        resources=["*"],
+    )
+
+    lambda_function_file = str(
+        Path(__file__).parent.parent / "auto_destruction/harakiri.py"
+    )
+    harakiri = aws_lambda.Function(
+        stack,
+        "AutoDestroy",
+        code=aws_lambda.Code.from_inline(open(lambda_function_file).read()),
+        handler="index.handler",
+        timeout=Duration.seconds(300),
+        log_retention=RetentionDays.ONE_MONTH,
+        environment={"StackName": stack.artifact_id},
+        runtime=aws_lambda.Runtime.PYTHON_3_11,
+    )
+    harakiri.add_to_role_policy(lambdaDeleteStackPolicy)
+
+    lambda_rule = Rule(
+        stack,
+        "TimeToDestroy",
+        schedule=Schedule.rate(Duration.minutes(deployment_timeout)),
+    )
+    lambda_rule.add_target(LambdaFunction(handler=harakiri))

easyecs/model/ecs.py

@@ -10,6 +10,7 @@
 class EcsFileMetadataModel(BaseModel):
     appname: str
     user: str = os.environ["USER"]
+    auto_destruction: Optional[int] = None
 
 
 class EcsFileStatementModel(BaseModel):

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "easyecs"
-version = "0.21.0"
+version = "0.22.0"
 description = ""
 authors = ["BONVARLET Benjamin <benjaminbonvarlet96@gmail.com>"]
 readme = "README.md"

tests/cloudformation/template/test_depends_on.py

@@ -79,6 +79,7 @@ def test_depends_on_container_has_depends_on(depends_on, mocker):
     container = MagicMock()
     container.depends_on = depends_on
     ecs_data.task_definition.containers = [container]
+    ecs_data.metadata.auto_destruction = None
 
     create_template(
         "test_service",

tests/cloudformation/template/test_efs.py

@@ -36,6 +36,7 @@ def test_efs_add_mount_points_if_container_has_efs_volumes(efs_volumes, count, m
     ecs_data.task_definition.efs_volumes = [MagicMock(), MagicMock()]
     container = MagicMock()
     container.efs_volumes = efs_volumes
+    ecs_data.metadata.auto_destruction = None
     ecs_data.task_definition.containers = [container]
 
     create_template(

tests/cloudformation/template/test_template_command.py

@@ -58,6 +58,7 @@ def test_command_is_always_sleep_when_tty(mocker):
 
     container = _create_mock_container()
     ecs_data = _create_mock_ecs_data(container)
+    ecs_data.metadata.auto_destruction = None
 
     create_template(
         "test_service",
@@ -95,6 +96,7 @@ def test_command_is_always_sleep_when_run(mocker):
     container = _create_mock_container()
     container.tty = True
     ecs_data = _create_mock_ecs_data(container)
+    ecs_data.metadata.auto_destruction = None
 
     create_template(
         "test_service",