path: root/src/main/java/org/zwobble/couscous/ast/MethodNode.java

package org.zwobble.couscous.ast;

import com.google.common.collect.ImmutableList;
import org.zwobble.couscous.types.Type;
import org.zwobble.couscous.ast.visitors.NodeMapper;
import org.zwobble.couscous.ast.visitors.NodeTransformer;
import org.zwobble.couscous.types.Types;

import java.util.List;
import java.util.Optional;
import java.util.function.Function;

import static org.zwobble.couscous.util.ExtraLists.eagerMap;
import static org.zwobble.couscous.util.ExtraLists.list;

public class MethodNode implements Node {
    public static Builder staticMethod(String name) {
        return builder(name).isStatic(true);
    }
    public static Builder builder(String name) {
        return new Builder(name);
    }

    public static class Builder {
        private final ImmutableList.Builder<AnnotationNode> annotations =
            ImmutableList.builder();
        private boolean isStatic = false;
        private boolean isAbstract = false;
        private final String name;
        private final ImmutableList.Builder<FormalArgumentNode> arguments =
            ImmutableList.builder();
        private Type returnType = Types.VOID;
        private final ImmutableList.Builder<StatementNode> body =
            ImmutableList.builder();
        private final ImmutableList.Builder<MethodSignature> overrides =
            ImmutableList.builder();

        public Builder(String name) {
            this.name = name;
        }
        
        public Builder annotations(List<AnnotationNode> annotations) {
            this.annotations.addAll(annotations);
            return this;
        }
        
        public Builder isStatic(boolean isStatic) {
            this.isStatic = isStatic;
            return this;
        }

        public Builder isAbstract() {
            this.isAbstract = true;
            return this;
        }
        
        public Builder argument(FormalArgumentNode argument) {
            arguments.add(argument);
            return this;
        }
        
        public Builder arguments(List<FormalArgumentNode> arguments) {
            this.arguments.addAll(arguments);
            return this;
        }

        public Builder returns(Type returnType) {
            this.returnType = returnType;
            return this;
        }
        
        public Builder statement(StatementNode statement) {
            this.body.add(statement);
            return this;
        }
        
        public Builder body(List<StatementNode> statements) {
            this.body.addAll(statements);
            return this;
        }
        
        public MethodNode build() {
            return new MethodNode(
                annotations.build(),
                isStatic,
                name,
                list(),
                arguments.build(),
                returnType,
                isAbstract ? Optional.empty() : Optional.of(body.build()),
                overrides.build());
        }
    }
    
    public static MethodNode method(
        List<AnnotationNode> annotations,
        boolean isStatic,
        String name,
        List<FormalTypeParameterNode> typeParameters,
        List<FormalArgumentNode> arguments,
        Type returnType,
        Optional<List<StatementNode>> body,
        List<MethodSignature> overrides)
    {
        return new MethodNode(annotations, isStatic, name, typeParameters, arguments, returnType, body, overrides);
    }
    
    private final List<AnnotationNode> annotations;
    private final boolean isStatic;
    private final String name;
    private final List<FormalTypeParameterNode> typeParameters;
    private final List<FormalArgumentNode> arguments;
    private final Type returnType;
    private final Optional<List<StatementNode>> body;
    private final List<MethodSignature> overrides;

    private MethodNode(
        List<AnnotationNode> annotations,
        boolean isStatic,
        String name,
        List<FormalTypeParameterNode> typeParameters,
        List<FormalArgumentNode> arguments,
        Type returnType,
        Optional<List<StatementNode>> body,
        List<MethodSignature> overrides)
    {
        this.annotations = annotations;
        this.isStatic = isStatic;
        this.name = name;
        this.typeParameters = typeParameters;
        this.arguments = arguments;
        this.returnType = returnType;
        this.body = body;
        this.overrides = overrides;
    }
    
    public List<AnnotationNode> getAnnotations() {
        return annotations;
    }
    
    public boolean isStatic() {
        return isStatic;
    }
    
    public String getName() {
        return name;
    }

    public List<FormalTypeParameterNode> getTypeParameters() {
        return typeParameters;
    }

    public List<FormalArgumentNode> getArguments() {
        return arguments;
    }

    public Type getReturnType() {
        return returnType;
    }
    
    public Optional<List<StatementNode>> getBody() {
        return body;
    }

    public List<MethodSignature> getOverrides() {
        return overrides;
    }

    public boolean isAbstract() {
        return !body.isPresent();
    }

    public MethodSignature signature() {
        return new MethodSignature(
            name,
            eagerMap(typeParameters, parameter -> parameter.getType()),
            eagerMap(arguments, argument -> argument.getType()),
            returnType);
    }

    public MethodNode mapBody(Function<List<StatementNode>, List<StatementNode>> function) {
        return new MethodNode(annotations, isStatic, name, typeParameters, arguments, returnType, body.map(function), overrides);
    }

    @Override
    public <T> T accept(NodeMapper<T> visitor) {
        return visitor.visit(this);
    }

    public MethodNode transform(NodeTransformer transformer) {
        return new MethodNode(
            transformer.transformAnnotations(annotations),
            isStatic,
            transformer.transformMethodName(signature()),
            transformer.transformFormalTypeParameters(typeParameters),
            transformer.transformFormalArguments(arguments),
            transformer.transform(returnType),
            body.map(transformer::transformStatements),
            eagerMap(overrides, transformer::transform));
    }

    @Override
    public String toString() {
        return "MethodNode(" +
            "annotations=" + annotations +
            ", isStatic=" + isStatic +
            ", name=" + name +
            ", typeParameters=" + typeParameters +
            ", arguments=" + arguments +
            ", returnType=" + returnType +
            ", body=" + body +
            ", overrides=" + overrides +
            ')';
    }

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

        MethodNode that = (MethodNode) o;

        if (isStatic != that.isStatic) return false;
        if (!annotations.equals(that.annotations)) return false;
        if (!name.equals(that.name)) return false;
        if (!typeParameters.equals(that.typeParameters)) return false;
        if (!arguments.equals(that.arguments)) return false;
        if (!returnType.equals(that.returnType)) return false;
        if (!body.equals(that.body)) return false;
        return overrides.equals(that.overrides);

    }

    @Override
    public int hashCode() {
        int result = annotations.hashCode();
        result = 31 * result + (isStatic ? 1 : 0);
        result = 31 * result + name.hashCode();
        result = 31 * result + typeParameters.hashCode();
        result = 31 * result + arguments.hashCode();
        result = 31 * result + returnType.hashCode();
        result = 31 * result + body.hashCode();
        result = 31 * result + overrides.hashCode();
        return result;
    }
}