feat(query): implement endByteForPattern

Author: ObserverOfTime

scripts/jextract.sh

@@ -111,6 +111,7 @@ jextract \
     --include-function ts_query_delete \
     --include-function ts_query_disable_capture \
     --include-function ts_query_disable_pattern \
+    --include-function ts_query_end_byte_for_pattern \
     --include-function ts_query_is_pattern_guaranteed_at_step \
     --include-function ts_query_is_pattern_non_local \
     --include-function ts_query_is_pattern_rooted \

src/main/java/io/github/treesitter/jtreesitter/Query.java

@@ -120,7 +120,7 @@ public final class Query implements AutoCloseable {
         }
 
         try (var alloc = Arena.ofConfined()) {
-            for (int i = 0, steps = 0; i < patternCount; ++i) {
+            for (int i = 0, steps; i < patternCount; ++i) {
                 var count = alloc.allocate(C_INT);
                 var tokens = ts_query_predicates_for_pattern(query, i, count);
                 if ((steps = count.get(C_INT, 0)) == 0) continue;
@@ -369,6 +369,17 @@ public void disableCapture(String name) throws NoSuchElementException {
         return ts_query_start_byte_for_pattern(query, index);
     }
 
+    /**
+     * Get the byte offset where the given pattern ends in the query's source.
+     *
+     * @throws IndexOutOfBoundsException If the index exceeds the
+     *                                   {@linkplain #getPatternCount pattern count}.
+     */
+    public @Unsigned int endByteForPattern(@Unsigned int index) throws IndexOutOfBoundsException {
+        checkIndex(index);
+        return ts_query_end_byte_for_pattern(query, index);
+    }
+
     /**
      * Check if the pattern with the given index has a single root node.
      *

src/test/java/io/github/treesitter/jtreesitter/QueryTest.java

@@ -148,6 +148,14 @@ void startByteForPattern() {
         });
     }
 
+    @Test
+    void endByteForPattern() {
+        assertQuery(query -> {
+            assertEquals(26, query.endByteForPattern(0));
+            assertThrows(IndexOutOfBoundsException.class, () -> query.endByteForPattern(2));
+        });
+    }
+
     @Test
     void isPatternRooted() {
         assertQuery(query -> {