com.artemis.generator.util.MethodBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of artemis-fluid-core Show documentation

fluid entity api.

There is a newer version: 2.3.0

package com.artemis.generator.util;

import com.artemis.generator.model.artemis.ComponentDescriptor;
import com.artemis.generator.model.type.AccessLevel;
import com.artemis.generator.model.type.MethodDescriptor;
import com.artemis.generator.model.type.ParameterDescriptor;

import java.lang.reflect.Type;

/**
 * Helper builder for MethodDescriptor.
 *
 * @author Daan van Yperen
 */
public class MethodBuilder {

    private final MethodDescriptor method;

    public MethodBuilder(Type returnType, String methodName) {
        method = new MethodDescriptor(returnType, methodName);
    }

    /**
     * Prep method.
     */
    public MethodDescriptor build()
    {
        return method;
    }

    public MethodBuilder returnFluid() {
        method.addStatement("return this");
        return this;
    }

    public MethodBuilder mapper(ComponentDescriptor component, String suffix) {
        mapper("", component, suffix);
        return this;
    }

    public MethodBuilder mapper(String prefix, ComponentDescriptor component, String suffix) {
        method.addStatement(prefix + "mappers.m"+component.getName() + suffix);
        return this;
    }

    public MethodBuilder parameter(Type type, String name) {
        method.addParameter(new ParameterDescriptor(type, name));
        return this;
    }

    public MethodBuilder setStatic(boolean value) {
        method.setStatic(value);
        return this;
    }

    /** Add body, excluding outer brackets. */
    public MethodBuilder body(String value) {
        String[] split = value.split("\n");
        for (String statement : split) {
            method.addStatement(statement);
        }
        return this;
    }

    public MethodBuilder statement(String statement) {
        method.addStatement(statement);
        return this;
    }

    public MethodBuilder accessLevel(AccessLevel level) {
        method.setAccessLevel(level);
        return this;
    }

    public MethodBuilder debugNotes(String message) {
        method.setDebugNotes(message);
        return this;
    }

    public MethodBuilder varArgs(boolean varargs) {
        method.setVarargs(varargs);
        return this;
    }

    public MethodBuilder javaDoc(String javadoc) {
        method.setJavadoc(javadoc);
        return this;
    }
}