tree_sitter_language_pack 1.0.0.rc.7 → 1.0.0.rc.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e0e70fba75e85bf3ca744874c8b278fe3fa821bfe35998495f5826b50f94c4c0
-  data.tar.gz: e77f82f0394870ae8a02718e0d52a0193379eff0696bb874571c1c61159e133f
+  metadata.gz: 2153215d6298710720f131215b09d0dac8b8f38da43c1f72b85b3cfaab8f0c01
+  data.tar.gz: 1f591520ce16bda97c40cfb217a233e8af5b6f3ac4e6a358bb4cfe97d4fde165
 SHA512:
-  metadata.gz: 97b9ef50b06634a695a6dbf2f14abe459a13dac86453e7543fb4f8aeaff580a0833f5940d4d0431d4afd152d5f2c45b18714da3b738f1bc87b621c79a71da2c6
-  data.tar.gz: e3b8796578f971068923eea849d7323f9df5d444d786fd6e3cf798af3317dbac282b2c84c4ce812352b846a129680ecbdb524faef68a67d3c4b536a53ce33b22
+  metadata.gz: 57ddab59c16c7dcea2d1cb90144d928ed5161f72d0eead1fb0fb6e906cdbdc11b8e3c6b46725783d8243a1b58600a2aae31c269a44a04314993214b9e56fbc0a
+  data.tar.gz: 1d660c50f5ccfddb836f90ae898bcb3257b62f0cbb7beeab78143c07ff82b2ef4be0bf399a3c9c9d736fc833b2e441dcf388ab0585a863e94f1d43ecc9b2bf4f

data/Cargo.toml

@@ -12,7 +12,10 @@ name = "ts_pack_ruby"
 crate-type = ["cdylib"]
 
 [dependencies]
-tree-sitter-language-pack = { workspace = true, features = ["serde"] }
+tree-sitter-language-pack = { workspace = true, features = [
+  "serde",
+  "download",
+] }
 serde_json = { workspace = true }
 tree-sitter = { workspace = true }
 magnus = "0.8"

data/src/lib.rs

@@ -75,10 +75,78 @@ 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}")))
 }
 
+/// Initialize the language pack with configuration.
+///
+/// Accepts a JSON string with optional fields:
+/// - `cache_dir` (string): override default cache directory
+/// - `languages` (array): language names to pre-download
+/// - `groups` (array): language groups to pre-download
+fn rb_init(ruby: &Ruby, config_json: String) -> Result<(), Error> {
+    let config: tree_sitter_language_pack::PackConfig = serde_json::from_str(&config_json)
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("invalid config JSON: {e}")))?;
+
+    tree_sitter_language_pack::init(&config).map_err(|e| Error::new(ruby.exception_runtime_error(), format!("{e}")))
+}
+
+/// Configure the language pack without downloading.
+///
+/// Accepts a JSON string with optional fields:
+/// - `cache_dir` (string): override default cache directory
+fn rb_configure(ruby: &Ruby, config_json: String) -> Result<(), Error> {
+    let config: tree_sitter_language_pack::PackConfig = serde_json::from_str(&config_json)
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("invalid config JSON: {e}")))?;
+
+    tree_sitter_language_pack::configure(&config)
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("{e}")))
+}
+
+/// Download specific languages to the cache.
+///
+/// Returns the number of newly downloaded languages.
+fn rb_download(ruby: &Ruby, names: Vec<String>) -> Result<usize, Error> {
+    let refs: Vec<&str> = names.iter().map(String::as_str).collect();
+    tree_sitter_language_pack::download(&refs).map_err(|e| Error::new(ruby.exception_runtime_error(), format!("{e}")))
+}
+
+/// Download all available languages from the remote manifest.
+///
+/// Returns the number of newly downloaded languages.
+fn rb_download_all(ruby: &Ruby) -> Result<usize, Error> {
+    tree_sitter_language_pack::download_all().map_err(|e| Error::new(ruby.exception_runtime_error(), format!("{e}")))
+}
+
+/// Get all language names available in the remote manifest.
+fn rb_manifest_languages(ruby: &Ruby) -> Result<Vec<String>, Error> {
+    tree_sitter_language_pack::manifest_languages()
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("{e}")))
+}
+
+/// Get all languages that are already downloaded and cached locally.
+fn rb_downloaded_languages() -> Vec<String> {
+    tree_sitter_language_pack::downloaded_languages()
+}
+
+/// Delete all cached parser shared libraries.
+fn rb_clean_cache(ruby: &Ruby) -> Result<(), Error> {
+    tree_sitter_language_pack::clean_cache().map_err(|e| Error::new(ruby.exception_runtime_error(), format!("{e}")))
+}
+
+/// Get the effective cache directory path as a string.
+fn rb_cache_dir(ruby: &Ruby) -> Result<String, Error> {
+    tree_sitter_language_pack::cache_dir()
+        .map_err(|e| Error::new(ruby.exception_runtime_error(), format!("{e}")))
+        .and_then(|path| {
+            path.to_str()
+                .ok_or_else(|| Error::new(ruby.exception_runtime_error(), "cache path is not valid UTF-8"))
+                .map(|s| s.to_string())
+        })
+}
+
 #[magnus::init]
 fn init(ruby: &Ruby) -> Result<(), Error> {
     let module = ruby.define_module("TreeSitterLanguagePack")?;
 
+    // Registry and parsing functions
     module.define_module_function("available_languages", function!(available_languages, 0))?;
     module.define_module_function("has_language", function!(has_language, 1))?;
     module.define_module_function("language_count", function!(language_count, 0))?;
@@ -86,6 +154,16 @@ fn init(ruby: &Ruby) -> Result<(), Error> {
     module.define_module_function("parse_string", function!(parse_string, 2))?;
     module.define_module_function("process", function!(process, 2))?;
 
+    // Download API functions
+    module.define_module_function("init", function!(rb_init, 1))?;
+    module.define_module_function("configure", function!(rb_configure, 1))?;
+    module.define_module_function("download", function!(rb_download, 1))?;
+    module.define_module_function("download_all", function!(rb_download_all, 0))?;
+    module.define_module_function("manifest_languages", function!(rb_manifest_languages, 0))?;
+    module.define_module_function("downloaded_languages", function!(rb_downloaded_languages, 0))?;
+    module.define_module_function("clean_cache", function!(rb_clean_cache, 0))?;
+    module.define_module_function("cache_dir", function!(rb_cache_dir, 0))?;
+
     let tree_class = module.define_class("Tree", ruby.class_object())?;
     tree_class.define_method("root_node_type", method!(TreeWrapper::root_node_type, 0))?;
     tree_class.define_method("root_child_count", method!(TreeWrapper::root_child_count, 0))?;

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: tree_sitter_language_pack
 version: !ruby/object:Gem::Version
-  version: 1.0.0.rc.7
+  version: 1.0.0.rc.9
 platform: ruby
 authors:
 - kreuzberg.dev
-autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-03-17 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rb_sys
@@ -42,7 +41,6 @@ licenses:
 - MIT
 metadata:
   cargo_crate_name: ts-pack-ruby
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -50,15 +48,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.2.0
+      version: 3.4.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.22
-signing_key:
+rubygems_version: 3.6.9
 specification_version: 4
 summary: Ruby bindings for tree-sitter-language-pack
 test_files: []