forked from Mirrors/helix
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
1.3 KiB
Rust
40 lines
1.3 KiB
Rust
use anyhow::{Context, Result};
|
|
use libloading::{Library, Symbol};
|
|
use tree_sitter::Language;
|
|
|
|
fn replace_dashes_with_underscores(name: &str) -> String {
|
|
let mut result = String::with_capacity(name.len());
|
|
for c in name.chars() {
|
|
if c == '-' {
|
|
result.push('_');
|
|
} else {
|
|
result.push(c);
|
|
}
|
|
}
|
|
result
|
|
}
|
|
#[cfg(unix)]
|
|
const DYLIB_EXTENSION: &str = "so";
|
|
|
|
#[cfg(windows)]
|
|
const DYLIB_EXTENSION: &str = "dll";
|
|
|
|
pub fn get_language(runtime_path: &std::path::Path, name: &str) -> Result<Language> {
|
|
let name = name.to_ascii_lowercase();
|
|
let mut library_path = runtime_path.join("grammars").join(&name);
|
|
// TODO: duplicated under build
|
|
library_path.set_extension(DYLIB_EXTENSION);
|
|
|
|
let library = unsafe { Library::new(&library_path) }
|
|
.with_context(|| format!("Error opening dynamic library {:?}", &library_path))?;
|
|
let language_fn_name = format!("tree_sitter_{}", replace_dashes_with_underscores(&name));
|
|
let language = unsafe {
|
|
let language_fn: Symbol<unsafe extern "C" fn() -> Language> = library
|
|
.get(language_fn_name.as_bytes())
|
|
.with_context(|| format!("Failed to load symbol {}", language_fn_name))?;
|
|
language_fn()
|
|
};
|
|
std::mem::forget(library);
|
|
Ok(language)
|
|
}
|