diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/lib.rs | 8 | ||||
| -rw-r--r-- | src/main.rs | 6 |
2 files changed, 11 insertions, 3 deletions
@@ -5,9 +5,13 @@ pub fn search<'a>(pattern: &str, contents: &'a str) -> Vec<&'a str> { .collect() } -pub fn contains_pattern(src: &str, pat: &str, ignore_case: bool) -> bool { +pub fn contains_pattern(src: &str, pat: &str, ignore_case: bool, word_regexp: bool) -> bool { let src = if ignore_case { src.to_lowercase() } else { src.to_owned() }; let pat = if ignore_case { pat.to_lowercase() } else { pat.to_owned() }; - src.contains(&pat) + if word_regexp { + src.split_whitespace().any(|p| p == pat) + } else { + src.contains(&pat) + } } diff --git a/src/main.rs b/src/main.rs index 3111766..33de082 100644 --- a/src/main.rs +++ b/src/main.rs @@ -18,6 +18,10 @@ struct Config { /// Ignores the case of the search string #[arg(short, long)] ignore_case: bool, + + /// Matches only whole words + #[arg(short, long)] + word_regexp: bool, } fn main() { @@ -48,7 +52,7 @@ fn run(config: Config) -> Result<(), Box<dyn Error>> { }; for line in reader.lines().map(|l| l.unwrap()) { - if !trgrep::contains_pattern(&line, &config.pattern, config.ignore_case) { + if !trgrep::contains_pattern(&line, &config.pattern, config.ignore_case, config.word_regexp) { continue; } println!("{line}"); |