jtool-grep 0.2.1

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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

//! CSV/TSV output formatter

use super::OutputFormatter;
use crate::types::GrepResult;
use anyhow::Result;

/// CSV output formatter
pub struct CsvFormatter {
    delimiter: char,
}

impl CsvFormatter {
    pub fn new() -> Self {
        Self { delimiter: ',' }
    }

    pub fn with_delimiter(delimiter: char) -> Self {
        Self { delimiter }
    }
}

impl Default for CsvFormatter {
    fn default() -> Self {
        Self::new()
    }
}

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

        // Header (only for single result)
        output.push_str(&format!(
            "notebook{}cell_index{}cell_number{}execution_count{}match_type{}line_index{}line_number{}matched_text{}line_content\n",
            self.delimiter, self.delimiter, self.delimiter, self.delimiter,
            self.delimiter, self.delimiter, self.delimiter, self.delimiter
        ));

        // Data rows
        for m in &result.matches {
            let exec_count = m.execution_count.map(|c| c.to_string()).unwrap_or_default();

            output.push_str(&format!(
                "\"{}\"{}{}{}{}{}\"{}\"{}\"{}\"{}{}{}{}{}\"{}\"{}\"{}\"",
                escape_csv(&result.notebook),
                self.delimiter,
                m.cell_index,
                self.delimiter,
                m.cell_number,
                self.delimiter,
                exec_count,
                self.delimiter,
                m.match_type,
                self.delimiter,
                m.line_index,
                self.delimiter,
                m.line_number,
                self.delimiter,
                escape_csv(&m.matched_text),
                self.delimiter,
                escape_csv(&m.line_content)
            ));
            output.push('\n');
        }

        Ok(output)
    }

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

        // Header
        output.push_str(&format!(
            "notebook{}cell_index{}cell_number{}execution_count{}match_type{}line_index{}line_number{}matched_text{}line_content\n",
            self.delimiter, self.delimiter, self.delimiter, self.delimiter,
            self.delimiter, self.delimiter, self.delimiter, self.delimiter
        ));

        // Data rows from all results
        for result in results {
            for m in &result.matches {
                let exec_count = m.execution_count.map(|c| c.to_string()).unwrap_or_default();

                output.push_str(&format!(
                    "\"{}\"{}{}{}{}{}\"{}\"{}\"{}\"{}{}{}{}{}\"{}\"{}\"{}\"",
                    escape_csv(&result.notebook),
                    self.delimiter,
                    m.cell_index,
                    self.delimiter,
                    m.cell_number,
                    self.delimiter,
                    exec_count,
                    self.delimiter,
                    m.match_type,
                    self.delimiter,
                    m.line_index,
                    self.delimiter,
                    m.line_number,
                    self.delimiter,
                    escape_csv(&m.matched_text),
                    self.delimiter,
                    escape_csv(&m.line_content)
                ));
                output.push('\n');
            }
        }

        Ok(output)
    }
}

/// Escape CSV field
fn escape_csv(s: &str) -> String {
    s.replace('"', "\"\"")
}