/**
 * 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.am.management.handlers.management.api.provider;

import io.gravitee.common.http.HttpHeaders;

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.brassely at graviteesource.com)
 * @author GraviteeSource Team
 */
@Provider
@PreMatching
public class UriBuilderRequestFilter implements ContainerRequestFilter {

    @Override
    public void filter( ContainerRequestContext ctx ) throws IOException {
        List<String> schemes = ctx.getHeaders().get(HttpHeaders.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(HttpHeaders.X_FORWARDED_HOST);
        if (hosts != null && !hosts.isEmpty()) {
            String host = hosts.get(0);
            UriBuilder builder = ctx.getUriInfo().getRequestUriBuilder();

            if (host.contains(":")) {
                // Forwarded host contains both host and port
                String [] parts = host.split(":");
                builder.host(parts[0]).port(Integer.parseInt(parts[1]));
            } else {
                builder.host(host);
            }

            ctx.setRequestUri(builder.build());
        }
    }
}

@@ -1382,16 +1382,165 @@
 PROTO);%0A
+        UriBuilder baseBuilder = ctx.getUriInfo().getBaseUriBuilder();%0A        UriBuilder requestBuilder = ctx.getUriInfo().getRequestUriBuilder();%0A%0A
         
@@ -1644,70 +1644,80 @@
     
-Uri
+base
 Builder
- builder = ctx.getUriInfo().getR
+.scheme(scheme);%0A            r
 equest
-Uri
 Builder
-(
+.scheme(scheme
 );%0A
+%0A
     
@@ -1743,37 +1743,50 @@
 estUri(b
+aseB
 uilder.
-scheme(scheme)
+build(), requestBuilder
 .build()
@@ -1972,82 +1972,8 @@
 (0);
-%0A            UriBuilder builder = ctx.getUriInfo().getRequestUriBuilder();
 %0A%0A  
@@ -2130,32 +2130,116 @@
                b
+aseBuilder.host(parts%5B0%5D).port(Integer.parseInt(parts%5B1%5D));%0A                requestB
 uilder.host(part
@@ -2316,16 +2316,63 @@
        b
+aseBuilder.host(host);%0A                requestB
 uilder.h
@@ -2428,16 +2428,44 @@
 estUri(b
+aseBuilder.build(), requestB
 uilder.b

Set correct proto and port for base URI and requestURI