From 74051a0b43dda2ff00d2e0908a95770fc241d504 Mon Sep 17 00:00:00 2001
From: Lee <liam@fascinated.cc>
Date: Sat, 13 Jan 2024 19:37:10 +0000
Subject: [PATCH] Add manage.py

---
 manage.py | 128 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 128 insertions(+)
 create mode 100644 manage.py

diff --git a/manage.py b/manage.py
new file mode 100644
index 0000000..5251518
--- /dev/null
+++ b/manage.py
@@ -0,0 +1,128 @@
+import sys
+import subprocess
+import yaml
+
+# Variables
+baseDir = "/home/traefik/"
+configFile = baseDir + "data/config.yml"
+
+# DO NOT TOUCH
+commands = ["add", "remove", "list"]
+command = len(sys.argv) > 1 and sys.argv[1]
+
+if command not in commands:
+  print("")
+  print("Usage: manage.py [command]")
+  print("")
+  print("Commands:")
+  print("  - add [name] [domain] [service host]")
+  print("  - remove [name]")
+  print("  - list")
+  exit()
+
+with open(configFile, "r") as config:
+  configYml = yaml.safe_load(config)
+
+http = configYml["http"]
+routers = http["routers"]
+services = http["services"]
+
+def restartTraefik():
+  print("Restarting Traefik...")
+  
+  # Restart Traefik in the base directory
+  subprocess.run(["docker", "compose", "restart", "traefik"], cwd=baseDir)
+
+def addDomain(name, domain, serviceHost):
+  # Check if name already exists
+  if name in routers:
+    print("Name \"%s\" already exists" % name)
+    exit()
+
+  print("Adding domain \"%s\" -> \"%s\"" % (domain, serviceHost))
+
+  # Add router
+  routers[name] = {
+    "entryPoints": ["https"],
+    "rule": "Host(`%s`)" % domain,
+    "middlewares": ["default-headers", "https-redirectscheme"],
+    "tls": {},
+    "service": name
+  }
+
+  # Add service
+  services[name] = {
+    "loadBalancer": {
+      "servers": [
+        {
+          "url": serviceHost
+        }
+      ]
+    }
+  }
+
+  # Write to file
+  with open(configFile, "w") as config:
+    yaml.dump(configYml, config)
+  
+  # Restart Traefik
+  restartTraefik()
+
+def removeDomain(name):
+  # Check if name exists
+  if name not in routers:
+    print("Name \"%s\" does not exist" % name)
+    exit()
+
+  print("Removing domain \"%s\"" % name)
+
+  # Remove router
+  del routers[name]
+
+  # Remove service
+  del services[name]
+
+  # Write to file
+  with open(configFile, "w") as config:
+    yaml.dump(configYml, config)
+  
+  # Restart Traefik
+  restartTraefik()
+
+def listDomains():
+  print("Listing domains:")
+
+  # domain -> service host
+  domains = {}
+
+  # Get domains (Host and HostRegexp)
+  for name, router in routers.items():
+    if "rule" in router:
+      rule = router["rule"]
+
+      # Host
+      if rule.startswith("Host"):
+        domain = rule.split("`")[1]
+        domains[domain] = services[name]["loadBalancer"]["servers"][0]["url"]
+
+      # HostRegexp
+      if rule.startswith("HostRegexp"):
+        domain = rule.split("`")[1]
+        domains[domain] = services[name]["loadBalancer"]["servers"][0]["url"]
+
+  # Print domains
+  for domain, serviceHost in domains.items():
+    print(" - %s -> %s" % (domain, serviceHost))
+
+match command:
+  case "add":
+    name = sys.argv[2]
+    domain = sys.argv[3]
+    serviceHost = sys.argv[4]
+    addDomain(name, domain, serviceHost)
+
+  case "remove":
+    name = sys.argv[2]
+    removeDomain(name)
+  case "list":
+    listDomains()
\ No newline at end of file