feat: locateFile for nodejs

Author: mohebifar

crates/web-tree-sitter-sg/src/lib.rs

@@ -4,7 +4,7 @@
 
 use core::cell::RefCell;
 use js_sys::{Array, Error, Function, JsString, Object, Promise, Reflect, Uint8Array};
-use wasm_bindgen::{prelude::*, JsCast};
+use wasm_bindgen::{JsCast, prelude::*};
 use wasm_bindgen_futures::JsFuture;
 
 trait JsValueExt {
@@ -41,15 +41,21 @@ thread_local! {
 pub struct TreeSitter;
 
 impl TreeSitter {
-    pub async fn init() -> Result<(), JsError> {
+    pub async fn init(locate_file: Option<Function>) -> Result<(), JsError> {
         #![allow(non_snake_case)]
 
         // Exit early if `web-tree-sitter` is already initialized
         if TREE_SITTER_INITIALIZED.with(|cell| *cell.borrow()) {
             return Ok(());
         }
 
-        JsFuture::from(Parser::init()).await.lift_error()?;
+        if let Some(locate_file) = locate_file {
+            let options = Object::new();
+            Reflect::set(&options, &"locateFile".into(), &locate_file.into()).unwrap();
+            JsFuture::from(Parser::init(Some(&options))).await.lift_error()?;
+        } else {
+            JsFuture::from(Parser::init(None)).await.lift_error()?;
+        }
 
         // Set `web-tree-sitter` to initialized
         TREE_SITTER_INITIALIZED.with(|cell| cell.replace(true));
@@ -154,7 +160,7 @@ extern {
     fn delete(this: &LoggerParams, val: &JsString);
 }
 
-#[wasm_bindgen(module="web-tree-sitter")]
+#[wasm_bindgen(module = "web-tree-sitter")]
 extern {
     #[derive(Clone, Debug, PartialEq)]
     pub type Language;
@@ -292,8 +298,7 @@ impl Default for Point {
     }
 }
 
-impl Eq for Point {
-}
+impl Eq for Point {}
 
 impl std::hash::Hash for Point {
     fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
@@ -538,8 +543,7 @@ impl Default for Range {
     }
 }
 
-impl Eq for Range {
-}
+impl Eq for Range {}
 
 impl std::hash::Hash for Range {
     fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
@@ -743,8 +747,7 @@ impl PartialEq<SyntaxNode> for SyntaxNode {
     }
 }
 
-impl Eq for SyntaxNode {
-}
+impl Eq for SyntaxNode {}
 
 impl std::hash::Hash for SyntaxNode {
     fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
@@ -850,14 +853,14 @@ extern {
     pub fn reset(this: &TreeCursor, node: &SyntaxNode);
 }
 
-#[wasm_bindgen(module="web-tree-sitter")]
+#[wasm_bindgen(module = "web-tree-sitter")]
 extern {
     #[derive(Clone, Debug)]
     pub type Parser;
 
     // Static Methods
     #[wasm_bindgen(static_method_of = Parser)]
-    pub fn init() -> Promise;
+    pub fn init(options: Option<&Object>) -> Promise;
 
     // Constructor