org.raml.builder.ParameterBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of raml-builder Show documentation

There is a newer version: 1.0.8

package org.raml.builder;

import org.raml.v2.internal.impl.commons.nodes.TypeDeclarationNode;
import org.raml.yagi.framework.nodes.KeyValueNode;
import org.raml.yagi.framework.nodes.KeyValueNodeImpl;
import org.raml.yagi.framework.nodes.Node;
import org.raml.yagi.framework.nodes.StringNodeImpl;

/**
 * Created. There, you have it.
 */
public class ParameterBuilder extends KeyValueNodeBuilder {

    private TypeBuilder type;
    private String displayName;
    private String description;
    private Boolean required;

    private ParameterBuilder(String name) {
        super(name);
    }

    public static ParameterBuilder parameter(String name) {

        return new ParameterBuilder(name);
    }

    public ParameterBuilder ofType(String name) {

        this.type = TypeBuilder.type(name);
        return this;
    }

    public ParameterBuilder ofType(TypeBuilder name) {

        this.type = name;
        return this;
    }

    @Override
    protected Node createValueNode() {
        return new TypeDeclarationNode();
    }

    @Override
    public KeyValueNode buildNode() {

        KeyValueNode node = super.buildNode();

        if ( type != null ) {

            node.getValue().addChild(new KeyValueNodeImpl(new StringNodeImpl("type"), type.buildNode()));
        }

        addProperty(node.getValue(), "displayName", displayName);
        addProperty(node.getValue(), "description", description);
        addProperty(node.getValue(), "required", required);

        return node;
    }

    public ParameterBuilder displayName(String displayName) {

        this.displayName = displayName;
        return this;
    }

    public ParameterBuilder description(String description) {

        this.description = description;
        return this;
    }

    public ParameterBuilder required(boolean required) {

        this.required = required;
        return this;
    }

    public boolean required() {
        return required;
    }
}