PLATFORMINFRA-193: rework byoh-agent flags

agent/.goreleaser.yaml

@@ -32,6 +32,9 @@ nfpms:
       - src: debian/systemd
         dst: /etc/systemd/system/
         type: tree
+      - src: debian/scripts
+        dst: /opt/byoh-agent/
+        type: tree
 
 changelog:
   sort: asc

agent/debian/scripts/byoh-agent-flags.py

@@ -0,0 +1,44 @@
+#!/bin/python3
+import json
+import os
+
+METADATA_FILE_PATH = '/etc/infra-k8s/host-metadata.json'
+FLAGS_FILE_DIR = '/etc/byoh-agent'
+FLAGS_FILE_PATH = f'{FLAGS_FILE_DIR}/byoh-agent-flags.env'
+
+def load_metadata():
+    with open(METADATA_FILE_PATH) as f:
+        return json.load(f)
+
+def create_config_dir():
+    os.makedirs(FLAGS_FILE_DIR, exist_ok=True)
+
+def construct_labels(metadata):
+    common_labels = [
+        "--label", f"byoh.infrastructure.cluster.x-k8s.io/type={metadata.get('node_role')}",
+        "--label", f"byoh.infrastructure.cluster.x-k8s.io/cluster={metadata.get('cluster_name')}",
+        "--label", f"byoh.infrastructure.cluster.x-k8s.io/hostname={os.getenv('HOSTNAME')}",
+    ]
+
+    extra_labels = []
+    for k, v in metadata.get('extra_labels', {}).items():
+        extra_labels += ["--label", f"{k}={v}"]
+
+    return common_labels + extra_labels
+
+def construct_args(metadata):
+    return construct_labels(metadata) + [
+        "--bootstrap-kubeconfig", "/etc/infra-k8s/kubeconfig",
+        "--skip-installation",
+        "--namespace", metadata.get('cluster_name'),
+        "--metricsbindaddress", ":8081",
+    ]
+
+def write_env_file(metadata):
+    # Writing structured form of flags as string into '/etc/byoh-agent/byoh-agent-flags.env'
+    with open(FLAGS_FILE_PATH, 'w') as f:
+        f.write(f"BYOH_AGENT_FLAGS=\"{' '.join(construct_args(metadata))}\"")
+
+metadata = load_metadata()
+create_config_dir()
+write_env_file(metadata)

agent/debian/systemd/byoh-agent-flag.path

@@ -0,0 +1,5 @@
+[Path]
+Unit=byoh-agent-flag.service
+PathModified=/etc/infra-k8s/host-metadata.json
+[Install]
+WantedBy=multi-user.target

agent/debian/systemd/byoh-agent-flag.service

@@ -0,0 +1,16 @@
+[Unit]
+Description=Prepares env with flags for byoh-agent service
+After=network.target
+StartLimitIntervalSec=60
+StartLimitBurst=5
+PartOf=byoh-agent.service
+
+[Service]
+Type=oneshot
+RemainAfterExit=yes
+ExecStart=/opt/byoh-agent/byoh-agent-flags.py
+Restart=on-failure
+RestartSec=5
+
+[Install]
+WantedBy=byoh-agent.service

agent/debian/systemd/byoh-agent-watcher.path

@@ -1,5 +1,6 @@
 [Path]
 Unit=byoh-agent-watcher.service
 PathModified=/root/.byoh/config
+PathModified=/etc/byoh-agent/byoh-agent-flags.env
 [Install]
 WantedBy=multi-user.target