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

Minor improvements to error handling and trim the output

Browse source
This commit is contained in:
runebaas 2019-10-27 12:12:03 +01:00
parent 4e9d1c91ed
commit d2d28b8011
No known key found for this signature in database
GPG key ID: 2677AF508D0300D6
1 changed files with 13 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/main.rs
View file

@ -25,17 +25,25 @@ fn main() {
// get fortunes
let filename = matches.value_of("file").unwrap().to_owned();
let fortune_file = fs::read_to_string(filename).unwrap();
let fortune_file = fs::read_to_string(filename).expect("Cannot read fortune file");
let fortunes: Vec<&str> = fortune_file.split('%').collect();
// filter by max length
let max_length = matches.value_of("length").unwrap().parse::<usize>().unwrap();
let filtered = fortunes.into_iter().filter(|x| x.len() < max_length).collect::<Vec<&str>>();
let max_length = matches
.value_of("length")
.unwrap()
.parse::<usize>()
.expect("Length is not a valid number");
let filtered = fortunes
.into_iter()
.filter(|x| x.replace(" ", "").len() < max_length)
.collect::<Vec<&str>>();
// get a random one
let total_fortunes = filtered.len();
let random_fortune = rand::thread_rng().gen_range(0, total_fortunes);
// print
println!("{}", filtered.get(random_fortune).unwrap());
// print the fortune
let the_fortune = filtered.get(random_fortune).unwrap().trim();
println!("{}", the_fortune);
}