stdin stuff
This commit is contained in:
parent
862caed98b
commit
4e05c1a6d4
131
src/main.rs
131
src/main.rs
|
@ -1,4 +1,4 @@
|
|||
use std::{process::exit, env::Args};
|
||||
use std::{process::exit, env::Args, io::{stdin, Read}, sync::mpsc, thread, time::Instant};
|
||||
|
||||
static DEFAULT_MAX: usize = 100;
|
||||
static DEFAULT_WIDTH: usize = 80;
|
||||
|
@ -10,23 +10,65 @@ fn main() {
|
|||
Err(e) => err(&e)
|
||||
};
|
||||
|
||||
// [#####....] ---- 50%
|
||||
if let Some(val) = config.val {
|
||||
let bar = generate_bar(val, config.max, config.width, &config.chars, None);
|
||||
println!("{}", bar);
|
||||
} else {
|
||||
let (tx, rx) = mpsc::sync_channel(256);
|
||||
thread::spawn(move || {
|
||||
let mut stdin = stdin();
|
||||
let mut out = [0; 1];
|
||||
loop {
|
||||
if let Ok(count) = stdin.read(&mut out) {
|
||||
tx.send(count);
|
||||
}
|
||||
}
|
||||
});
|
||||
let mut val = 0;
|
||||
let mut spin = 0;
|
||||
let now = Instant::now();
|
||||
loop {
|
||||
if let Ok(count) = rx.try_recv() {
|
||||
val += count;
|
||||
}
|
||||
let bar = generate_bar_clamp(val, config.max, config.width, &config.chars, Some(spin));
|
||||
print!("\r{}", bar);
|
||||
if val >= config.max {
|
||||
break;
|
||||
}
|
||||
spin = ((now.elapsed().as_millis() / 100) % 6) as usize;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let percent = config.val * 100 / config.max;
|
||||
fn generate_bar(val: usize, max: usize, width: usize, chars: &Chars, spin: Option<usize>) -> String {
|
||||
let percent = val * 100 / max;
|
||||
let percent_str = percent.to_string();
|
||||
let after_bar = format!(" {} {}% ", "-".repeat(6 - percent_str.len()), percent_str);
|
||||
let bar_len = config.width - after_bar.len();
|
||||
let after_bar = if let Some(spin) = spin {
|
||||
format!(" {} {}% {} ", "-".repeat(6 - percent_str.len()), percent_str, if val == max { chars.spin_done() } else { chars.spin(spin) })
|
||||
} else {
|
||||
format!(" {} {}% ", "-".repeat(6 - percent_str.len()), percent_str)
|
||||
};
|
||||
let bar_len = width - after_bar.len();
|
||||
let sep = bar_len * percent / 100;
|
||||
|
||||
let mut bar = String::new();
|
||||
|
||||
bar.push(if sep > 0 { config.first_char.1 } else { config.first_char.0 });
|
||||
bar.push(chars.first(sep > 0));
|
||||
for i in 1..bar_len-1 {
|
||||
bar.push(if sep > i { config.middle_char.1 } else { config.middle_char.0 });
|
||||
bar.push(chars.mid(sep > i));
|
||||
}
|
||||
bar.push(if sep > bar_len-1 { config.last_char.1 } else { config.last_char.0 });
|
||||
bar.push(chars.last(sep > bar_len - 1));
|
||||
|
||||
println!("{}{}", bar, after_bar);
|
||||
format!("{}{}", bar, after_bar)
|
||||
}
|
||||
|
||||
fn generate_bar_clamp(mut val: usize, max: usize, width: usize, chars: &Chars, spin: Option<usize>) -> String {
|
||||
if val > max {
|
||||
val = max;
|
||||
}
|
||||
|
||||
generate_bar(val, max, width, chars, spin)
|
||||
}
|
||||
|
||||
fn err(msg: &str) -> ! {
|
||||
|
@ -36,16 +78,48 @@ fn err(msg: &str) -> ! {
|
|||
}
|
||||
|
||||
fn show_help() {
|
||||
eprintln!("Usage: progress <value> [--max <max>] [--width <width>] [--fira]");
|
||||
eprintln!("Usage: progress [<value>] [--max <max>] [--width <width>] [--fira]");
|
||||
}
|
||||
|
||||
struct Chars {
|
||||
first: (char, char),
|
||||
mid: (char, char),
|
||||
last: (char, char),
|
||||
spin: [char; 6],
|
||||
spin_done: char
|
||||
}
|
||||
|
||||
impl Chars {
|
||||
fn new(first: (char, char), mid: (char, char), last: (char, char), spin: [char; 6], spin_done: char) -> Self {
|
||||
Self { first, mid, last, spin, spin_done }
|
||||
}
|
||||
|
||||
fn first(&self, filled: bool) -> char {
|
||||
if filled { self.first.1 } else { self.first.0 }
|
||||
}
|
||||
|
||||
fn mid(&self, filled: bool) -> char {
|
||||
if filled { self.mid.1 } else { self.mid.0 }
|
||||
}
|
||||
|
||||
fn last(&self, filled: bool) -> char {
|
||||
if filled { self.last.1 } else { self.last.0 }
|
||||
}
|
||||
|
||||
fn spin(&self, n: usize) -> char {
|
||||
self.spin[n % 6]
|
||||
}
|
||||
|
||||
fn spin_done(&self) -> char {
|
||||
self.spin_done
|
||||
}
|
||||
}
|
||||
|
||||
struct Config {
|
||||
val: usize,
|
||||
val: Option<usize>,
|
||||
max: usize,
|
||||
first_char: (char, char),
|
||||
middle_char: (char, char),
|
||||
last_char: (char, char),
|
||||
width: usize
|
||||
width: usize,
|
||||
chars: Chars
|
||||
}
|
||||
|
||||
impl Config {
|
||||
|
@ -53,17 +127,19 @@ impl Config {
|
|||
args.next();
|
||||
let mut val = None;
|
||||
let mut max = None;
|
||||
let mut first_char = ('[', '[');
|
||||
let mut middle_char = ('.', '#');
|
||||
let mut last_char = (']', ']');
|
||||
let mut width = None;
|
||||
let mut first_char = ('[', '[');
|
||||
let mut mid_char = ('.', '#');
|
||||
let mut last_char = (']', ']');
|
||||
let mut spin = ['/', '-', '\\', '/', '-', '\\'];
|
||||
|
||||
while let Some(next) = args.next() {
|
||||
if next.starts_with("--") {
|
||||
if next == "--fira" {
|
||||
first_char = ('\u{ee00}', '\u{ee03}');
|
||||
middle_char = ('\u{ee01}', '\u{ee04}');
|
||||
mid_char = ('\u{ee01}', '\u{ee04}');
|
||||
last_char = ('\u{ee02}', '\u{ee05}');
|
||||
spin = ['\u{ee06}', '\u{ee07}', '\u{ee08}', '\u{ee09}', '\u{ee0a}', '\u{ee0b}'];
|
||||
} else if next == "--max" {
|
||||
if let Some(_) = max { return Err("can only provide 1 max".to_string()) }
|
||||
let next = match args.next() {
|
||||
|
@ -99,17 +175,8 @@ impl Config {
|
|||
}
|
||||
}
|
||||
|
||||
let val = match val {
|
||||
Some(v) => v,
|
||||
None => return Err("must provide a value".to_string())
|
||||
};
|
||||
|
||||
let max = max.unwrap_or(DEFAULT_MAX);
|
||||
|
||||
if val > max {
|
||||
return Err("value cannot be > max".to_string());
|
||||
}
|
||||
|
||||
let width = width.unwrap_or_else(|| termsize::get().map_or(DEFAULT_WIDTH, |v| v.cols.into()));
|
||||
|
||||
if width < 11 {
|
||||
|
@ -117,12 +184,8 @@ impl Config {
|
|||
}
|
||||
|
||||
Ok(Config {
|
||||
val,
|
||||
max,
|
||||
first_char,
|
||||
middle_char,
|
||||
last_char,
|
||||
width
|
||||
val, max, width,
|
||||
chars: Chars::new(first_char, mid_char, last_char, spin, '✓'),
|
||||
})
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue