gonic/server/db/migrations.go

package db

import (
	"fmt"

	"github.com/jinzhu/gorm"
	"gopkg.in/gormigrate.v1"
)

// $ date '+%Y%m%d%H%M'

func migrateInitSchema() gormigrate.Migration {
	return gormigrate.Migration{
		ID: "202002192100",
		Migrate: func(tx *gorm.DB) error {
			return tx.AutoMigrate(
				Artist{},
				Track{},
				User{},
				Setting{},
				Play{},
				Album{},
				Playlist{},
				PlayQueue{},
			).
				Error
		},
	}
}

func migrateCreateInitUser() gormigrate.Migration {
	return gormigrate.Migration{
		ID: "202002192019",
		Migrate: func(tx *gorm.DB) error {
			const (
				initUsername = "admin"
				initPassword = "admin"
			)
			err := tx.
				Where("name=?", initUsername).
				First(&User{}).
				Error
			if !gorm.IsRecordNotFoundError(err) {
				return nil
			}

			return tx.Create(&User{
				Name:     initUsername,
				Password: initPassword,
				IsAdmin:  true,
			}).
				Error
		},
	}
}

func migrateMergePlaylist() gormigrate.Migration {
	return gormigrate.Migration{
		ID: "202002192222",
		Migrate: func(tx *gorm.DB) error {
			if !tx.HasTable("playlist_items") {
				return nil
			}

			return tx.Exec(`
				UPDATE playlists
				SET items=( SELECT group_concat(track_id) FROM (
					SELECT track_id
					FROM playlist_items
					WHERE playlist_items.playlist_id=playlists.id
					ORDER BY created_at
				) );
				DROP TABLE playlist_items;`,
			).
				Error
		},
	}
}

func migrateCreateTranscode() gormigrate.Migration {
	return gormigrate.Migration{
		ID: "202003111222",
		Migrate: func(tx *gorm.DB) error {
			return tx.AutoMigrate(
				TranscodePreference{},
			).
				Error
		},
	}
}

func migrateAddGenre() gormigrate.Migration {
	return gormigrate.Migration{
		ID: "202003121330",
		Migrate: func(tx *gorm.DB) error {
			return tx.AutoMigrate(
				Genre{},
				Album{},
				Track{},
			).
				Error
		},
	}
}

func migrateUpdateTranscodePrefIDX() gormigrate.Migration {
	return gormigrate.Migration{
		ID: "202003241509",
		Migrate: func(tx *gorm.DB) error {
			var hasIDX int
			tx.
				Select("1").
				Table("sqlite_master").
				Where("type = ?", "index").
				Where("name = ?", "idx_user_id_client").
				Count(&hasIDX)
			if hasIDX == 1 {
				// index already exists
				return nil
			}

			step := tx.Exec(`
				ALTER TABLE transcode_preferences RENAME TO transcode_preferences_orig;
			`)
			if err := step.Error; err != nil {
				return fmt.Errorf("step rename: %w", err)
			}

			step = tx.AutoMigrate(
				TranscodePreference{},
			)
			if err := step.Error; err != nil {
				return fmt.Errorf("step create: %w", err)
			}

			step = tx.Exec(`
				INSERT INTO transcode_preferences (user_id, client, profile)
					SELECT user_id, client, profile
					FROM transcode_preferences_orig;
				DROP TABLE transcode_preferences_orig;
			`)
			if err := step.Error; err != nil {
				return fmt.Errorf("step copy: %w", err)
			}
			return nil
		},
	}
}

func migrateAddAlbumIDX() gormigrate.Migration {
	return gormigrate.Migration{
		ID: "202004302006",
		Migrate: func(tx *gorm.DB) error {
			return tx.AutoMigrate(
				Album{},
			).
				Error
		},
	}
}

func migrateMultiGenre() gormigrate.Migration {
	return gormigrate.Migration{
		ID: "202012151806",
		Migrate: func(tx *gorm.DB) error {
			step := tx.AutoMigrate(
				Track{},
				Album{},
				Genre{},
				TrackGenre{},
				AlbumGenre{},
			)
			var genreCount int
			tx.
				Model(Genre{}).
				Count(&genreCount)
			if genreCount == 0 {
				return nil
			}

			if err := step.Error; err != nil {
				return fmt.Errorf("step auto migrate: %w", err)
			}
			step = tx.Exec(`
				INSERT INTO track_genres (track_id, genre_id)
					SELECT id, tag_genre_id
					FROM tracks
					WHERE tag_genre_id IS NOT NULL;
				UPDATE tracks SET tag_genre_id=NULL;
			`)

			if err := step.Error; err != nil {
				return fmt.Errorf("step migrate track genres: %w", err)
			}
			step = tx.Exec(`
				INSERT INTO album_genres (album_id, genre_id)
					SELECT id, tag_genre_id
					FROM albums
					WHERE tag_genre_id IS NOT NULL;
				UPDATE albums SET tag_genre_id=NULL;
			`)
			if err := step.Error; err != nil {
				return fmt.Errorf("step migrate album genres: %w", err)
			}
			return nil
		},
	}
}