Implemented TSTree.getLanguage()

Author: itsaky

android-tree-sitter/src/main/java/com/itsaky/androidide/treesitter/TSTree.java

@@ -41,6 +41,15 @@ public void edit(TSInputEdit edit) {
     Native.edit(pointer, edit);
   }
 
+  /**
+   * Get the language that was used to parse the syntax tree.
+   *
+   * @return The tree sitter language.
+   */
+  public TSLanguage getLanguage() {
+    return new TSLanguage(Native.getLanguage(this.pointer));
+  }
+
   /**
    * Get the pointer of this tree.
    *
@@ -52,8 +61,13 @@ long getPointer() {
 
   private static class Native {
     public static native void edit(long tree, TSInputEdit inputEdit);
+
     public static native void delete(long tree);
+
     public static native long copy(long tree);
+
     public static native TSNode rootNode(long tree);
+
+    public static native long getLanguage(long tree);
   }
 }

android-tree-sitter/src/test/java/com/itsaky/androidide/treesitter/TreeTest.java

@@ -18,6 +18,7 @@ public void testTreeCopy() throws UnsupportedEncodingException {
       try (final var tree =
           parser.parseString(
               "class Main { void main() {} }", TSInputEncoding.TSInputEncodingUTF8)) {
+        assertThat(tree.getLanguage()).isEqualTo(TSLanguages.java());
         assertThat(tree.copy().getRootNode().getNodeString())
             .isEqualTo(tree.getRootNode().getNodeString());
       }

lib/com_itsaky_androidide_treesitter_TSTree_Native.h

@@ -1,24 +1,36 @@
 #include "com_itsaky_androidide_treesitter_TSTree_Native.h"
 #include "ts_utils.h"
 
-JNIEXPORT void JNICALL Java_com_itsaky_androidide_treesitter_TSTree_00024Native_edit(
-  JNIEnv* env, jclass self, jlong tree, jobject inputEdit) {
-
+JNIEXPORT void JNICALL
+Java_com_itsaky_androidide_treesitter_TSTree_00024Native_edit(
+    JNIEnv* env, jclass self, jlong tree, jobject inputEdit) {
   TSInputEdit edit = _unmarshalInputEdit(env, inputEdit);
-  ts_tree_edit((TSTree*) tree, &edit);
+  ts_tree_edit((TSTree*)tree, &edit);
 }
 
-JNIEXPORT void JNICALL Java_com_itsaky_androidide_treesitter_TSTree_00024Native_delete(
-  JNIEnv* env, jclass self, jlong tree) {
+JNIEXPORT void JNICALL
+Java_com_itsaky_androidide_treesitter_TSTree_00024Native_delete(JNIEnv* env,
+                                                                jclass self,
+                                                                jlong tree) {
   ts_tree_delete((TSTree*)tree);
 }
 
-JNIEXPORT jlong JNICALL Java_com_itsaky_androidide_treesitter_TSTree_00024Native_copy(
-  JNIEnv* env, jclass self, jlong tree) {
-  return (jlong) ts_tree_copy((TSTree*)tree);
+JNIEXPORT jlong JNICALL
+Java_com_itsaky_androidide_treesitter_TSTree_00024Native_copy(JNIEnv* env,
+                                                              jclass self,
+                                                              jlong tree) {
+  return (jlong)ts_tree_copy((TSTree*)tree);
 }
 
-JNIEXPORT jobject JNICALL Java_com_itsaky_androidide_treesitter_TSTree_00024Native_rootNode(
-  JNIEnv* env, jclass self, jlong tree) {
+JNIEXPORT jobject JNICALL
+Java_com_itsaky_androidide_treesitter_TSTree_00024Native_rootNode(JNIEnv* env,
+                                                                  jclass self,
+                                                                  jlong tree) {
   return _marshalNode(env, ts_tree_root_node((TSTree*)tree));
+}
+
+JNIEXPORT jlong JNICALL
+Java_com_itsaky_androidide_treesitter_TSTree_00024Native_getLanguage(
+    JNIEnv* env, jclass self, jlong tree) {
+  return (jlong)ts_tree_language((TSTree*)tree);
 }