Allow context override

Signed-off-by: Aidan Jensen <aidan@artificial.com>

.vscode/launch.json

@@ -9,7 +9,11 @@
             "type": "go",
             "request": "launch",
             "mode": "auto",
-            "program": "${workspaceFolder}/cmd/fixer"
+            "program": "${workspaceFolder}/cmd/fixer",
+            "args": [
+                "-k8s-context",
+                "aidan-sandbox-aks-artificial-cluster",
+            ]
         }
     ]
 }

cmd/fixer/fixer.go

@@ -3,6 +3,7 @@ package main
 
 import (
 	"context"
+	"flag"
 	"os"
 
 	"github.com/artificialinc/cm-429-fixer/pkg/cm"
@@ -24,6 +25,9 @@ func init() {
 }
 
 func main() {
+	k8sContext := flag.String("k8s-context", "", "Kubernetes context to use")
+	flag.Parse()
+
 	zapCfg := zap.NewProductionConfig()
 	level, err := zapcore.ParseLevel(logLevel)
 	if err != nil {
@@ -37,6 +41,7 @@ func main() {
 
 	watcher := cm.NewWatcher(
 		cm.WithLogger(zapr.NewLogger(logger).WithName("watcher")),
+		cm.WithClient(cm.GetLocalClient(&cm.ClientOpts{Context: *k8sContext})),
 	)
 
 	ctx := context.Background()

pkg/cm/cm.go

@@ -20,13 +20,23 @@ import (
 	"k8s.io/client-go/tools/clientcmd"
 )
 
-func getLocalClient() versioned.Interface {
+// ClientOpts is a set of options for the client
+type ClientOpts struct {
+	Context string
+}
+
+// GetLocalClient returns a client for the local cluster
+func GetLocalClient(opts *ClientOpts) versioned.Interface {
 	// Get kubeconfig
 	loadingRules := clientcmd.NewDefaultClientConfigLoadingRules()
 	// if you want to change the loading rules (which files in which order), you can do so here
 
 	configOverrides := &clientcmd.ConfigOverrides{}
 
+	if opts != nil && opts.Context != "" {
+		configOverrides.CurrentContext = opts.Context
+	}
+
 	config := clientcmd.NewNonInteractiveDeferredLoadingClientConfig(loadingRules, configOverrides)
 
 	clientConfig, err := config.ClientConfig()
@@ -103,7 +113,7 @@ func NewWatcher(opts ...Option) *Watcher {
 	}
 
 	if w.c == nil {
-		c := getLocalClient()
+		c := GetLocalClient(nil)
 		w.c = c
 	}