summaryrefslogtreecommitdiffstats
path: root/src/display/components/total_bandwidth.rs
blob: c9a9676c6c867412c0e9368b032fa24fc6009569 (plain)
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
use ::tui::backend::Backend;
use ::tui::layout::{Alignment, Rect};
use ::tui::style::{Color, Modifier, Style};
use ::tui::terminal::Frame;
use ::tui::widgets::{Paragraph, Text, Widget};

use crate::display::{DisplayBandwidth, UIState};
use tui::layout::{Constraint, Direction};

pub struct HeaderDetails<'a> {
    pub state: &'a UIState,
    pub elapsed_time: std::time::Duration,
    pub paused: bool,
}

impl<'a> HeaderDetails<'a> {
    pub fn render(&self, frame: &mut Frame<impl Backend>, rect: Rect) {
        let parts = self.header_parts(rect);

        let color = if self.paused {
            Color::Yellow
        } else {
            Color::Green
        };

        if parts.get(0).is_some() {
            self.render_bandwidth(frame, parts[0], &color);
        }
        if parts.get(1).is_some() {
            self.render_paused(frame, parts[1], &color);
        }
        if parts.get(2).is_some() && self.state.cumulative_mode {
            self.render_elapsed_time(frame, parts[2], &color);
        }
    }

    fn render_bandwidth(&self, frame: &mut Frame<impl Backend>, rect: Rect, color: &Color) {
        let c_mode = self.state.cumulative_mode;
        let title_text = {
            [Text::styled(
                format!(
                    " Total Up / Down: {} / {}",
                    DisplayBandwidth {
                        bandwidth: self.state.total_bytes_uploaded as f64,
                        as_rate: !c_mode,
                    },
                    DisplayBandwidth {
                        bandwidth: self.state.total_bytes_downloaded as f64,
                        as_rate: !c_mode,
                    }
                ),
                Style::default().fg(*color).modifier(Modifier::BOLD),
            )]
        };

        Paragraph::new(title_text.iter())
            .alignment(Alignment::Left)
            .render(frame, rect);
    }

    fn render_elapsed_time(&self, frame: &mut Frame<impl Backend>, rect: Rect, color: &Color) {
        let elapsed_time_text = [Text::styled(
            format!(
                "Duration: {:02}:{:02}:{:02} ",
                self.elapsed_time.as_secs() / 3600,
                (self.elapsed_time.as_secs() % 3600) / 60,
                self.elapsed_time.as_secs() % 60
            ),
            Style::default().fg(*color).modifier(Modifier::BOLD),
        )];
        Paragraph::new(elapsed_time_text.iter())
            .alignment(Alignment::Right)
            .render(frame, rect);
    }

    fn render_paused(&self, frame: &mut Frame<impl Backend>, rect: Rect, color: &Color) {
        if self.paused {
            let paused_text = [Text::styled(
                format!("PAUSED"),
                Style::default().fg(*color).modifier(Modifier::BOLD),
            )];
            Paragraph::new(paused_text.iter())
                .alignment(Alignment::Center)
                .render(frame, rect);
        }
    }

    fn header_parts(&self, rect: Rect) -> Vec<Rect> {
        const MAX_BANDWIDTH_STRING_LENGTH: u16 = 31;
        let number_of_columns = rect.width / MAX_BANDWIDTH_STRING_LENGTH;
        let constraints: Vec<Constraint> = (0..number_of_columns)
            .map(|_| Constraint::Percentage(100 / number_of_columns))
            .collect();

        ::tui::layout::Layout::default()
            .direction(Direction::Horizontal)
            .margin(0)
            .constraints(constraints.as_ref())
            .split(rect)
    }
}