git-topology 0.1.2

Shared clustering and topology primitives for git-cognitive and git-semantic
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

use tree_sitter::Language;

#[derive(Debug, Clone, Copy)]
pub enum SupportedLanguage {
    Rust,
    Python,
    JavaScript,
    TypeScript,
    Java,
    C,
    Cpp,
    Go,
}

impl SupportedLanguage {
    pub fn tree_sitter_language(&self) -> Language {
        match self {
            SupportedLanguage::Rust => tree_sitter_rust::LANGUAGE.into(),
            SupportedLanguage::Python => tree_sitter_python::LANGUAGE.into(),
            SupportedLanguage::JavaScript => tree_sitter_javascript::LANGUAGE.into(),
            SupportedLanguage::TypeScript => tree_sitter_typescript::LANGUAGE_TYPESCRIPT.into(),
            SupportedLanguage::Java => tree_sitter_java::LANGUAGE.into(),
            SupportedLanguage::C => tree_sitter_c::LANGUAGE.into(),
            SupportedLanguage::Cpp => tree_sitter_cpp::LANGUAGE.into(),
            SupportedLanguage::Go => tree_sitter_go::LANGUAGE.into(),
        }
    }
}

pub fn detect_language(file_path: &str) -> Option<SupportedLanguage> {
    let extension = file_path.split('.').next_back()?.to_lowercase();
    match extension.as_str() {
        "rs" => Some(SupportedLanguage::Rust),
        "py" | "pyw" | "pyi" => Some(SupportedLanguage::Python),
        "js" | "mjs" | "cjs" => Some(SupportedLanguage::JavaScript),
        "ts" | "tsx" => Some(SupportedLanguage::TypeScript),
        "java" => Some(SupportedLanguage::Java),
        "c" | "h" => Some(SupportedLanguage::C),
        "cpp" | "cc" | "cxx" | "hpp" | "hxx" => Some(SupportedLanguage::Cpp),
        "go" => Some(SupportedLanguage::Go),
        _ => None,
    }
}

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

    #[test]
    fn detects_rust() {
        assert!(matches!(
            detect_language("src/main.rs"),
            Some(SupportedLanguage::Rust)
        ));
    }

    #[test]
    fn detects_python_variants() {
        assert!(matches!(
            detect_language("script.py"),
            Some(SupportedLanguage::Python)
        ));
        assert!(matches!(
            detect_language("types.pyi"),
            Some(SupportedLanguage::Python)
        ));
    }

    #[test]
    fn detects_typescript() {
        assert!(matches!(
            detect_language("app.ts"),
            Some(SupportedLanguage::TypeScript)
        ));
        assert!(matches!(
            detect_language("comp.tsx"),
            Some(SupportedLanguage::TypeScript)
        ));
    }

    #[test]
    fn detects_go() {
        assert!(matches!(
            detect_language("main.go"),
            Some(SupportedLanguage::Go)
        ));
    }

    #[test]
    fn unsupported_returns_none() {
        assert!(detect_language("README.md").is_none());
        assert!(detect_language("data.json").is_none());
        assert!(detect_language("Makefile").is_none());
    }

    #[test]
    fn case_insensitive_extension() {
        assert!(detect_language("main.RS").is_some());
        assert!(detect_language("script.PY").is_some());
    }
}