io.descoped.dc.api.node.builder.BodyPublisherBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of data-collector-api Show documentation
Data Collector API
The newest version!
package io.descoped.dc.api.node.builder;

import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
import io.descoped.dc.api.node.Base;
import io.descoped.dc.api.node.BodyPart;
import io.descoped.dc.api.node.BodyPublisher;
import io.descoped.dc.api.node.BodyPublisherProducer;
import io.descoped.dc.api.node.FormEncoding;

import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

@JsonDeserialize(using = NodeBuilderDeserializer.class)
public class BodyPublisherBuilder extends OperationPublisherBuilder {

    @JsonIgnore
    FormEncoding encoding;
    @JsonProperty("plainTextData")
    BodyPublisherProducerBuilder plainText;
    @JsonProperty("urlEncodedData")
    BodyPublisherProducerBuilder urlEncodedData;
    @JsonProperty("partsData")
    List parts = new ArrayList<>();

    public BodyPublisherBuilder() {
        super(BuilderType.BodyPublisher);
    }

    public BodyPublisherBuilder plainText(String text) {
        this.plainText = new StringBodyPublisherProducerBuilder(text);
        this.encoding = FormEncoding.TEXT_PLAIN;
        return this;
    }

    public BodyPublisherBuilder plainText(BodyPublisherProducerBuilder bodyPublisherProducerBuilder) {
        this.plainText = bodyPublisherProducerBuilder;
        this.encoding = FormEncoding.TEXT_PLAIN;
        return this;
    }

    public BodyPublisherBuilder urlEncoded(String data) {
        this.urlEncodedData = new StringBodyPublisherProducerBuilder(data);
        this.encoding = FormEncoding.APPLICATION_X_WWW_FORM_URLENCODED;
        return this;
    }

    public BodyPublisherBuilder urlEncoded(BodyPublisherProducerBuilder bodyPublisherProducerBuilder) {
        this.urlEncodedData = bodyPublisherProducerBuilder;
        this.encoding = FormEncoding.APPLICATION_X_WWW_FORM_URLENCODED;
        return this;
    }

    public BodyPublisherBuilder textPart(String name, String value) {
        this.encoding = FormEncoding.MULTIPART_FORM_DATA;
        this.parts.add(new BodyPart(name, value, StandardCharsets.UTF_8));
        return this;
    }

    public BodyPublisherBuilder formPart(String name, String filename, byte[] value) {
        this.encoding = FormEncoding.MULTIPART_FORM_DATA;
        this.parts.add(new BodyPart(name, filename, value, StandardCharsets.UTF_8));
        return this;
    }

    public BodyPublisherBuilder formPart(String name, String filename, String value) {
        this.encoding = FormEncoding.MULTIPART_FORM_DATA;
        this.parts.add(new BodyPart(name, filename, value, StandardCharsets.UTF_8));
        return this;
    }

    @Override
    public  R build(BuildContext buildContext) {
        BodyPublisherProducer plainTextProducer = plainText == null ? null : plainText.build(buildContext);
        BodyPublisherProducer urlEncodedDataProducer = urlEncodedData == null ? null : urlEncodedData.build(buildContext);
        return (R) new BodyPublisherNode(encoding, plainTextProducer, urlEncodedDataProducer, parts);
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        if (!super.equals(o)) return false;
        BodyPublisherBuilder that = (BodyPublisherBuilder) o;
        return encoding == that.encoding &&
                Objects.equals(plainText, that.plainText) &&
                Objects.equals(urlEncodedData, that.urlEncodedData) &&
                Objects.equals(parts, that.parts);
    }

    @Override
    public int hashCode() {
        return Objects.hash(super.hashCode(), encoding, plainText, urlEncodedData, parts);
    }

    @Override
    public String toString() {
        return "BodyPublisherBuilder{" +
                "encoding=" + encoding +
                ", plainText='" + plainText + '\'' +
                ", urlEncodedData='" + urlEncodedData + '\'' +
                ", parts=" + parts +
                '}';
    }

    static class BodyPublisherNode extends LeafNode implements BodyPublisher {

        private final FormEncoding encoding;
        private final BodyPublisherProducer plainText;
        private final BodyPublisherProducer urlEncodedData;
        private final List parts;

        public BodyPublisherNode(FormEncoding encoding, BodyPublisherProducer plainText, BodyPublisherProducer urlEncodedData, List parts) {
            this.encoding = encoding;
            this.plainText = plainText;
            this.urlEncodedData = urlEncodedData;
            this.parts = parts;
        }

        @Override
        public FormEncoding getEncoding() {
            return encoding;
        }

        @Override
        public BodyPublisherProducer getPlainText() {
            return plainText;
        }

        @Override
        public BodyPublisherProducer getUrlEncodedData() {
            return urlEncodedData;
        }

        @Override
        public List getParts() {
            return parts;
        }

        @Override
        public boolean equals(Object o) {
            if (this == o) return true;
            if (o == null || getClass() != o.getClass()) return false;
            BodyPublisherNode that = (BodyPublisherNode) o;
            return encoding == that.encoding &&
                    Objects.equals(plainText, that.plainText) &&
                    Objects.equals(urlEncodedData, that.urlEncodedData) &&
                    Objects.equals(parts, that.parts);
        }

        @Override
        public int hashCode() {
            return Objects.hash(encoding, plainText, urlEncodedData, parts);
        }

        @Override
        public String toString() {
            return "BodyPublisherNode{" +
                    "encoding=" + encoding +
                    ", plainText='" + plainText + '\'' +
                    ", urlEncodedData='" + urlEncodedData + '\'' +
                    ", parts=" + parts +
                    '}';
        }
    }

}