app.py

#!/usr/bin/env python3

import os
import base64
import configparser
from typing import Optional

from aws_cdk import core
from once.once_stack import OnceStack, CustomDomainStack


ONCE_CONFIG_FILE = os.getenv('ONCE_CONFIG_FILE', os.path.expanduser('~/.once'))

SECRET_KEY = os.getenv('SECRET_KEY')
CUSTOM_DOMAIN = os.getenv('CUSTOM_DOMAIN')
HOSTED_ZONE_NAME = os.getenv('HOSTED_ZONE_NAME')
HOSTED_ZONE_ID = os.getenv('HOSTED_ZONE_ID')


def generate_random_key() -> str:
    return base64.b64encode(os.urandom(128)).decode('utf-8')


def generate_config(secret_key: Optional[str] = None,
    custom_domain: str = None,
    hosted_zone_name: str = None,
    hosted_zone_id: str = None) -> configparser.ConfigParser:
    config = configparser.ConfigParser()

    config['once'] = {
        'secret_key': secret_key or generate_random_key(),
    }

    config['deployment'] = {}
    if all([custom_domain, hosted_zone_name, hosted_zone_id]):
        config['once']['base_url'] = f'https://{custom_domain}'
        config['deployment'] = {
            'custom_domain': custom_domain,
            'hosted_zone_name': hosted_zone_name,
            'hosted_zone_id': hosted_zone_id
        }
    return config


def get_config(config_gile: str = ONCE_CONFIG_FILE) -> configparser.ConfigParser:
    if not os.path.exists(ONCE_CONFIG_FILE):
        print(f'Generating configuration file at {ONCE_CONFIG_FILE}')
        with open(ONCE_CONFIG_FILE, 'w') as config_file:
            config = generate_config(
                secret_key=SECRET_KEY,
                custom_domain=CUSTOM_DOMAIN,
                hosted_zone_name=HOSTED_ZONE_NAME,
                hosted_zone_id=HOSTED_ZONE_ID)
            config.write(config_file)
    else:
        config = configparser.ConfigParser()
        config.read(ONCE_CONFIG_FILE)
    return config


def main():
    config = get_config()

    kwargs = {'secret_key': config['once']['secret_key']}
    if config.has_section('deployment'):
        kwargs.update(config['deployment'])

    app = core.App()
    once = OnceStack(app, 'once', **kwargs)
    app.synth()


if __name__ == '__main__':
    main()