VariableDeclaration.java

package me.tomassetti.turin.parser.ast.statements;

import com.google.common.collect.ImmutableList;
import me.tomassetti.turin.resolvers.SymbolResolver;
import me.tomassetti.turin.parser.ast.Node;
import me.tomassetti.turin.parser.ast.typeusage.TypeUsageNode;
import me.tomassetti.turin.parser.ast.expressions.Expression;
import me.tomassetti.turin.symbols.Symbol;
import me.tomassetti.turin.typesystem.TypeUsage;

import java.util.Optional;

public class VariableDeclaration extends Statement implements Symbol {
    private TypeUsageNode type;
    private String name;
    private Expression value;

    public TypeUsageNode getType() {
        return type;
    }

    public String getName() {
        return name;
    }

    public Expression getValue() {
        return value;
    }

    public VariableDeclaration(String name, Expression value) {
        this.name = name;
        this.value = value;
        this.value.setParent(this);
    }

    @Override
    public Optional<Symbol> findSymbol(String name, SymbolResolver resolver) {
        if (name.equals(this.name)) {
            return Optional.of(this);
        }
        return super.findSymbol(name, resolver);
    }

    @Override
    public TypeUsage calcType() {
        if (type == null) {
            return value.calcType();
        } else {
            return getType();
        }
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        VariableDeclaration that = (VariableDeclaration) o;

        if (!name.equals(that.name)) return false;
        if (type != null ? !type.equals(that.type) : that.type != null) return false;
        if (!value.equals(that.value)) return false;

        return true;
    }

    @Override
    public String toString() {
        return "VariableDeclaration{" +
                "type=" + type +
                ", name='" + name + '\'' +
                ", value=" + value +
                '}';
    }

    @Override
    public int hashCode() {
        int result = type != null ? type.hashCode() : 0;
        result = 31 * result + name.hashCode();
        result = 31 * result + value.hashCode();
        return result;
    }

    public VariableDeclaration(String name, Expression value, TypeUsageNode type) {
        this.name = name;
        this.type = type;
        this.type.setParent(this);
        this.value = value;
        this.value.setParent(this);
    }

    @Override
    public Iterable<Node> getChildren() {
        if (type != null) {
            return ImmutableList.of(type, value);
        } else {
            return ImmutableList.of(value);
        }
    }

    public TypeUsage varType(SymbolResolver resolver) {
        if (type != null) {
            return type;
        } else {
            return value.calcType();
        }
    }
}