added namespace support

joemiller · Nov 9, 2019 · e2c3b6b · e2c3b6b
1 parent 469e36a
commit e2c3b6b
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/cmd/erase.go b/cmd/erase.go
@@ -22,6 +22,12 @@ var eraseCmd = &cobra.Command{
 			return errors.New("Missing VAULT_ADDR environment variable")
 		}
 
+		vaultNamespace := os.Getenv("VAULT_NAMESPACE")
+		if vaultNamespace != "" {
+			vaultAddr += "/"
+			vaultAddr += vaultNamespace
+		}
+
 		if err := backend.Erase(vaultAddr); err != nil {
 			return err
 		}

diff --git a/cmd/get.go b/cmd/get.go
@@ -23,6 +23,12 @@ var getCmd = &cobra.Command{
 			return errors.New("Missing VAULT_ADDR environment variable")
 		}
 
+		vaultNamespace := os.Getenv("VAULT_NAMESPACE")
+		if vaultNamespace != "" {
+			vaultAddr += "/"
+			vaultAddr += vaultNamespace
+		}
+
 		token, err := backend.Get(vaultAddr)
 		if err != nil {
 			return err

diff --git a/cmd/store.go b/cmd/store.go
@@ -25,6 +25,12 @@ var storeCmd = &cobra.Command{
 			return errors.New("Missing VAULT_ADDR environment variable")
 		}
 
+		vaultNamespace := os.Getenv("VAULT_NAMESPACE")
+		if vaultNamespace != "" {
+			vaultAddr += "/"
+			vaultAddr += vaultNamespace
+		}
+
 		stdin, err := ioutil.ReadAll(os.Stdin)
 		if err != nil {
 			return errors.Wrap(err, "Failed to read token from STDIN")