netviz 0.1.0

A simple network traffic monitor and visualizer
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use ratatui::buffer::Buffer;
use ratatui::layout::Rect;
use ratatui::style::Color;
use ratatui::text::Span;
use ratatui::widgets::{Block, BorderType, Padding, Sparkline, SparklineBar, Widget};

use crate::define_directional_history;
use crate::history::{DEFAULT_HISTORY_SIZE, Direction, History};

define_directional_history!(PacketHistory, u64, DEFAULT_HISTORY_SIZE, {}, {
    pub fn add_bar(&mut self, value: u64) {
        self.history.add(value);
    }
});

impl Widget for &PacketHistory {
    fn render(self, area: Rect, buf: &mut Buffer) {
        let mut data = Vec::new();

        let (title, color) = match self.direction {
            Direction::Download => ("Incoming Packets", Color::Cyan),
            Direction::Upload => ("Outgoing Packets", Color::Magenta),
        };

        let block = Block::bordered()
            .title(Span::styled(title, color))
            .border_type(BorderType::Rounded)
            .padding(Padding::uniform(1));

        let block_area = block.inner(area);

        block.render(area, buf);

        for value in self.history.values().iter().rev() {
            for _ in 0..block_area
                .width
                .div_ceil(self.history.max_size as u16)
                .max(1)
            {
                data.push(SparklineBar::from(value))
            }
        }

        let sparkline = Sparkline::default().data(data).style(color);

        sparkline.render(block_area, buf);
    }
}