jtool-grep 0.3.0

notebook-specific grep tool for jtool
Documentation 
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
57
58
59
60

//! Template-based output formatter

use super::OutputFormatter;
use crate::types::{GrepResult, Match};
use anyhow::Result;

/// Template-based output formatter
pub struct TemplateFormatter {
    template: String,
}

impl TemplateFormatter {
    pub fn new(template: String) -> Self {
        Self { template }
    }

    /// Format a single match using the template
    fn format_match(&self, notebook: &str, m: &Match) -> String {
        let exec_str = if let Some(count) = m.execution_count {
            count.to_string()
        } else {
            String::new()
        };

        // Replace placeholders in the template
        self.template
            .replace("{notebook}", notebook)
            .replace("{cell}", &m.cell_index.to_string())
            .replace("{cell_num}", &m.cell_number.to_string())
            .replace("{exec}", &exec_str)
            .replace("{type}", &m.match_type.to_string())
            .replace("{line}", &m.line_index.to_string())
            .replace("{line_num}", &m.line_number.to_string())
            .replace("{match}", &m.matched_text)
            .replace("{full_line}", m.line_content.trim())
    }
}

impl OutputFormatter for TemplateFormatter {
    fn format_result(&self, result: &GrepResult) -> Result<String> {
        let mut output = String::new();

        for m in &result.matches {
            output.push_str(&self.format_match(&result.notebook, m));
            output.push('\n');
        }

        Ok(output)
    }

    fn format_results(&self, results: &[GrepResult]) -> Result<String> {
        let mut output = String::new();

        for result in results {
            output.push_str(&self.format_result(result)?);
        }

        Ok(output)
    }
}