src/main/java/org/zwobble/couscous/ast/sugar/SwitchNode.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

package org.zwobble.couscous.ast.sugar;

import org.zwobble.couscous.ast.ExpressionNode;
import org.zwobble.couscous.ast.Node;
import org.zwobble.couscous.ast.NodeTypes;
import org.zwobble.couscous.ast.StatementNode;
import org.zwobble.couscous.ast.visitors.NodeTransformer;

import java.util.List;

import static org.zwobble.couscous.util.ExtraIterables.lazyCons;
import static org.zwobble.couscous.util.ExtraLists.eagerMap;

public class SwitchNode implements StatementNode {
    private final ExpressionNode value;
    private final List<SwitchCaseNode> cases;

    public SwitchNode(ExpressionNode value, List<SwitchCaseNode> cases) {
        this.value = value;
        this.cases = cases;
    }

    public ExpressionNode getValue() {
        return value;
    }

    public List<SwitchCaseNode> getCases() {
        return cases;
    }

    @Override
    public int nodeType() {
        return NodeTypes.SWITCH;
    }

    @Override
    public Iterable<? extends Node> childNodes() {
        return lazyCons(value, cases);
    }

    @Override
    public StatementNode transformSubtree(NodeTransformer transformer) {
        return new SwitchNode(
            transformer.transformExpression(value),
            eagerMap(cases, caseNode -> caseNode.transformSubtree(transformer))
        );
    }
}