explicitly check for length of args

hashicorp · Dec 12, 2017 · 67e17fe · 67e17fe
1 parent 5a5f66d
commit 67e17fe
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/command/deploy.go b/command/deploy.go
@@ -83,14 +83,17 @@ func (c *DeployCommand) Run(args []string) int {
 
 	args = flags.Args()
 
-	if len(args) != 1 {
+	if len(args) == 1 {
+		templateFile = args[0]
+	} else if len(args) == 0 {
 		templateFile = "levant.nomad"
 		if _, err := os.Stat(templateFile); os.IsNotExist(err) {
 			c.UI.Error(c.Help())
 			return 1
 		}
 	} else {
-		templateFile = args[0]
+		c.UI.Error(c.Help())
+		return 1
 	}
 
 	logging.SetLevel(log)

diff --git a/command/render.go b/command/render.go
@@ -64,14 +64,17 @@ func (c *RenderCommand) Run(args []string) int {
 
 	args = flags.Args()
 
-	if len(args) != 1 {
+	if len(args) == 1 {
+		templateFile = args[0]
+	} else if len(args) == 0 {
 		templateFile = "levant.nomad"
 		if _, err := os.Stat(templateFile); os.IsNotExist(err) {
 			c.UI.Error(c.Help())
 			return 1
 		}
 	} else {
-		templateFile = args[0]
+		c.UI.Error(c.Help())
+		return 1
 	}
 
 	tpl, err = levant.RenderTemplate(templateFile, variables, &c.Meta.flagVars)