io.github.mike10004.vhs.harbridge.HarResponseDataTransformer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of virtual-har-server Show documentation
There is a newer version: 0.32
package io.github.mike10004.vhs.harbridge;

import com.google.common.base.Functions;
import com.google.common.collect.ImmutableList;
import com.google.common.io.ByteSource;
import com.google.common.net.HttpHeaders;
import com.google.common.net.MediaType;

import java.util.AbstractMap;
import java.util.Map;
import java.util.Objects;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.stream.Stream;

import static java.util.Objects.requireNonNull;

/**
 * Transformer of {@link HarResponseData} objects. A transformer is a builder
 * that collects operations to be performed on an underlying response data instance.
 * The {@link #transform()} method returns a response data instance that
 * reflects the set of transforms applied.
 * @see HarResponseData#transformer()
 */
public final class HarResponseDataTransformer {

    private final HarResponseData preTransformData;
    private Function bodyTransform = Functions.identity();
    private Function contentTypeTransform = Functions.identity();
    private Function>, Stream>> headersTransform = Functions.identity();

    HarResponseDataTransformer(HarResponseData preTransformData) {
        this.preTransformData = requireNonNull(preTransformData, "preTransformData");
    }

    public HarResponseDataTransformer body(Function bodyTransform) {
        this.bodyTransform = this.bodyTransform.andThen(bodyTransform);
        return this;
    }

    public HarResponseDataTransformer contentType(Function contentTypeSupplier) {
        this.contentTypeTransform = this.contentTypeTransform.andThen(contentTypeSupplier);
        return this;
    }

    public HarResponseDataTransformer headers(Function>, Stream>> headersTransform) {
        this.headersTransform = this.headersTransform.andThen(headersTransform);
        return this;
    }

    public HarResponseDataTransformer filterHeaders(Predicate> filter) {
        return headers(stream -> stream.filter(filter));
    }

    public HarResponseDataTransformer mapHeader(Function, Map.Entry> map) {
        return headers(stream -> stream.map(map));
    }

    public HarResponseDataTransformer mapHeaderWithName(String headerName, Function valueMap) {
        // TODO handle malformed header lists by removing all headers with specified name and appending single new name-value pair
        return mapHeader(entry -> {
            if (headerName.equalsIgnoreCase(entry.getKey())) {
                String newValue = valueMap.apply(entry.getValue());
                if (!Objects.equals(entry.getValue(), newValue)) {
                    return new AbstractMap.SimpleImmutableEntry<>(headerName, newValue);
                }
            }
            return entry;
        });
    }

    public HarResponseDataTransformer replaceHeader(String headerName, String newValue) {
        return mapHeaderWithName(headerName, oldValue -> newValue);
    }

    public HarResponseDataTransformer replaceContentType(MediaType contentType) {
        return replaceHeader(HttpHeaders.CONTENT_TYPE, contentType.toString())
                .contentType(old -> contentType);
    }

    public HarResponseData transform() {
        requireNonNull(preTransformData);
        return new LazyHarResponseData(() -> bodyTransform.apply(preTransformData.getBody()),
                () -> contentTypeTransform.apply(preTransformData.getContentType()),
                () -> headersTransform.apply(preTransformData.headers().stream()).collect(ImmutableList.toImmutableList()));
    }
}