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
use colored::*;
use std::cmp::min;
use std::fmt;
use std::time::Duration;
pub trait TryUnwrap {
type Val;
fn try_unwrap(self) -> Result<Self::Val, String>;
}
impl<T> TryUnwrap for Option<T> {
type Val = T;
fn try_unwrap(self) -> Result<Self::Val, String> {
if let Some(val) = self {
Ok(val)
} else {
Err("empty output".to_string())
}
}
}
impl<T, E: fmt::Display> TryUnwrap for Result<T, E> {
type Val = T;
fn try_unwrap(self) -> Result<Self::Val, String> {
self.map_err(|err| format!("{}", err))
}
}
const DEFAULT_WIDTH: usize = 30;
pub struct Line {
text: String,
duration: Option<Duration>,
state: Option<ColoredString>,
}
impl Line {
pub fn new(text: impl Into<String>) -> Self {
Self {
text: text.into(),
duration: None,
state: None,
}
}
pub fn with_duration(mut self, duration: Duration) -> Self {
self.duration = Some(duration);
self
}
pub fn with_state(mut self, state: impl Into<ColoredString>) -> Self {
self.state = Some(state.into());
self
}
}
impl fmt::Display for Line {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
let display_width = f.width().unwrap_or(DEFAULT_WIDTH);
let duration = self
.duration
.map(|duration| format!(" ({:.2?})", duration))
.unwrap_or_else(String::new);
write!(f, "{}{}", self.text, duration.bright_black())?;
if let Some(state) = &self.state {
let width = self.text.chars().count() + 1 + duration.chars().count();
let dots = display_width - min(display_width - 5, width) - 2;
let dots = ".".repeat(dots);
write!(f, " {}", dots.bright_black())?;
if state.contains('\n') {
for line in state.trim_matches('\n').lines() {
write!(f, "\n {}", line.bold())?;
}
} else {
write!(f, " {}", state.clone().bold())?;
}
}
Ok(())
}
}