serguzim/infrastructure
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve visualize.py to include provider

Browse source
This commit is contained in:
Tobias Reisinger 2024-10-14 22:53:08 +02:00
parent 20bb2c8fd7
commit b5ddefbd90
Signed by: serguzim
GPG key ID: 13AD60C237A28DFE
2 changed files with 22 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
modules/infrastructure/output.tf
View file

@ -3,6 +3,7 @@ output "hosts" {
for key, host in var.hosts :
key => {
"hostname" = host.hostname
"provider" = host.provider
"fqdn" = "${host.hostname}.serguzim.net"
"fqdn_vpn" = "${host.hostname}.vpn.serguzim.net"
"ipv4_address" = try(

40
visualize.py
View file

@ -31,35 +31,35 @@ def get_icon(svc):
fmt = icon_format.get(svc, 'svg')
return f'https://cdn.jsdelivr.net/gh/selfhst/icons/{fmt}/{svc}.{fmt}'
def service_key(svc, data):
return f"'serguzim.net'.{data['host']}.{svc}"
def host_key(host):
return f"'serguzim.net'.{host}"
def host_key(host, hosts):
return f"'serguzim.net'.{hosts[host]["provider"]}.{host}"
def service_key(svc, data, hosts):
return f"{host_key(data["host"], hosts)}.{svc}"
def service_key_find(svc_name, services):
def service_key_find(svc_name, services, hosts):
for svc, data in services.items():
if svc == svc_name:
return service_key(svc, data)
return service_key(svc, data, hosts)
return None
def parse_hosts(hosts):
result = []
for host, data in hosts.items():
result.append({
'key': host_key(host),
'key': host_key(host, hosts),
'name': host,
})
return result
def parse_services(services):
def parse_services(services, hosts):
result = []
postgresql_key = service_key_find("postgresql", services)
authentik_key = service_key_find("authentik", services)
postgresql_key = service_key_find("postgresql", services, hosts)
authentik_key = service_key_find("authentik", services, hosts)
for svc, data in services.items():
svc_key = service_key(svc, data)
svc_key = service_key(svc, data, hosts)
domains = []
for dns in data.get("dns") or []:
@ -83,19 +83,21 @@ if __name__ == '__main__':
data = json.loads(sys.stdin.read())
db_key = service_key_find("postgresql", data["services"])
auth_key = service_key_find("authentik", data["services"])
monitoring_key = service_key_find("gatus", data["services"])
services = data["services"]
hosts = data["hosts"]
db_key = service_key_find("postgresql", services, hosts)
auth_key = service_key_find("authentik", services, hosts)
monitoring_key = service_key_find("gatus", services, hosts)
jinja_loader = jinja2.FileSystemLoader(searchpath="./templates")
jinja_env = jinja2.Environment(loader=jinja_loader)
template = jinja_env.get_template("infrastructure.d2.j2")
text = template.render(
print(template.render(
grid_svcs=[db_key, auth_key],
svcs=parse_services(data["services"]),
hosts=parse_hosts(data["hosts"]),
svcs=parse_services(services, hosts),
hosts=parse_hosts(hosts),
db_key=db_key,
auth_key=auth_key,
monitoring_key=monitoring_key,
)
print(text)
))