add activate flag to install command

MrLYC · Dec 27, 2021 · 0293080 · 0293080
1 parent 458a596
commit 0293080
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/cmd/command/install.go b/cmd/command/install.go
@@ -8,11 +8,16 @@ import (
 	"github.com/mrlyc/cmdr/utils"
 )
 
+var installCmdFlag struct {
+	activate bool
+}
+
 // installCmd represents the install command
 var installCmd = &cobra.Command{
 	Use:   "install",
 	Short: "Install command into cmdr",
 	Run: func(cmd *cobra.Command, args []string) {
+		binDir := core.GetBinDir()
 		shimsDir := core.GetShimsDir()
 
 		runner := core.NewStepRunner(
@@ -22,6 +27,14 @@ var installCmd = &cobra.Command{
 			core.NewBinariesInstaller(shimsDir),
 		)
 
+		if installCmdFlag.activate {
+			runner.Add(
+				core.NewCommandDeactivator(),
+				core.NewBinariesActivator(binDir, shimsDir),
+				core.NewCommandActivator(),
+			)
+		}
+
 		utils.ExitWithError(runner.Run(cmd.Context()), "install failed")
 
 		define.Logger.Info("installed command", map[string]interface{}{
@@ -39,6 +52,7 @@ func init() {
 	flags.StringVarP(&simpleCmdFlag.name, "name", "n", "", "command name")
 	flags.StringVarP(&simpleCmdFlag.version, "version", "v", "", "command version")
 	flags.StringVarP(&simpleCmdFlag.location, "location", "l", "", "command location")
+	flags.BoolVarP(&installCmdFlag.activate, "activate", "a", false, "activate command")
 
 	installCmd.MarkFlagRequired("name")
 	installCmd.MarkFlagRequired("version")