From 910198d117bdac45276c8aaf66235891d169789a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Luis=20Salvador=20Rufo?=
 <salvador.joseluis@gmail.com>
Date: Tue, 16 Dec 2014 16:18:36 +0100
Subject: [PATCH] etcdctl: add environment support to certs args

---
 etcdctl/command/util.go | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/etcdctl/command/util.go b/etcdctl/command/util.go
index 14ca7320e..bcc95adbd 100644
--- a/etcdctl/command/util.go
+++ b/etcdctl/command/util.go
@@ -91,10 +91,26 @@ func getEndpoints(c *cli.Context) ([]string, error) {
 }
 
 func getTransport(c *cli.Context) (*http.Transport, error) {
+	cafile := c.GlobalString("ca-file")
+	certfile := c.GlobalString("cert-file")
+	keyfile := c.GlobalString("key-file")
+
+	// Use an environment variable if nothing was supplied on the
+	// command line
+	if cafile == "" {
+		cafile = os.Getenv("ETCDCTL_CA_FILE")
+	}
+	if certfile == "" {
+		certfile = os.Getenv("ETCDCTL_CERT_FILE")
+	}
+	if keyfile == "" {
+		keyfile = os.Getenv("ETCDCTL_KEY_FILE")
+	}
+
 	tls := transport.TLSInfo{
-		CAFile:   c.GlobalString("ca-file"),
-		CertFile: c.GlobalString("cert-file"),
-		KeyFile:  c.GlobalString("key-file"),
+		CAFile:   cafile,
+		CertFile: certfile,
+		KeyFile:  keyfile,
 	}
 	return transport.NewTransport(tls)