diff --git a/pkg/cobrautils/flag/path.go b/pkg/cobrautils/flag/path.go index 9a864df1d7..5dac065003 100644 --- a/pkg/cobrautils/flag/path.go +++ b/pkg/cobrautils/flag/path.go @@ -56,3 +56,9 @@ func PathP(f *pflag.FlagSet, name, shorthand string, value string, usage string) PathVarP(f, p, name, shorthand, value, usage) return p } + +// PathVarPF is like PathVarP, but returns the created flag. +func PathVarPF(f *pflag.FlagSet, p *string, name, shorthand string, value string, usage string) *pflag.Flag { + PathVarP(f, p, name, shorthand, value, usage) + return f.Lookup(name) +} diff --git a/pkg/cobrautils/flag/path_array.go b/pkg/cobrautils/flag/path_array.go index 7452f12943..1d9578a4ca 100644 --- a/pkg/cobrautils/flag/path_array.go +++ b/pkg/cobrautils/flag/path_array.go @@ -106,3 +106,9 @@ func PathArrayP(f *pflag.FlagSet, name, shorthand string, value []string, usage PathArrayVarP(f, &p, name, shorthand, value, usage) return &p } + +// PathArrayVarPF is like PathArrayVarP, but returns the created flag. +func PathArrayVarPF(f *pflag.FlagSet, p *[]string, name, shorthand string, value []string, usage string) *pflag.Flag { + PathArrayVarP(f, p, name, shorthand, value, usage) + return f.Lookup(name) +} diff --git a/pkg/cobrautils/flag/replace.go b/pkg/cobrautils/flag/replace.go index c12895b7f5..d68c1e08b8 100644 --- a/pkg/cobrautils/flag/replace.go +++ b/pkg/cobrautils/flag/replace.go @@ -27,15 +27,3 @@ func IntVarPF(f *pflag.FlagSet, p *int, name, shorthand string, value int, usage f.IntVarP(p, name, shorthand, value, usage) return f.Lookup(name) } - -// PathVarPF is like PathVarP, but returns the created flag. -func PathVarPF(f *pflag.FlagSet, p *string, name, shorthand string, value string, usage string) *pflag.Flag { - PathVarP(f, p, name, shorthand, value, usage) - return f.Lookup(name) -} - -// PathArrayVarPF is like PathArrayVarP, but returns the created flag. -func PathArrayVarPF(f *pflag.FlagSet, p *[]string, name, shorthand string, value []string, usage string) *pflag.Flag { - PathArrayVarP(f, p, name, shorthand, value, usage) - return f.Lookup(name) -}