LunaChocken
89 lines
2.3 KiB
Python
89 lines
2.3 KiB
Python
import os
|
|
import time
|
|
from prometheus_client import start_http_server, Gauge, REGISTRY
|
|
from dotenv import load_dotenv
|
|
|
|
# 1. Define Metrics with Labels
|
|
# Use labels for domain and feature to avoid creating 100s of unique metric names
|
|
DOMAIN_FEATURE = Gauge(
|
|
"caddy_domain_imports",
|
|
"Binary status of domain features (1=enabled, 0=disabled)",
|
|
["domain", "feature"],
|
|
)
|
|
|
|
GLOBAL_STATS = Gauge(
|
|
"caddy_global_stats", "Aggregated counts of features across all domains", ["type"]
|
|
)
|
|
|
|
|
|
load_dotenv()
|
|
|
|
CADDY_PATH = os.environ["CADDY_PATH"]
|
|
|
|
|
|
def load_caddyfile():
|
|
with open(CADDY_PATH, "r") as f:
|
|
return f.read()
|
|
|
|
|
|
def parse(data):
|
|
from libs.parse_caddyfile import parse as caddy_parse
|
|
|
|
return caddy_parse(data)
|
|
|
|
|
|
def mathing(data: dict):
|
|
total_domains = len(data.items())
|
|
# info = dict({"auth": 0, "wan": 0, "noauth": 0, "nowan": 0, "none": 0})
|
|
info = dict({"no_auth": 0, "no_wan": 0})
|
|
for domain, imports in data.items():
|
|
if len(imports) == 0:
|
|
if "none" not in info:
|
|
info["none"] = 1
|
|
info["none"] += 1
|
|
continue
|
|
for imp in imports:
|
|
if imp not in info:
|
|
info[imp] = 1
|
|
info[imp] += 1
|
|
if "auth" not in imports:
|
|
info["no_auth"] += 1
|
|
if "wan" not in imports:
|
|
info["no_wan"] += 1
|
|
info["total_domains"] = total_domains
|
|
return info
|
|
|
|
|
|
def update_metrics():
|
|
# Your existing parsing logic
|
|
try:
|
|
raw_data = load_caddyfile()
|
|
hosts = parse(raw_data)
|
|
info = mathing(hosts)
|
|
|
|
# --- Update Global Metrics ---
|
|
for key, value in info.items():
|
|
GLOBAL_STATS.labels(type=key).set(value)
|
|
|
|
# --- Update Per-Domain Metrics ---
|
|
tracked_features = ["auth", "wan"]
|
|
|
|
for domain, imports in hosts.items():
|
|
for feature in tracked_features:
|
|
is_enabled = 1 if feature in imports else 0
|
|
DOMAIN_FEATURE.labels(domain=domain, feature=feature).set(is_enabled)
|
|
|
|
print(f"Metrics updated at {time.ctime()}")
|
|
|
|
except Exception as e:
|
|
print(f"Error updating metrics: {e}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
start_http_server(port=8000, addr="0.0.0.0")
|
|
print("Prometheus server started on port 8000")
|
|
|
|
while True:
|
|
update_metrics()
|
|
time.sleep(60)
|