feat: fold class body

Tai-e · Oct 18, 2023 · b0e4b8e · b0e4b8e
1 parent ed2712a
commit b0e4b8e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,7 @@
 ### Changed
 
 - Fix store static field
+- Fold class body
 
 ## [0.1.3] - 2023-10-17
 

diff --git a/src/main/kotlin/pascal/taie/intellij/tir/editing/TirFoldingBuilder.kt b/src/main/kotlin/pascal/taie/intellij/tir/editing/TirFoldingBuilder.kt
@@ -7,14 +7,15 @@ import com.intellij.openapi.editor.Document
 import com.intellij.openapi.project.DumbAware
 import com.intellij.psi.PsiElement
 import com.intellij.psi.SyntaxTraverser
+import pascal.taie.intellij.tir.syntax.psi.TirClassBody
 import pascal.taie.intellij.tir.syntax.psi.TirMethodBody
 
 class TirFoldingBuilder : FoldingBuilderEx(), DumbAware {
     override fun buildFoldRegions(
         root: PsiElement, document: Document, quick: Boolean
     ) = SyntaxTraverser
         .psiTraverser(root)
-        .filter { it is TirMethodBody }
+        .filter { it is TirMethodBody || it is TirClassBody }
         .map { TirFoldingDescriptor(it, "{…}") }
         .toList().toTypedArray()