package cmd
import (
"database/sql"
"fmt"
"github.com/golang-migrate/migrate/v4"
"github.com/golang-migrate/migrate/v4/database/sqlite"
"github.com/golang-migrate/migrate/v4/source/iofs"
"github.com/spf13/cobra"
"gitrepo.ru/neonxp/gorum/migrations"
)
var migrateCmd = &cobra.Command{
Use: "migrate",
Short: "Migrate db",
Long: `Up and down migrations`,
}
func init() {
migrateCmd.AddCommand(
&cobra.Command{
Use: "up",
Short: "Migrate up",
Long: `Up migrations`,
RunE: up,
},
&cobra.Command{
Use: "down",
Short: "Migrate down",
Long: `Down migrations`,
RunE: down,
},
&cobra.Command{
Use: "drop",
Short: "Drop db",
Long: `Deletes everything in the database`,
RunE: drop,
},
)
}
func up(_ *cobra.Command, _ []string) error {
m, err := initMigrate()
if err != nil {
return fmt.Errorf("open migration failed: %w", err)
}
defer m.Close()
if err := m.Up(); err != nil && err != migrate.ErrNoChange {
return fmt.Errorf("do migration failed: %w", err)
}
return nil
}
func down(_ *cobra.Command, _ []string) error {
m, err := initMigrate()
if err != nil {
return fmt.Errorf("open migration failed: %w", err)
}
defer m.Close()
if err := m.Down(); err != nil && err != migrate.ErrNoChange {
return fmt.Errorf("do migration failed: %w", err)
}
return nil
}
func drop(_ *cobra.Command, _ []string) error {
m, err := initMigrate()
if err != nil {
return fmt.Errorf("open migration failed: %w", err)
}
defer m.Close()
if err := m.Drop(); err != nil && err != migrate.ErrNoChange {
return fmt.Errorf("do migration failed: %w", err)
}
return nil
}
func initMigrate() (*migrate.Migrate, error) {
db, err := sql.Open("sqlite3", dbFile)
if err != nil {
return nil, fmt.Errorf("open db failed: %w", err)
}
defer db.Close()
driver, err := sqlite.WithInstance(db, &sqlite.Config{})
if err != nil {
return nil, fmt.Errorf("failed create migration driver: %w", err)
}
sourceDriver, err := iofs.New(migrations.FS, ".")
if err != nil {
return nil, fmt.Errorf("failed open migrations: %w", err)
}
return migrate.NewWithInstance("fs", sourceDriver, "sqlite3", driver)
}
