Add CLI flags

Author: realkarych

main.go

@@ -148,6 +148,22 @@ func NewApp() *cli.App {
 			Name:  "migrate",
 			Usage: "Migrate to the latest version",
 			Flags: []cli.Flag{
+				&cli.StringFlag{
+					Name:  "only",
+					Usage: "apply only the specified migration version",
+				},
+				&cli.StringFlag{
+					Name:  "to",
+					Usage: "apply migrations up to specified version (inclusive)",
+				},
+				&cli.BoolFlag{
+					Name:  "one",
+					Usage: "apply exactly one pending migration",
+				},
+				&cli.BoolFlag{
+					Name:  "all",
+					Usage: "apply all pending migrations (default)",
+				},
 				&cli.BoolFlag{
 					Name:    "strict",
 					EnvVars: []string{"DBMATE_STRICT"},
@@ -163,14 +179,42 @@ func NewApp() *cli.App {
 			Action: action(func(db *dbmate.DB, c *cli.Context) error {
 				db.Strict = c.Bool("strict")
 				db.Verbose = c.Bool("verbose")
-				return db.Migrate()
+				if version := c.String("only"); version != "" {
+					migrations, err := db.FindMigrations()
+					if err != nil {
+						return err
+					}
+					return db.MigrateOnly(migrations, version)
+				} else if version := c.String("to"); version != "" {
+					return db.MigrateTo(version)
+				} else if c.Bool("one") {
+					return db.MigrateNext()
+				} else {
+					return db.Migrate() // default: apply all
+				}
 			}),
 		},
 		{
 			Name:    "rollback",
 			Aliases: []string{"down"},
 			Usage:   "Rollback the most recent migration",
 			Flags: []cli.Flag{
+				&cli.StringFlag{
+					Name:  "only",
+					Usage: "rollback only the specified migration version",
+				},
+				&cli.StringFlag{
+					Name:  "to",
+					Usage: "rollback migrations down to specified version (exclusive)",
+				},
+				&cli.BoolFlag{
+					Name:  "one",
+					Usage: "rollback exactly one migration (default)",
+				},
+				&cli.BoolFlag{
+					Name:  "all",
+					Usage: "rollback all migrations",
+				},
 				&cli.BoolFlag{
 					Name:    "verbose",
 					Aliases: []string{"v"},
@@ -180,7 +224,19 @@ func NewApp() *cli.App {
 			},
 			Action: action(func(db *dbmate.DB, c *cli.Context) error {
 				db.Verbose = c.Bool("verbose")
-				return db.Rollback()
+				if version := c.String("only"); version != "" {
+					migrations, err := db.FindMigrations()
+					if err != nil {
+						return err
+					}
+					return db.RollbackOnly(migrations, version)
+				} else if version := c.String("to"); version != "" {
+					return db.RollbackTo(version)
+				} else if c.Bool("all") {
+					return db.RollbackAll()
+				} else {
+					return db.Rollback() // default: rollback one migration
+				}
 			}),
 		},
 		{

pkg/dbmate/db.go

@@ -741,8 +741,8 @@ func (db *DB) Rollback() error {
 	return nil
 }
 
-// migrateOnly applies exactly one pending migration that matches the given version.
-func (db *DB) migrateOnly(migrations []Migration, version string) error {
+// MigrateOnly applies exactly one pending migration that matches the given version.
+func (db *DB) MigrateOnly(migrations []Migration, version string) error {
 	drv, err := db.Driver()
 	if err != nil {
 		return err
@@ -808,8 +808,8 @@ func (db *DB) migrateOnly(migrations []Migration, version string) error {
 	return nil
 }
 
-// rollbackOnly rolls back exactly one applied migration that matches the given version.
-func (db *DB) rollbackOnly(migrations []Migration, version string) error {
+// RollbackOnly rolls back exactly one applied migration that matches the given version.
+func (db *DB) RollbackOnly(migrations []Migration, version string) error {
 	drv, err := db.Driver()
 	if err != nil {
 		return err
@@ -883,7 +883,7 @@ func (db *DB) MigrateTo(target string) error {
 		if m.Applied || m.Version > target {
 			continue
 		}
-		if err := db.migrateOnly(migrations, m.Version); err != nil {
+		if err := db.MigrateOnly(migrations, m.Version); err != nil {
 			return err
 		}
 	}
@@ -922,7 +922,7 @@ func (db *DB) RollbackTo(target string) error {
 		if !m.Applied || m.Version <= target {
 			continue
 		}
-		if err := db.rollbackOnly(migrations, m.Version); err != nil {
+		if err := db.RollbackOnly(migrations, m.Version); err != nil {
 			return err
 		}
 	}