41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
def natural_key(name: str):
|
||
|
|
m = re.match(r"^([a-zA-Z-]+)-(\d+)$", name)
|
||
|
|
if m:
|
||
|
|
return (m.group(1), int(m.group(2)))
|
||
|
|
return (name, 0)
|
||
|
|
|
||
|
|
|
||
|
|
def map_to_pairs(items: dict[str, str]) -> str:
|
||
|
|
ordered = sorted(items.items(), key=lambda kv: natural_key(kv[0]))
|
||
|
|
return " ".join(f"{k}={v}" for k, v in ordered)
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> int:
|
||
|
|
payload = json.load(sys.stdin)
|
||
|
|
|
||
|
|
cp_map = payload.get("control_plane_vm_ipv4", {}).get("value", {})
|
||
|
|
wk_map = payload.get("worker_vm_ipv4", {}).get("value", {})
|
||
|
|
|
||
|
|
if not cp_map or not wk_map:
|
||
|
|
raise SystemExit("Missing control_plane_vm_ipv4 or worker_vm_ipv4 in terraform output")
|
||
|
|
|
||
|
|
ssh_user = os.environ.get("KUBEADM_SSH_USER", "").strip() or "micqdf"
|
||
|
|
|
||
|
|
print(f"SSH_USER={ssh_user}")
|
||
|
|
print("PRIMARY_CONTROL_PLANE=cp-1")
|
||
|
|
print(f"CONTROL_PLANES=\"{map_to_pairs(cp_map)}\"")
|
||
|
|
print(f"WORKERS=\"{map_to_pairs(wk_map)}\"")
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
raise SystemExit(main())
|