package net.hellmann.platform.authproxy.proxy.filter;

import com.netflix.util.Pair;
import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import org.springframework.stereotype.Component;
import org.springframework.web.util.UrlPathHelper;

import java.util.Optional;

@Component
public class LocationHeaderRewritingFilter extends ZuulFilter {

    private final UrlPathHelper urlPathHelper = new UrlPathHelper();

    @Override
    public String filterType() {
        return "post";
    }

    @Override
    public int filterOrder() {
        return 100;
    }

    @Override
    public boolean shouldFilter() {
        return extractLocationHeader(RequestContext.getCurrentContext()).isPresent();
    }

    @Override
    public Object run() {
        RequestContext ctx = RequestContext.getCurrentContext();
        Pair<String, String> lh = extractLocationHeader(ctx).get();
        lh.setSecond(lh.second().replace("http://", "https://"));
        return null;
    }

    private Optional<Pair<String, String>> extractLocationHeader(RequestContext ctx) {
        return ctx.getZuulResponseHeaders().stream().filter(p -> "Location".equals(p.first())).findFirst();
    }
}