daan/fortune-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add the -w flag

Browse source
This commit is contained in:
runebaas 2019-10-28 00:19:43 +01:00
parent 3a704e7860
commit 84b8328d31
No known key found for this signature in database
GPG key ID: 2677AF508D0300D6
1 changed files with 30 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
src/main.rs
View file

@ -1,11 +1,13 @@
use std::fs; use std::{fs, thread, time};
use std::convert::TryFrom;
use rand::prelude::*; use rand::prelude::*;
use clap::{Arg, App}; use clap::{Arg, App};
struct Options { struct Options {
filename: String, filename: String,
length: usize, length: usize,
long_fortunes: bool long_fortunes: bool,
wait: bool
} }
fn main() { fn main() {
@ -23,6 +25,7 @@ fn main() {
.value_name("file")) .value_name("file"))
.arg(Arg::with_name("length") .arg(Arg::with_name("length")
.short("n") .short("n")
.long("length")
.default_value("160") .default_value("160")
.help("Set the longest fortune length (in characters) considered to be 'short'. All fortunes longer than this are considered 'long'.") .help("Set the longest fortune length (in characters) considered to be 'short'. All fortunes longer than this are considered 'long'.")
.takes_value(true) .takes_value(true)
@ -31,7 +34,11 @@ fn main() {
.short("l") .short("l")
.long("long") .long("long")
.help("only return long fortunes") .help("only return long fortunes")
.takes_value(false)); .takes_value(false))
.arg(Arg::with_name("wait")
.short("w")
.long("wait")
.help("Wait before termination for an amount of time calculated from the number of characters in the message. (20 characters = 1 second, mim 6 seconds)"));
let options = parse_options(app); let options = parse_options(app);
// get fortunes // get fortunes
@ -47,6 +54,10 @@ fn main() {
// get a random one // get a random one
let the_fortune = get_random_fortune(filtered); let the_fortune = get_random_fortune(filtered);
println!("{}", the_fortune); println!("{}", the_fortune);
if options.wait {
wait(the_fortune)
}
} }
fn filter_long(x: &str, max_length: usize) -> bool { fn filter_long(x: &str, max_length: usize) -> bool {
@ -62,7 +73,8 @@ fn parse_options(app: App) -> Options {
let options: Options = Options { let options: Options = Options {
filename: matches.value_of("file").unwrap().to_owned(), filename: matches.value_of("file").unwrap().to_owned(),
length: matches.value_of("length").unwrap().parse::<usize>().expect("Length is not a valid number"), length: matches.value_of("length").unwrap().parse::<usize>().expect("Length is not a valid number"),
long_fortunes: matches.is_present("long") long_fortunes: matches.is_present("long"),
wait: matches.is_present("wait")
}; };
return options; return options;
@ -81,3 +93,17 @@ fn get_random_fortune(fortunes: Vec<String>) -> String {
return fortunes.get(random_fortune).unwrap().trim().to_owned(); return fortunes.get(random_fortune).unwrap().trim().to_owned();
} }
fn wait(fortune: String) {
let characters_per_second = 20; // as defined in the original fortune command
let minimum_wait = 6; // seconds
let mut time_to_wait = fortune.len() / characters_per_second;
if time_to_wait < minimum_wait {
time_to_wait = minimum_wait;
}
let fortune_length = u64::try_from(time_to_wait).unwrap();
let time_to_wait = time::Duration::from_secs(fortune_length);
thread::sleep(time_to_wait);
}