Only allow use-context on existing contexts (#296)

grafana · Jan 4, 2024 · a80f3ac · a80f3ac
1 parent 01faa49
commit a80f3ac
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -108,8 +108,15 @@ func GetContexts() error {
 }
 
 func UseContext(context string) error {
-	viper.Set(CURRENT_CONTEXT, context)
-	return Write()
+	contexts := map[string]interface{}{}
+	viper.UnmarshalKey("contexts", &contexts)
+	for k := range contexts {
+		if k == context {
+			viper.Set(CURRENT_CONTEXT, context)
+			return Write()
+		}
+	}
+	return fmt.Errorf("Context %s not found", context)
 }
 
 func CurrentContext() (*Context, error) {