Pass view mode to ItemSource::connectors().

greatscottgadgets · Aug 12, 2024 · 9da0e86 · 9da0e86
1 parent 21a0c97
commit 9da0e86
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/capture.rs b/src/capture.rs
@@ -1174,7 +1174,10 @@ pub trait ItemSource<Item, ViewMode> {
                    item: &Item,
                    detail: bool)
         -> Result<String, Error>;
-    fn connectors(&mut self, item: &Item) -> Result<String, Error>;
+    fn connectors(&mut self,
+                  view_mode: ViewMode,
+                  item: &Item)
+        -> Result<String, Error>;
     fn timestamp(&mut self, item: &Item) -> Result<Timestamp, Error>;
 }
 
@@ -1565,7 +1568,7 @@ impl ItemSource<TrafficItem, TrafficViewMode> for CaptureReader {
         })
     }
 
-    fn connectors(&mut self, item: &TrafficItem)
+    fn connectors(&mut self, _view_mode: TrafficViewMode, item: &TrafficItem)
         -> Result<String, Error>
     {
         use EndpointState::*;
@@ -1887,7 +1890,9 @@ impl ItemSource<DeviceItem, ()> for CaptureReader {
         })
     }
 
-    fn connectors(&mut self, item: &DeviceItem) -> Result<String, Error> {
+    fn connectors(&mut self, _view_mode: (), item: &DeviceItem)
+        -> Result<String, Error>
+    {
         use DeviceItem::*;
         let depth = match item {
             Device(..) => 0,

diff --git a/src/tree_list_model.rs b/src/tree_list_model.rs
@@ -966,7 +966,7 @@ where Item: 'static + Copy + Debug,
 
     pub fn connectors(&self, item: &Item) -> String {
         let mut cap = self.capture.borrow_mut();
-        match cap.connectors(item) {
+        match cap.connectors(self.view_mode, item) {
             Ok(string) => string,
             Err(e) => format!("Error: {e:?}")
         }