Setting Use to clear out ACL tokens when switching stacks (#10)

.changes/unreleased/Added-20240711-233913.yaml

@@ -0,0 +1,5 @@
+kind: Added
+body: Set Use to clear previous ACL tokens when called to avoid conflicts
+time: 2024-07-11T23:39:13.520023-05:00
+custom:
+    Author: Shackelford-Arden

cmd/use.go

@@ -4,9 +4,14 @@ import (
 	"fmt"
 
 	"github.com/Shackelford-Arden/hctx/cache"
+	"github.com/Shackelford-Arden/hctx/models"
 	"github.com/urfave/cli/v2"
 )
 
+// Use sets the appropriate environment variables to use
+// the selected Stack.
+// If caching is disabled, this will also unset any
+// existing env vars for tokens (ie NOMAD_TOKEN).
 func Use(ctx *cli.Context) error {
 
 	stackName := ctx.Args().First()
@@ -18,7 +23,7 @@ func Use(ctx *cli.Context) error {
 	}
 
 	currentStack := AppConfig.GetCurrentStack()
-	// Get current stacks tokens, if any and cache them
+	// If caching is enabled, set to cache.
 	if currentStack != nil && AppConfig.CacheAuth {
 		toCache := cache.GetCacheableValues()
 		updateErr := AppCache.Update(currentStack.Name, toCache)
@@ -27,10 +32,15 @@ func Use(ctx *cli.Context) error {
 		}
 	}
 
-	// rehydrate env w/ new stack cache, if present
-	newStackCache := AppCache.Get(selectedStack.Name)
+	useOut := unsetTokens(AppConfig.Shell)
+	var stackCache *models.StackCache
+	if AppConfig.CacheAuth {
+		stackCache = AppCache.Get(selectedStack.Name)
+	}
+
+	useOut += selectedStack.Use(AppConfig.Shell, stackCache, AppConfig.CacheAuth)
 
-	fmt.Print(selectedStack.Use(AppConfig.Shell, newStackCache, AppConfig.CacheAuth))
+	fmt.Println(useOut)
 
 	return nil
 }

cmd/utils.go

@@ -0,0 +1,21 @@
+package cmd
+
+func unsetTokens(shell string) string {
+
+	output := ""
+
+	switch shell {
+	case "bash":
+		fallthrough
+	case "zsh":
+		fallthrough
+	default:
+		output = `
+unset NOMAD_TOKEN
+unset CONSUL_TOKEN
+unset VAULT_TOKEN
+`
+	}
+
+	return output
+}