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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
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<String>,
/// Print a help text and exit
#[arg(long, action = clap::ArgAction::Help)]
help: Option<bool>,
/// Ignores the case of the search string
#[arg(short, long)]
ignore_case: bool,
/// Selects non-matching lines
#[arg(short = 'v', long)]
invert_match: bool,
/// Displays only a count of the number of lines that match the search string
#[arg(short, long)]
count: bool,
/// Prefixes each matching line with the line number
#[arg(short = 'n', long)]
line_number: bool,
/// Matches only whole words
#[arg(short, long)]
word_regexp: bool,
/// Displays only the filenames of files that contain matches
#[arg(short = 'l', long)]
files_with_matches: bool,
/// Suppresses the display of filenames
#[arg(short = 'h', long)]
no_filename: bool,
}
fn main() {
let mut config = Config::parse();
match config.files.len() {
0 => {
config.files.push(String::from("-"));
config.no_filename = true;
}
1 => config.no_filename = true,
_ => (),
}
if let Err(e) = run(config) {
eprintln!("Error: {e}");
process::exit(1);
}
}
fn run(config: Config) -> Result<(), Box<dyn Error>> {
'outer: for file in config.files {
let mut count: usize = 0;
// 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 (idx, line) in reader.lines().map(|l| l.unwrap()).enumerate() {
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;
}
if config.files_with_matches {
println!("{file}");
continue 'outer;
} else if config.count {
count += 1;
continue;
}
let formatted_output = if !config.no_filename && config.line_number {
format!("{}:{}:{}", file, idx + 1, line)
} else if !config.no_filename {
format!("{}:{}", file, line)
} else if config.line_number {
format!("{}:{}", idx + 1, line)
} else {
format!("{}", line)
};
println!("{}", formatted_output);
}
if config.count {
println!("{}:{}", file, count);
}
}
Ok(())
}
|
