feat: add registerDynamicLanguage to napi

HerringtonDarkholme · HerringtonDarkholme · commit dd2f65bccd63 · 2025-01-04T10:39:16.000-05:00
diff --git a/crates/napi/__test__/custom.spec.ts b/crates/napi/__test__/custom.spec.ts
@@ -0,0 +1,34 @@
+import test from 'ava'
+
+import {
+  registerDynamicLanguage,
+  parse,
+} from '../index'
+
+const { platform, arch } = process
+
+const isAppleSilicon = platform === 'darwin' && arch === 'arm64'
+if (isAppleSilicon) {
+  registerDynamicLanguage({
+    myjson: {
+      libraryPath: "../../benches/fixtures/json-mac.so",
+      languageSymbol: "tree_sitter_json",
+      extensions: ["myjson"],
+    }
+  })
+}
+
+test('test load custom lang', t => {
+  if (!isAppleSilicon) {
+    t.pass('This test is not available on this platform')
+    return
+  }
+  // @ts-expect-error TODO: change type
+  const sg = parse('myjson', '{"test": 123}')
+  const root = sg.root()
+  const node = root.find("123")!
+  t.truthy(node)
+  t.is(node.kind(), 'number')
+  const no = root.find("456")
+  t.falsy(no)
+})
diff --git a/crates/napi/__test__/index.spec.ts b/crates/napi/__test__/index.spec.ts
@@ -434,4 +434,4 @@ test('find node by range', t => {
   })
   const byKind = sg.root().find(js.kind('template_string'))
   t.is(node!.id(), byKind!.id())
-})
+})
diff --git a/crates/napi/index.d.ts b/crates/napi/index.d.ts
@@ -17,3 +17,6 @@ export {
 } from './types/api'
 // deprecated
 export * from './types/deprecated'
+
+// TODO
+export declare function registerDynamicLanguage(langs: any): void
diff --git a/crates/napi/index.js b/crates/napi/index.js
@@ -310,7 +310,7 @@ if (!nativeBinding) {
   throw new Error(`Failed to load native binding`)
 }
 
-const { parseFiles, Lang, SgNode, SgRoot, parse, parseAsync, kind, pattern, findInFiles, html, js, jsx, ts, tsx, css } = nativeBinding
+const { parseFiles, Lang, SgNode, SgRoot, parse, parseAsync, kind, pattern, findInFiles, registerDynamicLanguage, html, js, jsx, ts, tsx, css } = nativeBinding
 
 module.exports.parseFiles = parseFiles
 module.exports.Lang = Lang
@@ -321,6 +321,7 @@ module.exports.parseAsync = parseAsync
 module.exports.kind = kind
 module.exports.pattern = pattern
 module.exports.findInFiles = findInFiles
+module.exports.registerDynamicLanguage = registerDynamicLanguage
 module.exports.html = html
 module.exports.js = js
 module.exports.jsx = jsx
diff --git a/crates/napi/src/lib.rs b/crates/napi/src/lib.rs
@@ -5,11 +5,11 @@ mod find_files;
 mod napi_lang;
 mod sg_node;
 
-use ast_grep_core::language::Language;
-use ast_grep_core::AstGrep;
+use ast_grep_core::{AstGrep, Language};
 use ast_grep_language::SupportLang;
 use napi::bindgen_prelude::*;
 use napi_derive::napi;
+use napi_lang::register_dynamic_language as register_dynamic_language_impl;
 
 use doc::{JsDoc, NapiConfig};
 use find_files::{find_in_files_impl, FindConfig, FindInFiles, ParseAsync};
@@ -119,3 +119,11 @@ pub fn find_in_files(
   let lang: NapiLang = lang.parse()?;
   find_in_files_impl(lang, config, callback)
 }
+
+/// Register a dynamic language to ast-grep.
+/// `langs` is a Map of language name to its CustomLanguage registration.
+#[napi]
+pub fn register_dynamic_language(langs: serde_json::Value) -> Result<()> {
+  let langs = serde_json::from_value(langs)?;
+  register_dynamic_language_impl(langs)
+}
