#!/usr/bin/env python

from jinja2 import Environment, FileSystemLoader
from os import getenv


def main():
    # This value comes as a build env var: `SSM_CONFIG_PARAM_NAME`
    _env = Environment(loader=FileSystemLoader('/'), autoescape=True)
    _template = _env.get_template("/config-as-code.j2")
    _config_file = open("/config-as-code.yaml", "w")

    _config_file.write((_template.render(
        ECS_CLUSTER_ARN=getenv('cluster_arn'),
        AWS_REGION=getenv('aws_region'),
        JENKINS_URL=getenv('jenkins_url'),
        SUBNET_IDS=getenv('subnet_ids'),
        SECURITY_GROUP_IDS=getenv('security_group_ids'),
        EXECUTION_ROLE_ARN=getenv('execution_role_arn'),
        TASK_ROLE_ARN=getenv('task_role_arn'),
        LOG_GROUP=getenv('worker_log_group'),
        LOG_STREAM_PREFIX=getenv('worker_log_stream_prefix')
    )))

    _config_file.close()


if __name__ == '__main__':
    main()