feat: fold method body

CHANGELOG.md

@@ -8,6 +8,7 @@
 
 - Support invoke dynamic stmt
 - Support monitor stmts
+- Fold method body
 
 ## [0.1.2] - 2023-10-16
 

src/main/kotlin/pascal/taie/intellij/tir/editing/TirFoldingBuilder.kt

@@ -0,0 +1,29 @@
+package pascal.taie.intellij.tir.editing
+
+import com.intellij.lang.ASTNode
+import com.intellij.lang.folding.FoldingBuilderEx
+import com.intellij.lang.folding.FoldingDescriptor
+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.TirMethodBody
+
+class TirFoldingBuilder : FoldingBuilderEx(), DumbAware {
+    override fun buildFoldRegions(
+        root: PsiElement, document: Document, quick: Boolean
+    ) = SyntaxTraverser
+        .psiTraverser(root)
+        .filter { it is TirMethodBody }
+        .map { TirFoldingDescriptor(it, "{…}") }
+        .toList().toTypedArray()
+
+    override fun getPlaceholderText(node: ASTNode) = "…"
+
+    override fun isCollapsedByDefault(node: ASTNode) = false
+
+    class TirFoldingDescriptor(element: PsiElement, private val holder: String) :
+        FoldingDescriptor(element.node, element.textRange) {
+        override fun getPlaceholderText() = holder
+    }
+}

src/main/resources/META-INF/plugin.xml

@@ -35,6 +35,9 @@
         <lang.syntaxHighlighterFactory
                 language="tir"
                 implementationClass="pascal.taie.intellij.tir.highlight.TirSyntaxHighlighterFactory"/>
+        <lang.foldingBuilder
+                language="tir"
+                implementationClass="pascal.taie.intellij.tir.editing.TirFoldingBuilder"/>
         <additionalTextAttributes scheme="Darcula" file="/colorSchemes/tir.xml"/>
         <additionalTextAttributes scheme="Default" file="/colorSchemes/tir.xml"/>
         <colorSettingsPage