Hook up tree-sitter-ocaml for .mli files

ida_star
Wilfred Hughes 2021-08-16 22:08:55 +07:00
parent 96673ee2d6
commit 4fae2e9ef5
2 changed files with 8 additions and 0 deletions

@ -50,6 +50,11 @@ fn main() {
"vendor/tree-sitter-ocaml/ocaml/src",
&["scanner.cc"],
);
build(
"tree-sitter-ocaml-interface",
"vendor/tree-sitter-ocaml/interface/src",
&["scanner.cc"],
);
build(
"tree-sitter-rust",
"vendor/tree-sitter-rust/src",

@ -12,6 +12,7 @@ extern "C" {
fn tree_sitter_javascript() -> Language;
fn tree_sitter_json() -> Language;
fn tree_sitter_ocaml() -> Language;
fn tree_sitter_ocaml_interface() -> Language;
fn tree_sitter_rust() -> Language;
}
@ -47,6 +48,8 @@ pub fn parse<'a>(
unsafe { tree_sitter_javascript() }
} else if extension == "ml" {
unsafe { tree_sitter_ocaml() }
} else if extension == "mli" {
unsafe { tree_sitter_ocaml_interface() }
} else {
panic!("Unknown extension for tree-sitter parsers.")
};