Adding until to listen

Signed-off-by: fjtirado <[email protected]>

Author: fjtirado

fluent/agentic/src/test/java/io/serverlessworkflow/fluent/agentic/ChatBotIT.java

@@ -70,11 +70,13 @@ void chat_bot() {
                     t.listen(
                             "listenToMessages",
                             l ->
-                                l.one(c -> c.with(event -> event.type("org.acme.chatbot.request"))))
-                        .when(message -> !"".equals(message.get("message")), Map.class)
+                                l.until(message -> !"".equals(message.get("message")), Map.class)
+                                    .one(
+                                        c ->
+                                            c.with(
+                                                event -> event.type("org.acme.chatbot.request"))))
                         .agent(chatBot)
-                        .emit(emit -> emit.event(e -> e.type("org.acme.chatbot.reply")))
-                        .then("listenToMessages"))
+                        .emit(emit -> emit.event(e -> e.type("org.acme.chatbot.reply"))))
             .build();
 
     try (WorkflowApplication app = WorkflowApplication.builder().build()) {

fluent/func/src/main/java/io/serverlessworkflow/fluent/func/FuncListenTaskBuilder.java

@@ -15,15 +15,37 @@
  */
 package io.serverlessworkflow.fluent.func;
 
+import io.serverlessworkflow.api.types.AnyEventConsumptionStrategy;
+import io.serverlessworkflow.api.types.ListenTask;
+import io.serverlessworkflow.api.types.func.UntilPredicate;
 import io.serverlessworkflow.fluent.func.spi.ConditionalTaskBuilder;
 import io.serverlessworkflow.fluent.func.spi.FuncTransformations;
 import io.serverlessworkflow.fluent.spec.ListenTaskBuilder;
+import java.util.function.Predicate;
 
 public class FuncListenTaskBuilder extends ListenTaskBuilder
     implements ConditionalTaskBuilder<FuncListenTaskBuilder>,
         FuncTransformations<FuncListenTaskBuilder> {
 
+  private UntilPredicate untilPredicate;
+
   FuncListenTaskBuilder() {
     super();
   }
+
+  public <T> FuncListenTaskBuilder until(Predicate<T> predicate, Class<T> predClass) {
+    untilPredicate = new UntilPredicate().withPredicate(predicate, predClass);
+    return this;
+  }
+
+  @Override
+  public ListenTask build() {
+    ListenTask task = super.build();
+    AnyEventConsumptionStrategy anyEvent =
+        task.getListen().getTo().getAnyEventConsumptionStrategy();
+    if (untilPredicate != null && anyEvent != null) {
+      anyEvent.withUntil(untilPredicate);
+    }
+    return task;
+  }
 }