tokf 0.1.4

Config-driven CLI tool that compresses command output before it reaches an LLM context
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

use std::collections::HashMap;

use crate::config::types::MatchOutputRule;

use super::section::SectionMap;
use super::template;

/// Find the first `match_output` rule whose `contains` substring appears
/// in the combined output. Returns the matching rule, or `None`.
pub fn find_matching_rule<'a>(
    rules: &'a [MatchOutputRule],
    combined: &str,
) -> Option<&'a MatchOutputRule> {
    rules.iter().find(|rule| combined.contains(&rule.contains))
}

/// Render a `match_output` rule's output template, resolving `{line_containing}`
/// to the first line that contains the matched substring, and `{output}` to the
/// full combined output.
pub fn render_output(output_tmpl: &str, contains: &str, combined: &str) -> String {
    let mut vars = HashMap::new();
    if let Some(line) = combined.lines().find(|l| l.contains(contains)) {
        vars.insert("line_containing".to_string(), line.to_string());
    }
    vars.insert("output".to_string(), combined.to_string());
    template::render_template(output_tmpl, &vars, &SectionMap::new())
}

#[cfg(test)]
#[allow(clippy::unwrap_used, clippy::expect_used)]
mod tests {
    use super::*;

    // --- find_matching_rule ---

    #[test]
    fn first_match_wins() {
        let rules = vec![
            MatchOutputRule {
                contains: "up-to-date".to_string(),
                output: "ok (up-to-date)".to_string(),
            },
            MatchOutputRule {
                contains: "rejected".to_string(),
                output: "rejected!".to_string(),
            },
        ];
        let matched = find_matching_rule(&rules, "Everything up-to-date");
        assert_eq!(matched.unwrap().output, "ok (up-to-date)");
    }

    #[test]
    fn no_match_returns_none() {
        let rules = vec![MatchOutputRule {
            contains: "NOMATCH".to_string(),
            output: "nope".to_string(),
        }];
        assert!(find_matching_rule(&rules, "some output").is_none());
    }

    #[test]
    fn empty_rules() {
        assert!(find_matching_rule(&[], "anything").is_none());
    }

    #[test]
    fn case_sensitive() {
        let rules = vec![MatchOutputRule {
            contains: "Fatal".to_string(),
            output: "found".to_string(),
        }];
        assert!(find_matching_rule(&rules, "fatal: error").is_none());
        assert!(find_matching_rule(&rules, "Fatal: error").is_some());
    }

    // --- render_output ---

    #[test]
    fn resolves_line_containing() {
        let output = render_output(
            "\u{2717} {line_containing}",
            "fatal:",
            "some preamble\nfatal: bad revision\nmore stuff",
        );
        assert_eq!(output, "\u{2717} fatal: bad revision");
    }

    #[test]
    fn resolves_output_var() {
        let output = render_output("matched: {output}", "keyword", "line with keyword");
        assert_eq!(output, "matched: line with keyword");
    }

    #[test]
    fn plain_string_passthrough() {
        let output = render_output("ok (up-to-date)", "up-to-date", "Everything up-to-date");
        assert_eq!(output, "ok (up-to-date)");
    }

    #[test]
    fn no_matching_line_empty_var() {
        let output = render_output("\u{2717} {line_containing}", "fatal:", "no match here");
        // "fatal:" not found in any line → {line_containing} resolves to ""
        assert_eq!(output, "\u{2717} ");
    }
}