acc-server-manager/local/utl/db/db.go

package db

import (
	"acc-server-manager/local/migrations"
	"acc-server-manager/local/model"
	"acc-server-manager/local/utl/logging"
	"os"

	"go.uber.org/dig"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

func Start(di *dig.Container) {
	dbName := os.Getenv("DB_NAME")
	if dbName == "" {
		dbName = "acc.db"
	}

	db, err := gorm.Open(sqlite.Open(dbName), &gorm.Config{})
	if err != nil {
		logging.Panic("failed to connect database")
	}
	err = di.Provide(func() *gorm.DB {
		return db
	})
	if err != nil {
		logging.Panic("failed to bind database")
	}
	Migrate(db)
}

func Migrate(db *gorm.DB) {
	logging.Info("Migrating database")

	err := db.AutoMigrate(
		&model.ServiceControlModel{},
		&model.Config{},
		&model.Track{},
		&model.CarModel{},
		&model.CupCategory{},
		&model.DriverCategory{},
		&model.SessionType{},
		&model.StateHistory{},
		&model.SteamCredentials{},
		&model.Server{},
		&model.User{},
		&model.Role{},
		&model.Permission{},
	)

	if err != nil {
		logging.Error("GORM AutoMigrate failed: %v", err)
	}

	db.FirstOrCreate(&model.ServiceControlModel{ServiceControl: "Works"})

	Seed(db)
}

func runMigrations(db *gorm.DB) {
	logging.Info("Running custom database migrations...")

	if err := migrations.RunPasswordSecurityMigration(db); err != nil {
		logging.Error("Failed to run password security migration: %v", err)
	}

	logging.Info("Custom database migrations completed")
}

func Seed(db *gorm.DB) error {
	if err := seedTracks(db); err != nil {
		return err
	}
	if err := seedCarModels(db); err != nil {
		return err
	}
	if err := seedDriverCategories(db); err != nil {
		return err
	}
	if err := seedCupCategories(db); err != nil {
		return err
	}
	if err := seedSessionTypes(db); err != nil {
		return err
	}
	return nil
}

func seedTracks(db *gorm.DB) error {
	tracks := []model.Track{
		{Name: "monza", UniquePitBoxes: 29, PrivateServerSlots: 60},
		{Name: "zolder", UniquePitBoxes: 34, PrivateServerSlots: 50},
		{Name: "brands_hatch", UniquePitBoxes: 32, PrivateServerSlots: 50},
		{Name: "silverstone", UniquePitBoxes: 36, PrivateServerSlots: 60},
		{Name: "paul_ricard", UniquePitBoxes: 33, PrivateServerSlots: 80},
		{Name: "misano", UniquePitBoxes: 30, PrivateServerSlots: 50},
		{Name: "spa", UniquePitBoxes: 82, PrivateServerSlots: 82},
		{Name: "nurburgring", UniquePitBoxes: 30, PrivateServerSlots: 50},
		{Name: "barcelona", UniquePitBoxes: 29, PrivateServerSlots: 50},
		{Name: "hungaroring", UniquePitBoxes: 27, PrivateServerSlots: 50},
		{Name: "zandvoort", UniquePitBoxes: 25, PrivateServerSlots: 50},
		{Name: "kyalami", UniquePitBoxes: 40, PrivateServerSlots: 50},
		{Name: "mount_panorama", UniquePitBoxes: 36, PrivateServerSlots: 50},
		{Name: "suzuka", UniquePitBoxes: 51, PrivateServerSlots: 105},
		{Name: "laguna_seca", UniquePitBoxes: 30, PrivateServerSlots: 50},
		{Name: "imola", UniquePitBoxes: 30, PrivateServerSlots: 50},
		{Name: "oulton_park", UniquePitBoxes: 28, PrivateServerSlots: 50},
		{Name: "donington", UniquePitBoxes: 37, PrivateServerSlots: 50},
		{Name: "snetterton", UniquePitBoxes: 26, PrivateServerSlots: 50},
		{Name: "cota", UniquePitBoxes: 30, PrivateServerSlots: 70},
		{Name: "indianapolis", UniquePitBoxes: 30, PrivateServerSlots: 60},
		{Name: "watkins_glen", UniquePitBoxes: 30, PrivateServerSlots: 60},
		{Name: "valencia", UniquePitBoxes: 29, PrivateServerSlots: 50},
		{Name: "nurburgring_24h", UniquePitBoxes: 50, PrivateServerSlots: 110},
		{Name: "red_bull_ring", UniquePitBoxes: 50, PrivateServerSlots: 50},
	}

	for _, track := range tracks {
		if err := db.FirstOrCreate(&track).Error; err != nil {
			return err
		}
	}
	return nil
}

func seedCarModels(db *gorm.DB) error {
	carModels := []model.CarModel{
		{Value: 0, CarModel: "Porsche 991 GT3 R"},
		{Value: 1, CarModel: "Mercedes-AMG GT3"},
	}

	for _, cm := range carModels {
		if err := db.FirstOrCreate(&cm).Error; err != nil {
			return err
		}
	}
	return nil
}

func seedDriverCategories(db *gorm.DB) error {
	categories := []model.DriverCategory{
		{Value: 3, Category: "Platinum"},
		{Value: 2, Category: "Gold"},
		{Value: 1, Category: "Silver"},
		{Value: 0, Category: "Bronze"},
	}

	for _, cat := range categories {
		if err := db.FirstOrCreate(&cat).Error; err != nil {
			return err
		}
	}
	return nil
}

func seedCupCategories(db *gorm.DB) error {
	categories := []model.CupCategory{
		{Value: 0, Category: "Overall"},
		{Value: 1, Category: "ProAm"},
		{Value: 2, Category: "Am"},
		{Value: 3, Category: "Silver"},
		{Value: 4, Category: "National"},
	}

	for _, cat := range categories {
		if err := db.FirstOrCreate(&cat).Error; err != nil {
			return err
		}
	}
	return nil
}

func seedSessionTypes(db *gorm.DB) error {
	sessionTypes := []model.SessionType{
		{Value: 0, SessionType: "Practice"},
		{Value: 4, SessionType: "Qualifying"},
		{Value: 10, SessionType: "Race"},
	}

	for _, st := range sessionTypes {
		if err := db.FirstOrCreate(&st).Error; err != nil {
			return err
		}
	}
	return nil
}