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
102
|
use std::time::{Duration, Instant};
use ratatui::{
layout::{Alignment, Rect},
style::{Color, Modifier, Style},
terminal::Frame,
text::Span,
widgets::Paragraph,
};
use unicode_width::UnicodeWidthStr;
use crate::display::{DisplayBandwidth, UIState};
pub fn elapsed_time(last_start_time: Instant, cumulative_time: Duration, paused: bool) -> Duration {
if paused {
cumulative_time
} else {
cumulative_time + last_start_time.elapsed()
}
}
fn format_duration(d: Duration) -> String {
let s = d.as_secs();
let days = match s / 86400 {
0 => "".to_string(),
1 => "1 day, ".to_string(),
n => format!("{n} days, "),
};
format!(
"{days}{:02}:{:02}:{:02}",
(s / 3600) % 24,
(s / 60) % 60,
s % 60,
)
}
pub struct HeaderDetails<'a> {
pub state: &'a UIState,
pub elapsed_time: Duration,
pub paused: bool,
}
impl<'a> HeaderDetails<'a> {
pub fn render(&self, frame: &mut Frame, rect: Rect) {
let bandwidth = self.bandwidth_string();
let color = if self.paused {
Color::Yellow
} else {
Color::Green
};
// do not render time in tests, otherwise the output becomes non-deterministic
// see: https://github.com/imsnif/bandwhich/issues/303
if cfg!(not(test)) && self.state.cumulative_mode {
let elapsed_time = format_duration(self.elapsed_time);
// only render if there is enough width
if bandwidth.width() + 1 + elapsed_time.width() <= rect.width as usize {
self.render_elapsed_time(frame, rect, &elapsed_time, color);
}
}
self.render_bandwidth(frame, rect, &bandwidth, color);
}
fn render_bandwidth(&self, frame: &mut Frame, rect: Rect, bandwidth: &str, color: Color) {
let bandwidth_text = Span::styled(
bandwidth,
Style::default().fg(color).add_modifier(Modifier::BOLD),
);
let paragraph = Paragraph::new(bandwidth_text).alignment(Alignment::Left);
frame.render_widget(paragraph, rect);
}
fn bandwidth_string(&self) -> String {
let t = if self.state.cumulative_mode {
"Data"
} else {
"Rate"
};
let unit_family = self.state.unit_family;
let up = DisplayBandwidth {
bandwidth: self.state.total_bytes_uploaded as f64,
unit_family,
};
let down = DisplayBandwidth {
bandwidth: self.state.total_bytes_downloaded as f64,
unit_family,
};
let paused = if self.paused { " [PAUSED]" } else { "" };
format!(" Total {t} (Up / Down): {up} / {down}{paused}")
}
fn render_elapsed_time(&self, frame: &mut Frame, rect: Rect, elapsed_time: &str, color: Color) {
let elapsed_time_text = Span::styled(
elapsed_time,
Style::default().fg(color).add_modifier(Modifier::BOLD),
);
let paragraph = Paragraph::new(elapsed_time_text).alignment(Alignment::Right);
frame.render_widget(paragraph, rect);
}
}
|
