# HG changeset patch
# User Peter Sanchez <peter@netlandish.com>
# Date 1645291520 21600
#      Sat Feb 19 11:25:20 2022 -0600
# Node ID d23f6f83d00a569e6b51233e66e8ef5db478b743
# Parent  c6b69ef9bdd6e43c4ee35ceebb2d73ec879d1816
README update for Command function

diff --git a/README.md b/README.md
--- a/README.md
+++ b/README.md
@@ -95,39 +95,6 @@
 func main() {
 	// `db` is your database connection (*sql.DB)
 
-	var (
-		verbose, fakeIt   bool
-		direction, stopID string
-	)
-
-	mf := flag.NewFlagSet("migrate", flag.ContinueOnError)
-	mf.BoolVar(&verbose, "verbose", false, "verbose output")
-	mf.BoolVar(&fakeIt, "fake", false, "fake migrations")
-	mf.Usage = func() {
-        msg := fmt.Sprintf("usage: %s migrate [options...] [up | down] [migration ID]\n", os.Args[0])
-        fmt.Fprintf(flag.CommandLine.Output(), msg)
-        mf.PrintDefaults()
-	}
-
-	if os.Args[1] != "migrate" {
-		mf.Usage()
-		return fmt.Errorf("Unknown command: %s", os.Args[1])
-	}
-
-	mf.Parse(os.Args[2:])
-	args := mf.Args()
-
-	if len(args) > 0 {
-		direction = args[0]
-		if direction != "up" && direction != "down" {
-			return fmt.Errorf("Unknown migration direction: %s", direction)
-		}
-	}
-
-	if len(args) > 1 {
-		stopID = args[1]
-	}
-
 	// Custom functions should be watching for the context's Done() channel
 	migrations := []migrate.Migration{
 		migrate.FileMigration("0001_initial", "migrations/0001_initial.sql", "migrations/0001_initial_rollback.sql", 0),
@@ -145,12 +112,15 @@
 		},
 	}
 
-	ctx := context.Background()
-	engine := migrate.NewEngine(db, migrations, migrate.DOLLAR, verbose)
-	if direction == "down" {
-		return engine.Rollback(ctx, stopID)
+	if len(os.Args) > 0 {
+		if os.Args[1] != "migrate" {
+			err := migrate.Command(db, migrations, migrate.DOLLAR)
+			// handle err
+			os.Exit(0)
+		}
 	}
-	return engine.Migrate(ctx, stopID, fakeIt)
+
+	// rest of your main() code
 }
 
 ```