leindex 1.6.0

LeIndex MCP and semantic code search engine for AI tools and large codebases
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

use serde::{Deserialize, Serialize};

/// Output formatting mode for phase reports.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq, Default)]
pub enum FormatMode {
    /// Shortest output for tight token budgets.
    Ultra,
    /// Balanced detail and compactness.
    #[default]
    Balanced,
    /// Most detailed output.
    Verbose,
}

impl FormatMode {
    /// Parse mode from CLI/MCP string.
    pub fn parse(value: &str) -> Option<Self> {
        match value.to_ascii_lowercase().as_str() {
            "ultra" => Some(Self::Ultra),
            "balanced" => Some(Self::Balanced),
            "verbose" => Some(Self::Verbose),
            _ => None,
        }
    }

    /// Suggested max chars for each mode.
    pub fn default_max_chars(self) -> usize {
        match self {
            Self::Ultra => 4_000,
            Self::Balanced => 12_000,
            Self::Verbose => 24_000,
        }
    }
}

/// Token-aware formatter utilities.
pub struct TokenFormatter;

impl TokenFormatter {
    /// Truncate a string to a max character count while preserving UTF-8 boundaries.
    pub fn truncate(input: &str, max_chars: usize) -> String {
        if input.chars().count() <= max_chars {
            return input.to_string();
        }

        let mut out = String::new();
        for (i, ch) in input.chars().enumerate() {
            if i >= max_chars {
                break;
            }
            out.push(ch);
        }
        out.push_str("\n\n…[truncated]");
        out
    }
}

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

    #[test]
    fn truncate_keeps_short_strings() {
        assert_eq!(TokenFormatter::truncate("abc", 10), "abc");
    }

    #[test]
    fn truncate_limits_output() {
        let value = TokenFormatter::truncate("1234567890", 4);
        assert!(value.starts_with("1234"));
        assert!(value.contains("truncated"));
    }

    #[test]
    fn format_mode_parse_and_default_char_targets() {
        assert_eq!(FormatMode::parse("ultra"), Some(FormatMode::Ultra));
        assert_eq!(FormatMode::parse("balanced"), Some(FormatMode::Balanced));
        assert_eq!(FormatMode::parse("verbose"), Some(FormatMode::Verbose));
        assert_eq!(FormatMode::parse("invalid"), None);

        assert_eq!(FormatMode::Ultra.default_max_chars(), 4_000);
        assert_eq!(FormatMode::Balanced.default_max_chars(), 12_000);
        assert_eq!(FormatMode::Verbose.default_max_chars(), 24_000);
    }
}