From 5db452a63f1b8ff0319f08986a4a04324647738f Mon Sep 17 00:00:00 2001 From: Eyal Posener Date: Mon, 8 May 2017 07:32:13 +0300 Subject: [PATCH] Install fixups - remove root installation - install according to shell type Closes #7 --- cmd/cmd.go | 15 ++++------- cmd/install/{home.go => bash.go} | 7 ++--- cmd/install/install.go | 45 +++++++++++++++++++++++--------- cmd/install/root.go | 29 -------------------- 4 files changed, 42 insertions(+), 54 deletions(-) rename cmd/install/{home.go => bash.go} (95%) delete mode 100644 cmd/install/root.go diff --git a/cmd/cmd.go b/cmd/cmd.go index 8149aac..d0b341c 100644 --- a/cmd/cmd.go +++ b/cmd/cmd.go @@ -27,9 +27,9 @@ func Run(cmd string) { } fmt.Println(c.action() + "ing...") if c.install { - err = install.Install(cmd, c.root) + err = install.Install(cmd) } else { - err = install.Uninstall(cmd, c.root) + err = install.Uninstall(cmd) } if err != nil { fmt.Printf("%s failed! %s\n", c.action(), err) @@ -40,7 +40,7 @@ func Run(cmd string) { // prompt use for approval func prompt(action, cmd string) bool { - fmt.Printf("%s bash completion for %s? ", action, cmd) + fmt.Printf("%s completion for %s? ", action, cmd) var answer string fmt.Scanln(&answer) @@ -56,7 +56,6 @@ func prompt(action, cmd string) bool { type config struct { install bool uninstall bool - root bool yes bool } @@ -64,13 +63,9 @@ type config struct { func parseFlags(cmd string) config { var c config flag.BoolVar(&c.install, "install", false, - fmt.Sprintf("Install bash completion for %s command", cmd)) + fmt.Sprintf("Install 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") + fmt.Sprintf("Uninstall completion for %s command", cmd)) flag.BoolVar(&c.yes, "y", false, "Don't prompt user for typing 'yes'") flag.Parse() return c diff --git a/cmd/install/home.go b/cmd/install/bash.go similarity index 95% rename from cmd/install/home.go rename to cmd/install/bash.go index 2694e96..c8bff49 100644 --- a/cmd/install/home.go +++ b/cmd/install/bash.go @@ -11,9 +11,9 @@ import ( "path/filepath" ) -type home struct{} +type bash struct{} -func (home) Install(cmd, bin string) error { +func (bash) Install(cmd, bin string) error { bashRCFileName, err := bashRCFileName() if err != nil { return err @@ -32,7 +32,7 @@ func (home) Install(cmd, bin string) error { return err } -func (home) Uninstall(cmd, bin string) error { +func (bash) Uninstall(cmd, bin string) error { bashRC, err := bashRCFileName() if err != nil { return err @@ -57,6 +57,7 @@ func (home) Uninstall(cmd, bin string) error { } return os.Remove(backup) + } func completeCmd(cmd, bin string) string { diff --git a/cmd/install/install.go b/cmd/install/install.go index c20e11a..bb44ad8 100644 --- a/cmd/install/install.go +++ b/cmd/install/install.go @@ -1,6 +1,8 @@ package install import ( + "errors" + "fmt" "os" "path/filepath" ) @@ -12,33 +14,47 @@ type installer interface { // Install complete command given: // cmd: is the command name -// asRoot: if true the completion will be installed in /etc/bash_complete.d -// otherwise the complete command will be added to the ~/.bashrc file. -func Install(cmd string, asRoot bool) error { +func Install(cmd string) error { + shell := shellType() + if shell == "" { + return errors.New("must install through a terminatl") + } + i := getInstaller(shell) + if i == nil { + return fmt.Errorf("shell %s not supported", shell) + } bin, err := getBinaryPath() if err != nil { return err } - return getInstaller(asRoot).Install(cmd, bin) + return i.Install(cmd, bin) } // Uninstall complete command given: // cmd: is the command name -// asRoot: if true the completion will be removed from /etc/bash_complete.d -// otherwise the complete command will be removed from the ~/.bashrc file. -func Uninstall(cmd string, asRoot bool) error { +func Uninstall(cmd string) error { + shell := shellType() + if shell == "" { + return errors.New("must uninstall through a terminatl") + } + i := getInstaller(shell) + if i == nil { + return fmt.Errorf("shell %s not supported", shell) + } bin, err := getBinaryPath() if err != nil { return err } - return getInstaller(asRoot).Uninstall(cmd, bin) + return i.Uninstall(cmd, bin) } -func getInstaller(asRoot bool) installer { - if asRoot { - return root{} +func getInstaller(shell string) installer { + switch shell { + case "bash": + return bash{} + default: + return nil } - return home{} } func getBinaryPath() (string, error) { @@ -48,3 +64,8 @@ func getBinaryPath() (string, error) { } return filepath.Abs(bin) } + +func shellType() string { + shell := os.Getenv("SHELL") + return filepath.Base(shell) +} diff --git a/cmd/install/root.go b/cmd/install/root.go deleted file mode 100644 index 66e23b3..0000000 --- a/cmd/install/root.go +++ /dev/null @@ -1,29 +0,0 @@ -package install - -import "os" - -type root struct{} - -func (r root) Install(cmd string, bin string) error { - completeLink := getBashCompletionDLink(cmd) - err := r.Uninstall(cmd, bin) - if err != nil { - return err - } - return os.Symlink(bin, completeLink) -} - -func (root) Uninstall(cmd string, bin string) error { - completeLink := getBashCompletionDLink(cmd) - if _, err := os.Stat(completeLink); err == nil { - err := os.Remove(completeLink) - if err != nil { - return err - } - } - return nil -} - -func getBashCompletionDLink(cmd string) string { - return "/etc/bash_completion.d/" + cmd -}