summaryrefslogtreecommitdiff
path: root/minigrep/src
diff options
context:
space:
mode:
authorAdam Carpenter <53hornet@gmail.com>2019-02-27 19:23:56 -0500
committerAdam Carpenter <53hornet@gmail.com>2019-02-27 19:23:56 -0500
commitb43fca42427dbd276ca2dd1d712611b6dea4338c (patch)
treec9a747e6e99d74c39a857e420738652bbf96079d /minigrep/src
parent3402ea2a8c91e187afdb5e8ca085d779f5877674 (diff)
downloadlearning-rust-b43fca42427dbd276ca2dd1d712611b6dea4338c.tar.xz
learning-rust-b43fca42427dbd276ca2dd1d712611b6dea4338c.zip
Added environment variable support for case sensitivity.
Diffstat (limited to 'minigrep/src')
-rw-r--r--minigrep/src/lib.rs58
1 files changed, 55 insertions, 3 deletions
diff --git a/minigrep/src/lib.rs b/minigrep/src/lib.rs
index b55e924..9f0adbd 100644
--- a/minigrep/src/lib.rs
+++ b/minigrep/src/lib.rs
@@ -1,9 +1,11 @@
+use std::env;
use std::error::Error;
use std::fs;
pub struct Config {
pub query: String,
pub filename: String,
+ pub case_sensitive: bool,
}
impl Config {
@@ -16,13 +18,34 @@ impl Config {
let query = args[1].clone();
let filename = args[2].clone();
- Ok(Config { query, filename })
+ let mut case_sensitive = env::var("CASE_INSENSITIVE").is_err();
+
+ if args.nth(3) == "-i" {
+ case_sensitive = false;
+ }
+ else if args[3] == "-s" {
+ case_sensitive = true;
+ }
+
+ Ok(Config { query, filename, case_sensitive })
}
}
pub fn run(config: Config) -> Result<(), Box<dyn Error>> {
let contents = fs::read_to_string(config.filename)?;
+
+ let results = if config.case_sensitive {
+ search(&config.query, &contents)
+ }
+ else {
+ search_case_insensitive(&config.query, &contents)
+ };
+
+ for line in results {
+ println!("{}", line);
+ }
+
Ok(())
}
@@ -38,21 +61,50 @@ fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
results
}
+fn search_case_insensitive<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
+ let query = query.to_lowercase();
+ let mut results = Vec::new();
+
+ for line in contents.lines() {
+ if line.to_lowercase().contains(&query) {
+ results.push(line);
+ }
+ }
+
+ results
+}
+
#[cfg(test)]
mod tests {
use super::*;
#[test]
- fn one_result() {
+ fn case_sensitive() {
let query = "duct";
let contents = "\
Rust:
safe, fast, productive.
-Pick three.";
+Pick three.
+Duct tape.";
assert_eq!(
vec!["safe, fast, productive."],
search(query, contents)
);
}
+
+ #[test]
+ fn case_insensitive() {
+ let query = "rUsT";
+ let contents = "\
+Rust:
+safe, fast, productive.
+Pick three.
+Trust me.";
+
+ assert_eq!(
+ vec!["Rust:", "Trust me."],
+ search_case_insensitive(query, contents)
+ );
+ }
}