Fix quarkiverse#1180: also look into outgoing header while propagating

Signed-off-by: gabriel-farache <[email protected]>

Author: gabriel-farache

client/runtime/src/main/java/io/quarkiverse/openapi/generator/providers/AbstractAuthenticationPropagationHeadersFactory.java

@@ -7,6 +7,8 @@
 import jakarta.ws.rs.core.MultivaluedMap;
 
 import org.eclipse.microprofile.rest.client.ext.ClientHeadersFactory;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import io.quarkiverse.openapi.generator.OpenApiGeneratorConfig;
 
@@ -23,6 +25,8 @@ public abstract class AbstractAuthenticationPropagationHeadersFactory implements
     protected OpenApiGeneratorConfig generatorConfig;
     protected HeadersProvider headersProvider;
 
+    private static final Logger LOGGER = LoggerFactory.getLogger(AbstractAuthenticationPropagationHeadersFactory.class);
+
     protected AbstractAuthenticationPropagationHeadersFactory(BaseCompositeAuthenticationProvider compositeProvider,
             OpenApiGeneratorConfig generatorConfig,
             HeadersProvider headersProvider) {
@@ -36,6 +40,11 @@ public MultivaluedMap<String, String> update(MultivaluedMap<String, String> inco
             MultivaluedMap<String, String> clientOutgoingHeaders) {
         MultivaluedMap<String, String> propagatedHeaders = new MultivaluedHashMap<>();
         MultivaluedMap<String, String> providedHeaders = headersProvider.getStringHeaders(generatorConfig);
+
+        LOGGER.debug("Incoming headers keys{}", incomingHeaders.keySet());
+        LOGGER.debug("Outgoing headers keys{}", clientOutgoingHeaders.keySet());
+        LOGGER.debug("Provided headers keys{}", providedHeaders.keySet());
+
         compositeProvider.getAuthenticationProviders().stream()
                 .filter(AbstractAuthProvider.class::isInstance)
                 .map(AbstractAuthProvider.class::cast)
@@ -46,9 +55,14 @@ public MultivaluedMap<String, String> update(MultivaluedMap<String, String> inco
                     // get priority to the headers coming from the headers provider.
                     List<String> headerValue = providedHeaders.get(headerName);
                     if (headerValue == null) {
-                        // lastly look into the incoming headers.
+                        // next, look into the incoming headers.
                         headerValue = incomingHeaders.get(headerName);
                     }
+                    if (headerValue == null) {
+                        // lastly look into the outgoing headers.
+                        headerValue = clientOutgoingHeaders.get(headerName);
+                    }
+
                     if (headerValue != null) {
                         propagatedHeaders.put(propagationHeaderName(authProvider.getOpenApiSpecId(),
                                 authProvider.getName(),

client/runtime/src/main/java/io/quarkiverse/openapi/generator/providers/BearerAuthenticationProvider.java

@@ -10,6 +10,9 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 /**
  * Provides bearer token authentication or any other valid scheme.
  *
@@ -20,6 +23,7 @@ public class BearerAuthenticationProvider extends AbstractAuthProvider {
     private static final Logger LOGGER = LoggerFactory.getLogger(BearerAuthenticationProvider.class);
 
     private final String scheme;
+    private static final Logger LOGGER = LoggerFactory.getLogger(BearerAuthenticationProvider.class);
 
     public BearerAuthenticationProvider(final String openApiSpecId, final String name, final String scheme,
             List<OperationAuthInfo> operations, CredentialsProvider credentialsProvider) {
@@ -29,6 +33,7 @@ public BearerAuthenticationProvider(final String openApiSpecId, final String nam
 
     @Override
     public void filter(ClientRequestContext requestContext) throws IOException {
+        LOGGER.debug("Headers keys set in incoming requestContext: {}", requestContext.getHeaders().keySet());
         String bearerToken = getBearerToken(requestContext);
 
         if (isTokenPropagation()) {
@@ -43,6 +48,7 @@ public void filter(ClientRequestContext requestContext) throws IOException {
                     getName(), getCanonicalAuthConfigPropertyName(BEARER_TOKEN, getOpenApiSpecId(), getName()),
                     getHeaderForPropagation());
         }
+        LOGGER.debug("Header keys set in filtered requestContext: {}", requestContext.getHeaders().keySet());
     }
 
     private String getBearerToken(ClientRequestContext requestContext) {