ratkit 0.2.15

A comprehensive collection of reusable TUI components for ratatui including resizable splits, tree views, markdown rendering, toast notifications, dialogs, and terminal embedding
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

//! Language detection for source files.

use std::path::Path;

use crate::widgets::code_widget::foundation::CodeLanguage;

/// Detects a source language from override, file path, and shebang.
pub fn detect_language(
    path: Option<&Path>,
    content: &str,
    override_language: Option<&str>,
) -> CodeLanguage {
    if let Some(language) = override_language {
        return CodeLanguage::from_token(language);
    }

    if let Some(language) = detect_from_shebang(content) {
        return language;
    }

    path.and_then(|path| path.extension())
        .and_then(|extension| extension.to_str())
        .map(CodeLanguage::from_token)
        .unwrap_or(CodeLanguage::Unknown)
}

/// Detects a language from a Unix shebang.
fn detect_from_shebang(content: &str) -> Option<CodeLanguage> {
    let first_line = content.lines().next()?.trim();
    if !first_line.starts_with("#!") {
        return None;
    }

    let shebang = first_line.trim_start_matches("#!");
    for token in shebang.split(|character: char| character == '/' || character.is_whitespace()) {
        let language = CodeLanguage::from_token(token);
        if is_shebang_language(&language) {
            return Some(language);
        }
    }
    None
}

/// Returns true for languages that should be trusted from shebang tokens.
fn is_shebang_language(language: &CodeLanguage) -> bool {
    matches!(
        language,
        CodeLanguage::Python | CodeLanguage::JavaScript | CodeLanguage::Shell
    )
}

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

    #[test]
    fn detects_rust_from_extension() {
        let language = detect_language(Some(Path::new("src/main.rs")), "", None);
        assert_eq!(language, CodeLanguage::Rust);
    }

    #[test]
    fn detects_python_from_shebang() {
        let language = detect_language(None, "#!/usr/bin/env python3\nprint('x')", None);
        assert_eq!(language, CodeLanguage::Python);
    }

    #[test]
    fn explicit_language_overrides_path() {
        let language = detect_language(Some(Path::new("main.py")), "", Some("rust"));
        assert_eq!(language, CodeLanguage::Rust);
    }
}