fix: opened files are not highlighted with tree sitter

This happens because the tree-sitter nodes and trees are inaccessible when the files are opened

.idea/deploymentTargetDropDown.xml

@@ -1,17 +1,17 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
   <component name="deploymentTargetDropDown">
-    <targetSelectedWithDropDown>
+    <runningDeviceTargetSelectedWithDropDown>
       <Target>
-        <type value="QUICK_BOOT_TARGET" />
+        <type value="RUNNING_DEVICE_TARGET" />
         <deviceKey>
           <Key>
-            <type value="VIRTUAL_DEVICE_PATH" />
+            <type value="SERIAL_NUMBER" />
-            <value value="$USER_HOME$/.android/avd/Resizable_Experimental_API_33.avd" />
+            <value value="b36e6c44" />
           </Key>
         </deviceKey>
       </Target>
-    </targetSelectedWithDropDown>
+    </runningDeviceTargetSelectedWithDropDown>
-    <timeTargetWasSelectedWithDropDown value="2023-10-14T04:10:20.140892617Z" />
+    <timeTargetWasSelectedWithDropDown value="2023-10-15T12:47:54.859420162Z" />
   </component>
 </project>

editor-treesitter/src/main/java/io/github/rosemoe/sora/editor/ts/TsAnalyzeManager.kt

@@ -37,6 +37,7 @@ import com.itsaky.androidide.treesitter.api.TreeSitterQueryCapture
 import com.itsaky.androidide.treesitter.api.TreeSitterQueryMatch
 import com.itsaky.androidide.treesitter.string.UTF16String
 import com.itsaky.androidide.treesitter.string.UTF16StringFactory
+import com.itsaky.androidide.utils.ILogger
 import io.github.rosemoe.sora.data.ObjectAllocator
 import io.github.rosemoe.sora.editor.ts.spans.DefaultSpanFactory
 import io.github.rosemoe.sora.editor.ts.spans.TsSpanFactory
