atlasexec: amount arg and --to-tag flag to migrate down (#59)

atlasexec/atlas.go

@@ -71,7 +71,9 @@ type (
 		DirURL          string
 		URL             string
 		RevisionsSchema string
+		Amount          uint64
 		ToVersion       string
+		ToTag           string
 		Vars            Vars
 
 		// Not yet supported
@@ -325,6 +327,12 @@ func (c *Client) MigrateDown(ctx context.Context, params *MigrateDownParams) (*M
 	if params.ToVersion != "" {
 		args = append(args, "--to-version", params.ToVersion)
 	}
+	if params.ToTag != "" {
+		args = append(args, "--to-tag", params.ToTag)
+	}
+	if params.Amount > 0 {
+		args = append(args, strconv.FormatUint(params.Amount, 10))
+	}
 	args = append(args, params.Vars.AsArgs()...)
 	r, err := c.runCommand(ctx, args)
 	if cliErr := (cliError{}); errors.As(err, &cliErr) && cliErr.stderr == "" {

atlasexec/atlas_test.go

@@ -881,6 +881,20 @@ func TestMigrateDown(t *testing.T) {
 			},
 			expect: "migrate down --format {{ json . }} --to-version 12345",
 		},
+		{
+			name: "with tag version",
+			params: &atlasexec.MigrateDownParams{
+				ToTag: "12345",
+			},
+			expect: "migrate down --format {{ json . }} --to-tag 12345",
+		},
+		{
+			name: "with amount",
+			params: &atlasexec.MigrateDownParams{
+				Amount: 10,
+			},
+			expect: "migrate down --format {{ json . }} 10",
+		},
 	} {
 		t.Run(tt.name, func(t *testing.T) {
 			_, err := c.MigrateDown(context.Background(), tt.params)