Refactor visualize.py to use jinja2
This commit is contained in:
parent
73a1b21cad
commit
20bb2c8fd7
3 changed files with 114 additions and 73 deletions
109
visualize.py
109
visualize.py
|
@ -3,6 +3,8 @@
|
|||
import json
|
||||
import sys
|
||||
|
||||
import jinja2
|
||||
|
||||
icon_overrides = {
|
||||
"acme_dns": "lets-encrypt",
|
||||
"extra_services": None,
|
||||
|
@ -24,41 +26,6 @@ icon_format = {
|
|||
"tiny-tiny-rss": "webp",
|
||||
}
|
||||
|
||||
template_host = """
|
||||
'serguzim.net'.{host}: {{
|
||||
label: {host}
|
||||
}}
|
||||
"""
|
||||
|
||||
template_service = """
|
||||
{key}: {{
|
||||
label: {label}
|
||||
label.near: top-left
|
||||
icon: {icon}
|
||||
icon.near: top-right
|
||||
}}
|
||||
"""
|
||||
|
||||
default_d2 = """
|
||||
vars: {
|
||||
d2-config: {
|
||||
layout-engine: elk
|
||||
# Terminal theme code
|
||||
theme-id: 101
|
||||
}
|
||||
}
|
||||
|
||||
external: {
|
||||
scaleway: {
|
||||
s3
|
||||
}
|
||||
|
||||
restic: {
|
||||
icon: https://cdn.jsdelivr.net/gh/selfhst/icons/webp/restic.webp
|
||||
}
|
||||
}
|
||||
"""
|
||||
|
||||
def get_icon(svc):
|
||||
svc = icon_overrides.get(svc, svc) or 'docker'
|
||||
fmt = icon_format.get(svc, 'svg')
|
||||
|
@ -66,6 +33,9 @@ def get_icon(svc):
|
|||
|
||||
def service_key(svc, data):
|
||||
return f"'serguzim.net'.{data['host']}.{svc}"
|
||||
def host_key(host):
|
||||
return f"'serguzim.net'.{host}"
|
||||
|
||||
|
||||
def service_key_find(svc_name, services):
|
||||
for svc, data in services.items():
|
||||
|
@ -75,8 +45,11 @@ def service_key_find(svc_name, services):
|
|||
|
||||
def parse_hosts(hosts):
|
||||
result = []
|
||||
for host in hosts.keys():
|
||||
result.append(template_host.format(host=host))
|
||||
for host, data in hosts.items():
|
||||
result.append({
|
||||
'key': host_key(host),
|
||||
'name': host,
|
||||
})
|
||||
return result
|
||||
|
||||
def parse_services(services):
|
||||
|
@ -85,42 +58,23 @@ def parse_services(services):
|
|||
postgresql_key = service_key_find("postgresql", services)
|
||||
authentik_key = service_key_find("authentik", services)
|
||||
|
||||
result.append(f"{postgresql_key}.grid-columns: 3")
|
||||
result.append(f"{postgresql_key}.grid-gap: 0")
|
||||
result.append(f"{authentik_key}.grid-columns: 3")
|
||||
result.append(f"{authentik_key}.grid-gap: 0")
|
||||
|
||||
for svc, data in services.items():
|
||||
svc_key = service_key(svc, data)
|
||||
|
||||
domains = []
|
||||
if data.get("dns"):
|
||||
domains = []
|
||||
for dns in data["dns"]:
|
||||
domain = ""
|
||||
if dns.get("target") != "@":
|
||||
domain += f"{dns["target"]}."
|
||||
domain += dns['domain']
|
||||
domains.append(f"- {domain}")
|
||||
for dns in data.get("dns") or []:
|
||||
domain = ""
|
||||
if dns.get("target") != "@":
|
||||
domain += f"{dns["target"]}."
|
||||
domain += dns['domain']
|
||||
domains.append(f"- {domain}")
|
||||
|
||||
result.append(template_service.format(
|
||||
key=svc_key,
|
||||
label="\\n".join([svc] + domains),
|
||||
icon=get_icon(svc)
|
||||
))
|
||||
data['key'] = svc_key
|
||||
data['label'] = "\\n".join([svc] + domains)
|
||||
data['icon'] = get_icon(svc)
|
||||
|
||||
for backup in data.get("backup") or []:
|
||||
result.append(f'({svc_key} -> external.restic.{data['host']}).style.stroke: "#0f0"')
|
||||
result.append(data)
|
||||
|
||||
if data.get("database"):
|
||||
result.append(f"({svc_key} -> {postgresql_key}).style.stroke: '#00f'")
|
||||
result.append(f"{postgresql_key}.{svc}")
|
||||
if data.get("auth"):
|
||||
result.append(f"({svc_key} -> {authentik_key}).style.stroke: '#FD4B2D'")
|
||||
result.append(f"{authentik_key}.{svc}")
|
||||
if data.get("s3"):
|
||||
result.append(f"({svc_key} -> external.scaleway.s3).style.stroke: '#bbb'")
|
||||
result.append(f"external.scaleway.s3.{svc}")
|
||||
return result
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
@ -128,11 +82,20 @@ if __name__ == '__main__':
|
|||
services = []
|
||||
|
||||
data = json.loads(sys.stdin.read())
|
||||
|
||||
hosts = parse_hosts(data["hosts"])
|
||||
services = parse_services(data["services"])
|
||||
|
||||
print("\n".join(
|
||||
[default_d2]
|
||||
+ hosts
|
||||
+ services))
|
||||
db_key = service_key_find("postgresql", data["services"])
|
||||
auth_key = service_key_find("authentik", data["services"])
|
||||
monitoring_key = service_key_find("gatus", data["services"])
|
||||
|
||||
jinja_loader = jinja2.FileSystemLoader(searchpath="./templates")
|
||||
jinja_env = jinja2.Environment(loader=jinja_loader)
|
||||
template = jinja_env.get_template("infrastructure.d2.j2")
|
||||
text = template.render(
|
||||
grid_svcs=[db_key, auth_key],
|
||||
svcs=parse_services(data["services"]),
|
||||
hosts=parse_hosts(data["hosts"]),
|
||||
db_key=db_key,
|
||||
auth_key=auth_key,
|
||||
monitoring_key=monitoring_key,
|
||||
)
|
||||
print(text)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue