Added API to get capture names from TSQuery

Author: itsaky

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

@@ -19,10 +19,11 @@
 
 public class TSQuery implements AutoCloseable {
   final long pointer;
-
   private int errorOffset;
   private int errorType;
 
+  private String[] captureNames = null;
+
   /**
    * Create a new query from a string containing one or more S-expression patterns. The query is
    * associated with a particular language, and can only be run on syntax nodes parsed with that
@@ -72,6 +73,16 @@ public int getStringCount() {
     return Native.stringCount(this.pointer);
   }
 
+  public String[] getCaptureNames() {
+    if (captureNames == null) {
+      captureNames = new String[getCaptureCount()];
+      for (int i = 0; i < getCaptureCount(); i++) {
+        captureNames[i] = getCaptureNameForId(i);
+      }
+    }
+    return captureNames;
+  }
+
   public int getStartByteForPattern(int pattern) {
     validatePatternIndex(pattern);
     return Native.startByteForPattern(this.pointer, pattern);

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

@@ -55,6 +55,14 @@ public void queryTest() throws Exception {
         assertThat(match.getCaptures()).isNotNull();
         assertThat(match.getCaptures()).hasLength(1);
         assertThat(cursor.nextMatch()).isNull();
+
+        var captureNames = query.getCaptureNames();
+        assertThat(captureNames).hasLength(query.getCaptureCount());
+        captureNames = new String[query.getCaptureCount()];
+        for (int i = 0; i < captureNames.length; i++) {
+          captureNames[i] = query.getCaptureNameForId(0);
+        }
+
         query.close();
         cursor.close();
 
@@ -66,6 +74,7 @@ public void queryTest() throws Exception {
         assertThat(match.getCaptures()).isNotNull();
         assertThat(match.getCaptures()).hasLength(1);
         assertThat(cursor.nextMatch()).isNull();
+
         query.close();
         cursor.close();
       } catch (Throwable err) {