From 8e3d4596d783b93d61bea6c9bdf2e1ebb7421a32 Mon Sep 17 00:00:00 2001 From: Dan Ballard Date: Fri, 11 Sep 2020 16:19:52 -0700 Subject: [PATCH] minigrep from rust book ch 12 --- .gitignore | 1 + poem.txt | 9 ++++++ src/lib.rs | 93 +++++++++++++++++++++++++++++++++++++++++++++++++++++ src/main.rs | 18 +++++++++++ 4 files changed, 121 insertions(+) create mode 100644 .gitignore create mode 100644 poem.txt create mode 100644 src/lib.rs create mode 100644 src/main.rs diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ea8c4bf --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +/target diff --git a/poem.txt b/poem.txt new file mode 100644 index 0000000..b0fc12f --- /dev/null +++ b/poem.txt @@ -0,0 +1,9 @@ +I’m nobody! Who are you? +Are you nobody, too? +Then there’s a pair of us - don’t tell! +They’d banish us, you know. + +How dreary to be somebody! +How public, like a frog +To tell your name the livelong day +To an admiring bog! \ No newline at end of file diff --git a/src/lib.rs b/src/lib.rs new file mode 100644 index 0000000..40d13ee --- /dev/null +++ b/src/lib.rs @@ -0,0 +1,93 @@ +use std::fs; +use std::env; +use std::error::Error; + +pub struct Config { + pub query: String, + pub filename: String, + pub case_sensitive: bool, +} + +impl Config { + pub fn new(args: &[String]) -> Result { + if args.len() < 3 { + return Err("not enough arguments"); + } + let query = args[1].clone(); + let filename = args[2].clone(); + + let case_sensitive = env::var("CASE_INSENSITIVE").is_err(); + + Ok(Config {query, filename, case_sensitive}) + } +} + +pub fn run(config: Config) -> Result<(), Box> { + 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(()) +} + +pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> { + let mut results = Vec::new(); + + for line in contents.lines() { + if line.contains(query) { + results.push(line); + } + } + results +} + +pub fn search_case_insensitive<'a>(query: &str, contents: &'a str) -> Vec<&'a str> { + let mut results = Vec::new(); + let query = query.to_lowercase(); + + 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(){ + let query = "duct"; + let contents = "\ +Rust: +safe, fast, productive. +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) + ); + } +} diff --git a/src/main.rs b/src/main.rs new file mode 100644 index 0000000..c63eaa2 --- /dev/null +++ b/src/main.rs @@ -0,0 +1,18 @@ +use std::env; +use std::process; + +use minigrep::Config; + +fn main() { + let args: Vec = env::args().collect(); + + let config = Config::new(&args).unwrap_or_else(|err| { + eprintln!("Problem parsing arguments: {}", err); + process::exit(1); + }); + + if let Err(e) = minigrep::run(config) { + eprintln!("Application error: {}", e); + process::exit(1) + } +} \ No newline at end of file