Add tentez switch

README.md

@@ -111,6 +111,7 @@ aws_listener_rules:
 
 ```console
 # rollback
+# if you want to skip the pause step, add `--no-pause` flag.
 $ tentez -f ./examples/example.yaml rollback
 1. pause
 2. switch old:new = 100:0
@@ -135,6 +136,34 @@ Switched at 2022-03-12 12:05:30
 Apply complete!
 ```
 
+```console
+# switch weights
+# this command overrides steps of the config file.
+# if you want to skip the pause step, add `--no-pause` flag.
+$ tentez -f ./examples/example.yaml switch --weights 70,30
+1. pause
+2. switch old:new = 70:30
+  1. tentez-web
+  2. tentez-api
+  3. tentez-foo
+
+1 / 2 steps
+Pause
+enter "yes", continue steps.
+If you'd like to interrupt steps, enter "quit".
+> yes
+continue step
+
+2 / 2 steps
+Switch old:new = 70:30
+1. tentez-web switched!
+2. tentez-api switched!
+3. tentez-foo switched!
+Switched at 2023-05-25 13:18:25
+
+Apply complete!
+```
+
 ```console
 # show version
 $ tentez version

internal/cli/apply.go

@@ -12,7 +12,6 @@ var applyCmd = &cobra.Command{
 # show plan and apply
 $ tentez -f ./examples/example.yaml apply`,
 	RunE: func(cmd *cobra.Command, args []string) error {
-		filename := cmd.Flag("filename").Value.String()
 		t, err := tentez.NewFromYaml(filename)
 		if err != nil {
 			return err

internal/cli/get.go

@@ -15,7 +15,6 @@ var getCmd = &cobra.Command{
 # Show current state of targets
 $ tentez -f ./examples/example.yaml get`,
 	RunE: func(cmd *cobra.Command, args []string) error {
-		filename := cmd.Flag("filename").Value.String()
 		t, err := tentez.NewFromYaml(filename)
 		if err != nil {
 			return err

internal/cli/plan.go

@@ -12,7 +12,6 @@ var planCmd = &cobra.Command{
 # show plan
 $ tentez -f ./examples/example.yaml plan`,
 	RunE: func(cmd *cobra.Command, args []string) error {
-		filename := cmd.Flag("filename").Value.String()
 		t, err := tentez.NewFromYaml(filename)
 		if err != nil {
 			return err

internal/cli/rollback.go

@@ -12,18 +12,20 @@ var rollbackCmd = &cobra.Command{
 # rollback
 $ tentez -f ./examples/example.yaml rollback`,
 	RunE: func(cmd *cobra.Command, args []string) error {
-		filename := cmd.Flag("filename").Value.String()
 		t, err := tentez.NewFromYaml(filename)
+
 		if err != nil {
 			return err
 		}
 
-		return t.Rollback(true)
+		return t.Rollback(!noPause)
 	},
 }
 
 func init() {
 	rollbackCmd.Flags().StringVarP(&filename, "filename", "f", "", "config file for tentez")
+	rollbackCmd.Flags().BoolVar(&noPause, "no-pause", false, "skip pause")
+
 	if err := rollbackCmd.MarkFlagRequired("filename"); err != nil {
 		panic(err)
 	}

internal/cli/root.go

@@ -9,6 +9,7 @@ import (
 )
 
 var filename = ""
+var noPause = false
 
 var rootCmd = &cobra.Command{
 	Use:   "tentez",

internal/cli/switch.go

@@ -0,0 +1,42 @@
+package cli
+
+import (
+	"fmt"
+
+	"github.com/FeLvi-zzz/tentez"
+	"github.com/spf13/cobra"
+)
+
+var switchCmd = &cobra.Command{
+	Use:   "switch",
+	Short: "Switch targets weights",
+	Long: `
+# show plan and switch
+$ tentez -f ./examples/example.yaml switch --weight 30,70`,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		t, err := tentez.NewFromYaml(filename)
+		if err != nil {
+			return err
+		}
+
+		if len(weights) != 2 {
+			return fmt.Errorf("the length of weights must be 2, got %v, len: %d", weights, len(weights))
+		}
+
+		return t.Switch(weights, !noPause)
+	},
+}
+
+var weights []int
+
+func init() {
+	switchCmd.Flags().StringVarP(&filename, "filename", "f", "", "config file for tentez")
+	switchCmd.Flags().IntSliceVarP(&weights, "weights", "w", []int{}, "weights for switch, 'old,new'")
+	switchCmd.Flags().BoolVar(&noPause, "no-pause", false, "skip pause")
+
+	if err := switchCmd.MarkFlagRequired("filename"); err != nil {
+		panic(err)
+	}
+
+	rootCmd.AddCommand(switchCmd)
+}

tentez.go

@@ -10,6 +10,7 @@ type Tentez interface {
 	Apply(isForce bool) error
 	Get() (map[TargetType]TargetsData, error)
 	Rollback(hasPause bool) error
+	Switch(weights []int, hasPause bool) error
 }
 
 type tentez struct {
@@ -129,3 +130,26 @@ func (t tentez) Rollback(hasPause bool) (err error) {
 	}
 	return t.Apply(true)
 }
+
+func (t tentez) Switch(weights []int, hasPause bool) (err error) {
+	t.Steps = []Step{}
+
+	if hasPause {
+		t.Steps = append(t.Steps, Step{
+			Type: StepTypePause,
+		})
+	}
+
+	t.Steps = append(t.Steps, Step{
+		Type: StepTypeSwitch,
+		Weight: Weight{
+			Old: int32(weights[0]),
+			New: int32(weights[1]),
+		},
+	})
+
+	if err = t.Plan(); err != nil {
+		return err
+	}
+	return t.Apply(false)
+}