src/main/java/org/zwobble/couscous/ast/AssignmentNode.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

package org.zwobble.couscous.ast;

import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper;

import static org.zwobble.couscous.ast.VariableReferenceNode.reference;

import lombok.Value;

@Value
public class AssignmentNode implements ExpressionNode {
    public static StatementNode assignStatement(AssignableExpressionNode target, ExpressionNode value) {
        return new ExpressionStatementNode(assign(target, value));
    }
    
    public static AssignmentNode assign(AssignableExpressionNode target, ExpressionNode value) {
        return new AssignmentNode(target, value);
    }
    
    public static StatementNode assignStatement(VariableNode target, ExpressionNode value) {
        return new ExpressionStatementNode(assign(target, value));
    }
    
    public static AssignmentNode assign(VariableNode target, ExpressionNode value) {
        return new AssignmentNode(reference(target), value);
    }
    
    AssignableExpressionNode target;
    ExpressionNode value;
    
    @Override
    public <T> T accept(ExpressionNodeMapper<T> visitor) {
        return visitor.visit(this);
    }

    @Override
    public TypeName getType() {
        return value.getType();
    }
}