fixed capture output and errors in runprogramtask

Author: zoroz

src/FlubuCore/Tasks/Process/IRunProgramTask.cs

@@ -33,6 +33,18 @@ public interface IRunProgramTask : ITaskOfT<int, IRunProgramTask>, IExternalProc
         /// <returns></returns>
         IRunProgramTask CaptureErrorOutput();
 
+        /// <summary>
+        ///     Capture output of the running program.
+        /// </summary>
+        /// <returns></returns>
+        IRunProgramTask CaptureOutput(bool capture);
+
+        /// <summary>
+        ///     Capture error output of the running program.
+        /// </summary>
+        /// <returns></returns>
+        IRunProgramTask CaptureErrorOutput(bool capture);
+
         /// <summary>
         ///     Gets the whole output of the executed command.
         /// </summary>

src/FlubuCore/Tasks/Process/RunProgramTask.cs

@@ -24,9 +24,9 @@ public class RunProgramTask : TaskBase<int, IRunProgramTask>, IRunProgramTask
 
         private string _workingFolder;
 
-        private bool _captureOutput;
+        private bool _captureOutput = true;
 
-        private bool _captureErrorOutput;
+        private bool _captureErrorOutput = true;
 
         private LogLevel _outputLogLevel = LogLevel.Info;
 
@@ -110,13 +110,27 @@ public IRunProgramTask CaptureOutput()
             return this;
         }
 
+        /// <inheritdoc />
+        public IRunProgramTask CaptureOutput(bool capture)
+        {
+            _captureOutput = capture;
+            return this;
+        }
+
         /// <inheritdoc />
         public IRunProgramTask CaptureErrorOutput()
         {
             _captureErrorOutput = true;
             return this;
         }
 
+        /// <inheritdoc />
+        public IRunProgramTask CaptureErrorOutput(bool capture)
+        {
+            _captureErrorOutput = false;
+            return this;
+        }
+
         /// <inheritdoc />
         /// <summary>
         /// Get the output produced by executable.
@@ -222,25 +236,35 @@ protected override int DoExecute(ITaskContextInternal context)
             ICommand command = _commandFactory.Create(cmd, _arguments.Select(x => x.arg));
             string workingFolder = _workingFolder ?? rootDir;
             command
-                .CaptureStdErr()
-                .CaptureStdOut()
-                .WorkingDirectory(workingFolder)
-                .OnErrorLine(l =>
-                {
-                    if (_outputLogLevel >= LogLevel.Error)
-                        DoLogInfo(l);
+                .WorkingDirectory(workingFolder);
 
-                    if (_captureErrorOutput)
-                        _errorOutput.AppendLine(l);
-                })
-                .OnOutputLine(l =>
-                {
-                    if (_outputLogLevel >= LogLevel.Info)
-                        DoLogInfo(l);
+            if (_captureOutput)
+            {
+                command
+                    .CaptureStdOut()
+                    .OnOutputLine(l =>
+                    {
+                        if (_outputLogLevel >= LogLevel.Info)
+                            DoLogInfo(l);
+
+                        if (_captureOutput)
+                            _output.AppendLine(l);
+                    });
+            }
 
-                    if (_captureOutput)
-                        _output.AppendLine(l);
-                });
+            if (_captureErrorOutput)
+            {
+                command
+                    .CaptureStdErr()
+                    .OnErrorLine(l =>
+                    {
+                        if (_outputLogLevel >= LogLevel.Error)
+                            DoLogInfo(l);
+
+                        if (_captureErrorOutput)
+                            _errorOutput.AppendLine(l);
+                    });
+            }
 
             string commandArgs = null;
             ProcessAdditionalOptions(context);