From 11da09934e82f1218fd7c284d823ee68bf4ae58a Mon Sep 17 00:00:00 2001
From: Oleh <45392385+olehs0@users.noreply.github.com>
Date: Sun, 5 Apr 2020 19:22:27 +0300
Subject: feat(network): hide dns queries by default (#161)

* add filter by dns queries

* update snapshots with cargo fmt

* add tests with filtering improvement
---
 src/display/components/help_text.rs | 17 +++++++++++++----
 src/display/ui.rs                   |  4 ++--
 2 files changed, 15 insertions(+), 6 deletions(-)

(limited to 'src/display')

diff --git a/src/display/components/help_text.rs b/src/display/components/help_text.rs
index 49de3b8..2b49551 100644
--- a/src/display/components/help_text.rs
+++ b/src/display/components/help_text.rs
@@ -6,22 +6,31 @@ use ::tui::widgets::{Paragraph, Text, Widget};
 
 pub struct HelpText {
     pub paused: bool,
+    pub show_dns: bool,
 }
 
-const TEXT_WHEN_PAUSED: &str = " Press <SPACE> to resume.";
-const TEXT_WHEN_NOT_PAUSED: &str = " Press <SPACE> to pause.";
+const TEXT_WHEN_PAUSED: &str = " Press <SPACE> to resume";
+const TEXT_WHEN_NOT_PAUSED: &str = " Press <SPACE> to pause";
+const TEXT_WHEN_DNS_NOT_SHOWN: &str = " (DNS queries hidden).";
+const TEXT_WHEN_DNS_SHOWN: &str = " (DNS queries shown).";
 
 impl HelpText {
     pub fn render(&self, frame: &mut Frame<impl Backend>, rect: Rect) {
         let text = {
-            let content = if self.paused {
+            let pause_content = if self.paused {
                 TEXT_WHEN_PAUSED
             } else {
                 TEXT_WHEN_NOT_PAUSED
             };
 
+            let dns_content = if self.show_dns {
+                TEXT_WHEN_DNS_SHOWN
+            } else {
+                TEXT_WHEN_DNS_NOT_SHOWN
+            };
+
             [Text::styled(
-                content,
+                format!("{}{}", pause_content, dns_content),
                 Style::default().modifier(Modifier::BOLD),
             )]
         };
diff --git a/src/display/ui.rs b/src/display/ui.rs
index cb4e7bd..4530664 100644
--- a/src/display/ui.rs
+++ b/src/display/ui.rs
@@ -77,7 +77,7 @@ where
             ));
         }
     }
-    pub fn draw(&mut self, paused: bool) {
+    pub fn draw(&mut self, paused: bool, show_dns: bool) {
         let state = &self.state;
         let children = self.get_tables_to_display();
         self.terminal
@@ -87,7 +87,7 @@ where
                     state: &state,
                     paused,
                 };
-                let help_text = HelpText { paused };
+                let help_text = HelpText { paused, show_dns };
                 let layout = Layout {
                     header: total_bandwidth,
                     children,
-- 
cgit v1.2.3