-
Notifications
You must be signed in to change notification settings - Fork 0
/
complete.go
94 lines (74 loc) · 1.58 KB
/
complete.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
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
package compflag
import (
"errors"
"flag"
"fmt"
"io"
"os"
"strconv"
"strings"
)
const (
exitCode = 0
envCompLine = "COMP_LINE"
envCompPoint = "COMP_POINT"
flagUsage = "completion-triggering flag"
)
var ErrUnknownShell = errors.New("shell unknown")
// Complete handles completion process started by shell, returns false if no completion was requested.
func Complete(opts ...Option) (ok bool) {
line, ok := os.LookupEnv(envCompLine)
if !ok || line == "" {
return false
}
pos, err := strconv.Atoi(os.Getenv(envCompPoint))
if err != nil {
return false
}
var (
token = extractToken(line, pos)
sb strings.Builder
cfg options
)
for _, o := range opts {
o(&cfg)
}
cfg.validate()
cfg.flags.VisitAll(func(f *flag.Flag) {
if _, ok := cfg.hidden[f.Name]; ok {
return
}
if token == "" || strings.HasPrefix(f.Name, token) {
_, _ = sb.WriteString("-" + f.Name + "\n")
}
})
_, _ = io.WriteString(cfg.writer, sb.String())
return true
}
func Var(name string, opts ...Option) {
var cfg options
for _, o := range opts {
o(&cfg)
}
cfg.validate()
cfg.flags.Func(name, flagUsage, func(arg string) (err error) {
switch arg {
case "bash", "zsh":
if Complete(append(opts, WithHidden(name))...) {
cfg.exit(exitCode)
}
default:
err = fmt.Errorf("%w: %s", ErrUnknownShell, arg)
}
return
})
}
func extractToken(line string, pos int) (val string) {
if pos > len(line) {
pos = len(line)
}
if t := strings.LastIndexByte(line[:pos], ' '); t != -1 {
line = strings.TrimSpace(line[t:pos])
}
return strings.TrimPrefix(line, "-")
}