Skip to content

Commit

Permalink
Add --output to gengen command
Browse files Browse the repository at this point in the history
  • Loading branch information
jackc committed Aug 8, 2023
1 parent 42057ab commit 4a122cc
Showing 1 changed file with 16 additions and 1 deletion.
17 changes: 16 additions & 1 deletion main.go
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,7 @@ var cliOptions struct {
migrationsPath string
configPaths []string
editNewMigration bool
gengenOutputFile string

connString string
host string
Expand Down Expand Up @@ -320,6 +321,7 @@ Migrations can only go forward to the latest version.
cmdGengen.Flags().StringSliceVarP(&cliOptions.configPaths, "config", "c", []string{}, "config path (default is ./tern.conf)")
cmdGengen.Flags().StringVarP(&cliOptions.versionTable, "version-table", "", "", "version table name (default is public.schema_version)")
cmdGengen.Flags().StringVarP(&cliOptions.migrationsPath, "migrations", "m", "", "migrations path (default is .)")
cmdGengen.Flags().StringVarP(&cliOptions.gengenOutputFile, "output", "o", "", "output file")

cmdVersion := &cobra.Command{
Use: "version",
Expand Down Expand Up @@ -682,7 +684,20 @@ order by version asc;
`))

err = gengenTemplate.Execute(os.Stdout, map[string]any{
var out *os.File
if cliOptions.gengenOutputFile == "" {
out = os.Stdout
} else {
var err error
out, err = os.Create(cliOptions.gengenOutputFile)
if err != nil {
fmt.Fprintln(os.Stderr, err)
os.Exit(1)
}
defer out.Close()
}

err = gengenTemplate.Execute(out, map[string]any{
"Version": VERSION,
"VersionTable": config.VersionTable,
"Migrations": migrator.Migrations,
Expand Down

0 comments on commit 4a122cc

Please sign in to comment.