diff options
author | j-james | 2022-11-15 08:00:02 +0000 |
---|---|---|
committer | j-james | 2022-11-15 08:01:55 +0000 |
commit | ef73dc1db2f0389912e442b0f6cd5fa6712d4026 (patch) | |
tree | 44d5673de3be8e6caf879914498572ea83f23cfa /minigrep/src | |
parent | 06615702d85d5640984ab0e6a7b3155df4d261a3 (diff) |
Add the I/O project from the Rust book
Diffstat (limited to 'minigrep/src')
-rw-r--r-- | minigrep/src/lib.rs | 107 | ||||
-rw-r--r-- | minigrep/src/main.rs | 23 |
2 files changed, 130 insertions, 0 deletions
diff --git a/minigrep/src/lib.rs b/minigrep/src/lib.rs new file mode 100644 index 0000000..8196b1f --- /dev/null +++ b/minigrep/src/lib.rs @@ -0,0 +1,107 @@ +use std::env; +use std::fs; +use std::error::Error; + +pub struct Config { + pub query: String, + pub filename: String, + pub case_sensitive: bool, +} + +// doesn't need to be pub? +impl Config { + pub fn new(args: &[String]) -> Result<Config, &'static str> { + 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(); + + return 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); + } + + return Ok(()); +} + +// why do we need lifetime annotations here, and why are they not on query +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); + } + } + return results; +} + +pub 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); + } + } + + return results; +} + +// ??? +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn case_sensitive() { + 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) + ); + } + + #[test] + fn no_result() { + let query = "asdfgh"; + let contents = "\ +Rust: +safe, fast, productive. +Pick three."; + assert_eq!(Vec::<&str>::new(), search(query, contents)); + + } +} diff --git a/minigrep/src/main.rs b/minigrep/src/main.rs new file mode 100644 index 0000000..5066347 --- /dev/null +++ b/minigrep/src/main.rs @@ -0,0 +1,23 @@ +use std::env; +use std::process; + +use minigrep::Config; + +fn main() { + let args: Vec<String> = env::args().collect(); + // & is a reference, so we don't worry abt the borrow checker until args goes out of scope + let config = Config::new(&args).unwrap_or_else(|err| { + eprintln!("Problem parsing arguments: {}", err); + process::exit(1); + }); + + // println!("Searching for {}", config.query); + // println!("In file {}", config.filename); + + // passing a variable without an ampersand moves it + if let Err(e) = minigrep::run(config) { + // this syntax is a little bit weird - you're almost implicitly running run() + eprintln!("Application error: {}", e); + process::exit(1); + } +} |