use std::error::Error; use std::process; use std::fs; use std::io::{self, BufReader, BufRead}; use clap::Parser; #[derive(Parser)] #[command(author, version, about, long_about = None)] // Read from `Cargo.toml` #[clap(disable_help_flag = true)] struct Config { /// The pattern to look for pattern: String, /// The path to files to read. A path of "-" stands for standard input. /// /// If no FILE is given, read standard input. files: Vec, /// Print a help text and exit #[arg(long, action = clap::ArgAction::Help)] help: Option, /// Ignores the case of the search string #[arg(short, long)] ignore_case: bool, /// Selects non-matching lines #[arg(short = 'v', long)] invert_match: bool, /// Matches only whole words #[arg(short, long)] word_regexp: bool, } fn main() { let mut config = Config::parse(); if config.files.is_empty() { config.files.push(String::from("-")); } if let Err(e) = run(config) { eprintln!("Error: {e}"); process::exit(1); } } fn run(config: Config) -> Result<(), Box> { for file in config.files { // On-Stack Dynamic Dispatch let (mut stdin_read, mut file_read); // We need to ascribe the type to get dynamic dispatch. let reader: &mut dyn BufRead = if file == "-" { stdin_read = BufReader::new(io::stdin()); &mut stdin_read } else { file_read = BufReader::new(fs::File::open(&file)?); &mut file_read }; for line in reader.lines().map(|l| l.unwrap()) { let match_flag = trgrep::contains_pattern(&line, &config.pattern, config.ignore_case, config.word_regexp); let match_flag = if config.invert_match { !match_flag } else { match_flag }; if !match_flag { continue; } println!("{line}"); } } Ok(()) }