Add error handling for missing contexts in Get-, Remove-, and Set-Con…

…textSetting functions
PSModule · Nov 21, 2024 · 35f5b55 · 35f5b55
1 parent f68c9cc
commit 35f5b55
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/src/functions/public/ContextSetting/Get-ContextSetting.ps1 b/src/functions/public/ContextSetting/Get-ContextSetting.ps1
@@ -28,6 +28,11 @@ function Get-ContextSetting {
 
     $null = Get-ContextVault
     $contextObj = Get-Context -ID $ID
+
+    if (-not $contexts) {
+        throw "Context [$ID] not found"
+    }
+
     Write-Verbose "Returning setting: [$Name]"
     $contextObj.$Name
 }
diff --git a/src/functions/public/ContextSetting/Remove-ContextSetting.ps1 b/src/functions/public/ContextSetting/Remove-ContextSetting.ps1
@@ -48,6 +48,10 @@ filter Remove-ContextSetting {
     $null = Get-ContextVault
     $contextObj = Get-Context -ID $ID
 
+    if (-not $contexts) {
+        throw "Context [$ID] not found"
+    }
+
     if ($PSCmdlet.ShouldProcess("[$($contextObj.Name)]", "Remove [$Name]")) {
         Write-Verbose "Setting [$Name] in [$($contextObj.Name)]"
         $contextObj.PSObject.Properties.Remove($Name)

diff --git a/src/functions/public/ContextSetting/Set-ContextSetting.ps1 b/src/functions/public/ContextSetting/Set-ContextSetting.ps1
@@ -42,6 +42,10 @@ function Set-ContextSetting {
     $null = Get-ContextVault
     $contextObj = Get-Context -ID $ID
 
+    if (-not $contexts) {
+        throw "Context [$ID] not found"
+    }
+
     if ($PSCmdlet.ShouldProcess($Name, "Set value [$Value]")) {
         Write-Verbose "Setting [$Name] to [$Value] in [$($contextObj.Name)]"
         if ($contextObj.PSObject.Properties[$Name]) {