feat: Updates to align with v0.2.5 of the spec

Author: fjuma

impl/src/main/java/org/wildfly/extras/a2a/server/apps/jakarta/A2ARequestFilter.java

@@ -6,8 +6,10 @@
 import java.io.InputStream;
 
 import io.a2a.spec.CancelTaskRequest;
+import io.a2a.spec.DeleteTaskPushNotificationConfigRequest;
 import io.a2a.spec.GetTaskPushNotificationConfigRequest;
 import io.a2a.spec.GetTaskRequest;
+import io.a2a.spec.ListTaskPushNotificationConfigRequest;
 import io.a2a.spec.SendMessageRequest;
 import io.a2a.spec.SendStreamingMessageRequest;
 import io.a2a.spec.SetTaskPushNotificationConfigRequest;
@@ -67,7 +69,9 @@ private static boolean isNonStreamingRequest(String requestBody) {
                 requestBody.contains(CancelTaskRequest.METHOD) ||
                 requestBody.contains(SendMessageRequest.METHOD) ||
                 requestBody.contains(SetTaskPushNotificationConfigRequest.METHOD) ||
-                requestBody.contains(GetTaskPushNotificationConfigRequest.METHOD);
+                requestBody.contains(GetTaskPushNotificationConfigRequest.METHOD) ||
+                requestBody.contains(ListTaskPushNotificationConfigRequest.METHOD) ||
+                requestBody.contains(DeleteTaskPushNotificationConfigRequest.METHOD);
     }
 
     private static void putAcceptHeader(ContainerRequestContext requestContext, String mediaType) {

impl/src/main/java/org/wildfly/extras/a2a/server/apps/jakarta/A2AServerResource.java

@@ -25,6 +25,7 @@
 import io.a2a.server.util.async.Internal;
 import io.a2a.spec.AgentCard;
 import io.a2a.spec.CancelTaskRequest;
+import io.a2a.spec.DeleteTaskPushNotificationConfigRequest;
 import io.a2a.spec.GetTaskPushNotificationConfigRequest;
 import io.a2a.spec.GetTaskRequest;
 import io.a2a.spec.IdJsonMappingException;
@@ -37,6 +38,7 @@
 import io.a2a.spec.JSONRPCErrorResponse;
 import io.a2a.spec.JSONRPCRequest;
 import io.a2a.spec.JSONRPCResponse;
+import io.a2a.spec.ListTaskPushNotificationConfigRequest;
 import io.a2a.spec.MethodNotFoundError;
 import io.a2a.spec.MethodNotFoundJsonMappingException;
 import io.a2a.spec.NonStreamingJSONRPCRequest;
@@ -146,11 +148,15 @@ private JSONRPCResponse<?> processNonStreamingRequest(NonStreamingJSONRPCRequest
         } else if (request instanceof CancelTaskRequest) {
             return jsonRpcHandler.onCancelTask((CancelTaskRequest) request);
         } else if (request instanceof SetTaskPushNotificationConfigRequest) {
-            return jsonRpcHandler.setPushNotification((SetTaskPushNotificationConfigRequest) request);
+            return jsonRpcHandler.setPushNotificationConfig((SetTaskPushNotificationConfigRequest) request);
         } else if (request instanceof GetTaskPushNotificationConfigRequest) {
-            return jsonRpcHandler.getPushNotification((GetTaskPushNotificationConfigRequest) request);
+            return jsonRpcHandler.getPushNotificationConfig((GetTaskPushNotificationConfigRequest) request);
         } else if (request instanceof SendMessageRequest) {
             return jsonRpcHandler.onMessageSend((SendMessageRequest) request);
+        } else if (request instanceof ListTaskPushNotificationConfigRequest) {
+            return jsonRpcHandler.listPushNotificationConfig((ListTaskPushNotificationConfigRequest) request);
+        } else if (request instanceof DeleteTaskPushNotificationConfigRequest) {
+            return jsonRpcHandler.deletePushNotificationConfig((DeleteTaskPushNotificationConfigRequest) request);
         } else {
             return generateErrorResponse(request, new UnsupportedOperationError());
         }

tck/src/main/java/org/wildfly/extras/a2a/server/jakarta/tck/AgentCardProducer.java

@@ -40,6 +40,7 @@ public AgentCard agentCard() {
                                 .tags(Collections.singletonList("hello world"))
                                 .examples(List.of("hi", "hello world"))
                                 .build()))
+                .protocolVersion("0.2.5")
                 .build();
     }
 }

tests/common/src/main/java/org/wildfly/extras/a2a/server/jakarta/test/common/A2ATestResource.java

@@ -19,6 +19,7 @@
 import jakarta.ws.rs.core.Response;
 
 import io.a2a.server.apps.common.TestUtilsBean;
+import io.a2a.spec.PushNotificationConfig;
 import io.a2a.spec.Task;
 import io.a2a.spec.TaskArtifactUpdateEvent;
 import io.a2a.spec.TaskStatusUpdateEvent;
@@ -102,4 +103,29 @@ public Response enqueueTaskArtifactUpdateEvent(@PathParam("taskId") String taskI
     public Response getStreamingSubscribedCount() {
         return Response.ok(String.valueOf(streamingSubscribedCount.get()), TEXT_PLAIN).build();
     }
+
+    @DELETE
+    @Path("/task/{taskId}/config/{configId}")
+    public Response deleteTaskPushNotificationConfig(@PathParam("taskId") String taskId, @PathParam("configId") String configId) {
+        Task task = testUtilsBean.getTask(taskId);
+        if (task == null) {
+            return Response.status(404).build();
+        }
+        testUtilsBean.deleteTaskPushNotificationConfig(taskId, configId);
+        return Response.ok()
+                .header(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON)
+                .build();
+    }
+
+    @POST
+    @Path("/task/{taskId}")
+    @Consumes(MediaType.APPLICATION_JSON)
+    public Response savePushNotificationConfigInStore(@PathParam("taskId") String taskId, String body) throws Exception {
+        PushNotificationConfig notificationConfig = Utils.OBJECT_MAPPER.readValue(body, PushNotificationConfig.class);
+        if (notificationConfig == null) {
+            return Response.status(404).build();
+        }
+        testUtilsBean.saveTaskPushNotificationConfig(taskId, notificationConfig);
+        return Response.ok().build();
+    }
 }