src/util/language.rs


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

use std::path::Path;

pub fn detect_language(filename: &Path, contents: &str) -> &'static str {
    detect_from_extension(filename)
        .or_else(|| detect_from_shebang(contents))
        .unwrap_or("")
}

fn detect_from_extension(filename: &Path) -> Option<&'static str> {
    let ext = filename
        .extension()
        .and_then(|e| e.to_str())
        .map(|e| e.to_ascii_lowercase());
    let ext_str = ext.as_deref();
    match ext_str {
        Some("rs") => Some("rust"),
        Some("py") => Some("python"),
        Some("json") => Some("json"),
        _ => None,
    }
}

fn detect_from_shebang(contents: &str) -> Option<&'static str> {
    if let Some(first_line) = contents.lines().next() {
        if first_line.contains("python") {
            return Some("python");
        }
        if first_line.contains("bash") {
            return Some("bash");
        }
    }
    None
}