Move metrics related code into metrics.rs

To improve readability metrics related code moves to `metrics.rs`.

src/metrics.rs

@@ -0,0 +1,91 @@
+use crate::Config;
+use log::warn;
+use rocket::{get, State};
+use std::{
+    net::SocketAddr,
+    sync::atomic::{AtomicUsize, Ordering},
+};
+
+#[derive(Debug, Default)]
+pub struct Metrics {
+    pub requests_received: AtomicUsize,
+    pub requests_invalid: AtomicUsize,
+    pub hooks_successful: AtomicUsize,
+    pub hooks_forbidden: AtomicUsize,
+    pub hooks_unmatched: AtomicUsize,
+    pub commands_executed: AtomicUsize,
+    pub commands_execution_failed: AtomicUsize,
+    pub commands_successful: AtomicUsize,
+    pub commands_failed: AtomicUsize,
+}
+
+#[get("/metrics")]
+pub async fn metrics(
+    address: SocketAddr,
+    metrics: &State<Metrics>,
+    config: &State<Config>,
+) -> Option<String> {
+    // Are metrics configured?
+    if let Some(metrics_config) = &config.metrics {
+        // Are metrics enabled?
+        if metrics_config.enabled {
+            // Is a filter configured?
+            if let Some(filter) = &metrics_config.ip_filter {
+                // Does the request match the filter?
+                if filter.validate(&address.ip()) {
+                    return Some(metrics.get_metrics());
+                }
+            } else {
+                return Some(metrics.get_metrics());
+            }
+        }
+    }
+
+    warn!("Forbidden request for metrics: {:?}", address);
+
+    None
+}
+
+impl Metrics {
+    fn get_metrics(&self) -> String {
+        format!(
+            r"# HELP webhookey_requests_received Number of requests received
+# TYPE webhookey_requests_received gauge
+webhookey_requests_received {}
+# HELP webhookey_requests_invalid Number of invalid requests received
+# TYPE webhookey_requests_invalid gauge
+webhookey_requests_invalid {}
+# HELP webhookey_hooks_successful Number of successfully executed hooks
+# TYPE webhookey_hooks_successful gauge
+webhookey_hooks_sucessful {}
+# HELP webhookey_hooks_forbidden Number of forbidden requests
+# TYPE webhookey_hooks_forbidden gauge
+webhookey_hooks_forbidden {}
+# HELP webhookey_hooks_unmatched Number of unmatched requests
+# TYPE webhookey_hooks_unmatched gauge
+webhookey_hooks_unmatched {}
+# HELP webhookey_commands_executed Number of commands executed
+# TYPE webhookey_commands_executed gauge
+webhookey_commands_executed {}
+# HELP webhookey_commands_execution_failed Number of commands failed to execute
+# TYPE webhookey_commands_execution_failed gauge
+webhookey_commands_execution_failed {}
+# HELP webhookey_commands_successful Number of executed commands returning return code 0
+# TYPE webhookey_commands_successful gauge
+webhookey_commands_successful {}
+# HELP webhookey_commands_failed Number of executed commands returning different return code than 0
+# TYPE webhookey_commands_failed gauge
+webhookey_commands_failed {}
+",
+            self.requests_received.load(Ordering::Relaxed),
+            self.requests_invalid.load(Ordering::Relaxed),
+            self.hooks_successful.load(Ordering::Relaxed),
+            self.hooks_forbidden.load(Ordering::Relaxed),
+            self.hooks_unmatched.load(Ordering::Relaxed),
+            self.commands_executed.load(Ordering::Relaxed),
+            self.commands_execution_failed.load(Ordering::Relaxed),
+            self.commands_successful.load(Ordering::Relaxed),
+            self.commands_failed.load(Ordering::Relaxed),
+        )
+    }
+}