feat(install): allow define install/uninstall dir

allow define install/uninstall dir

Signed-off-by: mritd <[email protected]>

Author: mritd

cmd/install.go

@@ -21,16 +21,18 @@ import (
 	"github.com/spf13/cobra"
 )
 
+var installDir string
 var installCmd = &cobra.Command{
 	Use:   "install",
 	Short: "Install mmh",
 	Long: `
 Install mmh.`,
 	Run: func(cmd *cobra.Command, args []string) {
-		utils.Install()
+		utils.Install(installDir)
 	},
 }
 
 func init() {
+	installCmd.PersistentFlags().StringVar(&installDir, "dir", "/usr/bin", "install dir")
 	RootCmd.AddCommand(installCmd)
 }

cmd/uninstall.go

@@ -21,16 +21,18 @@ import (
 	"github.com/spf13/cobra"
 )
 
+var uninstallDir string
 var uninstallCmd = &cobra.Command{
 	Use:   "uninstall",
 	Short: "Uninstall mmh",
 	Long: `
 Uninstall mmh.`,
 	Run: func(cmd *cobra.Command, args []string) {
-		utils.Uninstall()
+		utils.Uninstall(uninstallDir)
 	},
 }
 
 func init() {
+	uninstallCmd.PersistentFlags().StringVar(&uninstallDir, "dir", "/usr/bin", "uninstall dir")
 	RootCmd.AddCommand(uninstallCmd)
 }

pkg/utils/install.go

@@ -24,34 +24,32 @@ import (
 	"path"
 )
 
-const InstallBaseDir = "/usr/bin"
+func Install(dir string) {
 
-var BinPaths = []string{
-	path.Join(InstallBaseDir, "mcp"),
-	path.Join(InstallBaseDir, "mec"),
-	path.Join(InstallBaseDir, "mgo"),
-}
-
-func Install() {
+	var BinPaths = []string{
+		path.Join(dir, "mcp"),
+		path.Join(dir, "mec"),
+		path.Join(dir, "mgo"),
+	}
 
-	Uninstall()
+	Uninstall(dir)
 
 	fmt.Println("Install")
 	currentPath, err := exec.LookPath(os.Args[0])
 	CheckAndExit(err)
 	f, err := os.Open(currentPath)
 	CheckAndExit(err)
 	defer f.Close()
-	target, err := os.OpenFile(path.Join(InstallBaseDir, "mmh"), os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0755)
+	target, err := os.OpenFile(path.Join(dir, "mmh"), os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0755)
 	CheckAndExit(err)
 	defer target.Close()
 
-	fmt.Printf("Install %s\n", path.Join(InstallBaseDir, "mmh"))
+	fmt.Printf("Install %s\n", path.Join(dir, "mmh"))
 	_, err = io.Copy(target, f)
 	CheckAndExit(err)
 	for _, bin := range BinPaths {
 		fmt.Printf("Install %s\n", bin)
-		err = os.Symlink(path.Join(InstallBaseDir, "mmh"), bin)
+		err = os.Symlink(path.Join(dir, "mmh"), bin)
 		CheckAndExit(err)
 	}
 }

pkg/utils/uninstall.go

@@ -22,7 +22,14 @@ import (
 	"path"
 )
 
-func Uninstall() {
+func Uninstall(dir string) {
+
+	var BinPaths = []string{
+		path.Join(dir, "mcp"),
+		path.Join(dir, "mec"),
+		path.Join(dir, "mgo"),
+	}
+
 	CheckRoot()
 
 	fmt.Println("Uninstall")
@@ -31,6 +38,6 @@ func Uninstall() {
 		fmt.Printf("Remove %s\n", bin)
 		os.Remove(bin)
 	}
-	fmt.Printf("Remove %s\n", path.Join(InstallBaseDir, "mmh"))
-	os.Remove(path.Join(InstallBaseDir, "mmh"))
+	fmt.Printf("Remove %s\n", path.Join(dir, "mmh"))
+	os.Remove(path.Join(dir, "mmh"))
 }