final details

Author: santisq

src/PSParallelPipeline/Commands/InvokeParallelCommand.cs

@@ -79,7 +79,7 @@ protected override void BeginProcessing()
     protected override void ProcessRecord()
     {
         Dbg.Assert(_worker is not null);
-        this.ThrowIfInputObjectIsScriptBlock(InputObject);
+        InputObject.ThrowIfInputObjectIsScriptBlock(this);
 
         try
         {

src/PSParallelPipeline/ExceptionHelper.cs

@@ -56,7 +56,7 @@ internal static void ThrowIfVariableIsScriptBlock(this PSCmdlet cmdlet, object?
             value));
     }
 
-    internal static void ThrowIfInputObjectIsScriptBlock(this PSCmdlet cmdlet, object? value)
+    internal static void ThrowIfInputObjectIsScriptBlock(this object? value, PSCmdlet cmdlet)
     {
         if (ValueIsNotScriptBlock(value))
         {

src/PSParallelPipeline/PSTask.cs

@@ -36,15 +36,16 @@ static internal async Task<PSTask> CreateAsync(
         TaskSettings settings)
     {
         PSTask ps = new(runspacePool);
-        HookStreams(ps._internalStreams, runspacePool.Streams);
+        SetStreams(ps._internalStreams, runspacePool.Streams);
         ps.Runspace = await runspacePool.GetRunspaceAsync();
-        ps.AddInput(input);
-        ps.AddScript(settings.Script);
-        ps.AddUsingStatements(settings.UsingStatements);
-        return ps;
+
+        return ps
+            .AddInput(input)
+            .AddScript(settings.Script)
+            .AddUsingStatements(settings.UsingStatements);
     }
 
-    private static void HookStreams(
+    private static void SetStreams(
         PSDataStreams streams,
         PSOutputStreams outputStreams)
     {
@@ -63,7 +64,7 @@ private static Task InvokePowerShellAsync(
             powerShell.BeginInvoke<PSObject, PSObject>(null, output),
             powerShell.EndInvoke);
 
-    private void AddInput(object? inputObject)
+    private PSTask AddInput(object? inputObject)
     {
         if (inputObject is not null)
         {
@@ -72,17 +73,24 @@ private void AddInput(object? inputObject)
                 .AddArgument("_")
                 .AddArgument(inputObject);
         }
+
+        return this;
     }
 
-    private void AddScript(string script) =>
+    private PSTask AddScript(string script)
+    {
         _powershell.AddScript(script, useLocalScope: true);
+        return this;
+    }
 
-    private void AddUsingStatements(Dictionary<string, object?> usingParams)
+    private PSTask AddUsingStatements(Dictionary<string, object?> usingParams)
     {
         if (usingParams.Count > 0)
         {
             _powershell.AddParameter("--%", usingParams);
         }
+
+        return this;
     }
 
     internal async Task InvokeAsync()

src/PSParallelPipeline/RunspacePool.cs

@@ -56,7 +56,8 @@ internal void PushRunspace(Runspace runspace)
         _semaphore.Release();
     }
 
-    internal CancellationTokenRegistration RegisterCancellation(Action callback) => _token.Register(callback);
+    internal CancellationTokenRegistration RegisterCancellation(Action callback) =>
+        _token.Register(callback);
 
     private Runspace CreateRunspace()
     {