Merge pull request #21 from santisq/20-add-support-for-subexpressions-with-using-statements

add support for member accessing using statement

Author: santisq

Module/private/InvocationManager.ps1

@@ -137,9 +137,25 @@ class InvocationManager : IDisposable {
                     $value))
             }
 
-            if ($usingstatement.SubExpression -is [IndexExpressionAst]) {
-                $idx = $usingstatement.SubExpression.Index.SafeGetValue()
-                $value = $value[$idx]
+            if ($usingstatement.SubExpression -isnot [VariableExpressionAst]) {
+                [Stack[Ast]] $subexpressionStack = $usingstatement.SubExpression.FindAll({
+                        $args[0] -is [IndexExpressionAst] -or
+                        $args[0] -is [MemberExpressionAst] },
+                    $false)
+
+                while ($subexpressionStack.Count) {
+                    $subexpression = $subexpressionStack.Pop()
+                    if ($subexpression -is [IndexExpressionAst]) {
+                        $idx = $subexpression.Index.SafeGetValue()
+                        $value = $value[$idx]
+                        continue
+                    }
+
+                    if ($subexpression -is [MemberExpressionAst]) {
+                        $member = $subexpression.Member.SafeGetValue()
+                        $value = $value.$member
+                    }
+                }
             }
 
             $usingParams.Add($key, $value)

tests/PSParallelPipeline.tests.ps1

@@ -140,5 +140,15 @@
             1 | Invoke-Parallel { $using:arr[-1] } | Should -BeExactly 10
             1 | Invoke-Parallel { $using:hash['FOO'] } | Should -BeExactly 'bar'
         }
+
+        It 'Allows member accessing on using: statemets' {
+            $hash = @{
+                foo = @{
+                    bar = [pscustomobject]@{ Index = 0..10 }
+                }
+            }
+
+            1 | Invoke-Parallel { $using:hash['foo']['bar'].Index[5] } | Should -BeExactly 5
+        }
     }
 }