Added support for 'ts_node_child_by_field_name'

Author: itsaky

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

@@ -1,5 +1,7 @@
 package com.itsaky.androidide.treesitter;
 
+import java.nio.charset.StandardCharsets;
+
 public class TSNode {
   private int context0;
   private int context1;
@@ -32,6 +34,11 @@ public TSNode getNamedChild(int index) {
     return TreeSitter.nodeNamedChild(this, index);
   }
 
+  public TSNode getChildByFieldName(String fieldName) {
+    final var bytes = fieldName.getBytes(StandardCharsets.UTF_8);
+    return TreeSitter.getChildByFieldName(this, bytes, bytes.length);
+  }
+
   public int getChildCount() {
     return TreeSitter.nodeChildCount(this);
   }

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

@@ -47,6 +47,17 @@ public class TreeSitter {
    */
   public static native TSNode nodeNamedChild(TSNode node, int child);
 
+  /**
+   * Find the child node of the given node by field name.
+   *
+   * @param node The node.
+   * @param fieldName The field name of the child.
+   * @param fieldNameLength The length of `fieldName`.
+   * @return The found node.
+   */
+  public static native TSNode getChildByFieldName(
+      TSNode node, byte[] fieldName, int fieldNameLength);
+
   /**
    * Get the end byte of the given node.
    *
@@ -139,6 +150,7 @@ public class TreeSitter {
    * @param node The node to check.
    */
   public static native boolean nodeHasError(TSNode node);
+
   // -------------------------------------------
   // ---------- Section: TSParser --------------
   // -------------------------------------------

android-tree-sitter/src/main/jniLibs/arm64-v8a/libts.so

@@ -80,6 +80,8 @@ public void testGetChildren() throws UnsupportedEncodingException {
         isNull = function.isNull();
         assertFalse(isNull);
 
+        var body = function.getChildByFieldName("body");
+
         var parent = function.getParent();
         type = parent.getType();
         assertEquals("module", type);

android-tree-sitter/src/main/jniLibs/armeabi-v7a/libts.so

@@ -248,6 +248,15 @@ JNIEXPORT jobject JNICALL Java_com_itsaky_androidide_treesitter_TreeSitter_nodeN
            env, ts_node_named_child(_unmarshalNode(env, node), (uint32_t)child));
 }
 
+JNIEXPORT jobject JNICALL Java_com_itsaky_androidide_treesitter_TreeSitter_getChildByFieldName(
+  JNIEnv* env, jclass self, jobject node, jbyteArray name, jint length) {
+  jbyte* nameStr = env->GetByteArrayElements(name, NULL);
+  jobject found = _marshalNode(
+           env, ts_node_child_by_field_name(_unmarshalNode(env, node), reinterpret_cast<const char*>(nameStr), (uint32_t) length));
+  env->ReleaseByteArrayElements(name, nameStr, JNI_ABORT);
+  return found;
+}
+
 JNIEXPORT jstring JNICALL Java_com_itsaky_androidide_treesitter_TreeSitter_nodeString(
   JNIEnv* env, jclass self, jobject node) {
   char* nodeString = ts_node_string(_unmarshalNode(env, node));