package sqlgen import ( "fmt" "os" ) const ( DRIVER_SQLITE = "sqlite" DRIVER_POSTGRES = "postgres" ) func Main() { usage := func() { fmt.Fprintf(os.Stderr, ` %s DRIVER DEFS_PATH OUTPUT_PATH Drivers are one of: %s %s `, os.Args[0], DRIVER_SQLITE, DRIVER_POSTGRES) os.Exit(1) } if len(os.Args) != 4 { usage() } var ( driver = os.Args[1] defsPath = os.Args[2] outputPath = os.Args[3] ) switch driver { case DRIVER_SQLITE, DRIVER_POSTGRES: // OK default: fmt.Fprintf(os.Stderr, "Unknown driver: %s", driver) usage() } err := render(driver, defsPath, outputPath) if err != nil { fmt.Fprintf(os.Stderr, "Error: %v", err) os.Exit(1) } }