Skip to main content

semantic/parser/
parser_language.rs

1// SPDX-License-Identifier: Apache-2.0
2//! Supported language definitions.
3
4/// Supported programming languages.
5#[derive(Clone, Copy, Debug, Hash, PartialEq, Eq)]
6pub enum Language {
7    Rust,
8    Python,
9    JavaScript,
10    TypeScript,
11    Go,
12    C,
13    Cpp,
14    Java,
15    Unknown,
16}
17
18impl Language {
19    /// Detect language from file extension.
20    pub fn from_path(path: &std::path::Path) -> Self {
21        match path.extension().and_then(|e| e.to_str()) {
22            Some("rs") => Language::Rust,
23            Some("py") | Some("pyi") => Language::Python,
24            Some("js") | Some("jsx") | Some("mjs") | Some("cjs") => Language::JavaScript,
25            Some("ts") | Some("tsx") => Language::TypeScript,
26            Some("go") => Language::Go,
27            Some("c") | Some("h") => Language::C,
28            Some("cpp") | Some("cc") | Some("hpp") | Some("cxx") => Language::Cpp,
29            Some("java") => Language::Java,
30            _ => Language::Unknown,
31        }
32    }
33
34    /// Get the tree-sitter language parser. Public mirror of [`parser`]
35    /// for consumers outside the `semantic` crate (notably
36    /// [`crate::symbol_resolver`] and the `repo` re-export).
37    pub fn parser_handle(&self) -> Option<tree_sitter::Language> {
38        self.parser()
39    }
40
41    /// Get the tree-sitter language parser.
42    pub(crate) fn parser(&self) -> Option<tree_sitter::Language> {
43        match self {
44            #[cfg(feature = "lang-rust")]
45            Language::Rust => Some(tree_sitter_rust::LANGUAGE.into()),
46            #[cfg(feature = "lang-python")]
47            Language::Python => Some(tree_sitter_python::LANGUAGE.into()),
48            #[cfg(feature = "lang-javascript")]
49            Language::JavaScript => Some(tree_sitter_javascript::LANGUAGE.into()),
50            #[cfg(feature = "lang-typescript")]
51            Language::TypeScript => Some(tree_sitter_typescript::LANGUAGE_TYPESCRIPT.into()),
52            #[cfg(feature = "lang-go")]
53            Language::Go => Some(tree_sitter_go::LANGUAGE.into()),
54            #[cfg(feature = "lang-c")]
55            Language::C => Some(tree_sitter_c::LANGUAGE.into()),
56            #[cfg(feature = "lang-cpp")]
57            Language::Cpp => Some(tree_sitter_cpp::LANGUAGE.into()),
58            #[cfg(feature = "lang-java")]
59            Language::Java => Some(tree_sitter_java::LANGUAGE.into()),
60            _ => None,
61        }
62    }
63}