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
extern crate regex;
use std::fs;
use std::error::Error;
pub struct Config {
pub query: String,
pub filename: String,
}
impl Config {
pub fn new(args: &[String]) -> Result<Config, &'static str> {
if args.len() < 3 { return Err("Not enough arguments"); }
let query: String = args[1].clone().to_string();
let filename: String = args[2].clone().to_string();
Ok( Config { query: query, filename: filename } )
}
}
pub fn run(config: Config) -> Result<(), Box<dyn Error>> {
let contents: String = fs::read_to_string(config.filename)?;
let mut matching = Vec::new();
matching = search_regex(&config.query, &contents)?;
if matching.len() == 0 {
println!("[ WARNING ] No occurrences found");
return Ok(());
}
println!("-----------------START---------------");
for line in matching {
println!("{}", line);
}
println!("-----------------END-----------------");
Ok(())
}
fn search_regex(query: &str, contents: &str) -> Result<Vec<String>, Box<dyn Error>> {
let mut results: Vec<String> = Vec::new();
let re = regex::Regex::new(query)?;
let mut current_line: i32 = 0;
for line in contents.lines() {
current_line += 1;
let match_item = re.find(line);
if match_item.is_some() {
let unwrapped_result = match_item.unwrap();
let data: String = format!("{}::{}:{} -> {}", current_line, unwrapped_result.start(), unwrapped_result.end(), &line[unwrapped_result.start()..unwrapped_result.end()]);
results.push(data);
}
}
Ok(results)
}