package main

import (
	"github.com/mattes/migrate"
	_ "github.com/mattes/migrate/database/stub" // TODO remove again
	_ "github.com/mattes/migrate/source/file"
)

func gotoCmd(m *migrate.Migrate, v uint) {
	if err := m.Migrate(v); err != nil {
		log.fatalErr(err)
	}
}

func upCmd(m *migrate.Migrate, limit int) {
	if limit >= 0 {
		if err := m.Steps(limit); err != nil {
			log.fatalErr(err)
		}
	} else {
		if err := m.Up(); err != nil {
			log.fatalErr(err)
		}
	}
}

func downCmd(m *migrate.Migrate, limit int) {
	if limit >= 0 {
		if err := m.Steps(-limit); err != nil {
			log.fatalErr(err)
		}
	} else {
		if err := m.Down(); err != nil {
			log.fatalErr(err)
		}
	}
}

func dropCmd(m *migrate.Migrate) {
	if err := m.Drop(); err != nil {
		log.fatalErr(err)
	}
}

func versionCmd(m *migrate.Migrate) {
	v, err := m.Version()
	if err != nil {
		log.fatalErr(err)
	}
	log.Println(v)
}