io.muserver.ContextHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mu-server Show documentation
A simple but powerful web server framework
The newest version!
package io.muserver;

import java.net.URI;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * A handler that wraps a list of other handlers and serves them at a certain path prefix (or context).
 * Use this when you want to serve multiple handlers with the same path prefix.
 */
public class ContextHandler implements MuHandler {

    private final boolean hasContext;
    private final String contextPath;
    private final List muHandlers;
    private final String slashContextSlash;
    private final String slashContext;

    /**
     * Creates a new handler
     * @param contextPath The patch
     * @param muHandlers The handlers
     */
    ContextHandler(String contextPath, List muHandlers) {
        String slashTrimmed = Mutils.trim(Mutils.coalesce(contextPath, "").trim(), "/");
        this.hasContext = !slashTrimmed.isEmpty();
        this.contextPath = Stream.of(slashTrimmed.split("/"))
            .map(Mutils::urlEncode)
            .collect(Collectors.joining("/"));
        this.muHandlers = muHandlers;
        this.slashContextSlash = "/" + this.contextPath + "/";
        this.slashContext = "/" + this.contextPath;
    }

    @Override
    public boolean handle(MuRequest request, MuResponse response) throws Exception {
        String rp = request.relativePath();
        if (hasContext && rp.equals(slashContext)) {
            URI cur = request.uri();
            URI newUri;
            if (cur.getRawQuery() == null) {
                newUri = cur.resolve(cur.getRawPath() + "/");
            } else {
                newUri = cur.resolve(cur.getRawPath() + "/?" + cur.getRawQuery());
            }
            response.redirect(newUri);
            return true;
        }
        if (rp.startsWith(slashContextSlash) || !hasContext) {
            String originalContextPath = request.contextPath();
            String originalRelativePath = request.relativePath();
            if (hasContext) {
                ((NettyRequestAdapter) request).addContext(contextPath);
            }
            for (MuHandler muHandler : muHandlers) {
                if (muHandler.handle(request, response)) {
                    return true;
                }
            }
            ((NettyRequestAdapter) request).setPaths(originalContextPath, originalRelativePath);
        }
        return false;
    }

    @Override
    public String toString() {
        return "ContextHandler{" +
            "context='" + contextPath + '\'' +
            ", children='" + muHandlers + '\'' +
            '}';
    }
}