main.rs
rust
use minigrep::Config;
use std::{env, process};
fn main() {
let args: Vec<String> = env::args().collect();
let config = Config::new(&args).unwrap_or_else(|err| {
eprintln!("Problem parsing arguments: {}", err);
process::exit(1);
});
if let Err(err) = minigrep::run(config) {
eprintln!("Application err : {}", err);
process::exit(1);
}
}
lib.rs
rust
use std::{env, error::Error, fs};
pub struct Config {
pub query: String,
pub file_name: String,
pub case_sensitive: bool,
}
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 file_name = args[2].clone();
let case_sensitive = env::var("CASE_INSENSITIVE").is_err();
Ok(Config {
query,
file_name,
case_sensitive,
})
}
}
pub fn run(config: Config) -> Result<(), Box<dyn Error>> {
let content = fs::read_to_string(config.file_name)?;
if config.case_sensitive {
for line in search(&config.query, &content) {
println!("line = {}", line);
}
} else {
for line in search_case_insensitive(&config.query, &content) {
println!("line = {}", line);
}
}
Ok(())
}
fn search_case_insensitive<'a>(query: &str, content: &'a str) -> Vec<&'a str> {
let mut results = Vec::new();
let query = query.to_lowercase();
for line in content.lines() {
if line.to_lowercase().contains(&query) {
results.push(line);
}
}
results
}
fn search<'a>(query: &str, content: &'a str) -> Vec<&'a str> {
let mut result = Vec::new();
for line in content.lines() {
if line.contains(query) {
result.push(line);
}
}
result
}