batless 0.5.0

A non-blocking, LLM-friendly code viewer inspired by bat
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

//! JSON output formatting

use crate::config::BatlessConfig;
use crate::error::BatlessResult;
use crate::file_info::FileInfo;
use crate::formatter::OutputMode;
use crate::formatters::Formatter;
use serde_json::json;

/// JSON formatter for structured output
pub struct JsonFormatter;

impl Formatter for JsonFormatter {
    fn format(
        &self,
        file_info: &FileInfo,
        file_path: &str,
        config: &BatlessConfig,
    ) -> BatlessResult<String> {
        let lines_value: serde_json::Value = if config.json_line_numbers {
            file_info
                .lines
                .iter()
                .enumerate()
                .map(|(i, text)| json!({"n": i + 1, "text": text}))
                .collect()
        } else {
            json!(file_info.lines)
        };

        let json_output = json!({
            "file": file_path,
            "language": file_info.language,
            "lines": lines_value,
            "total_lines": file_info.total_lines,
            "total_lines_exact": file_info.total_lines_exact,
            "total_bytes": file_info.total_bytes,
            "encoding": file_info.encoding,
            "truncated": file_info.truncated,
            "truncated_by_lines": file_info.truncated_by_lines,
            "truncated_by_bytes": file_info.truncated_by_bytes,
            "truncated_by_context": file_info.truncated_by_context,
            "summary_lines": file_info.summary_lines,
            "syntax_errors": file_info.syntax_errors,
            "identifiers": if config.include_tokens {
                file_info.tokens.clone()
            } else {
                None
            },
            "identifier_total": file_info.token_total,
            "file_hash": file_info.file_hash,
            "estimated_llm_tokens": file_info.estimated_llm_tokens,
            "token_model": file_info.token_model,
            "compression_ratio": file_info.compression_ratio,
            "mode": "json"
        });

        Ok(serde_json::to_string_pretty(&json_output)?)
    }

    fn output_mode(&self) -> OutputMode {
        OutputMode::Json
    }
}

/// Compact JSON formatter for minimal output
pub struct CompactJsonFormatter;

impl Formatter for CompactJsonFormatter {
    fn format(
        &self,
        file_info: &FileInfo,
        file_path: &str,
        _config: &BatlessConfig,
    ) -> BatlessResult<String> {
        let json_output = json!({
            "file": file_path,
            "language": file_info.language,
            "total_lines": file_info.total_lines,
            "total_bytes": file_info.total_bytes,
            "encoding": file_info.encoding,
            "truncated": file_info.truncated,
        });

        Ok(serde_json::to_string(&json_output)?)
    }

    fn output_mode(&self) -> OutputMode {
        OutputMode::Json
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_json_formatter_mode() {
        let formatter = JsonFormatter;
        assert!(matches!(formatter.output_mode(), OutputMode::Json));
    }

    #[test]
    fn test_compact_formatter() {
        let formatter = CompactJsonFormatter;
        assert!(matches!(formatter.output_mode(), OutputMode::Json));
    }
}