ripwc 0.1.1

Easy to use wc replacement
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

#[allow(unused_imports)]
use std::path::PathBuf;
use clap::Clap;
use args::Opts;
mod args;
use std::io::Read;
use std::fs::File;
use std::error::Error;

fn main() {
	let opts: Opts = Opts::parse();
	match get_input(&opts) {
		Ok(input) => {
			if opts.c {
				println!("Total bytes in file: {}", input.as_bytes().len())
			}
			if opts.l {
				println!("Total lines in file: {}", count_up(input.split("\n")));
			}
			if opts.w {
				println!("Total words in file: {}", count_up(input.split(" ")));
			}
			if opts.m {
				println!("Total chars in file: {}", count_up(input.chars()))
			}
		},
		Err(err) => {
			eprintln!("{}", err.to_string());
			std::process::exit(1);
		},
	}
}

fn count_up(input: impl Iterator) -> i32 {
	input.fold(0, |acc, _| acc + 1)
}

fn get_input(opts: &Opts) -> Result<String, Box<dyn Error>> {
	let mut input = String::new();

	if let Some(file) = &opts.file {
		File::open(&file)?.read_to_string(&mut input)?;
	} else {
		std::io::stdin().read_to_string(&mut input)?;
	}

	Ok(input)
}