-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSwitchCase.java
More file actions
100 lines (80 loc) · 2.97 KB
/
Copy pathSwitchCase.java
File metadata and controls
100 lines (80 loc) · 2.97 KB
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package jtree.nodes;
import static jtree.util.Utils.*;
import java.util.List;
import java.util.function.Consumer;
import jtree.util.Either;
import jtree.util.Utils;
import lombok.EqualsAndHashCode;
import lombok.Getter;
import lombok.Setter;
import lombok.NonNull;
@EqualsAndHashCode
@Getter @Setter
public class SwitchCase extends Node {
protected @NonNull List<? extends Expression> labels;
protected @NonNull Either<? extends List<? extends Statement>, ? extends Statement> body;
public SwitchCase(Either<? extends List<? extends Statement>, ? extends Statement> body) {
this(emptyList(), body);
}
public SwitchCase(@NonNull List<? extends Statement> body) {
this(Either.first(body));
}
public SwitchCase(@NonNull Statement body) {
this(Either.second(body));
}
public SwitchCase(List<? extends Expression> labels, @NonNull List<? extends Statement> body) {
this(labels, Either.first(body));
}
public SwitchCase(List<? extends Expression> labels, @NonNull Statement body) {
this(labels, Either.second(body));
}
public SwitchCase(List<? extends Expression> labels, Either<? extends List<? extends Statement>, ? extends Statement> body) {
setLabels(labels);
setBody(body);
}
@Override
public SwitchCase clone() {
return new SwitchCase(clone(getLabels()), clone(getBody()));
}
@Override
public String toCode() {
var labels = getLabels();
return (labels.isEmpty()? "default" : "case " + joinNodes(", ", labels))
+ body.unravel(stmts -> ":" + (stmts.isEmpty()
? ""
: stmts.size() == 1 && stmts.get(0) instanceof Block
? " " + stmts.get(0).toCode()
: "\n" + join("", stmts, stmt -> stmt.toCode().indent(4))),
stmt -> " -> " + stmt.toCode());
}
public void setBody(Either<? extends List<? extends Statement>, ? extends Statement> body) {
this.body = body.map(Utils::newList, (@NonNull var stmt) -> {
if(!(stmt instanceof Block || stmt instanceof ThrowStmt || stmt instanceof ExpressionStmt)) {
throw new IllegalArgumentException("single-statement case bodies can only be blocks, throw statements, or expression statements.");
}
return stmt;
});
}
public final void setBody(@NonNull List<? extends Statement> body) {
setBody(Either.first(body));
}
public final void setBody(@NonNull Block body) {
setBody(Either.second(body));
}
public final void setBody(@NonNull ThrowStmt body) {
setBody(Either.second(body));
}
public final void setBody(@NonNull ExpressionStmt body) {
setBody(Either.second(body));
}
public final void setBody(Expression body) {
setBody(Either.second(new ExpressionStmt(body)));
}
@Override
public <N extends INode> void accept(TreeVisitor visitor, Node parent, Consumer<N> replacer) {
if(visitor.visitSwitchCase(this, parent, cast(replacer))) {
visitList(visitor, getLabels());
getBody().accept(stmts -> visitList(visitor, stmts), stmt -> stmt.<Statement>accept(visitor, this, newstmt -> setBody(Either.second(newstmt))));
}
}
}