Skip to content

Commit

Permalink
Add additional argument helpers
Browse files Browse the repository at this point in the history
  • Loading branch information
anujc25 committed Jan 23, 2025
1 parent 5a47482 commit 978c5bd
Showing 1 changed file with 34 additions and 0 deletions.
34 changes: 34 additions & 0 deletions command/arguments.go
Original file line number Diff line number Diff line change
Expand Up @@ -86,6 +86,40 @@ func NamesArg(names *[]string) Arg {
}
}

func Argument(name string, val *string) Arg {
return Arg{
Name: name,
Arity: 1,
Set: func(_ *cobra.Command, args []string, offset int) error {
*val = args[offset]
return nil
},
}
}

func OptionalArgument(name string, val *string) Arg {
arg := Argument(name, val)
arg.Optional = true
return arg
}

func RemainingArguments(name string, values *[]string) Arg {
return Arg{
Name: name,
Arity: -1,
Set: func(_ *cobra.Command, args []string, offset int) error {
*values = args[offset:]
return nil
},
}
}

func OptionalRemainingArguments(name string, values *[]string) Arg {
arg := RemainingArguments(name, values)
arg.Optional = true
return arg
}

func BareDoubleDashArgs(values *[]string) Arg {
return Arg{
Arity: -1,
Expand Down

0 comments on commit 978c5bd

Please sign in to comment.