Install fixups

- remove root installation
- install according to shell type

Closes #7

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

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 {

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)
+}

cmd/install/root.go

@@ -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
-}