Merge pull request #453 from dahlbyk/prompt-error

Handle Get/Write-GitStatus Errors
dahlbyk · Mar 12, 2017 · 99c1fc5 · 99c1fc5
2 parents c5447d6 + a134285
commit 99c1fc5
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/src/GitPrompt.ps1 b/src/GitPrompt.ps1
@@ -82,6 +82,9 @@ $global:GitPromptSettings = [pscustomobject]@{
     StashBackgroundColor                        = $null
     StashForegroundColor                        = [ConsoleColor]::Red
 
+    ErrorForegroundColor                        = [ConsoleColor]::Red
+    ErrorBackgroundColor                        = $null
+
     ShowStatusWhenZero                          = $true
 
     AutoRefreshIndex                            = $true
@@ -327,8 +330,18 @@ function Global:Write-VcsStatus {
 
 # Add scriptblock that will execute for Write-VcsStatus
 $PoshGitVcsPrompt = {
-    $Global:GitStatus = Get-GitStatus
-    Write-GitStatus $GitStatus
+    try {
+        $Global:GitStatus = Get-GitStatus
+        Write-GitStatus $GitStatus
+    }
+    catch {
+        $s = $Global:GitPromptSettings
+        if ($s) {
+            Write-Prompt $s.BeforeText -BackgroundColor $s.BeforeBackgroundColor -ForegroundColor $s.BeforeForegroundColor
+            Write-Prompt "Error: $_" -BackgroundColor $s.ErrorBackgroundColor -ForegroundColor $s.ErrorForegroundColor
+            Write-Prompt $s.AfterText -BackgroundColor $s.AfterBackgroundColor -ForegroundColor $s.AfterForegroundColor
+        }
+    }
 }
 
 $Global:VcsPromptStatuses += $PoshGitVcsPrompt