package main

import (
	"fmt"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"log/slog"
)

type ArtistProfile struct {
	gorm.Model
	SpotifyID  string `gorm:"unique"`
	Name       string
	Popularity int
	Genres     []Genre `gorm:"many2many:artist_genres;"`
}

type Genre struct {
	gorm.Model
	Name string `gorm:"unique"`
}

func setupTestDatabase(name string) *gorm.DB {
	slog.Info("[GOMUSIC] Setting up new test database in memory")
	// Open a named DB instance so each test has a clean environment
	dbName := fmt.Sprintf("file:%s?mode=memory&cache=shared", name)
	db, err := gorm.Open(sqlite.Open(dbName), &gorm.Config{})
	if err != nil {
		panic("Failed to open database")
	}

	// Run model migrations here to keep DB consistent
	db.AutoMigrate(&ArtistProfile{}, &Genre{})
	return db
}

func setupDatabase() *gorm.DB {
	slog.Info("[GOMUSIC] Setting up database and running auto migrations")
	db, err := gorm.Open(sqlite.Open("gomusic.db"), &gorm.Config{})
	if err != nil {
		panic("Failed to open database")
	}

	// Run model migrations here to keep DB consistent
	db.AutoMigrate(&ArtistProfile{}, &Genre{})
	return db
}