78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
#!/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()
|