use std::path::Path;
use tree_sitter::Language as TreeSitterLanguage;
pub use blends_domain::Language;
pub trait LanguageExt {
#[must_use]
fn from_path(path: &Path) -> Option<Language>;
#[must_use]
fn tree_sitter(self) -> TreeSitterLanguage;
#[cfg(test)]
fn node_types(self) -> Option<&'static str>;
}
impl LanguageExt for Language {
fn from_path(path: &Path) -> Option<Language> {
path.extension()
.and_then(|extension| extension.to_str())
.map(str::to_ascii_lowercase)
.as_deref()
.and_then(Self::from_extension)
}
fn tree_sitter(self) -> TreeSitterLanguage {
match self {
Self::CSharp => tree_sitter_c_sharp::LANGUAGE.into(),
Self::Elixir => tree_sitter_elixir::LANGUAGE.into(),
Self::Go => tree_sitter_go::LANGUAGE.into(),
Self::Hcl => tree_sitter_hcl::LANGUAGE.into(),
Self::Java => tree_sitter_java::LANGUAGE.into(),
Self::JavaScript => tree_sitter_javascript::LANGUAGE.into(),
Self::Json => tree_sitter_json::LANGUAGE.into(),
Self::Kotlin => tree_sitter_kotlin_ng::LANGUAGE.into(),
Self::Php => tree_sitter_php::LANGUAGE_PHP.into(),
Self::Ruby => tree_sitter_ruby::LANGUAGE.into(),
Self::Rust => tree_sitter_rust::LANGUAGE.into(),
Self::Scala => tree_sitter_scala::LANGUAGE.into(),
Self::Swift => tree_sitter_swift::LANGUAGE.into(),
Self::TypeScript => tree_sitter_typescript::LANGUAGE_TSX.into(),
Self::Python => tree_sitter_python::LANGUAGE.into(),
Self::Yaml => tree_sitter_yaml::LANGUAGE.into(),
}
}
#[cfg(test)]
fn node_types(self) -> Option<&'static str> {
match self {
Self::CSharp => Some(tree_sitter_c_sharp::NODE_TYPES),
Self::Elixir => Some(tree_sitter_elixir::NODE_TYPES),
Self::Go => Some(tree_sitter_go::NODE_TYPES),
Self::Hcl => Some(tree_sitter_hcl::NODE_TYPES),
Self::Java => Some(tree_sitter_java::NODE_TYPES),
Self::JavaScript => Some(tree_sitter_javascript::NODE_TYPES),
Self::Json => Some(tree_sitter_json::NODE_TYPES),
Self::Kotlin => Some(tree_sitter_kotlin_ng::NODE_TYPES),
Self::Php => Some(tree_sitter_php::PHP_NODE_TYPES),
Self::Ruby => Some(tree_sitter_ruby::NODE_TYPES),
Self::Rust => Some(tree_sitter_rust::NODE_TYPES),
Self::Scala => Some(tree_sitter_scala::NODE_TYPES),
Self::Swift => Some(tree_sitter_swift::NODE_TYPES),
Self::TypeScript => Some(tree_sitter_typescript::TSX_NODE_TYPES),
Self::Python => Some(tree_sitter_python::NODE_TYPES),
Self::Yaml => Some(tree_sitter_yaml::NODE_TYPES),
}
}
}
#[cfg(test)]
mod tests {
use super::{Language, LanguageExt};
use std::path::Path;
use test_case::test_case;
#[test_case("cs" => Some(Language::CSharp))]
#[test_case("ex" => Some(Language::Elixir))]
#[test_case("exs" => Some(Language::Elixir))]
#[test_case("go" => Some(Language::Go))]
#[test_case("hcl" => Some(Language::Hcl))]
#[test_case("tf" => Some(Language::Hcl))]
#[test_case("java" => Some(Language::Java))]
#[test_case("js" => Some(Language::JavaScript))]
#[test_case("jsx" => Some(Language::JavaScript))]
#[test_case("json" => Some(Language::Json))]
#[test_case("kt" => Some(Language::Kotlin))]
#[test_case("ktm" => Some(Language::Kotlin))]
#[test_case("kts" => Some(Language::Kotlin))]
#[test_case("php" => Some(Language::Php))]
#[test_case("py" => Some(Language::Python))]
#[test_case("rb" => Some(Language::Ruby))]
#[test_case("rs" => Some(Language::Rust))]
#[test_case("scala" => Some(Language::Scala))]
#[test_case("swift" => Some(Language::Swift))]
#[test_case("ts" => Some(Language::TypeScript))]
#[test_case("tsx" => Some(Language::TypeScript))]
#[test_case("yaml" => Some(Language::Yaml))]
#[test_case("yml" => Some(Language::Yaml))]
#[test_case("unknown" => None)]
fn detects_by_extension(extension: &str) -> Option<Language> {
Language::from_extension(extension)
}
#[test_case("services/user/Api.java" => Some(Language::Java))]
#[test_case("integration/TestSuite.JAVA" => Some(Language::Java) ; "uppercase extension")]
#[test_case("infra/main.tf" => Some(Language::Hcl))]
#[test_case("config/vars.hcl" => Some(Language::Hcl))]
#[test_case("lib/app/router.ex" => Some(Language::Elixir))]
#[test_case("cmd/main.go" => Some(Language::Go))]
#[test_case("test/app/router_test.exs" => Some(Language::Elixir))]
#[test_case("Controllers/HouseController.cs" => Some(Language::CSharp))]
#[test_case("Models/User.CS" => Some(Language::CSharp) ; "uppercase cs extension")]
#[test_case("src/app/index.js" => Some(Language::JavaScript))]
#[test_case("src/app/index.JS" => Some(Language::JavaScript) ; "uppercase js extension")]
#[test_case("src/app/component.jsx" => Some(Language::JavaScript))]
#[test_case("src/app/app.ts" => Some(Language::TypeScript))]
#[test_case("src/app/component.tsx" => Some(Language::TypeScript))]
#[test_case("src/views/Home.swift" => Some(Language::Swift))]
#[test_case("pkg/module.py" => Some(Language::Python))]
#[test_case("scripts/build.PY" => Some(Language::Python) ; "uppercase py extension")]
#[test_case("config/settings.json" => Some(Language::Json))]
#[test_case("app/models/user.rb" => Some(Language::Ruby))]
#[test_case("src/main/scala/App.scala" => Some(Language::Scala))]
#[test_case("app/Main.kt" => Some(Language::Kotlin))]
#[test_case("build.gradle.kts" => Some(Language::Kotlin))]
#[test_case("deploy/values.yaml" => Some(Language::Yaml))]
#[test_case("ci/pipeline.yml" => Some(Language::Yaml))]
#[test_case("deploy/values.YAML" => Some(Language::Yaml) ; "uppercase yaml extension")]
#[test_case("src/main.rs" => Some(Language::Rust))]
#[test_case("src/unknown/randomfile.unknown" => None)]
#[test_case("folder/with/no_extension" => None)]
fn detects_language_by_path(path: &str) -> Option<Language> {
Language::from_path(Path::new(path))
}
}