cargo_zm/
lib.rs

1use std::{fs, error::Error, env};
2
3pub fn search_case_insensitive<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
4    let mut results = Vec::new();
5    let query = query.to_lowercase();
6
7    for line in contents.lines(){
8        if line.to_lowercase().contains(&query){
9            results.push(line);
10        }
11    }
12    results
13}
14
15pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
16    contents.lines().filter(|line| line.contains(query)).collect()
17}
18
19pub fn run(config: Config) -> Result<(), Box<dyn Error>>{
20    let contents = fs::read_to_string(config.file_path)?;
21    let results = if config.ignore_case{
22        search_case_insensitive(&config.query, &contents)
23    }else {
24        search(&config.query, &contents)
25    };
26    for line in results{
27        println!("{}", line);
28    }
29    Ok(())
30}
31
32pub struct Config {
33    pub query: String,
34    pub file_path: String,
35    pub ignore_case: bool,
36}
37impl Config {
38    pub fn build(mut args: impl Iterator<Item = String>,)
39    -> Result<Config, &'static str>{
40        args.next();  // 第一个值是程序的名称用不到
41
42        let query = match args.next() {
43            Some(arg) => arg,
44            None => return Err("Didn't get a query string"),
45        };
46        let file_path = match args.next(){
47            Some(arg) => arg,
48            None => return Err("Didn't get a file path"),
49        };
50        let ignore_case = env::var("IGNORE_CASE").is_ok();
51        Ok(Config {query, file_path, ignore_case})
52    }
53}
54