Method ProvideCommand

otgorm/module.go:57–135

ProvideCommand provides migration and seed command.

(command *cobra.Command)

// ProvideCommand provides migration and seed command.
func (m Module) ProvideCommand(command *cobra.Command) {
	var (
		force      bool
		rollbackId string
		logger     = logging.WithLevel(m.logger)
	)
	migrateCmd := &cobra.Command{
		Use:   "migrate [database]",
		Short: "Migrate gorm tables",
		Long:  `Run all gorm table migrations.`,
		RunE: func(cmd *cobra.Command, args []string) error {
			connection := "default"
			if len(args) > 0 {
				connection = args[0]
			}

			if m.env.IsProduction() && !force {
				e := fmt.Errorf("migrations and rollback in production requires force flag to be set")
				return e
			}

			migrations := m.collectMigrations(connection)

			if rollbackId != "" {
				if err := migrations.Rollback(rollbackId); err != nil {
					return fmt.Errorf("unable to rollback: %w", err)
				}

				logger.Info("rollback successfully completed")
				return nil
			}

			if err := migrations.Migrate(); err != nil {
				return fmt.Errorf("unable to migrate: %w", err)
			}

			logger.Info("migration successfully completed")
			return nil
		},
	}
	migrateCmd.Flags().BoolVarP(&force, "force", "f", false, "migrations and rollback in production requires force flag to be set")
	migrateCmd.Flags().StringVarP(&rollbackId, "rollback", "r", "", "rollback to the given migration id")
	migrateCmd.Flag("rollback").NoOptDefVal = "-1"

	seedCmd := &cobra.Command{
		Use:   "seed [database]",
		Short: "seed the database",
		Long:  `use the provided seeds to bootstrap fake data in database`,
		RunE: func(cmd *cobra.Command, args []string) error {
			connection := "default"
			if len(args) > 0 {
				connection = args[0]
			}

			if m.env.IsProduction() && !force {
				return fmt.Errorf("seeding in production requires force flag to be set")
			}