Add metrics section (#39)

pythops · Jan 15, 2025 · 29c1b69 · 29c1b69
1 parent 0ed04df
commit 29c1b69
Show file tree

Hide file tree

Showing 8 changed files with 604 additions and 91 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Readme.md b/Readme.md
@@ -11,6 +11,7 @@
 - Real-time traffic inspection and visualization.
 - Comprehensive Traffic Statistics.
 - Firewall functionalities.
+- Metrics explorer.
 - Fuzzy search.
 
 ## 💡 Prerequisites

diff --git a/oryx-tui/Cargo.toml b/oryx-tui/Cargo.toml
@@ -18,7 +18,7 @@ aya = "0.13"
 oryx-common = { path = "../oryx-common" }
 mio = { version = "1", features = ["os-poll", "os-ext"] }
 itertools = "0.14"
-dirs = "5"
+dirs = "6"
 kanal = "0.1.0-pre8"
 mimalloc = "0.1"
 clap = { version = "4", features = ["derive", "cargo"] }
@@ -28,6 +28,7 @@ log = "0.4"
 env_logger = "0.11"
 serde_json = "1"
 serde = { version = "1", features = ["derive"] }
+regex = "1"
 
 [[bin]]
 name = "oryx"

diff --git a/oryx-tui/src/app.rs b/oryx-tui/src/app.rs
@@ -28,6 +28,7 @@ pub enum ActivePopup {
     UpdateFilters,
     PacketInfos,
     NewFirewallRule,
+    NewMetricExplorer,
 }
 
 #[derive(Debug)]

diff --git a/oryx-tui/src/handler.rs b/oryx-tui/src/handler.rs
@@ -70,6 +70,10 @@ pub fn handle_key_events(
                             .handle_keys(key_event, event_sender.clone())?;
                         app.is_editing = false;
                     }
+                    ActivePopup::NewMetricExplorer => {
+                        app.section.metrics.handle_popup_keys(key_event)?;
+                        app.is_editing = false;
+                    }
                     _ => {}
                 }
             }
@@ -92,6 +96,12 @@ pub fn handle_key_events(
                         app.is_editing = false;
                     }
                 }
+                ActivePopup::NewMetricExplorer => {
+                    if app.section.metrics.handle_popup_keys(key_event).is_ok() {
+                        app.active_popup = None;
+                        app.is_editing = false;
+                    }
+                }
                 _ => {}
             },
 
@@ -104,6 +114,9 @@ pub fn handle_key_events(
                         .firewall
                         .handle_keys(key_event, event_sender.clone())?;
                 }
+                ActivePopup::NewMetricExplorer => {
+                    app.section.metrics.handle_popup_keys(key_event)?;
+                }
                 _ => {}
             },
         }
@@ -158,11 +171,21 @@ pub fn handle_key_events(
 
         KeyCode::Char('n') | KeyCode::Char('e') => {
             if app.section.focused_section == FocusedSection::Firewall
-                && app.section.handle_keys(key_event, event_sender).is_ok()
+                && app
+                    .section
+                    .handle_keys(key_event, event_sender.clone())
+                    .is_ok()
             {
                 app.is_editing = true;
                 app.active_popup = Some(ActivePopup::NewFirewallRule);
             }
+
+            if app.section.focused_section == FocusedSection::Metrics
+                && app.section.handle_keys(key_event, event_sender).is_ok()
+            {
+                app.is_editing = true;
+                app.active_popup = Some(ActivePopup::NewMetricExplorer);
+            }
         }
 
         KeyCode::Char('i') => {