first commit

routes/edit_routes.py

@@ -0,0 +1,57 @@
+from flask import Blueprint, render_template, request
+import subprocess
+from auth.auth_middleware import requires_auth  # Updated import
+
+edit_bp = Blueprint('edit', __name__)
+
+@edit_bp.route('/edit', methods=['GET', 'POST'])
+@requires_auth
+def edit_haproxy_config():
+    if request.method == 'POST':
+        edited_config = request.form['haproxy_config']
+        # Save the edited config to the haproxy.cfg file
+        with open('/etc/haproxy/haproxy.cfg', 'w') as f:
+            f.write(edited_config)
+
+        check_output = ""
+        
+        if 'save_check' in request.form:
+            # Run haproxy -c -V -f to check the configuration
+            check_result = subprocess.run(['haproxy', '-c', '-V', '-f', '/etc/haproxy/haproxy.cfg'], capture_output=True, text=True)
+            check_output = check_result.stdout
+
+            # Check if there was an error, and if so, append it to the output
+            if check_result.returncode != 0:
+                error_message = check_result.stderr
+                check_output += f"\n\nError occurred:\n{error_message}"
+
+        elif 'save_reload' in request.form:
+            # Run haproxy -c -V -f to check the configuration
+            check_result = subprocess.run(['haproxy', '-c', '-V', '-f', '/etc/haproxy/haproxy.cfg'], capture_output=True, text=True)
+            check_output = check_result.stdout
+
+            # Check if there was an error, and if so, append it to the output
+            if check_result.returncode != 0:
+                error_message = check_result.stderr
+                check_output += f"\n\nError occurred:\n{error_message}"
+            else:
+                # If no error, run systemctl restart haproxy to reload HAProxy
+                reload_result = subprocess.run(['systemctl', 'restart', 'haproxy'], capture_output=True, text=True)
+                check_output += f"\n\nHAProxy Restart Output:\n{reload_result.stdout}"
+                
+                # Also add stderr if there are any warnings or errors during restart
+                if reload_result.stderr:
+                    check_output += f"\nRestart Stderr:\n{reload_result.stderr}"
+
+        return render_template('edit.html', config_content=edited_config, check_output=check_output)
+
+    # GET request - Read the current contents of haproxy.cfg
+    try:
+        with open('/etc/haproxy/haproxy.cfg', 'r') as f:
+            config_content = f.read()
+    except FileNotFoundError:
+        config_content = "# HAProxy configuration file not found\n# Please create /etc/haproxy/haproxy.cfg"
+    except PermissionError:
+        config_content = "# Permission denied reading HAProxy configuration file"
+    
+    return render_template('edit.html', config_content=config_content)