io.github.mike10004.vhs.bmp.HostRewriter 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.bmp;

import com.google.common.net.HostAndPort;
import com.google.common.net.HttpHeaders;
import io.netty.handler.codec.http.HttpRequest;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

interface HostRewriter {

    void replaceHost(HttpRequest request);

    static HostRewriter from(HostAndPort replacement) {
        String rewrittenHost = replacement.toString();
        return new HostRewriter() {

            private final Logger log = LoggerFactory.getLogger(HostRewriter.class.getName() + ".from.HostAndPort");

            @Override
            public void replaceHost(HttpRequest request) {
                log.debug("replacing URI and host header {} -> {}", request.headers().get(HttpHeaders.HOST), rewrittenHost);
                request.setUri(rewrittenHost);
                request.headers().set(com.google.common.net.HttpHeaders.HOST, rewrittenHost);
            }
        };
    }
}