repgrep 0.16.1

An interactive command line replacer for `ripgrep`.
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

use std::io::{self, BufRead, BufReader, Read, Write};

use anyhow::{anyhow, Result};

use crate::rg::de::RgMessage;

pub fn read_messages<R: Read>(rdr: R) -> Result<Vec<RgMessage>> {
    let mut saw_match_message = false;

    let mut rg_messages: Vec<RgMessage> = vec![];
    let reader = BufReader::new(rdr);
    for (i, line) in reader.lines().enumerate() {
        // For large result lists show some progress in the terminal.
        if i > 0 && i % 1000 == 0 {
            let _ = io::stdout().write_all(format!("\rMatches found: ~{}", i).as_bytes());
            let _ = io::stdout().flush();
        }

        let rg_msg: RgMessage =
            serde_json::from_str(&line?).map_err(|e| anyhow!("Failed to parse JSON: {}", e))?;

        if !saw_match_message && matches!(rg_msg, RgMessage::Match { .. }) {
            saw_match_message = true;
        }

        rg_messages.push(rg_msg);
    }

    // We expect at least one message.
    if !saw_match_message {
        Err(anyhow!("No matches returned from rg!"))
    } else {
        Ok(rg_messages)
    }
}