au.rs

extern crate rayon;
extern crate regex;
extern crate scan_dir;

use rayon::prelude::*;
use regex::Regex;
use scan_dir::ScanDir;

use std::env;
use std::fs::File;
use std::io::{BufReader, BufRead};

fn main() {
    let query = env::args().skip(1).next().unwrap().to_owned();
    let regex = Regex::new(&*query).unwrap();

    let files: Vec<_> = ScanDir::files()
        .walk("src", |iter| iter.map(|(entry, _)| entry.path()).collect())
        .unwrap();

    files
        .par_iter()
        .map(File::open)
        .map(Result::unwrap)
        .map(BufReader::new)
        .map(BufRead::lines)
        .map(|lines| {
            lines
                .flat_map(Result::ok)
                .filter(|line| regex.is_match(line))
                .collect::<Vec<_>>()
        })
        .for_each(|file| println!("{:?}", file));
}