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 clap::{Arg, App};
struct Options {
filename: String,
length: usize,
long_fortunes: bool
long_fortunes: bool,
wait: bool
}
fn main() {
@ -23,6 +25,7 @@ fn main() {
.value_name("file"))
.arg(Arg::with_name("length")
.short("n")
.long("length")
.default_value("160")
.help("Set the longest fortune length (in characters) considered to be 'short'. All fortunes longer than this are considered 'long'.")
.takes_value(true)
@ -31,7 +34,11 @@ fn main() {
.short("l")
.long("long")
.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);
// get fortunes
@ -47,6 +54,10 @@ fn main() {
// get a random one
let the_fortune = get_random_fortune(filtered);
println!("{}", the_fortune);
if options.wait {
wait(the_fortune)
}
}
fn filter_long(x: &str, max_length: usize) -> bool {
@ -62,7 +73,8 @@ fn parse_options(app: App) -> Options {
let options: Options = Options {
filename: matches.value_of("file").unwrap().to_owned(),
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;
@ -81,3 +93,17 @@ fn get_random_fortune(fortunes: Vec<String>) -> String {
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);
}