Add support for querying the syntaxdot version

tensordot · Feb 2, 2021 · c0c7fe4 · c0c7fe4
1 parent 227788b
commit c0c7fe4
Show file tree

Hide file tree

Showing 4 changed files with 38 additions and 110 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -29,4 +29,7 @@ prost-build = "0.6"
 pretty_assertions = "0.6"
 
 [features]
-model-tests = []
+model-tests = []
+
+[patch.crates-io]
+tch = { git = "https://github.com/LaurentMazare/tch-rs.git" }
diff --git a/include/syntaxdot.h b/include/syntaxdot.h
@@ -90,6 +90,13 @@ void syntaxdot_set_num_intraop_threads(int32_t n_threads);
  */
 void syntaxdot_set_num_intraop_threads(int32_t n_threads);
 
+/**
+ * Get the syntaxdot version.
+ *
+ * The returned string must not be deallocated.
+ */
+char const *syntaxdot_version();
+
 /**
  * Free a <tt>ByteBuffer</tt>.
  * @param buf The buffer to free.

diff --git a/src/lib.rs b/src/lib.rs
@@ -10,13 +10,16 @@ use annotator::Annotator;
 
 mod error;
 use error::AnnotatorError;
+use std::ffi::CString;
+use std::os::raw::c_char;
 
 pub mod sentences;
 
 mod util;
 
 lazy_static! {
     static ref ANNOTATORS: ConcurrentHandleMap<Annotator> = ConcurrentHandleMap::new();
+    static ref SYNTAXDOT_VERSION: CString = CString::new(syntaxdot::VERSION).unwrap();
 }
 
 define_bytebuffer_destructor!(syntaxdot_free_bytebuffer);
@@ -81,6 +84,14 @@ pub extern "C" fn syntaxdot_set_num_intraop_threads(n_threads: i32) {
     tch::set_num_threads(n_threads);
 }
 
+/// Get the syntaxdot version.
+///
+/// The returned string must not be deallocated.
+#[no_mangle]
+pub extern "C" fn syntaxdot_version() -> *const c_char {
+    SYNTAXDOT_VERSION.as_ptr()
+}
+
 #[cfg(test)]
 mod tests {
     use std::ffi::CString;