package database

import (
	"log"
	"tonav-go/models"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

var DB *gorm.DB

func InitDB(dbPath string) {
	var err error
	DB, err = gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
	if err != nil {
		log.Fatalf("Failed to connect database: %v", err)
	}

	// 自动迁移
	err = DB.AutoMigrate(&models.Category{}, &models.Service{}, &models.User{}, &models.Setting{})
	if err != nil {
		log.Fatalf("Failed to migrate database: %v", err)
	}
}