From 0bf6bef042d68687a768e52c1ae128b6ee017bcf Mon Sep 17 00:00:00 2001
From: Sukharev Maxim <maxim.sukharev@lazada.com>
Date: Sun, 2 Jul 2017 12:28:40 +0700
Subject: [PATCH 1/2] Reset & refresh commands

Reset rolls back all migrations
Refresh rolls back all migrations and applies all available migrations again
it's useful for development to put db in latest state without full
recreation of db.
---
 cmd/goose/main.go |  2 ++
 goose.go          | 11 +++++++++
 reset.go          | 59 +++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 72 insertions(+)
 create mode 100644 reset.go

diff --git a/cmd/goose/main.go b/cmd/goose/main.go
index c589029..fbbead4 100644
--- a/cmd/goose/main.go
+++ b/cmd/goose/main.go
@@ -116,6 +116,8 @@ Commands:
     down                 Roll back the version by 1
     down-to VERSION      Roll back to a specific VERSION
     redo                 Re-run the latest migration
+    reset                Roll back all migrations
+    refresh              Roll back all migrations and apply all available migrations again
     status               Dump the migration status for the current DB
     version              Print the current version of the database
     create NAME [sql|go] Creates new migration file with next version
diff --git a/goose.go b/goose.go
index 3861e41..330d5f9 100644
--- a/goose.go
+++ b/goose.go
@@ -68,6 +68,17 @@ func Run(command string, db *sql.DB, dir string, args ...string) error {
 		if err := Redo(db, dir); err != nil {
 			return err
 		}
+	case "reset":
+		if err := Reset(db, dir); err != nil {
+			return err
+		}
+	case "refresh":
+		if err := Reset(db, dir); err != nil {
+			return err
+		}
+		if err := Up(db, dir); err != nil {
+			return err
+		}
 	case "status":
 		if err := Status(db, dir); err != nil {
 			return err
diff --git a/reset.go b/reset.go
new file mode 100644
index 0000000..c7e0b4d
--- /dev/null
+++ b/reset.go
@@ -0,0 +1,59 @@
+package goose
+
+import (
+	"database/sql"
+	"log"
+	"sort"
+)
+
+// Reset rolls back all migrations
+func Reset(db *sql.DB, dir string) error {
+	migrations, err := CollectMigrations(dir, minVersion, maxVersion)
+	if err != nil {
+		return err
+	}
+	statuses, err := dbMigrationsStatus(db)
+	if err != nil {
+		return err
+	}
+	sort.Sort(sort.Reverse(migrations))
+
+	for _, migration := range migrations {
+		if !statuses[migration.Version] {
+			continue
+		}
+		if err = migration.Down(db); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func dbMigrationsStatus(db *sql.DB) (map[int64]bool, error) {
+	rows, err := GetDialect().dbVersionQuery(db)
+	if err != nil {
+		return map[int64]bool{}, createVersionTable(db)
+	}
+	defer rows.Close()
+
+	// The most recent record for each migration specifies
+	// whether it has been applied or rolled back.
+
+	result := make(map[int64]bool)
+
+	for rows.Next() {
+		var row MigrationRecord
+		if err = rows.Scan(&row.VersionID, &row.IsApplied); err != nil {
+			log.Fatal("error scanning rows:", err)
+		}
+
+		if _, ok := result[row.VersionID]; ok {
+			continue
+		}
+
+		result[row.VersionID] = row.IsApplied
+	}
+
+	return result, nil
+}

From b800f677cbdaf4dae6ee9bf76377ed46ed8e83bc Mon Sep 17 00:00:00 2001
From: Sukharev Maxim <maxim.sukharev@lazada.com>
Date: Thu, 6 Jul 2017 10:17:33 +0700
Subject: [PATCH 2/2] remove refresh command

---
 cmd/goose/main.go | 1 -
 goose.go          | 7 -------
 2 files changed, 8 deletions(-)

diff --git a/cmd/goose/main.go b/cmd/goose/main.go
index fbbead4..c59079e 100644
--- a/cmd/goose/main.go
+++ b/cmd/goose/main.go
@@ -117,7 +117,6 @@ Commands:
     down-to VERSION      Roll back to a specific VERSION
     redo                 Re-run the latest migration
     reset                Roll back all migrations
-    refresh              Roll back all migrations and apply all available migrations again
     status               Dump the migration status for the current DB
     version              Print the current version of the database
     create NAME [sql|go] Creates new migration file with next version
diff --git a/goose.go b/goose.go
index 330d5f9..51ca6ec 100644
--- a/goose.go
+++ b/goose.go
@@ -72,13 +72,6 @@ func Run(command string, db *sql.DB, dir string, args ...string) error {
 		if err := Reset(db, dir); err != nil {
 			return err
 		}
-	case "refresh":
-		if err := Reset(db, dir); err != nil {
-			return err
-		}
-		if err := Up(db, dir); err != nil {
-			return err
-		}
 	case "status":
 		if err := Status(db, dir); err != nil {
 			return err