From a919c4d4aa5a917e4cbed9eed809505da05a30c7 Mon Sep 17 00:00:00 2001
From: Katia Aresti <karesti@redhat.com>
Date: Thu, 4 Jul 2024 13:49:40 +0200
Subject: [PATCH] Infinispan Extension adds SASL to service providers

---
 .../deployment/InfinispanClientProcessor.java | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/extensions/infinispan-client/deployment/src/main/java/io/quarkus/infinispan/client/deployment/InfinispanClientProcessor.java b/extensions/infinispan-client/deployment/src/main/java/io/quarkus/infinispan/client/deployment/InfinispanClientProcessor.java
index a52ff559827ff9..60edd8dd08944d 100644
--- a/extensions/infinispan-client/deployment/src/main/java/io/quarkus/infinispan/client/deployment/InfinispanClientProcessor.java
+++ b/extensions/infinispan-client/deployment/src/main/java/io/quarkus/infinispan/client/deployment/InfinispanClientProcessor.java
@@ -56,6 +56,14 @@
 import org.jboss.jandex.DotName;
 import org.jboss.jandex.IndexView;
 import org.jboss.jandex.Type;
+import org.wildfly.security.WildFlyElytronBaseProvider;
+import org.wildfly.security.sasl.digest.WildFlyElytronSaslDigestProvider;
+import org.wildfly.security.sasl.external.WildFlyElytronSaslExternalProvider;
+import org.wildfly.security.sasl.gs2.WildFlyElytronSaslGs2Provider;
+import org.wildfly.security.sasl.gssapi.WildFlyElytronSaslGssapiProvider;
+import org.wildfly.security.sasl.oauth2.WildFlyElytronSaslOAuth2Provider;
+import org.wildfly.security.sasl.plain.WildFlyElytronSaslPlainProvider;
+import org.wildfly.security.sasl.scram.WildFlyElytronSaslScramProvider;
 
 import io.quarkus.arc.deployment.AdditionalBeanBuildItem;
 import io.quarkus.arc.deployment.BeanContainerBuildItem;
@@ -303,6 +311,11 @@ InfinispanPropertiesBuildItem setup(ApplicationArchivesBuildItem applicationArch
                         "org.infinispan.client.hotrod.impl.consistenthash.SegmentConsistentHash")
                         .build());
 
+        reflectiveClass.produce(
+                ReflectiveClassBuildItem.builder(
+                        "org.infinispan.client.hotrod.impl.consistenthash.SegmentConsistentHash")
+                        .build());
+
         return new InfinispanPropertiesBuildItem(propertiesMap);
     }
 
@@ -465,6 +478,19 @@ void registerServiceBinding(Capabilities capabilities, BuildProducer<ServiceProv
                     new ServiceProviderBuildItem(SERVICE_BINDING_INTERFACE_NAME,
                             InfinispanServiceBindingConverter.class.getName()));
         }
+
+        List<Class<? extends WildFlyElytronBaseProvider>> saslClasses = List.of(
+                WildFlyElytronSaslPlainProvider.class,
+                WildFlyElytronSaslDigestProvider.class,
+                WildFlyElytronSaslExternalProvider.class,
+                WildFlyElytronSaslOAuth2Provider.class,
+                WildFlyElytronSaslScramProvider.class,
+                WildFlyElytronSaslGssapiProvider.class,
+                WildFlyElytronSaslGs2Provider.class);
+
+        saslClasses.forEach(clazz -> buildProducer.produce(new ServiceProviderBuildItem(clazz.getName(),
+                clazz.getName())));
+
     }
 
     class RemoteCacheBean {