From ccf6a49cde0f5478ce32d3182988c4b091d50010 Mon Sep 17 00:00:00 2001
From: Johannes Mey <johannes.mey@tu-dresden.de>
Date: Sat, 27 Nov 2021 01:32:55 +0100
Subject: [PATCH] how could I forget the folding of annotation type bodies?

---
 .../org/jastadd/tooling/aspect/AspectFoldingBuilder.java    | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/main/java/org/jastadd/tooling/aspect/AspectFoldingBuilder.java b/src/main/java/org/jastadd/tooling/aspect/AspectFoldingBuilder.java
index 9fc57f8..5e28dbf 100644
--- a/src/main/java/org/jastadd/tooling/aspect/AspectFoldingBuilder.java
+++ b/src/main/java/org/jastadd/tooling/aspect/AspectFoldingBuilder.java
@@ -40,6 +40,12 @@ public class AspectFoldingBuilder extends FoldingBuilderEx {
         new TextRange(enumBody.getTextRange().getStartOffset() + 1,
           enumBody.getTextRange().getEndOffset() - 1)));
     }
+    Collection<JastAddAspectAnnotationTypeBody> annotationTypeBodies = PsiTreeUtil.findChildrenOfType(root, JastAddAspectAnnotationTypeBody.class);
+    for (final JastAddAspectAnnotationTypeBody annotationTypeBody : annotationTypeBodies) {
+      descriptors.add(new FoldingDescriptor(annotationTypeBody.getNode(),
+        new TextRange(annotationTypeBody.getTextRange().getStartOffset() + 1,
+          annotationTypeBody.getTextRange().getEndOffset() - 1)));
+    }
     Collection<JastAddAspectBlock> aspectBlocks = PsiTreeUtil.findChildrenOfType(root, JastAddAspectBlock.class);
     for (final JastAddAspectBlock block : aspectBlocks) {
       descriptors.add(new FoldingDescriptor(block.getNode(),
-- 
GitLab