From 341ebf3194d7e66a4424f397ae6756b9aeb58aa9 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 31 Jul 2016 19:28:41 +0100 Subject: Use only where appropriate --- .../java/org/zwobble/couscous/ast/visitors/NodeTransformer.java | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to 'src/main/java/org/zwobble/couscous/ast') diff --git a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java index d44bb39..b75106a 100644 --- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java +++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java @@ -15,6 +15,7 @@ import java.util.Set; import java.util.function.Function; import static org.zwobble.couscous.util.ExtraIterables.lazyMap; +import static org.zwobble.couscous.util.ExtraIterables.only; import static org.zwobble.couscous.util.ExtraLists.*; public class NodeTransformer { @@ -103,10 +104,7 @@ public class NodeTransformer { return transformExpression((ExpressionNode) node); } else if (node instanceof StatementNode) { List statements = ExtraLists.copyOf(transformStatement((StatementNode) node)); - if (statements.size() != 1) { - throw new UnsupportedOperationException("TODO: add support for compound statements"); - } - return statements.get(0); + return only(statements, () -> new UnsupportedOperationException("TODO: add support for compound statements")); } else if (node instanceof TypeNode) { return transformTypeDeclaration((TypeNode) node); } else { -- cgit v1.2.3