/**
 * Copyright (C) 2015 The Gravitee team (http://gravitee.io)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.gravitee.management.rest.provider;

import javax.ws.rs.container.ContainerRequestContext;
import javax.ws.rs.container.ContainerRequestFilter;
import javax.ws.rs.container.PreMatching;
import javax.ws.rs.core.UriBuilder;
import javax.ws.rs.ext.Provider;
import java.io.IOException;
import java.util.List;

/**
 * @author David BRASSELY (david at gravitee.io)
 * @author GraviteeSource Team
 */
@Provider
@PreMatching
public class UriBuilderRequestFilter implements ContainerRequestFilter {

    @Override
    public void filter( ContainerRequestContext ctx ) throws IOException {
        List<String> schemes = ctx.getHeaders().get("x-forwarded-proto");
        if (schemes != null && !schemes.isEmpty()) {
            String scheme = schemes.get(0);
            UriBuilder builder = ctx.getUriInfo().getRequestUriBuilder();
            ctx.setRequestUri(builder.scheme(scheme).build());
        }

        List<String> hosts = ctx.getHeaders().get("x-forwarded-host");
        if (hosts != null && !hosts.isEmpty()) {
            String host = hosts.get(0);
            UriBuilder builder = ctx.getUriInfo().getRequestUriBuilder();
            ctx.setRequestUri(builder.host(host).build());
        }
    }
}

@@ -671,16 +671,61 @@
 vider;%0A%0A
+import io.gravitee.common.http.HttpHeaders;%0A%0A
 import j
@@ -770,16 +770,16 @@
 ontext;%0A
-
 import j
@@ -1316,27 +1316,37 @@
 get(
-%22x-forwarded-proto%22
+HttpHeaders.X_FORWARDED_PROTO
 );%0A 
@@ -1643,26 +1643,36 @@
 get(
-%22x-forwarded-host%22
+HttpHeaders.X_FORWARDED_HOST
 );%0A 
@@ -1825,32 +1825,329 @@
 stUriBuilder();%0A
+%0A            if (host.contains(%22:%22)) %7B%0A                // Forwarded host contains both host and port%0A                String %5B%5D parts = host.split(%22:%22);%0A                builder.host(parts%5B0%5D).port(Integer.parseInt(parts%5B1%5D));%0A            %7D else %7B%0A                builder.host(host);%0A            %7D%0A%0A
             ctx.
@@ -2172,19 +2172,8 @@
 der.
-host(host).
 buil

Set correct location when having x-forwarded-host composed of both host and port