123 lines
3.4 KiB
Python
123 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
try:
|
|
import yaml
|
|
except ImportError as exc: # pragma: no cover
|
|
raise SystemExit(
|
|
"Missing dependency: PyYAML. Install it with `python -m pip install pyyaml`."
|
|
) from exc
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(
|
|
description="Render docker compose .env from deploy config.yaml"
|
|
)
|
|
parser.add_argument(
|
|
"--input",
|
|
default="deploy/docker/config.yaml",
|
|
help="Path to config.yaml (default: deploy/docker/config.yaml)",
|
|
)
|
|
parser.add_argument(
|
|
"--output",
|
|
default="deploy/docker/.env",
|
|
help="Output dotenv file path (default: deploy/docker/.env)",
|
|
)
|
|
parser.add_argument(
|
|
"--section",
|
|
default="compose_env",
|
|
help="Top-level mapping section to export (default: compose_env)",
|
|
)
|
|
parser.add_argument(
|
|
"--stdout",
|
|
action="store_true",
|
|
help="Print rendered dotenv to stdout instead of writing file",
|
|
)
|
|
return parser.parse_args()
|
|
|
|
|
|
def load_config(path: Path) -> dict[str, Any]:
|
|
if not path.exists():
|
|
raise SystemExit(f"Config file not found: {path}")
|
|
|
|
data = yaml.safe_load(path.read_text(encoding="utf-8"))
|
|
if not isinstance(data, dict):
|
|
raise SystemExit("config.yaml root must be a mapping/object")
|
|
return data
|
|
|
|
|
|
def encode_env_value(value: Any) -> str:
|
|
if value is None:
|
|
return '""'
|
|
if isinstance(value, bool):
|
|
return "true" if value else "false"
|
|
if isinstance(value, (int, float)):
|
|
return str(value)
|
|
if not isinstance(value, str):
|
|
raise SystemExit(f"compose_env only supports scalar values, got: {type(value).__name__}")
|
|
|
|
if value == "":
|
|
return '""'
|
|
|
|
needs_quotes = any(ch in value for ch in [' ', '#', '"', "'", '\t', '\n', '\r']) or value.startswith('$')
|
|
if not needs_quotes:
|
|
return value
|
|
|
|
escaped = (
|
|
value.replace('\\', '\\\\')
|
|
.replace('"', '\\"')
|
|
.replace('\n', '\\n')
|
|
.replace('\r', '\\r')
|
|
.replace('\t', '\\t')
|
|
)
|
|
return f'"{escaped}"'
|
|
|
|
|
|
def render_env(section_name: str, values: dict[str, Any], source_path: Path) -> str:
|
|
lines = [
|
|
f"# Generated from {source_path.as_posix()}::{section_name}",
|
|
"# Do not edit this file directly; edit config.yaml and re-render.",
|
|
"",
|
|
]
|
|
|
|
for key, value in values.items():
|
|
if not isinstance(key, str) or not key:
|
|
raise SystemExit(f"Invalid env key: {key!r}")
|
|
lines.append(f"{key}={encode_env_value(value)}")
|
|
|
|
lines.append("")
|
|
return "\n".join(lines)
|
|
|
|
|
|
def main() -> int:
|
|
args = parse_args()
|
|
source_path = Path(args.input)
|
|
output_path = Path(args.output)
|
|
|
|
data = load_config(source_path)
|
|
section = data.get(args.section)
|
|
if not isinstance(section, dict):
|
|
raise SystemExit(
|
|
f"Section `{args.section}` must exist in config.yaml and must be a mapping/object"
|
|
)
|
|
|
|
rendered = render_env(args.section, section, source_path)
|
|
|
|
if args.stdout:
|
|
sys.stdout.write(rendered)
|
|
return 0
|
|
|
|
output_path.parent.mkdir(parents=True, exist_ok=True)
|
|
output_path.write_text(rendered, encoding="utf-8", newline="\n")
|
|
print(f"Wrote {output_path}")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|