feat: add state APIs in TSLanguage

Author: itsaky

.idea/kotlinc.xml

@@ -155,4 +155,22 @@ Java_com_itsaky_androidide_treesitter_TSLanguage_00024Native_dlclose(JNIEnv *env
                                                                      jlong libhandle) {
   if (libhandle == 0) return;
   dlclose((void *) libhandle);
+}
+
+extern "C"
+JNIEXPORT jint JNICALL
+Java_com_itsaky_androidide_treesitter_TSLanguage_00024Native_stateCount(JNIEnv *env,
+                                                                        jclass clazz,
+                                                                        jlong pointer) {
+  return (jint) ts_language_state_count((TSLanguage*) pointer);
+}
+
+extern "C"
+JNIEXPORT jshort JNICALL
+Java_com_itsaky_androidide_treesitter_TSLanguage_00024Native_nextState(JNIEnv *env,
+                                                                       jclass clazz,
+                                                                       jlong pointer,
+                                                                       jshort state_id,
+                                                                       jshort symbol) {
+  return (jshort) ts_language_next_state((TSLanguage*) pointer, state_id, symbol);
 }

android-tree-sitter/src/main/cpp/ts_language.cc

@@ -127,6 +127,23 @@ public TSSymbolType getSymbolType(int symbol) {
     return TSSymbolType.forId(Native.symType(this.pointer, symbol));
   }
 
+  /**
+   * Get the number of valid states in this language.
+   */
+  public int getStateCount() {
+    checkAccess();
+    return Native.stateCount(pointer);
+  }
+
+  /**
+   * Get the next parse state. Combine this with lookahead iterators to generate
+   * completion suggestions or valid symbols in error nodes.
+   */
+  public short getNextState(short stateId, short symbol) {
+    checkAccess();
+    return Native.nextState(pointer, stateId, symbol);
+  }
+
   public int getLanguageVersion() {
     checkAccess();
     return Native.langVer(this.pointer);
@@ -246,5 +263,9 @@ private static class Native {
     private static native long[] loadLanguage(String sharedLib, String func);
 
     private static native void dlclose(long libhandle);
+
+    public static native int stateCount(long pointer);
+
+    public static native short nextState(long pointer, short stateId, short symbol);
   }
 }