Support return

Signed-off-by: Matheus Cruz <[email protected]>

Author: mcruzdev

impl/core/src/main/java/io/serverlessworkflow/impl/executors/RunShellExecutor.java

@@ -23,6 +23,7 @@
 import io.serverlessworkflow.impl.WorkflowContext;
 import io.serverlessworkflow.impl.WorkflowDefinition;
 import io.serverlessworkflow.impl.WorkflowModel;
+import io.serverlessworkflow.impl.WorkflowModelFactory;
 import io.serverlessworkflow.impl.WorkflowUtils;
 import io.serverlessworkflow.impl.expressions.ExpressionUtils;
 import java.io.IOException;
@@ -109,12 +110,21 @@ public void init(RunShell taskConfiguration, WorkflowDefinition definition) {
               process.getInputStream().transferTo(the(stdout));
               process.getErrorStream().transferTo(the(stderr));
               int exitCode = process.waitFor();
-              return definition
-                  .application()
-                  .modelFactory()
-                  .fromAny(
-                      new ProcessResult(
-                          exitCode, stdout.toString().trim(), stderr.toString().trim()));
+
+              RunTaskConfiguration.ProcessReturnType returnType = taskConfiguration.getReturn();
+
+              WorkflowModelFactory modelFactory = definition.application().modelFactory();
+
+              return switch (returnType) {
+                case ALL ->
+                    modelFactory.fromAny(
+                        new ProcessResult(
+                            exitCode, stdout.toString().trim(), stderr.toString().trim()));
+                case NONE -> modelFactory.fromNull();
+                case CODE -> modelFactory.from(exitCode);
+                case STDOUT -> modelFactory.from(stdout.toString().trim());
+                case STDERR -> modelFactory.from(stderr.toString().trim());
+              };
             } else {
               return input;
             }

impl/test/src/test/resources/workflows-samples/run-shell/echo-jq.yaml

@@ -8,3 +8,4 @@ do:
       run:
         shell:
           command: ${ "echo Hello, \(.user.name)" }
+        return: all

impl/test/src/test/resources/workflows-samples/run-shell/echo-with-env.yaml

@@ -11,4 +11,4 @@ do:
           environment:
             FIRST_NAME: John
             LAST_NAME: ${.lastName}
-
+        return: all

impl/test/src/test/resources/workflows-samples/run-shell/echo.yaml

@@ -7,4 +7,5 @@ do:
   - runShell:
       run:
         shell:
-          command: 'echo "Hello, anonymous"'
+          command: 'echo "Hello, anonymous"'
+        return: all

impl/test/src/test/resources/workflows-samples/run-shell/touch-cat.yaml

@@ -7,9 +7,9 @@ do:
   - runShell:
       run:
         shell:
-#          write hello world to a file and then cat it
           command: echo "hello world" > /tmp/hello.txt && cat /tmp/hello.txt
           environment:
             FIRST_NAME: John
             LAST_NAME: ${.lastName}
+        return: all