Skip to main content

Stack #1: ECR Build & Deploy

structure
auto-trading-bot
├── auto_trading_bot_ecr
│   ├── __init__.py
│   └── auto_trading_bot_ecr_stack.py

Create ECR to hold the docker image​

ECR creation
./auto_trading_bot_ecr/auto_trading_bot_ecr_stack.py
# Create ECR to hold docker image
ecr = aws_ecr.Repository(
    self, "ECR",
    repository_name=f"{props['namespace']}",
    removal_policy=RemovalPolicy.DESTROY
)

Create a CodeBuild project​

CodeBuild Project configuration
./auto_trading_bot_ecr/auto_trading_bot_ecr_stack.py
# CodeBuild Project configuration
docker_build_from_github = aws_codebuild.Project(self, "DockerBuild",
   project_name=props["docker_build_project_name"],
   build_spec=aws_codebuild.BuildSpec.from_source_filename('docker-build/docker_build_buildspec.yml'),
   environment=aws_codebuild.BuildEnvironment(
       privileged=True,
       build_image=aws_codebuild.LinuxBuildImage.AMAZON_LINUX_2_5,
   ),
   source=aws_codebuild.Source.git_hub(
       owner=props["GIT_HUB_OWNER"],
       repo=props["GIT_HUB_REPO"],
       branch_or_ref=props["GIT_HUB_BRANCH"],
       webhook=True
   ),
   environment_variables={
       "ACCOUNT_ID": aws_codebuild.BuildEnvironmentVariable(value=Aws.ACCOUNT_ID),
       "ECR_URI": aws_codebuild.BuildEnvironmentVariable(value=ecr.repository_uri),
       "IMAGE_NAME": aws_codebuild.BuildEnvironmentVariable(value=ecr.repository_name),
       "FUNCTION_NAME": aws_codebuild.BuildEnvironmentVariable(value=props["function_name"])
   },
   description="CodeBuild for Lambda run on container",
   timeout=Duration.minutes(15)
)

Grant Lambda permission for updating function​

lambda permission
./auto_trading_bot_ecr/auto_trading_bot_ecr_stack.py
# Grant CodeBuild perssion to update Lambda main function
lambda_function_update_policy = {
   "Sid": "LambdaFunctionUpdatePolicy",
   "Effect": "Allow",
   "Action": ["lambda:UpdateFunctionCode"],
   "Resource": f"arn:aws:lambda:{Aws.REGION}:{Aws.ACCOUNT_ID}:function:{props["function_name"]}"
 }
# Add policy to CodeBuild project
docker_build_from_github.add_to_role_policy(aws_iam.PolicyStatement.from_json(lambda_function_update_policy))

Grant ECR permission​

ECR permission
./auto_trading_bot_ecr/auto_trading_bot_ecr_stack.py
# CodeBuild permissions to interact with ECR
ecr.grant_pull_push(docker_build_from_github)

Create AWS S3 source for CodePipeline​

AWS S3 Source
./auto_trading_bot_ecr/auto_trading_bot_ecr_stack.py
# Pipeline requires versioned bucket
bot_bucket = aws_s3.Bucket(
    self, "SourceBucket",
    bucket_name=props['bucket_name'],
    versioned=True,
    removal_policy=RemovalPolicy.DESTROY)

aws_s3_deployment.BucketDeployment(self, "DeployFiles",
    sources=[aws_s3_deployment.Source.asset("./s3source")],
    destination_bucket=bot_bucket
)

Create CodePipeline​

CodePipeline
./auto_trading_bot_ecr/auto_trading_bot_ecr_stack.py
# Define CopePipeline action
bot_pipeline = aws_codepipeline.Pipeline(self, "Pipeline",
  pipeline_name=props['pipeline_name'],
  pipeline_type=aws_codepipeline.PipelineType.V2
)
source_output = aws_codepipeline.Artifact(artifact_name="source")
bot_pipeline.add_stage(
  stage_name="Source",
  actions=[
    aws_codepipeline_actions.S3SourceAction(
      bucket=bot_bucket,
      bucket_key='docker-build.zip',
      action_name='S3Source',
      run_order=1,
      output=source_output
    )
  ]
)
bot_pipeline.add_stage(
   stage_name="Build",
   actions=[
     aws_codepipeline_actions.CodeBuildAction(
       action_name='DockerBuildImages',
       input=source_output,
       project=docker_build_from_github,
       run_order=2
     )
   ]
)

Store the SSM parameters​

SSM Parameters

Store these two parameters in SSM for the security reason.

./auto_trading_bot_ecr/auto_trading_bot_ecr_stack.py
# SSM parameter to get webhook allowed IP list
webhook_params = aws_ssm.StringListParameter(self, "WebhookParameter",
  parameter_name=props["WEBHOOK_SSM_NAME"],
  string_list_value=props["WEBHOOK_IP_ALLOWED_LIST"],
  description="TradingView Webhook allowed IP list"
)
# SSM parameter to get message token and message ID for Dicord
message_params = aws_ssm.StringParameter(self, "MessageParameter",
  parameter_name=props["MESSAGE_NAME"],
  string_value=json.dumps(props["MESSAGE_CONFIG"]),
  description="Token and ID to send message"
)

Pass the required objects to Stack #2​

output value

Output required parameters to the second stack where ECR resource needed for Lambda main function running the docker image.

./auto_trading_bot_ecr/auto_trading_bot_ecr_stack.py
  self.output_props = props.copy()
  self.output_props['ECR']=ecr
# Pass objects to another stack
@property
def outputs(self):
    return self.output_props