migrate/main.go

package main

import (
	"flag"
	"fmt"
	"github.com/mattes/migrate/migrate"
	"os"
	"strconv"
)

var url = flag.String("url", "", "Driver connection URL, like schema://url")
var migrationsPath = flag.String("path", "", "Path to migrations")

func main() {
	flag.Parse()
	command := flag.Arg(0)

	switch command {
	case "create":
		verifyMigrationsPath(*migrationsPath)
		createCmd(*url, *migrationsPath, flag.Arg(1))

	case "migrate":
		verifyMigrationsPath(*migrationsPath)

		relativeN := flag.Arg(1)
		relativeNInt, err := strconv.Atoi(relativeN)
		if err != nil {
			fmt.Println(err)
			os.Exit(1)
		}
		migrateCmd(*url, *migrationsPath, relativeNInt)

	case "up":
		verifyMigrationsPath(*migrationsPath)
		upCmd(*url, *migrationsPath)

	case "down":
		verifyMigrationsPath(*migrationsPath)
		downCmd(*url, *migrationsPath)

	case "redo":
		verifyMigrationsPath(*migrationsPath)
		redoCmd(*url, *migrationsPath)

	case "reset":
		verifyMigrationsPath(*migrationsPath)
		resetCmd(*url, *migrationsPath)

	case "version":
		verifyMigrationsPath(*migrationsPath)
		versionCmd(*url, *migrationsPath)

	case "help":
		helpCmd()

	default:
		helpCmd()
	}
}

func verifyMigrationsPath(path string) {
	if path == "" {
		fmt.Println("Please specify path")
		os.Exit(1)
	}
}

func createCmd(url, migrationsPath, name string) {
	migrationFile, err := migrate.Create(url, migrationsPath, name)
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

	fmt.Printf("Version %v migration files created in %v:\n", migrationFile.Version, migrationsPath)
	fmt.Println(migrationFile.UpFile.FileName)
	fmt.Println(migrationFile.DownFile.FileName)

}

func upCmd(url, migrationsPath string) {
	if err := migrate.Up(url, migrationsPath); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

}

func downCmd(url, migrationsPath string) {
	if err := migrate.Down(url, migrationsPath); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

}

func redoCmd(url, migrationsPath string) {
	if err := migrate.Redo(url, migrationsPath); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

}

func resetCmd(url, migrationsPath string) {
	if err := migrate.Reset(url, migrationsPath); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

}

func versionCmd(url, migrationsPath string) {
	version, err := migrate.Version(url, migrationsPath)
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	fmt.Println(version)
}

func migrateCmd(url, migrationsPath string, relativeN int) {
	if err := migrate.Migrate(url, migrationsPath, relativeN); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

func helpCmd() {
	os.Stderr.WriteString(
		`usage: migrate [-path=<path>] [-url=<url>] <command> [<args>]

Commands:
   create <name>  Create a new migration
   up             Apply all -up- migrations
   down           Apply all -down- migrations
   reset          Down followed by Up
   redo           Roll back most recent migration, then apply it again
   version        Show current migration version
   migrate <n>    Apply migrations -n|+n
   help           Show this help
`)
}