diff --git a/src/main/java/org/jastadd/tooling/aspect/AspectBlock.java b/src/main/java/org/jastadd/tooling/aspect/AspectBlock.java index af11eb3130ab23a400e2a1cf100cc19fe360320d..f6c6d7b56a313e8ed01eda0314da647d8abf3788 100644 --- a/src/main/java/org/jastadd/tooling/aspect/AspectBlock.java +++ b/src/main/java/org/jastadd/tooling/aspect/AspectBlock.java @@ -38,6 +38,8 @@ public class AspectBlock extends AbstractBlock { myInjectedBlockBuilder.addInjectedBlocks(blocks, myNode, Wrap.createWrap(WrapType.NONE, false), null, Indent.getIndent(Indent.Type.NONE, false, true)); } else if (myNode.getPsi() instanceof JastAddAspectBlock) { myInjectedBlockBuilder.addInjectedBlocks(blocks, myNode, Wrap.createWrap(WrapType.NONE, false), null, Indent.getIndent(Indent.Type.NONE, false, true)); + } else if (myNode.getPsi() instanceof JastAddAspectExpression) { + myInjectedBlockBuilder.addInjectedBlocks(blocks, myNode, Wrap.createWrap(WrapType.NONE, false), null, Indent.getIndent(Indent.Type.NONE, false, true)); } else { ASTNode child = myNode.getFirstChildNode(); while (child != null) { @@ -60,6 +62,8 @@ public class AspectBlock extends AbstractBlock { return Indent.getNormalIndent(); } else if (parentPsi instanceof JastAddAspectAspectConstructorDeclaration && (myNode.getPsi() instanceof JastAddAspectExplicitConstructorInvocation || myNode.getPsi() instanceof JastAddAspectBlockStatement)) { return Indent.getNormalIndent(); + } else if (parentPsi instanceof JastAddAspectCollectionAttribute && (childType == AspectTypes.LBRACKET || childType == AspectTypes.ROOT)) { + return Indent.getNormalIndent(); } else if (parentPsi instanceof JastAddAspectCollectionContribution && (childType == AspectTypes.WHEN || childType == AspectTypes.TO || childType == AspectTypes.FOR)) { return Indent.getNormalIndent(); } @@ -69,8 +73,6 @@ public class AspectBlock extends AbstractBlock { } if (nonBlankSuccecssor != null && nonBlankSuccecssor.getElementType() == AspectTypes.ASSIGN) { return Indent.getContinuationIndent(); - } else if (nonBlankSuccecssor != null) { - //System.out.println(nonBlankSuccecssor.getElementType().getDebugName()); } } return Indent.getNoneIndent();