Enable force to download install/uninstall script when tag is master

innobead · Aug 17, 2020 · d42045e · d42045e
1 parent 153030a
commit d42045e
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/cmd/kubefire/cmd/install.go b/cmd/kubefire/cmd/install.go
@@ -7,17 +7,24 @@ import (
 	"os/exec"
 )
 
+var forceDownload bool
+
 var InstallCmd = &cobra.Command{
 	Use:   "install",
 	Short: "Install prerequisites",
+	PreRun: func(cmd *cobra.Command, args []string) {
+		if !forceDownload {
+			forceDownload = config.TagVersion == "master"
+		}
+	},
 	RunE: func(cmd *cobra.Command, args []string) error {
 		// download install script
 		scripts := []script.Type{
 			script.InstallPrerequisites,
 		}
 
 		for _, s := range scripts {
-			if err := script.Download(s, config.TagVersion, true); err != nil {
+			if err := script.Download(s, config.TagVersion, forceDownload); err != nil {
 				return err
 			}
 
@@ -30,6 +37,11 @@ var InstallCmd = &cobra.Command{
 	},
 }
 
+func init() {
+	flags := InstallCmd.Flags()
+	flags.BoolVar(&forceDownload, "force", false, "force to install")
+}
+
 func createSetupInstallCommandEnvsFunc() func(cmd *exec.Cmd) error {
 	return func(cmd *exec.Cmd) error {
 		cmd.Env = append(

diff --git a/cmd/kubefire/cmd/uninstall.go b/cmd/kubefire/cmd/uninstall.go
@@ -9,8 +9,13 @@ import (
 var UninstallCmd = &cobra.Command{
 	Use:   "uninstall",
 	Short: "Uninstall prerequisites",
+	PreRun: func(cmd *cobra.Command, args []string) {
+		if !forceDownload {
+			forceDownload = config.TagVersion == "master"
+		}
+	},
 	RunE: func(cmd *cobra.Command, args []string) error {
-		if err := script.Download(script.UninstallPrerequisites, config.TagVersion, true); err != nil {
+		if err := script.Download(script.UninstallPrerequisites, config.TagVersion, forceDownload); err != nil {
 			return err
 		}
 
@@ -21,3 +26,8 @@ var UninstallCmd = &cobra.Command{
 		return nil
 	},
 }
+
+func init() {
+	flags := UninstallCmd.Flags()
+	flags.BoolVar(&forceDownload, "force", false, "force to uninstall")
+}