feat: add new WithRestConfig option

Signed-off-by: Ilya Mayorov <[email protected]>

Author: faroyam

client/client.go

@@ -129,6 +129,10 @@ func (c *Client) WithMetricLabels(labels map[string]string) {
 	c.metricLabels = labels
 }
 
+func (c *Client) WithRestConfig(config *rest.Config) {
+	c.restConfig = config
+}
+
 func (c *Client) DefaultNamespace() string {
 	return c.defaultNamespace
 }
@@ -160,7 +164,12 @@ func (c *Client) Init() error {
 	configType := "out-of-cluster"
 	var defaultNs string
 
-	if c.server == "" {
+	switch {
+	case c.restConfig != nil:
+		config = c.restConfig
+		defaultNs = "default"
+		configType = "server"
+	case c.server == "":
 		// Try to load from kubeconfig in flags or from ~/.kube/config
 		var outOfClusterErr error
 		config, defaultNs, outOfClusterErr = getOutOfClusterConfig(c.contextName, c.configPath)
@@ -191,7 +200,7 @@ func (c *Client) Init() error {
 			}
 			configType = "in-cluster"
 		}
-	} else {
+	default:
 		// use specific server to connect to API
 		config = &rest.Config{
 			Host: c.server,