fluzzi32/traefik-frontend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Fede Luzzi
2024-08-05 10:58:57 -03:00
parent ab6f27524d
commit 2353362c04
11 changed files with 649 additions and 231 deletions
Download Patch File Download Diff File Expand all files Collapse all files

215
app.py
View File

@@ -2,128 +2,146 @@ from flask import Flask, render_template
import re
import docker
import requests
import yaml
import os
from waitress import serve
app = Flask(__name__)
TRAEFIK_API_URL = "https://norman.lan/api/http/routers"
REGEX_PATTERNS = [r".*\.gederico\.dynu\.net"]
DEFAULT_GROUP_PRIORITY = 100
DEFAULT_GROUP = 'Applications'
DEFAULT_GROUP_ICON = 'fas fa-box'
DEFAULT_TITLE = 'Traefik Routers'
DEFAULT_ICON = 'fas fa-bars'
TRAEFIK_API_URL = os.getenv('TRAEFIK_API_URL')
# Check if TRAEFIK_API_URL is set
if not TRAEFIK_API_URL:
raise ValueError("TRAEFIK_API_URL environment variable is required and not set.")
USER_COLORS_FILE = os.getenv('USER_COLORS_FILE', False)
USER_CONFIG_FILE = os.getenv('USER_CONFIG_FILE', False)
router_configs, group_configs, global_configs = {}, {}, {}
client = docker.from_env()
def get_routers(containers):
def get_defaults(global_configs):
return {
'REGEX_PATTERNS': global_configs.get('url_regex', [r".*"]),
'DEFAULT_GROUP_PRIORITY': global_configs.get('group_priority', 100),
'DEFAULT_GROUP': global_configs.get('default_group', 'Applications'),
'DEFAULT_GROUP_ICON': global_configs.get('group_icon', 'fas fa-box'),
'DEFAULT_GROUP_COLLAPSED': global_configs.get('group_collapsed', 'false'),
'DEFAULT_TITLE': global_configs.get('title', 'Traefik Routers'),
'DEFAULT_ROUTER_ICON': global_configs.get('router_icon', 'fas fa-bars'),
'DEFAULT_PROTOCOL': global_configs.get('entrypoint', 'http'),
'DEFAULT_TARGET': global_configs.get('target', '_blank'),
}
def load_yaml(file_path):
with open(file_path, 'r') as file:
return yaml.safe_load(file)
def merge_colors(default_colors, user_colors):
for mode in default_colors:
if mode in user_colors:
default_colors[mode].update(user_colors[mode])
return default_colors
def categorize_configs(config):
for key, value in config.items():
if key == "routers":
router_configs.update(value)
elif key == "groups":
group_configs.update(value)
elif key == "global":
global_configs.update(value)
def fetch_containers():
containers = client.containers.list(all=True)
for container in containers:
labels = container.attrs.get('Config', {}).get('Labels', {})
for label, value in labels.items():
if label.startswith('traefik-frontend.http.routers.'):
router_name = label.split('.')[3]
if router_name not in router_configs:
router_configs[router_name] = {}
key = label.split('.')[4]
router_configs[router_name][key] = value
elif label.startswith('traefik-frontend.groups.'):
group_name = label.split('.')[2]
if group_name not in group_configs:
group_configs[group_name] = {}
key = label.split('.')[3]
group_configs[group_name][key] = value
elif label.startswith('traefik-frontend.'):
key = label.split('.')[1]
global_configs[key] = value
def get_routers(defaults):
response = requests.get(TRAEFIK_API_URL, verify=False) # Ignore SSL certificate verification
if response.status_code == 200:
routers = response.json()
filtered_routers = filter_routers(routers, containers)
filtered_routers = filter_routers(routers, defaults)
return filtered_routers
return []
def filter_routers(routers, containers):
def filter_routers(routers, defaults):
filtered_routers = []
for router in routers:
for pattern in REGEX_PATTERNS:
for pattern in defaults["REGEX_PATTERNS"]:
if re.match(pattern, router['rule'].split('`')[1]):
if not is_router_hidden(router['name'], containers):
router['description'] = get_router_description(router['name'], containers)
router['display_name'] = get_router_display_name(router['name'], containers)
router['icon'] = get_router_icon(router['name'], containers)
router['group'] = get_router_group(router['name'], containers)
if not is_router_hidden(router['name']):
router_name = router['name'].split('@')[0]
router['description'] = router_configs.get(router_name, {}).get('description', False)
router['display_name'] = router_configs.get(router_name, {}).get('router_name', router_name).upper()
router['icon'] = router_configs.get(router_name, {}).get('icon', defaults['DEFAULT_ROUTER_ICON'])
router['group'] = router_configs.get(router_name, {}).get('group', defaults['DEFAULT_GROUP'])
router['protocol'] = router_configs.get(router_name, {}).get('entrypoint', defaults['DEFAULT_PROTOCOL'])
router['target'] = router_configs.get(router_name, {}).get('target', defaults['DEFAULT_TARGET'])
filtered_routers.append(router)
break
return filtered_routers
def is_router_hidden(router_name, containers):
def is_router_hidden(router_name):
service_name = router_name.split('@')[0]
for container in containers:
labels = container.attrs.get('Config', {}).get('Labels', {})
hidden_label = f'traefik-frontend.http.routers.{service_name}.hidden'
if labels.get(hidden_label) == 'true':
return True
return False
def get_router_description(router_name, containers):
service_name = router_name.split('@')[0]
for container in containers:
labels = container.attrs.get('Config', {}).get('Labels', {})
description_label = f'traefik-frontend.http.routers.{service_name}.description'
if description_label in labels:
return labels[description_label]
return 'No description available'
def get_router_display_name(router_name, containers):
service_name = router_name.split('@')[0]
for container in containers:
labels = container.attrs.get('Config', {}).get('Labels', {})
display_name_label = f'traefik-frontend.http.routers.{service_name}.router_name'
if display_name_label in labels:
return labels[display_name_label].upper()
return service_name.upper()
def get_router_group(router_name, containers):
service_name = router_name.split('@')[0]
for container in containers:
labels = container.attrs.get('Config', {}).get('Labels', {})
group_label = f'traefik-frontend.http.routers.{service_name}.group'
if group_label in labels:
return labels[group_label]
return DEFAULT_GROUP
def get_router_icon(router_name, containers):
service_name = router_name.split('@')[0]
for container in containers:
labels = container.attrs.get('Config', {}).get('Labels', {})
icon_label = f'traefik-frontend.http.routers.{service_name}.icon'
if icon_label in labels:
return labels[icon_label]
return DEFAULT_ICON
def get_groups(containers):
groups = {DEFAULT_GROUP: {'priority': DEFAULT_GROUP_PRIORITY, 'collapsed': False, 'routers': [], 'icon': DEFAULT_GROUP_ICON}}
for container in containers:
labels = container.attrs.get('Config', {}).get('Labels', {})
for label, value in labels.items():
if label.startswith('traefik-frontend.groups.'):
group_name = label.split('.')[2]
priority_label = f'traefik-frontend.groups.{group_name}.priority'
collapsed_label = f'traefik-frontend.groups.{group_name}.collapsed'
icon_label = f'traefik-frontend.groups.{group_name}.icon'
if group_name not in groups:
groups[group_name] = {
'priority': DEFAULT_GROUP_PRIORITY,
'collapsed': False,
'routers': [],
'icon': DEFAULT_GROUP_ICON
}
if priority_label in labels:
groups[group_name]['priority'] = int(labels[priority_label])
if collapsed_label in labels:
groups[group_name]['collapsed'] = labels[collapsed_label].lower() == 'true'
if icon_label in labels:
groups[group_name]['icon'] = labels[icon_label]
return router_configs.get(service_name, {}).get('hidden') == 'true'
def get_groups(defaults):
groups = {
defaults['DEFAULT_GROUP']: {
'priority': int(defaults['DEFAULT_GROUP_PRIORITY']),
'collapsed': defaults['DEFAULT_GROUP_COLLAPSED'].lower() == 'true' if isinstance(defaults['DEFAULT_GROUP_COLLAPSED'], str) else defaults['DEFAULT_GROUP_COLLAPSED'],
'routers': [],
'icon': defaults['DEFAULT_GROUP_ICON']
}
}
for group_name, config in group_configs.items():
groups[group_name] = {
'priority': int(config.get('priority', defaults['DEFAULT_GROUP_PRIORITY'])),
'collapsed': config.get('collapsed', defaults["DEFAULT_GROUP_COLLAPSED"]).lower() == 'true' if isinstance(config.get('collapsed', defaults["DEFAULT_GROUP_COLLAPSED"]), str) else config.get('collapsed', defaults["DEFAULT_GROUP_COLLAPSED"]),
'routers': [],
'icon': config.get('icon', defaults['DEFAULT_GROUP_ICON'])
}
return groups
def get_title(containers):
for container in containers:
labels = container.attrs.get('Config', {}).get('Labels', {})
title_label = 'traefik-frontend.title'
if title_label in labels:
return labels[title_label]
return DEFAULT_TITLE
@app.route('/static/css/colors.css')
def colors_css():
DEFAULT_COLORS_FILE = '/app/colors.yml'
default_colors = load_yaml(DEFAULT_COLORS_FILE)
if os.path.exists(USER_COLORS_FILE) and USER_COLORS_FILE:
user_colors = load_yaml(USER_COLORS_FILE)
merged_colors = merge_colors(default_colors["colors"], user_colors["colors"])
else:
merged_colors = default_colors["colors"]
return render_template('colors.css', colors=merged_colors), {'Content-Type': 'text/css'}
@app.route('/')
def index():
containers = client.containers.list(all=True) # Get all containers once
title = get_title(containers)
routers = get_routers(containers)
groups = get_groups(containers)
if os.path.exists(USER_CONFIG_FILE) and USER_CONFIG_FILE:
user_config = load_yaml(USER_CONFIG_FILE)
categorize_configs(user_config.get("config", {}))
fetch_containers()
defaults = get_defaults(global_configs)
title = defaults["DEFAULT_TITLE"]
routers = get_routers(defaults)
groups = get_groups(defaults)
for router in routers:
group_name = router['group']
if group_name not in groups:
@@ -133,5 +151,4 @@ def index():
return render_template('index.html', title=title, groups=sorted_groups)
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0')
serve(app, host='0.0.0.0', port=5000)