migrations.go RawHistoryFolder

1package sqlite
2
3import (
4	"context"
5	"database/sql"
6	"embed"
7	"fmt"
8
9	"github.com/bnema/dumber/internal/logging"
10	"github.com/pressly/goose/v3"
11)
12
13//go:embed migrations/*.sql
14var embedMigrations embed.FS
15
16// RunMigrations applies all pending migrations to the database.
17// Uses goose with embedded SQL files for automatic schema management.
18func RunMigrations(ctx context.Context, db *sql.DB) error {
19	log := logging.FromContext(ctx)
20
21	// Configure goose
22	goose.SetBaseFS(embedMigrations)
23
24	if err := goose.SetDialect("sqlite3"); err != nil {
25		return fmt.Errorf("failed to set goose dialect: %w", err)
26	}
27
28	// Disable verbose logging from goose
29	goose.SetLogger(goose.NopLogger())
30
31	// Get current version before migration
32	currentVersion, err := goose.GetDBVersion(db)
33	if err != nil {
34		log.Debug().Err(err).Msg("could not get current db version (may be new database)")
35		currentVersion = 0
36	}
37
38	// Run migrations
39	if migrateErr := goose.Up(db, "migrations"); migrateErr != nil {
40		return fmt.Errorf("failed to run migrations: %w", migrateErr)
41	}
42
43	// Get new version after migration
44	newVersion, err := goose.GetDBVersion(db)
45	if err != nil {
46		return fmt.Errorf("failed to get db version after migration: %w", err)
47	}
48
49	if newVersion > currentVersion {
50		log.Info().
51			Int64("from_version", currentVersion).
52			Int64("to_version", newVersion).
53			Msg("database migrations applied")
54	} else {
55		log.Debug().Int64("version", newVersion).Msg("database schema up to date")
56	}
57
58	return nil
59}
60
61// GetMigrationStatus returns the current migration version.
62func GetMigrationStatus(db *sql.DB) (int64, error) {
63	goose.SetBaseFS(embedMigrations)
64	if err := goose.SetDialect("sqlite3"); err != nil {
65		return 0, err
66	}
67	return goose.GetDBVersion(db)
68}