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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
// Package compflag provides a handful of standard library-compatible flags with bash complition capabilities.
//
// Usage
//
// import "github.com/posener/complete/v2/compflag"
//
// var (
// // Define flags...
// foo = compflag.String("foo", "", "")
// )
//
// func main() {
// compflag.Parse()
// // Main function.
// }
//
// Alternatively, the library can just be used with the standard library flag package:
//
// import (
// "flag"
// "github.com/posener/complete/v2/compflag"
// )
//
// var (
// // Define flags...
// foo = compflag.String("foo", "", "")
// bar = flag.String("bar", "", "")
// )
//
// func main() {
// complete.CommandLine()
// flag.Parse()
// // Main function.
// }
package compflag
import (
"flag"
"os"
"strconv"
"time"
"github.com/posener/complete/v2"
)
// FlagSet is bash completion enabled flag.FlagSet.
type FlagSet flag.FlagSet
// Parse parses command line arguments.
func (fs *FlagSet) Parse(args []string) error {
return (*flag.FlagSet)(fs).Parse(args)
}
func (fs *FlagSet) Visit(fn func(*flag.Flag)) { (*flag.FlagSet)(fs).Visit(fn) }
func (fs *FlagSet) VisitAll(fn func(*flag.Flag)) { (*flag.FlagSet)(fs).VisitAll(fn) }
func (fs *FlagSet) Arg(i int) string { return (*flag.FlagSet)(fs).Arg(i) }
func (fs *FlagSet) Args() []string { return (*flag.FlagSet)(fs).Args() }
func (fs *FlagSet) NArg() int { return (*flag.FlagSet)(fs).NArg() }
func (fs *FlagSet) NFlag() int { return (*flag.FlagSet)(fs).NFlag() }
func (fs *FlagSet) Name() string { return (*flag.FlagSet)(fs).Name() }
func (fs *FlagSet) PrintDefaults() { (*flag.FlagSet)(fs).PrintDefaults() }
func (fs *FlagSet) Lookup(name string) *flag.Flag { return (*flag.FlagSet)(fs).Lookup(name) }
func (fs *FlagSet) Parsed() bool { return (*flag.FlagSet)(fs).Parsed() }
// Complete performs bash completion if needed.
func (fs *FlagSet) Complete() {
complete.Complete(fs.Name(), complete.FlagSet((*flag.FlagSet)(CommandLine)))
}
var CommandLine = (*FlagSet)(flag.CommandLine)
// Parse parses command line arguments. It also performs bash completion when needed.
func Parse() {
CommandLine.Complete()
CommandLine.Parse(os.Args[1:])
}
func predictBool(value bool, prefix string) []string {
// If false, typing the bool flag is expected to turn it on, so there is nothing to complete
// after the flag.
if !value {
return nil
}
// Otherwise, suggest only to turn it off.
return []string{"false"}
}
func parseString(s string) (string, error) { return s, nil }
func formatString(v string) string { return v }
func parseInt(s string) (int, error) { return strconv.Atoi(s) }
func formatInt(v int) string { return strconv.Itoa(v) }
func parseBool(s string) (bool, error) { return strconv.ParseBool(s) }
func formatBool(v bool) string { return strconv.FormatBool(v) }
func parseDuration(s string) (time.Duration, error) { return time.ParseDuration(s) }
func formatDuration(v time.Duration) string { return v.String() }
|
