Add EntityAgent support to JpaInjectionServices SPI

pom.xml

@@ -100,7 +100,7 @@
         <cdi.api.version>5.0.0.Beta1</cdi.api.version>
         <ejb.api.version>4.0.1</ejb.api.version>
         <jaxrs.api.version>4.0.0</jaxrs.api.version>
-        <jpa.api.version>3.2.0</jpa.api.version>
+        <jpa.api.version>4.0.0-M4</jpa.api.version>
         <jta.api.version>2.0.1</jta.api.version>
         <interceptor.api.version>2.2.0</interceptor.api.version>
         <servlet.api.version>6.1.0</servlet.api.version>

weld-spi/src/main/java/org/jboss/weld/injection/spi/JpaInjectionServices.java

@@ -17,8 +17,10 @@
 package org.jboss.weld.injection.spi;
 
 import jakarta.enterprise.inject.spi.InjectionPoint;
+import jakarta.persistence.EntityAgent;
 import jakarta.persistence.EntityManager;
 import jakarta.persistence.EntityManagerFactory;
+import jakarta.persistence.PersistenceAgent;
 import jakarta.persistence.PersistenceContext;
 import jakarta.persistence.PersistenceUnit;
 
@@ -61,4 +63,18 @@ public interface JpaInjectionServices extends Service {
      */
     ResourceReferenceFactory<EntityManagerFactory> registerPersistenceUnitInjectionPoint(InjectionPoint injectionPoint);
 
+    /**
+     * Register a persistence agent injection point. The implementation validates the injection point. If the validation passes,
+     * an instance of {@link ResourceReferenceFactory} is returned which may be used at runtime for creating instances of the
+     * resource.
+     *
+     * @param injectionPoint the injection point metadata
+     * @return factory for obtaining {@link EntityAgent} instances
+     * @throws IllegalArgumentException if the injection point is not annotated with {@link PersistenceAgent},
+     *         if the injection point is a method that doesn't follow JavaBean conventions
+     * @throws IllegalStateException if no suitable persistence units can be resolved
+     * @since 7.0
+     */
+    ResourceReferenceFactory<EntityAgent> registerPersistenceAgentInjectionPoint(InjectionPoint injectionPoint);
+
 }

weld-spi/src/main/java/org/jboss/weld/injection/spi/helpers/ForwardingJpaInjectionServices.java

@@ -17,6 +17,7 @@
 package org.jboss.weld.injection.spi.helpers;
 
 import jakarta.enterprise.inject.spi.InjectionPoint;
+import jakarta.persistence.EntityAgent;
 import jakarta.persistence.EntityManager;
 import jakarta.persistence.EntityManagerFactory;
 
@@ -48,6 +49,10 @@ public ResourceReferenceFactory<EntityManagerFactory> registerPersistenceUnitInj
         return delegate().registerPersistenceUnitInjectionPoint(injectionPoint);
     }
 
+    public ResourceReferenceFactory<EntityAgent> registerPersistenceAgentInjectionPoint(InjectionPoint injectionPoint) {
+        return delegate().registerPersistenceAgentInjectionPoint(injectionPoint);
+    }
+
     @Override
     public String toString() {
         return delegate().toString();

weld-spi/src/test/java/org/jboss/weld/bootstrap/api/test/MockJpaServices.java

@@ -17,6 +17,7 @@
 package org.jboss.weld.bootstrap.api.test;
 
 import jakarta.enterprise.inject.spi.InjectionPoint;
+import jakarta.persistence.EntityAgent;
 import jakarta.persistence.EntityManager;
 import jakarta.persistence.EntityManagerFactory;
 
@@ -34,4 +35,9 @@ public ResourceReferenceFactory<EntityManager> registerPersistenceContextInjecti
     public ResourceReferenceFactory<EntityManagerFactory> registerPersistenceUnitInjectionPoint(InjectionPoint injectionPoint) {
         return new MockResourceFactory<EntityManagerFactory>();
     }
+
+    @Override
+    public ResourceReferenceFactory<EntityAgent> registerPersistenceAgentInjectionPoint(InjectionPoint injectionPoint) {
+        return new MockResourceFactory<EntityAgent>();
+    }
 }