gru/haproxy-dashboard
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

revert

Browse Source
This commit is contained in:
Mateusz Gruszczyński
2025-11-01 23:35:22 +01:00
parent adf451a202
commit 43be5cdabf
1 changed files with 69 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files

146
routes/main_routes.py
View File

@@ -1,5 +1,5 @@
from flask import Blueprint, render_template, request
from auth.auth_middleware import requires_auth
from auth.auth_middleware import requires_auth # Updated import
from utils.haproxy_config import update_haproxy_config, is_frontend_exist, count_frontends_and_backends
main_bp = Blueprint('main', __name__)
@@ -8,95 +8,84 @@ main_bp = Blueprint('main', __name__)
@requires_auth
def index():
if request.method == 'POST':
f = request.form.get
gl = request.form.getlist
has = lambda k: k in request.form
frontend_name = request.form['frontend_name']
frontend_ip = request.form['frontend_ip']
frontend_port = request.form['frontend_port']
lb_method = request.form['lb_method']
protocol = request.form['protocol']
backend_name = request.form['backend_name']
add_header = 'add_header' in request.form if 'add_header' in request.form else ''
header_name = request.form['header_name']
header_value = request.form['header_value']
frontend_name = (f('frontend_name', '') or '').strip()
frontend_ip = (f('frontend_ip', '') or '').strip()
frontend_port = (f('frontend_port', '') or '').strip()
lb_method = f('lb_method', 'roundrobin')
protocol = f('protocol', '') # 'http' / 'tcp'
# Get all backend servers data
backend_server_names = request.form.getlist('backend_server_names[]')
backend_server_ips = request.form.getlist('backend_server_ips[]')
backend_server_ports = request.form.getlist('backend_server_ports[]')
backend_server_maxconns = request.form.getlist('backend_server_maxconns[]')
backend_name = (f('backend_name', '') or '').strip()
is_acl = 'add_acl' in request.form
acl_name = request.form['acl'] if 'acl' in request.form else ''
acl_action = request.form['acl_action'] if 'acl_action' in request.form else ''
acl_backend_name = request.form['backend_name_acl'] if 'backend_name_acl' in request.form else ''
use_ssl = 'ssl_checkbox' in request.form
ssl_cert_path = request.form['ssl_cert_path']
https_redirect = 'ssl_redirect_checkbox' in request.form
is_dos = 'add_dos' in request.form if 'add_dos' in request.form else ''
ban_duration = request.form["ban_duration"]
limit_requests = request.form["limit_requests"]
forward_for = 'forward_for_check' in request.form
add_header = has('add_header')
header_name = (f('header_name', '') or '').strip() if add_header else ''
header_value= (f('header_value', '') or '').strip() if add_header else ''
is_forbidden_path = 'add_acl_path' in request.form
forbidden_name = request.form["forbidden_name"]
allowed_ip = request.form["allowed_ip"]
forbidden_path = request.form["forbidden_path"]
backend_server_names = gl('backend_server_names[]')
backend_server_ips = gl('backend_server_ips[]')
backend_server_ports = gl('backend_server_ports[]')
backend_server_maxconns= gl('backend_server_maxconns[]')
sql_injection_check = 'sql_injection_check' in request.form if 'sql_injection_check' in request.form else ''
is_xss = 'xss_check' in request.form if 'xss_check' in request.form else ''
is_remote_upload = 'remote_uploads_check' in request.form if 'remote_uploads_check' in request.form else ''
is_acl = has('add_acl')
acl_name = (f('acl', '') or '').strip() if is_acl else ''
acl_action = (f('acl_action', '') or '').strip() if is_acl else ''
acl_backend_name= (f('backend_name_acl', '') or '').strip() if is_acl else ''
use_ssl = has('ssl_checkbox')
ssl_cert_path = (f('ssl_cert_path', '') or '').strip() if use_ssl else ''
https_redirect= has('ssl_redirect_checkbox') and use_ssl
is_dos = has('add_dos')
ban_duration = (f('ban_duration', '') or '').strip() if is_dos else ''
limit_requests= (f('limit_requests', '') or '').strip() if is_dos else ''
forward_for = has('forward_for_check') and protocol == 'http'
is_forbidden_path = has('add_acl_path') and protocol == 'http'
forbidden_name = (f('forbidden_name', '') or '').strip() if is_forbidden_path else ''
allowed_ip = (f('allowed_ip', '') or '').strip() if is_forbidden_path else ''
forbidden_path = (f('forbidden_path', '') or '').strip() if is_forbidden_path else ''
if protocol == 'http':
sql_injection_check = has('sql_injection_check')
is_xss = has('xss_check')
is_remote_upload = has('remote_uploads_check')
is_webshells = has('webshells_check')
else:
sql_injection_check = False
is_xss = False
is_remote_upload = False
is_webshells = False
add_path_based = has('add_path_based') and protocol == 'http'
redirect_domain_name = (f('redirect_domain_name', '') or '').strip() if add_path_based else ''
root_redirect = (f('root_redirect', '/') or '/').strip() if add_path_based else ''
redirect_to = (f('redirect_to', '') or '').strip() if add_path_based else ''
if protocol == 'http':
health_check = has('health_check')
health_check_link = (f('health_check_link', '') or '').strip() if health_check else ''
health_check_tcp = False
elif protocol == 'tcp':
health_check = False
health_check_link = ""
health_check_tcp = has('health_check2')
else:
health_check = False
health_check_link = ""
health_check_tcp = False
sticky_session = has('sticky_session')
sticky_session_type = (f('sticky_session_type', '') or '').strip() if sticky_session else ''
add_path_based = 'add_path_based' in request.form
redirect_domain_name = request.form["redirect_domain_name"]
root_redirect = request.form["root_redirect"]
redirect_to = request.form["redirect_to"]
is_webshells = 'webshells_check' in request.form if 'webshells_check' in request.form else ''
# Combine backend server info into a list of tuples (name, ip, port, maxconns)
backend_servers = []
max_len = max(len(backend_server_ips), len(backend_server_ports), len(backend_server_names), len(backend_server_maxconns))
for i in range(max_len):
name = backend_server_names[i] if i < len(backend_server_names) else f"server{i+1}"
ip = backend_server_ips[i] if i < len(backend_server_ips) else ''
port = backend_server_ports[i] if i < len(backend_server_ports) else ''
maxc = backend_server_maxconns[i] if i < len(backend_server_maxconns) else None
if ip and port:
backend_servers.append((name, ip, port, maxc))
for i in range(len(backend_server_ips)):
name = backend_server_names[i] if i < len(backend_server_names) else f"server{i+1}"
ip = backend_server_ips[i] if i < len(backend_server_ips) else ''
port = backend_server_ports[i] if i < len(backend_server_ports) else ''
maxconn = backend_server_maxconns[i] if i < len(backend_server_maxconns) else None
if not frontend_name or not frontend_ip or not frontend_port or not backend_name:
return render_template('index.html', message="Brak wymaganych pól (nazwa/IP/port frontendu, nazwa backendu).")
if ip and port: # Only add if we have IP and port
backend_servers.append((name, ip, port, maxconn))
# Check if frontend or port already exists
if is_frontend_exist(frontend_name, frontend_ip, frontend_port):
return render_template('index.html', message="Frontend or Port already exists. Cannot add duplicate.")
# Get health check related fields if the protocol is HTTP
health_check = False
health_check_link = ""
if protocol == 'http':
health_check = 'health_check' in request.form
if health_check:
health_check_link = request.form['health_check_link']
health_check_tcp = False
if protocol == 'tcp':
health_check_tcp = 'health_check2' in request.form
# Get sticky session related fields
sticky_session = False
sticky_session_type = ""
if 'sticky_session' in request.form:
sticky_session = True
sticky_session_type = request.form['sticky_session_type']
# Update the HAProxy config file
message = update_haproxy_config(
frontend_name, frontend_ip, frontend_port, lb_method, protocol, backend_name,
backend_servers, health_check, health_check_tcp, health_check_link, sticky_session,
@@ -110,7 +99,6 @@ def index():
return render_template('index.html')
@main_bp.route('/home')
@requires_auth
def home():