#!/usr/bin/env python3
import sys
from pathlib import Path

# Add project root to path
ROOT_DIR = Path(__file__).resolve().parents[1]
sys.path.append(str(ROOT_DIR))

from installer import config as inst_config


def main():
    vault_path = (
        ROOT_DIR
        / "infrastructure"
        / "inventory"
        / "group_vars"
        / "all"
        / "vault_k3s.yml"
    )
    token = inst_config._try_read_ansible_vault_value(vault_path, "vault_k3s_token")

    if token:
        print(f"Syncing K3S_TOKEN to conf/prole.cfg...")
        inst_config._update_prole_cfg_value("Global", "PROLE_K3S_TOKEN", token)
        inst_config._update_prole_cfg_value("Inputs", "init_cluster.k3s_token", token)
        inst_config._update_prole_cfg_value("Service Cluster (k3s)", "K3S_TOKEN", token)
    else:
        print("Warning: Could not extract vault_k3s_token from Ansible vault.")


if __name__ == "__main__":
    main()