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
|
package complete
import (
"errors"
"flag"
"fmt"
"os"
"strings"
"github.com/posener/complete/install"
)
func runCommandLine(cmd string) {
c := parseFlags(cmd)
err := c.validate()
if err != nil {
os.Stderr.WriteString(err.Error() + "\n")
os.Exit(1)
}
if !c.yes && !prompt(c.action(), cmd) {
fmt.Println("Cancelling...")
os.Exit(2)
}
fmt.Println(c.action() + "ing...")
if c.install {
err = install.Install(cmd, c.root)
} else {
err = install.Uninstall(cmd, c.root)
}
if err != nil {
fmt.Printf("%s failed! %s\n", c.action(), err)
os.Exit(3)
}
fmt.Println("Done!")
}
func prompt(action, cmd string) bool {
fmt.Printf("%s bash completion for %s? ", action, cmd)
var answer string
fmt.Scanln(&answer)
switch strings.ToLower(answer) {
case "y", "yes":
return true
default:
return false
}
}
type config struct {
install bool
uninstall bool
root bool
yes bool
}
func parseFlags(cmd string) config {
var c config
flag.BoolVar(&c.install, "install", false,
fmt.Sprintf("Install bash completion for %s command", cmd))
flag.BoolVar(&c.uninstall, "uninstall", false,
fmt.Sprintf("Uninstall bash completion for %s command", cmd))
flag.BoolVar(&c.root, "root", false,
"(Un)Install as root:\n"+
" (Un)Install at /etc/bash_completion.d/ (user should have write permissions to that directory).\n"+
" If not set, a complete command will be added(removed) to ~/.bashrc")
flag.BoolVar(&c.yes, "y", false, "Don't prompt user for typing 'yes'")
flag.Parse()
return c
}
func (c config) validate() error {
if c.install && c.uninstall {
return errors.New("Install and uninstall are exclusive")
}
if !c.install && !c.uninstall {
return errors.New("Must specify -install or -uninstall")
}
return nil
}
func (c config) action() string {
if c.install {
return "Install"
}
return "Uninstall"
}
|
