generate-config.py

import os
import sys
from os import mkdir
from os.path import exists

from mako.template import Template

regions = 'us-east-1,us-east-2,eu-west-1,us-west-1,us-west-2,ap-south-1,ap-southeast-1,ap-southeast-2,ca-central-1,eu-central-1,eu-north-1,eu-west-2,eu-west-3,sa-east-1,ap-northeast-1,ap-northeast-2'.split(
    ',')


def _mkdir(path):
    try:
        mkdir(path)
    except:
        pass


def save_template_file(path, data, directory=None):
    basename = path[(path.rindex('/') + 1) if '/' in path else 0:]
    with open(path if not directory else f'{directory}/{basename}', 'w') as f:
        src = path[:path.rindex('.') + 1] + 'template.' + path[path.rindex('.') + 1:]
        f.write(Template(filename=src).render(**data))


def go(env):
    project_name = os.environ.get('PROJECT_NAME', 'aws-bills-to-telegram')
    public_bucket = os.environ.get('PUBLIC_BUCKET', 'jeshan-oss-public-files')
    private_bucket = os.environ.get('PRIVATE_BUCKET', 'jeshan-oss-private-files')

    save_template_file('configure-aws-cli.py', {'PROJECT_NAME': project_name})
    save_template_file('upload-private-config.sh', {'PROJECT_NAME': project_name, 'PRIVATE_BUCKET': private_bucket})
    save_template_file('upload-public-templates.sh',
                       {'PROJECT_NAME': project_name, 'PRIVATE_BUCKET': private_bucket, 'PUBLIC_BUCKET': public_bucket})
    save_template_file('deployment-pipeline.yaml', {'PROJECT_NAME': project_name}, 'templates/')
    _mkdir('config/app/deployment')
    save_template_file('pipeline.yaml', {'PUBLIC_BUCKET': public_bucket, 'PRIVATE_BUCKET': private_bucket},
                       'config/app/deployment/')

    _mkdir('config')
    _mkdir('config/app/')
    _mkdir('config/app/deployment')
    _mkdir(f'config/app/{env}')
    if not exists(f'config/app/{env}/config.yaml'):
        with open(f'config/app/{env}/config.yaml', 'w') as f:
            f.write(f"""profile: {env}
""")

    with open('config/config.yaml', 'w') as f:
        f.write(f"""project_code: {project_name}
region: us-east-1

dlq_name: lambda-default-dlq
events_topic_name: cloudformation-stack-events
""")

    with open('config/app/deployment/pipeline.yaml', 'w') as f:
        f.write(f"""template_path: deployment-pipeline.yaml

parameters:
  ProjectName: {project_name}
  PrivateBucket: {private_bucket}
  PublicBucket: {public_bucket}
""")

    with open(f'config/app/{env}/base.yaml', 'w') as f:
        f.write(f"""template_path: deployment-target-account.yaml

parameters:
  DeploymentAccount: !environment_variable ACCOUNT_ID
  StackEventsTopic: {{{{stack_group_config.events_topic_name}}}}
  ProjectName: {project_name}
""")

    region = 'us-east-1'
    with open(f'config/app/{env}/{region}.yaml', 'w') as f:
        f.write(f"""template_path: template.yaml

region: {region}

parameters:
  IntervalHours: '24'
  DlqName: {{{{stack_group_config.dlq_name}}}}
  StackEventsTopic: {{{{stack_group_config.events_topic_name}}}}
  TelegramBotToken: !aws ssm::get_parameter::'Name':'bot-token','WithDecryption':True::Parameter.Value::us-east-1
  TelegramChatId: !aws ssm::get_parameter::'Name':'/bills-to-telegram/chat-id'::Parameter.Value::us-east-1
""")


if __name__ == '__main__':
    env_name = sys.argv[1] if len(sys.argv) > 1 else 'dev'
    go(env_name)