91 lines
2.1 KiB
Python
91 lines
2.1 KiB
Python
import requests
|
|
from typing import NamedTuple
|
|
import re
|
|
import sys
|
|
import operator
|
|
import tomllib
|
|
from pathlib import Path
|
|
from typing import TypedDict
|
|
|
|
TRAEFIK_HOST_RE = re.compile(r"Host\(`([a-z0-9\.-]+)`\)")
|
|
|
|
CONFIG_FILE = Path.cwd() / "config.toml"
|
|
|
|
class Route(NamedTuple):
|
|
name: str
|
|
destination: str
|
|
hostname: str
|
|
|
|
|
|
class Source(TypedDict):
|
|
type: str
|
|
url: str
|
|
destination: str
|
|
|
|
|
|
class Config(TypedDict):
|
|
source: list[Source]
|
|
|
|
|
|
def parse_traefik_rule(rule: str) -> list[str]:
|
|
if "Host(" not in rule:
|
|
return []
|
|
|
|
return TRAEFIK_HOST_RE.findall(rule)
|
|
|
|
|
|
def get_traefik_routes(traefik_host: str, traefik_route: str):
|
|
api_response = requests.get(f"{traefik_host}/api/http/routers").json()
|
|
routes = set()
|
|
|
|
for router in api_response:
|
|
hosts = parse_traefik_rule(router["rule"])
|
|
|
|
if not hosts:
|
|
print(f"Failed to find host for {router['rule']}", file=sys.stderr)
|
|
continue
|
|
|
|
if len(hosts) > 1:
|
|
print(f"WARNING: Found multiple hosts for rule: {router['rule']}", file=sys.stderr)
|
|
|
|
routes.add(Route(
|
|
router["service"],
|
|
traefik_route,
|
|
hosts[0]
|
|
))
|
|
|
|
return routes
|
|
|
|
def get_dokku_routes(dokku_exporter_url: str, dokku_route: str):
|
|
api_response = requests.get(dokku_exporter_url).json()
|
|
|
|
routes = set()
|
|
|
|
for app in api_response:
|
|
for vhost in app["vhosts"]:
|
|
routes.add(Route(
|
|
app["app"],
|
|
dokku_route,
|
|
vhost
|
|
))
|
|
|
|
return routes
|
|
|
|
|
|
def main():
|
|
config: Config = tomllib.loads(CONFIG_FILE.read_text())
|
|
|
|
routes = []
|
|
for source in config["source"]:
|
|
match source["type"]:
|
|
case "traefik":
|
|
routes.extend(get_traefik_routes(source["url"], source["destination"]))
|
|
case "dokku":
|
|
routes.extend(get_dokku_routes(source["url"], source["destination"]))
|
|
|
|
for route in sorted(routes, key=operator.attrgetter("hostname")):
|
|
print(f"{route.hostname}\t{route.destination}; # {route.name}")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|