-
Notifications
You must be signed in to change notification settings - Fork 36
/
hidden.go
68 lines (59 loc) · 1.74 KB
/
hidden.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package kcobra
import (
"github.com/enfabrica/enkit/lib/kflags"
"github.com/spf13/cobra"
"time"
)
type HiddenFlagSet struct {
inner *FlagSet
flags []string
showHidden bool
}
func (hfs *HiddenFlagSet) Hide(name string) {
hfs.inner.MarkHidden(name)
hfs.flags = append(hfs.flags, name)
}
func (hfs *HiddenFlagSet) BoolVar(p *bool, name string, value bool, usage string) {
hfs.inner.BoolVar(p, name, value, usage)
hfs.Hide(name)
}
func (hfs *HiddenFlagSet) DurationVar(p *time.Duration, name string, value time.Duration, usage string) {
hfs.inner.DurationVar(p, name, value, usage)
hfs.Hide(name)
}
func (hfs *HiddenFlagSet) StringArrayVar(p *[]string, name string, value []string, usage string) {
hfs.inner.StringArrayVar(p, name, value, usage)
hfs.Hide(name)
}
func (hfs *HiddenFlagSet) StringVar(p *string, name string, value string, usage string) {
hfs.inner.StringVar(p, name, value, usage)
hfs.Hide(name)
}
func (hfs *HiddenFlagSet) ByteFileVar(p *[]byte, name string, defaultFile string, usage string, mods ...kflags.ByteFileModifier) {
hfs.inner.ByteFileVar(p, name, defaultFile, usage, mods...)
hfs.Hide(name)
}
func (hfs *HiddenFlagSet) IntVar(p *int, name string, value int, usage string) {
hfs.inner.IntVar(p, name, value, usage)
hfs.Hide(name)
}
func (hfs *HiddenFlagSet) Help(cmd *cobra.Command, args []string) bool {
if !hfs.showHidden {
return true
}
for _, fl := range hfs.flags {
o := hfs.inner.Lookup(fl)
if o == nil {
continue
}
o.Hidden = false
}
return true
}
func HideFlags(fs *FlagSet) *HiddenFlagSet {
retval := &HiddenFlagSet{
inner: fs,
}
fs.BoolVar(&retval.showHidden, "help-all", false, "Show all the flags available, even those that are less useful and normally hidden.")
return retval
}