From 6ba4fd3f9375397ed8812ce76213b7c7e477b4b5 Mon Sep 17 00:00:00 2001
From: Minei3oat <Minei3oat@users.noreply.github.com>
Date: Fri, 29 Nov 2024 01:08:06 +0100
Subject: [PATCH] Add metrics endpoint for Netfilter Regex

---
 backend/routers/nfregex.py | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/backend/routers/nfregex.py b/backend/routers/nfregex.py
index 9766a42..7c6f05e 100644
--- a/backend/routers/nfregex.py
+++ b/backend/routers/nfregex.py
@@ -2,7 +2,7 @@ from base64 import b64decode
 import re
 import secrets
 import sqlite3
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter, Response, HTTPException
 from pydantic import BaseModel
 from modules.nfregex.nftables import FiregexTables
 from modules.nfregex.firewall import STATUS, FirewallManager
@@ -275,3 +275,26 @@ async def add_new_service(form: ServiceAddForm):
     await firewall.reload()
     await refresh_frontend()
     return {'status': 'ok', 'service_id': srv_id}
+
+@app.get('/metrics', response_class = Response)
+async def metrics():
+    """Aggregate metrics"""
+    stats = db.query("""
+        SELECT
+            s.name,
+            s.status,
+            r.regex,
+            r.is_blacklist,
+            r.mode,
+            r.is_case_sensitive,
+            r.blocked_packets,
+            r.active
+        FROM services s LEFT JOIN regexes r ON s.service_id = r.service_id;
+    """)
+    metrics = []
+    sanitize = lambda s : s.replace('\\', '\\\\').replace('"', '\\"').replace('\n', '\\n')
+    for stat in stats:
+        props = f'service_name="{sanitize(stat["name"])}",regex="{sanitize(b64decode(stat["regex"]).decode())}",mode="{stat["mode"]}",is_case_sensitive="{stat["is_case_sensitive"]}"'
+        metrics.append(f'firegex_blocked_packets{{{props}}} {stat["blocked_packets"]}')
+        metrics.append(f'firegex_active{{{props}}} {int(stat["active"] and stat["status"] == "active")}')
+    return "\n".join(metrics)