diff --git a/src/main/java/org/jastadd/tooling/aspect/AspectFoldingBuilder.java b/src/main/java/org/jastadd/tooling/aspect/AspectFoldingBuilder.java index 9fc57f8ecf188095bada03893af729b7c79fc495..5e28dbfe6259e790e2dc3f12669d00bbfbb79245 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(),