From 0106df70c4ee36c368dacffa17403b966acd0ffa Mon Sep 17 00:00:00 2001
From: Kamil Smigielski <kamil.smigielski@allegro.pl>
Date: Thu, 25 Jul 2024 11:41:08 +0200
Subject: [PATCH] test

---
 .../listeners/filters/JwtFilterFactory.kt     | 36 +++++++++++++++----
 1 file changed, 30 insertions(+), 6 deletions(-)

diff --git a/envoy-control-core/src/main/kotlin/pl/allegro/tech/servicemesh/envoycontrol/snapshot/resource/listeners/filters/JwtFilterFactory.kt b/envoy-control-core/src/main/kotlin/pl/allegro/tech/servicemesh/envoycontrol/snapshot/resource/listeners/filters/JwtFilterFactory.kt
index f91f28e19..5c940a6cc 100644
--- a/envoy-control-core/src/main/kotlin/pl/allegro/tech/servicemesh/envoycontrol/snapshot/resource/listeners/filters/JwtFilterFactory.kt
+++ b/envoy-control-core/src/main/kotlin/pl/allegro/tech/servicemesh/envoycontrol/snapshot/resource/listeners/filters/JwtFilterFactory.kt
@@ -25,7 +25,8 @@ class JwtFilterFactory(
     private val properties: JwtFilterProperties
 ) {
 
-    private val jwtProviderBuilders: Map<ProviderName, JwtProvider.Builder> = getJwtProviderBuilders()
+    private val jwtProviderBuilders: Map<ProviderName, JwtProvider> = getJwtProviderBuilders()
+    private val jwtProviderBuilders2: Map<ProviderName, JwtProvider> = getJwtProviderBuilders2()
     private val clientToOAuthProviderName: Map<String, String> =
         properties.providers.entries.flatMap { (providerName, provider) ->
             provider.matchings.keys.map { client -> client to providerName }
@@ -34,11 +35,9 @@ class JwtFilterFactory(
     fun createJwtFilter(group: Group): HttpFilter? {
         val configuredJwtProviders =
             if (group.listenersConfig?.addJwtFailureStatus != false && properties.failedStatusInMetadataEnabled) {
-                jwtProviderBuilders.mapValues {
-                    it.value.setFailedStatusInMetadata(properties.failedStatusInMetadata).build()
-                }
+                jwtProviderBuilders2
             } else {
-                jwtProviderBuilders.mapValues { it.value.clearFailedStatusInMetadata().build() }
+                jwtProviderBuilders
             }
 
         return if (shouldCreateFilter(group)) {
@@ -68,7 +67,7 @@ class JwtFilterFactory(
     private fun containsClientsWithSelector(it: IncomingEndpoint) =
         clientToOAuthProviderName.keys.intersect(it.clients.map { it.name }).isNotEmpty()
 
-    private fun getJwtProviderBuilders(): Map<ProviderName, JwtProvider.Builder> =
+    private fun getJwtProviderBuilders(): Map<ProviderName, JwtProvider> =
         properties.providers.entries.associate {
             it.key to createProviderBuilder(it.value)
         }
@@ -88,6 +87,31 @@ class JwtFilterFactory(
         .setForward(properties.forwardJwt)
         .setForwardPayloadHeader(properties.forwardPayloadHeader)
         .setPayloadInMetadata(properties.payloadInMetadata)
+        .build()
+
+    private fun getJwtProviderBuilders2(): Map<ProviderName, JwtProvider> =
+        properties.providers.entries.associate {
+            it.key to createProviderBuilder(it.value)
+        }
+
+    private fun createProviderBuilder2(provider: OAuthProvider) = JwtProvider.newBuilder()
+        .setRemoteJwks(
+            RemoteJwks.newBuilder().setHttpUri(
+                HttpUri.newBuilder()
+                    .setUri(provider.jwksUri.toString())
+                    .setCluster(provider.clusterName)
+                    .setTimeout(
+                        Durations.fromMillis(provider.connectionTimeout.toMillis())
+                    ).build()
+            )
+                .setCacheDuration(Durations.fromMillis(provider.cacheDuration.toMillis()))
+        )
+        .setForward(properties.forwardJwt)
+        .setForwardPayloadHeader(properties.forwardPayloadHeader)
+        .setPayloadInMetadata(properties.payloadInMetadata)
+        .setFailedStatusInMetadata(properties.failedStatusInMetadata)
+        .build()
+
 
     private fun createRules(endpoints: List<IncomingEndpoint>): Set<RequirementRule> {
         return endpoints.mapNotNull(this::createRuleForEndpoint).toSet()