Added code so the current directory will be tracked in the compat session.

Author: Bruce Payette

Tests/CompatibilitySession.Tests.ps1

@@ -193,4 +193,15 @@ Describe "Test the Windows PowerShell Compatibility Session functions" {
             Remove-Item -Recurse $fullModulePath
         }
     }
+
+    It "Should mirror directory changes in the compatibility session" {
+        # Verify that the initial directories are sync'ed
+        Invoke-WinCommand { $pwd.Path } | Should -BeExactly $pwd.Path
+        # Change location and verify that the compat session directory also changed
+        Push-Location ..
+        Invoke-WinCommand { $pwd.Path } | Should -BeExactly $pwd.Path
+        # Change back and verify again
+        Pop-Location
+        Invoke-WinCommand { $pwd.Path } | Should -BeExactly $pwd.Path
+    }
 }

WindowsCompatibility/WindowsCompatibility.psm1

@@ -93,6 +93,27 @@ $DefaultConfigurationName = 'Microsoft.PowerShell'
 # Specifies the default name of the computer on which to create the compatibility session
 $DefaultComputerName = 'localhost'
 
+# Location Changed handler that keeps the compatibility session PWD in sync with the parent PWD
+# This only applies on localhost.
+$locationChangedHandler = {
+    [PSSession] $session = Initialize-WinSession @PSBoundParameters -PassThru
+    if ($session.ComputerName -eq "localhost")
+    {
+        $newPath = $_.newPath
+        Invoke-Command -Session $session { Set-Location $using:newPath}
+    }
+}
+
+$ExecutionContext.InvokeCommand.LocationChangedAction = $locationChangedHandler
+
+# Remove the location changed handler if the module is removed.
+$MyInvocation.MyCommand.ScriptBlock.Module.OnRemove = {
+    if ($ExecutionContext.InvokeCommand.LocationChangedAction -eq $locationChangedHandler)
+    {
+        $ExecutionContext.InvokeCommand.LocationChangedAction = $null
+    }
+}
+
 function Initialize-WinSession
 {
     [CmdletBinding()]
@@ -143,13 +164,24 @@ function Initialize-WinSession
     }
     Write-Verbose -Verbose:$verboseFlag "The compatibility session name is '$script:SessionName'."
 
-    # BUGBUG - need to deal with the possibilities of multiple sessions
     $session = Get-PSSession | Where-Object {
         $_.ComputerName      -eq $ComputerName -and
         $_.ConfigurationName -eq $ConfigurationName -and
         $_.Name              -eq $script:SessionName
     }
 
+    # Deal with the possibilities of multiple sessions. This might arise
+    # from the user hitting ctrl-C. We'll make the assumption that the
+    # first one returned is the correct one and we'll remove the rest.
+    $session, $rest = $session
+    if ($rest)
+    {
+        foreach ($s in $rest)
+        {
+            Remove-PSSession  $s
+        }
+    }
+
     if ($session -and $session.State -ne "Opened")
     {
         Write-Verbose -Verbose:$verboseFlag "Removing closed compatibility session."
@@ -175,8 +207,11 @@ function Initialize-WinSession
             $newPSSessionParameters.EnableNetworkAccess = $true
         }
         Write-Verbose -Verbose:$verboseFlag "Creating a new compatibility session."
-        ##BUGBUG need to deal with the case where there might be multiple sessions because someone hit ctrl-C
         $session = New-PSSession @newPSSessionParameters
+        if ($session.ComputerName -eq "localhost")
+        {
+            Invoke-Command $session { Set-Location $using:PWD }
+        }
     }
     else
     {