80 lines
1.5 KiB
Go
80 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"text/template"
|
|
|
|
goose "github.com/Altoros/gorm-goose/lib/gorm-goose"
|
|
)
|
|
|
|
// global options. available to any subcommands.
|
|
var flagPath = flag.String("path", "db", "folder containing db info")
|
|
var flagEnv = flag.String("env", "development", "which DB environment to use")
|
|
var flagPgSchema = flag.String("pgschema", "", "which postgres-schema to migrate (default = none)")
|
|
|
|
// helper to create a DBConf from the given flags
|
|
func dbConfFromFlags() (dbconf *goose.DBConf, err error) {
|
|
return goose.NewDBConf(*flagPath, *flagEnv, *flagPgSchema)
|
|
}
|
|
|
|
var commands = []*Command{
|
|
upCmd,
|
|
downCmd,
|
|
redoCmd,
|
|
statusCmd,
|
|
createCmd,
|
|
dbVersionCmd,
|
|
}
|
|
|
|
func main() {
|
|
|
|
flag.Usage = usage
|
|
flag.Parse()
|
|
|
|
args := flag.Args()
|
|
if len(args) == 0 || args[0] == "-h" {
|
|
flag.Usage()
|
|
return
|
|
}
|
|
|
|
var cmd *Command
|
|
name := args[0]
|
|
for _, c := range commands {
|
|
if strings.HasPrefix(c.Name, name) {
|
|
cmd = c
|
|
break
|
|
}
|
|
}
|
|
|
|
if cmd == nil {
|
|
fmt.Printf("error: unknown command %q\n", name)
|
|
flag.Usage()
|
|
os.Exit(1)
|
|
}
|
|
|
|
cmd.Exec(args[1:])
|
|
}
|
|
|
|
func usage() {
|
|
fmt.Print(usagePrefix)
|
|
flag.PrintDefaults()
|
|
usageTmpl.Execute(os.Stdout, commands)
|
|
}
|
|
|
|
var usagePrefix = `
|
|
gorm-goose is a database migration management system for Go projects.
|
|
|
|
Usage:
|
|
gorm-goose [options] <subcommand> [subcommand options]
|
|
|
|
Options:
|
|
`
|
|
var usageTmpl = template.Must(template.New("usage").Parse(
|
|
`
|
|
Commands:{{range .}}
|
|
{{.Name | printf "%-10s"}} {{.Summary}}{{end}}
|
|
`))
|