Stylized Prompt Code

Function prompt([string]$PromptText = "PoSH")
{
    $lastSuccess = $?

    $color = @{
        Reset  = "`e[0m"
        Red    = "`e[31;1m"
        Green  = "`e[32;1m"
        Yellow = "`e[33;1m"
        Grey   = "`e[37;0m"
    }

    # set color of PS based on success of last execution
    if ($lastSuccess -eq $false)
    {
        $lastExit = $color.Red
    }
    else
    {
        $lastExit = $color.Green
    }


    # get the execution time of the last command
    $lastCmdTime = ""
    $lastCmd = Get-History -Count 1
    if ($null -ne $lastCmd)
    {
        $cmdTime = $lastCmd.Duration.TotalMilliseconds
        $units = "ms"
        $timeColor = $color.Green
        if ($cmdTime -gt 250 -and $cmdTime -lt 1000)
        {
            $timeColor = $color.Yellow
        }
        elseif ($cmdTime -ge 1000)
        {
            $timeColor = $color.Red
            $units = "s"
            $cmdTime = $lastCmd.Duration.TotalSeconds
            if ($cmdTime -ge 60)
            {
                $units = "m"
                $cmdTIme = $lastCmd.Duration.TotalMinutes
            }
        }

        $lastCmdTime = "$($color.Grey)[$timeColor$($cmdTime.ToString("#.##"))$units$($color.Grey)]$($color.Reset) "
    }

    # get git branch information if in a git folder or subfolder
    $gitBranch = ""
    $path = Get-Location
    while ($path -ne "")
    {
        if (Test-Path (Join-Path $path .git))
        {
            # need to do this so the stderr doesn't show up in $error
            $ErrorActionPreferenceOld = $ErrorActionPreference
            $ErrorActionPreference = 'Ignore'
            $branch = git rev-parse --abbrev-ref --symbolic-full-name '@{u}'
            $ErrorActionPreference = $ErrorActionPreferenceOld

            # handle case where branch is local
            if ($lastexitcode -ne 0 -or $null -eq $branch)
            {
                $branch = git rev-parse --abbrev-ref HEAD
            }

            $branchColor = $color.Green

            if ($branch -match "/master")
            {
                $branchColor = $color.Red
            }
            $gitBranch = " $($color.Grey)[$branchColor$branch$($color.Grey)]$($color.Reset)"
            break
        }

        $path = Split-Path -Path $path -Parent
    }

    # truncate the current location if too long
    $currentDirectory = $executionContext.SessionState.Path.CurrentLocation.Path
    $consoleWidth = [Console]::WindowWidth
    $maxPath = [int]($consoleWidth / 2)
    if ($currentDirectory.Length -gt $maxPath)
    {
        $currentDirectory = "`u{2026}" + $currentDirectory.SubString($currentDirectory.Length - $maxPath)
    }

    "$lastCmdTime$currentDirectory$gitBranch`n$($lastExit)$PromptText$($color.Reset)$('>' * ($nestedPromptLevel + 1)) "
}