function Write-BranchName () {
    try {
        $branch = git rev-parse --abbrev-ref HEAD

        if ($branch -eq "HEAD") {
            # we're probably in detached HEAD state, so print the SHA
            $branch = git rev-parse --short HEAD
            Write-Host " ($branch)" -ForegroundColor "red"
        }
        else {
            # we're on an actual branch, so print it
            Write-Host " ($branch)" -ForegroundColor "blue"
        }
    } catch {
        # we'll end up here if we're in a newly initiated git repo
        Write-Host " (no branches yet)" -ForegroundColor "yellow"
    }
}

function prompt {
    $meta = "$($env:UserName + "@" + $env:ComputerName) "
    $path = "$($ExecutionContext.SessionState.Path.CurrentLocation -replace [RegEx]::Escape($home),  "~")"
    $userPrompt = "$('$' * ($nestedPromptLevel + 1)) "

    Write-Host "`n" -NoNewline

    if (Test-Path .git) {
        Write-Host $meta -NoNewline -ForegroundColor "cyan"
        Write-Host $path -NoNewline -ForegroundColor "green"
        Write-BranchName
    }
    else {
        Write-Host $meta -NoNewline -ForegroundColor "cyan"
        # we're not in a repo so don't bother displaying branch name/sha
        Write-Host $path -ForegroundColor "green"
    }

    return $userPrompt
}