summaryrefslogtreecommitdiff
path: root/minigrep/src
diff options
context:
space:
mode:
authorj-james2022-11-15 08:00:02 +0000
committerj-james2022-11-15 08:01:55 +0000
commitef73dc1db2f0389912e442b0f6cd5fa6712d4026 (patch)
tree44d5673de3be8e6caf879914498572ea83f23cfa /minigrep/src
parent06615702d85d5640984ab0e6a7b3155df4d261a3 (diff)
Add the I/O project from the Rust book
Diffstat (limited to 'minigrep/src')
-rw-r--r--minigrep/src/lib.rs107
-rw-r--r--minigrep/src/main.rs23
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);
+ }
+}