fixtures.go

package schemaherocli

import (
	"github.com/schemahero/schemahero/pkg/database"

	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

func Fixtures() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "fixtures",
		Short: "fixtures creates sql statements from a schemahero definition",
		Long:  `...`,
		PreRun: func(cmd *cobra.Command, args []string) {
			viper.BindPFlags(cmd.Flags())
		},
		RunE: func(cmd *cobra.Command, args []string) error {
			v := viper.GetViper()

			db := database.Database{
				InputDir:    v.GetString("input-dir"),
				OutputDir:   v.GetString("output-dir"),
				Driver:      v.GetString("driver"),
				URI:         v.GetString("uri"),
				VaultURIRef: v.GetString("vault-uri-ref"),
			}

			return db.CreateFixturesSync()
		},
	}

	cmd.Flags().String("driver", "", "name of the database driver to use")
	cmd.Flags().String("dbname", "", "schemahero database name to write in the yaml")
	cmd.Flags().String("input-dir", "", "directory to read schema files from")
	cmd.Flags().String("output-dir", "", "directory to write fixture files to")

	cmd.MarkFlagRequired("driver")
	cmd.MarkFlagRequired("dbname")
	cmd.MarkFlagRequired("input-dir")
	cmd.MarkFlagRequired("output-dir")

	return cmd
}