github.com/paweljw/pop/v5@v5.4.6/migration_box.go

package pop

import (
	"strings"

	"github.com/gobuffalo/packd"
	"github.com/gobuffalo/pop/v5/logging"
	"github.com/pkg/errors"
)

// MigrationBox is a wrapper around packr.Box and Migrator.
// This will allow you to run migrations from a packed box
// inside of a compiled binary.
type MigrationBox struct {
	Migrator
	Box packd.Walkable
}

// NewMigrationBox from a packr.Box and a Connection.
func NewMigrationBox(box packd.Walkable, c *Connection) (MigrationBox, error) {
	fm := MigrationBox{
		Migrator: NewMigrator(c),
		Box:      box,
	}

	runner := func(f packd.File) func(mf Migration, tx *Connection) error {
		return func(mf Migration, tx *Connection) error {
			content, err := MigrationContent(mf, tx, f, true)
			if err != nil {
				return errors.Wrapf(err, "error processing %s", mf.Path)
			}
			if content == "" {
				return nil
			}
			err = tx.RawQuery(content).Exec()
			if err != nil {
				return errors.Wrapf(err, "error executing %s, sql: %s", mf.Path, content)
			}
			return nil
		}
	}

	err := fm.findMigrations(runner)
	if err != nil {
		return fm, err
	}

	return fm, nil
}

func (fm *MigrationBox) findMigrations(runner func(f packd.File) func(mf Migration, tx *Connection) error) error {
	return fm.Box.Walk(func(p string, f packd.File) error {
		info, err := f.FileInfo()
		if err != nil {
			return err
		}
		match, err := ParseMigrationFilename(info.Name())
		if err != nil {
			if strings.HasPrefix(err.Error(), "unsupported dialect") {
				log(logging.Warn, "ignoring migration file with %s", err.Error())
				return nil
			}
			return err
		}
		if match == nil {
			log(logging.Warn, "ignoring file %s because it does not match the migration file pattern", info.Name())
			return nil
		}
		mf := Migration{
			Path:      p,
			Version:   match.Version,
			Name:      match.Name,
			DBType:    match.DBType,
			Direction: match.Direction,
			Type:      match.Type,
			Runner:    runner(f),
		}
		fm.Migrations[mf.Direction] = append(fm.Migrations[mf.Direction], mf)
		return nil
	})
}