fix: use of specified context to obtain cluster name (#339)

Signed-off-by: chenk <[email protected]>
aquasecurity · May 7, 2024 · f6c5fb0 · f6c5fb0
1 parent f31b54a
commit f6c5fb0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/k8s/k8s.go b/pkg/k8s/k8s.go
@@ -579,7 +579,12 @@ func (c *cluster) ClusterNameVersion() (string, string, error) {
 	}
 	clusterName := "k8s.io/kubernetes"
 	if len(rawCfg.Contexts) > 0 {
-		clusterName = rawCfg.Contexts[rawCfg.CurrentContext].Cluster
+		if c.currentContext != "" {
+			rawCfg.CurrentContext = c.currentContext
+		}
+		if clusterContext, ok := rawCfg.Contexts[rawCfg.CurrentContext]; ok {
+			clusterName = clusterContext.Cluster
+		}
 	}
 	version, err := c.clientset.ServerVersion()
 	if err != nil {