tree_sitter_language_pack 1.2.1 → 1.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e39e2e8f16d680a03503e0fc7a5340cc8900a9c11250d423dcafc70dbb38a308
-  data.tar.gz: 22863be51d8d97eef161733d4864edac0a07ad96cef21672b3e4da9950c039c4
+  metadata.gz: 73a601edb6bb285bf825a5b588c6fb9ae58fa8f1a0bda7c4d537cdfc9f72a912
+  data.tar.gz: 331e45aafcaf276595cfa679aa5de2524204b5ba4bcb138d4c1b423471576964
 SHA512:
-  metadata.gz: 6e9afb8bbf4644808bac37295187463bbc9d8f702fc6e3e239a530044e31d95e57f966b1d97e6c8077832102edc439c19bbfc0261cb7aaedd16fba931b6e014b
-  data.tar.gz: 9248459136e6233d742c6832618663cc5bb209d0b56e583abcc45f0ba1aae9bc3ebd56e75cba149dbf92b8e061fa3c8db9637a56770c4f113a3b3a530edd42f6
+  metadata.gz: e197f722769c8cbb45aabbe38d713bfb234036c74d523d521041814916e5518ef6f4acd5c6b66ab0f1317404d7fc608b5043915aa93331c876769363674627d8
+  data.tar.gz: b3efc14816133eec048d9a988fba3b0620a991a3d3c9426eb6968bdce14925d6eeaa20bfb4394d0e561f828f7661e720d344cff09ec3bd421dc2697aa0fcc752

data/src/lib.rs

@@ -100,6 +100,42 @@ fn process(ruby: &Ruby, source: String, config_json: String) -> Result<String, E
         .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("serialization failed: {e}")))
 }
 
+/// Extract patterns from source code using a JSON configuration.
+///
+/// The config JSON must contain:
+/// - `language` (string): the language name
+/// - `patterns` (object): named patterns to run, each with a `query` field
+///
+/// Returns a JSON string with extraction results.
+fn extract(ruby: &Ruby, source: String, config_json: String) -> Result<String, Error> {
+    let config: tree_sitter_language_pack::ExtractionConfig = serde_json::from_str(&config_json)
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("invalid config JSON: {e}")))?;
+
+    let result = tree_sitter_language_pack::extract_patterns(&source, &config)
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("{e}")))?;
+
+    serde_json::to_string(&result)
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("serialization failed: {e}")))
+}
+
+/// Validate extraction patterns without running them.
+///
+/// The config JSON must contain:
+/// - `language` (string): the language name
+/// - `patterns` (object): named patterns to validate
+///
+/// Returns a JSON string with validation results.
+fn validate_extraction(ruby: &Ruby, config_json: String) -> Result<String, Error> {
+    let config: tree_sitter_language_pack::ExtractionConfig = serde_json::from_str(&config_json)
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("invalid config JSON: {e}")))?;
+
+    let result = tree_sitter_language_pack::validate_extraction(&config)
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("{e}")))?;
+
+    serde_json::to_string(&result)
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("serialization failed: {e}")))
+}
+
 /// Initialize the language pack with configuration.
 ///
 /// Accepts a JSON string with optional fields:
@@ -187,6 +223,8 @@ fn init(ruby: &Ruby) -> Result<(), Error> {
     module.define_module_function("get_language_ptr", function!(get_language_ptr, 1))?;
     module.define_module_function("parse_string", function!(parse_string, 2))?;
     module.define_module_function("process", function!(process, 2))?;
+    module.define_module_function("extract", function!(extract, 2))?;
+    module.define_module_function("validate_extraction", function!(validate_extraction, 1))?;
 
     // Download API functions
     module.define_module_function("init", function!(rb_init, 1))?;

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tree_sitter_language_pack
 version: !ruby/object:Gem::Version
-  version: 1.2.1
+  version: 1.3.0
 platform: ruby
 authors:
 - kreuzberg.dev