fix(lang-treesitter): new AST is unexpectedly edited by main thread s…

…ometimes
Rosemoe · Feb 7, 2025 · f67c11f · f67c11f
1 parent 3783c7b
commit f67c11f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/language-treesitter/src/main/java/io/github/rosemoe/sora/editor/ts/TsAnalyzeManager.kt b/language-treesitter/src/main/java/io/github/rosemoe/sora/editor/ts/TsAnalyzeManager.kt
@@ -178,7 +178,7 @@ open class TsAnalyzeManager(val languageSpec: TsLanguageSpec, var theme: TsTheme
             if (thread == this && messageQueue.isEmpty()) {
                 val oldTree = (styles.spans as LineSpansGenerator?)?.safeTree
                 val newTree = SafeTsTree(tree!!.copy())
-                styles.spans = LineSpansGenerator(
+                val newSpans = LineSpansGenerator(
                     newTree,
                     reference!!.lineCount,
                     reference!!.reference,
@@ -190,6 +190,7 @@ open class TsAnalyzeManager(val languageSpec: TsLanguageSpec, var theme: TsTheme
                 val oldBlocks = styles.blocks
                 updateCodeBlocks()
                 currentReceiver?.setStyles(this@TsAnalyzeManager, styles) {
+                    styles.spans = newSpans
                     oldTree?.close()
                     if (oldBlocks != null) {
                         ObjectAllocator.recycleBlockLines(oldBlocks)