@@ -158,6 +159,8 @@ open class TsAnalyzeManager(val languageSpec: TsLanguageSpec, var theme: TsTheme
 
     @Synchronized
     fun nextThreadId() = ++threadId
+
+    private val log = ILogger.newInstance("TsAnalyzeManager")
   }
 
   inner class TsLooperThread : Thread() {
@@ -219,14 +222,15 @@ open class TsAnalyzeManager(val languageSpec: TsLanguageSpec, var theme: TsTheme
     }
 
     fun updateCodeBlocks() {
-      if (languageSpec.blocksQuery.patternCount == 0 || !languageSpec.blocksQuery.canAccess()) {
+      if (languageSpec.blocksQuery.patternCount == 0
+        || !languageSpec.blocksQuery.canAccess()
+        || tree?.canAccess() != true
+      ) {
         return
       }
+
       val blocks = mutableListOf<CodeBlock>()
       TSQueryCursor.create().use {
-        if (tree?.canAccess() != true) {
-          throw IllegalStateException("Cannot access tree")
-        }
 
         val rootNode = tree!!.rootNode
         it.exec(languageSpec.blocksQuery, rootNode)
@@ -274,8 +278,9 @@ open class TsAnalyzeManager(val languageSpec: TsLanguageSpec, var theme: TsTheme
             val hasChanges = rootNode.hasChanges()
             (rootNode as? TreeSitterNode?)?.recycle()
             blocks.clear()
-            throw IllegalStateException(
+            log.info(
-              "Tree closed while querying, rootNode.hasChanges=$hasChanges")
+              "TsLooperThread.updateCodeBlock: Tree closed or edited while querying for code blocks. rootNode.hasChanges=$hasChanges")
+            break
           }
 
           match = it.nextMatch()

editor-treesitter/src/main/java/io/github/rosemoe/sora/editor/ts/TsBracketPairs.kt

@@ -30,17 +30,17 @@ import com.itsaky.androidide.treesitter.TSTree
 import com.itsaky.androidide.treesitter.api.TreeSitterNode
 import com.itsaky.androidide.treesitter.api.TreeSitterQueryCapture
 import com.itsaky.androidide.treesitter.api.TreeSitterQueryMatch
+import com.itsaky.androidide.utils.ILogger
 import io.github.rosemoe.sora.lang.brackets.BracketsProvider
 import io.github.rosemoe.sora.lang.brackets.PairedBracket
 import io.github.rosemoe.sora.text.Content
-import java.lang.Math.max
+import kotlin.math.max
 
-class TsBracketPairs(
+class TsBracketPairs(private val tree: TSTree, private val languageSpec: TsLanguageSpec) :
-  private val tree: TSTree,
+  BracketsProvider {
-  private val languageSpec: TsLanguageSpec
-) : BracketsProvider {
 
   companion object {
+    private val log = ILogger.newInstance("TsBracketPairs")
 
     val OPEN_NAME = "editor.brackets.open"
     val CLOSE_NAME = "editor.brackets.close"
@@ -48,62 +48,56 @@ class TsBracketPairs(
   }
 
   override fun getPairedBracketAt(text: Content, index: Int): PairedBracket? {
-    if (languageSpec.bracketsQuery.patternCount > 0 && languageSpec.bracketsQuery.canAccess() && tree.canAccess()) {
+    if (languageSpec.bracketsQuery.patternCount <= 0
-      TSQueryCursor.create().use { cursor ->
+      || !languageSpec.bracketsQuery.canAccess()
-        cursor.setByteRange(max(0, index - 1) * 2, index * 2 + 1)
+      || !tree.canAccess()
-        if (!tree.canAccess()) {
+    ) {
-          throw IllegalStateException("Cannot access tree")
+      return null
-        }
+    }
-        val rootNode = tree.rootNode
+
-        cursor.exec(languageSpec.bracketsQuery, rootNode)
+    TSQueryCursor.create().use { cursor ->
-        (rootNode as? TreeSitterNode?)?.recycle()
+      cursor.setByteRange(max(0, index - 1) * 2, index * 2 + 1)
-        var match = cursor.nextMatch()
+      val rootNode = tree.rootNode
-        var matched = false
+      cursor.exec(languageSpec.bracketsQuery, rootNode)
-        while (match != null && !matched) {
+      (rootNode as? TreeSitterNode?)?.recycle()
-          if (languageSpec.bracketsPredicator.doPredicate(
+      var match = cursor.nextMatch()
-              languageSpec.predicates,
+      var matched = false
-              text,
+      while (match != null && !matched) {
-              match
+        if (languageSpec.bracketsPredicator.doPredicate(languageSpec.predicates, text, match)) {
-            )
+          var startNode: TSNode? = null
-          ) {
+          var endNode: TSNode? = null
-            var startNode: TSNode? = null
+
-            var endNode: TSNode? = null
+          for (capture in match.captures) {
-            for (capture in match.captures) {
+            val captureName = languageSpec.bracketsQuery.getCaptureNameForId(capture.index)
-              val captureName =
+            if (captureName == OPEN_NAME || captureName == CLOSE_NAME) {
-                languageSpec.bracketsQuery.getCaptureNameForId(capture.index)
+              val node = capture.node
-              if (captureName == OPEN_NAME || captureName == CLOSE_NAME) {
+              if (index >= node.startByte / 2 && index <= node.endByte / 2) {
-                val node = capture.node
+                matched = true
-                if (index >= node.startByte / 2 && index <= node.endByte / 2) {
+              }
-                  matched = true
+              if (captureName == OPEN_NAME) {
-                }
+                startNode = node
-                if (captureName == OPEN_NAME) {
+              } else {
-                  startNode = node
+                endNode = node
-                } else {
-                  endNode = node
-                }
               }
-
-              (capture as? TreeSitterQueryCapture?)?.recycle()
             }
-            if (matched && startNode != null && endNode != null) {
-              return PairedBracket(
-                startNode.startByte / 2,
-                (startNode.endByte - startNode.startByte) / 2,
-                endNode.startByte / 2,
-                (endNode.endByte - endNode.startByte) / 2
-              )
-            }
-          }
-          (match as? TreeSitterQueryMatch?)?.recycle()
 
-          if (!tree.canAccess() || rootNode.hasChanges()) {
+            (capture as? TreeSitterQueryCapture?)?.recycle()
-            val hasChanges = rootNode.hasChanges()
+          }
-            (rootNode as? TreeSitterNode?)?.recycle()
+          if (matched && startNode != null && endNode != null) {
-            throw IllegalStateException(
+            return PairedBracket(startNode.startByte / 2,
-              "Tree closed or edited while querying, rootNode.hasChanges=$hasChanges")
+              (startNode.endByte - startNode.startByte) / 2, endNode.startByte / 2,
+              (endNode.endByte - endNode.startByte) / 2)
           }
-          match = cursor.nextMatch()
         }
+        (match as? TreeSitterQueryMatch?)?.recycle()
+
+        if (!tree.canAccess() || rootNode.hasChanges()) {
+          val hasChanges = rootNode.hasChanges()
+          (rootNode as? TreeSitterNode?)?.recycle()
+          log.info("Tree closed or edited while querying, rootNode.hasChanges=$hasChanges")
+          return null
+        }
+        match = cursor.nextMatch()
       }
     }
     return null

editor-treesitter/src/main/java/io/github/rosemoe/sora/editor/ts/TsScopedVariables.kt

@@ -32,6 +32,7 @@ import com.itsaky.androidide.treesitter.api.TreeSitterNode
 import com.itsaky.androidide.treesitter.api.TreeSitterQueryCapture
 import com.itsaky.androidide.treesitter.api.TreeSitterQueryMatch
 import com.itsaky.androidide.treesitter.string.UTF16String
+import com.itsaky.androidide.utils.ILogger
 import java.util.Stack
 
 /**
@@ -47,6 +48,10 @@ class TsScopedVariables(tree: TSTree, text: UTF16String, val spec: TsLanguageSpe
 
   private val rootScope: Scope
 
+  companion object {
+    private val log = ILogger.newInstance("TsScopedVariables")
+  }
+
   init {
     if (!tree.canAccess()) {
       throw IllegalStateException("Cannot access tree")
@@ -69,8 +74,8 @@ class TsScopedVariables(tree: TSTree, text: UTF16String, val spec: TsLanguageSpe
             val hasChanges = rootNode.hasChanges()
             (rootNode as? TreeSitterNode?)?.recycle()
             captures.clear()
-            throw IllegalStateException(
+            log.info("Tree editor or closed while querying, rootNode.hasChanges=$hasChanges")
-              "Tree closed while querying, rootNode.hasChanges=$hasChanges")
+            break
           }
 
           match = cursor.nextMatch()

editor/src/main/java/com/itsaky/androidide/editor/language/treesitter/TreeSitterLanguage.kt

@@ -74,7 +74,6 @@ abstract class TreeSitterLanguage(context: Context, lang: TSLanguage, type: Stri
       langScheme.styles.forEach { tsTheme.putStyleRule(it.key, it.value.makeStyle()) }
 
       analyzer.langScheme = languageScheme
-      analyzer.rerun()
     }
   }