From fc0af2f9007c063f402a873f106388355daeb879 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Tue, 5 Nov 2024 21:03:33 +0000
Subject: [PATCH] Added single-target command-line config generator.  Mainly
 using for testing ATM.

---
 templates/generate | 140 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 140 insertions(+)
 create mode 100755 templates/generate

diff --git a/templates/generate b/templates/generate
new file mode 100755
index 00000000..e4925cec
--- /dev/null
+++ b/templates/generate
@@ -0,0 +1,140 @@
+#!/usr/bin/env python3
+
+import _jsonnet as j
+import json
+import yaml
+import logging
+import os
+import sys
+import zipfile
+
+logger = logging.getLogger("generate")
+logging.basicConfig(level=logging.INFO, format='%(message)s')
+
+private_json = "Put your GCP private.json here"
+
+class Generator:
+
+    def __init__(self, config, base="./templates/", version="0.0.0"):
+
+        self.jsonnet_base = base
+        self.config = config
+        self.version = f"\"{version}\"".encode("utf-8")
+
+    def process(self, config):
+
+        res = j.evaluate_snippet("config", config, import_callback=self.load)
+        return json.loads(res)
+
+    def load(self, dir, filename):
+
+        logger.debug("Request jsonnet: %s %s", dir, filename)
+
+        if filename == "config.json" and dir == "":
+            path = os.path.join(".", dir, filename)
+            return str(path), self.config
+
+        if filename == "version.jsonnet" and dir == "./templates/values/":
+            path = os.path.join(".", dir, filename)
+            return str(path), self.version
+
+        if dir:
+            candidates = [
+                os.path.join(".", dir, filename),
+                os.path.join(".", filename)
+            ]
+        else:
+            candidates = [
+                os.path.join(".", filename)
+            ]
+
+        try:
+
+            if filename == "vertexai/private.json":
+
+                return candidates[0], private_json.encode("utf-8")
+
+            for c in candidates:
+                logger.debug("Try: %s", c)
+
+                if os.path.isfile(c):
+                    with open(c, "rb") as f:
+                        logger.debug("Loading: %s", c)
+                        return str(c), f.read()
+
+            raise RuntimeError(
+                f"Could not load file={filename} dir={dir}"
+            )
+                
+        except:
+
+            path = os.path.join(self.jsonnet_base, filename)
+            logger.debug("Try: %s", path)
+            with open(path, "rb") as f:
+                logger.debug("Loaded: %s", path)
+                return str(path), f.read()
+
+def main():
+
+    if len(sys.argv) != 3:
+        print()
+        print("Usage:")
+        print("  generate <outfile> <version> < input.json")
+        print()
+        raise RuntimeError("Arg error")
+
+    outfile = sys.argv[1]
+    version = sys.argv[2]
+
+    cfg = sys.stdin.read()
+    cfg = json.loads(cfg)
+    
+    logger.info(f"Outputting to {outfile}...")
+
+    with zipfile.ZipFile(outfile, mode='w') as out:
+
+        def output(name, content):
+            logger.info(f"Adding {name}...")
+            out.writestr(name, content)
+
+        fname = "tg-launch.yaml"
+
+        platform = "docker-compose"
+
+        with open(f"./templates/config-to-{platform}.jsonnet", "r") as f:
+            wrapper = f.read()
+
+        gen = Generator(json.dumps(cfg).encode("utf-8"), version=version)
+
+        processed = gen.process(wrapper)
+
+        y = yaml.dump(processed)
+
+        output(fname, y)
+
+        # Placeholder for the private.json file.  Won't put actual credentials
+        # here.
+        output("docker-compose/vertexai/private.json", private_json)
+
+        # Grafana config
+        with open("grafana/dashboards/dashboard.json") as f:
+            output(
+                "docker-compose/grafana/dashboards/dashboard.json", f.read()
+            )
+
+        with open("grafana/provisioning/dashboard.yml") as f:
+            output(
+                "docker-compose/grafana/provisioning/dashboard.yml", f.read()
+            )
+
+        with open("grafana/provisioning/datasource.yml") as f:
+            output(
+                "docker-compose/grafana/provisioning/datasource.yml", f.read()
+            )
+
+        # Prometheus config
+        with open("prometheus/prometheus.yml") as f:
+            output("docker-compose/prometheus/prometheus.yml", f.read())
+
+main()
+