1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156
use std::error::Error; use std::fs; use std::env; 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(()) } pub struct Config { pub query: String, pub filename: String, pub case_sensitive: bool, } impl Config { pub fn new(mut args: std::env::Args) -> Result<Config, &'static str> { args.next(); let query = match args.next() { Some(arg) => arg, None => return Err("Didn't get a query string"), }; let filename = match args.next(){ Some(arg) => arg, None => return Err("Didn't get a query string"), }; let case_sensitive = match args.next(){ Some(arg) => { if arg == "i" { false }else { env::var("CASE_INSENSITIVE").is_err() } }, None => env::var("CASE_INSENSITIVE").is_err(), }; Ok(Config {query,filename,case_sensitive}) } } pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> { contents.lines() .filter(|line| line.contains(query)) .collect() } pub fn search_case_insensitive<'a>(query: &str, contents: &'a str) -> Vec<&'a str> { contents.lines() .filter(|line| line.to_lowercase().contains(&query.to_lowercase())) .collect() } #[cfg(test)] mod tests { use super::*; use std::process; #[test] fn create_config_works() { let args = vec![String::from("system"),String::from("apple"),String::from("poem.txt")]; let sample_config = Config::new(&args).unwrap_or_else(|err| { println!("Problem parsing arguments: {}", err); process::exit(1); }); assert_eq!("apple",sample_config.query); } #[test] fn run_works() { let args = vec![String::from("system"),String::from("apple"),String::from("poem.txt")]; let sample_config = Config::new(&args).unwrap_or_else(|err| { println!("Problem parsing arguments: {}", err); process::exit(1); }); let mut file_was_read = true; let mut specific_error = String::from("none"); if let Err(e) = run(sample_config) { println!("**** error reading file: {}",e); file_was_read = false; specific_error = String::from(e.to_string()); } assert!(file_was_read,specific_error); } #[test] fn case_sensitive() { let query = "duct"; let contents = "\ Rust:\n safe, fast, productive.\n Pick three.\n Duct tape."; assert_eq!(vec!["safe, fast, productive."], search(query, contents)); } #[test] fn case_insensitive() { let query = "rUsT"; let contents = "\ Rust:\n safe, fast, productive.\n Pick three.\n Trust me."; assert_eq!(vec!["Rust:", "Trust me."], search_case_insensitive(query, contents) ); } }