Adds escape codes to the prompt for the shell integration

ShellIntegration.ps1

# Reference:
# https://devblogs.microsoft.com/commandline/shell-integration-in-the-windows-terminal/

param
(
    [ValidateSet('WindowsTerminal', 'ITerm2')]
    [String]$TerminalProgram = 'WindowsTerminal'
)

# Restore hooked functions in case this script is executed accidentally twice
if ($global:shellIntegrationGlobals) {
    $function:global:PSConsoleHostReadLine = $global:shellIntegrationGlobals.originalPSConsoleHostReadLine
    $function:global:Prompt = $global:shellIntegrationGlobals.originalPrompt
}

$global:shellIntegrationGlobals = @{
    terminalProgram = $TerminalProgram
    originalPSConsoleHostReadLine = $function:global:PSConsoleHostReadLine
    originalPrompt = $function:global:Prompt
    lastCommand = $null

    getExitCode = {
        param ($lastCommandStatus)
        if ($lastCommandStatus -eq $true) {
            return 0
        }

        if ($Error[0]) {
            $lastHistory = Get-History -Count 1
            $isPowerShellError = $Error[0].InvocationInfo.HistoryId -eq $lastHistory.Id
        }

        if ($isPowerShellError) {
            return 1
        }
        else {
            return $LastExitCode
        }
    }
}

$function:global:PSConsoleHostReadLine = {
    $commandExecuted = "`e]133;C`a"
    $command = $global:shellIntegrationGlobals.originalPSConsoleHostReadLine.Invoke()

    $commandExecuted | Write-Host -NoNewLine
    $command

    $global:shellIntegrationGlobals.lastCommand = $command
}

$function:global:Prompt = {
    $lastCommandStatus = $?

    if ($global:shellIntegrationGlobals.lastCommand) {
        $exitCode = $global:shellIntegrationGlobals.getExitCode.Invoke($lastCommandStatus)
        $commandFinished = "`e]133;D;$exitCode`a"
    }
    else {
        $commandFinished = "`e]133;D`a"
    }

    $currentLocation = $ExecutionContext.SessionState.Path.CurrentLocation.Path
    switch ($global:shellIntegrationGlobals.terminalProgram) {
        'WindowsTerminal' { $setWorkingDirectory = "`e]9;9;`"$currentLocation`"`a" }
        'ITerm2' { $setWorkingDirectory = "`e]1337;CurrentDir=$currentLocation`a" }
    }

    $promptStarted = "`e]133;A`a"
    $commandStarted = "`e]133;B`a"
    $prompt = $global:shellIntegrationGlobals.originalPrompt.Invoke()

    $commandFinished + $promptStarted + $setWorkingDirectory + $prompt + $commandStarted
}

You can leverage the "shell integration" feature in the Windows Terminal by adding the following code to your profile after the definition of your prompt function.

& 'PathTo/ShellIntegration.ps1'

The code adds the escape sequences to the prompt but keeps your prompt customization. It also emits FTCS_COMMAND_EXECUTED so autoMarkPrompts in the Windows Terminal setting needs to be turned off.

"profiles":
{
    "defaults": 
    {
        "experimental.autoMarkPrompts": false,
        "experimental.rightClickContextMenu": true,
        "experimental.showMarksOnScrollbar": true
    },
}

Refer to this post about the shell integration feature in the Windows Terminal itself.

In the VSCode terminal, you don't need to add this code. The shell integration is activated by default by dot-sourcing a similar builtin script.

. "$(code --locate-shell-integration-path pwsh)"

https://code.visualstudio.com/docs/terminal/shell-integration

Supported iTerm2 on macOS.

& 'PathTo/ShellIntegration.ps1' -TerminalProgram ITerm2