made AddPrimaryArg have a return value

ComedicChimera · May 23, 2021 · c9916e4 · c9916e4
1 parent 04e5097
commit c9916e4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/olive.go b/olive.go
@@ -90,12 +90,13 @@ func (c *Command) AddSubcommand(name, desc string, helpEnabled bool) *Command {
 }
 
 // AddPrimaryArg adds a primary argument to the command
-func (c *Command) AddPrimaryArg(name, desc string, required bool) {
+func (c *Command) AddPrimaryArg(name, desc string, required bool) *PrimaryArgument {
 	if len(c.subcommands) > 0 {
 		log.Fatalf("command `%s` cannot both take a primary argument and have subcommands", c.Name)
 	}
 
 	c.primaryArg = &PrimaryArgument{name: name, desc: desc, required: required}
+	return c.primaryArg
 }
 
 // AddFlag adds a flag to the command