package database

import (
    "go-rest-api/internal/models"
    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
)

func InitDB(dsn string) (*gorm.DB, error) {
    db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
    if err != nil {
        return nil, err
    }
    
    // Auto migrate models
    err = db.AutoMigrate(
        &models.User{},
        // Добавьте другие модели здесь по мере расширения
    )
    if err != nil {
        return nil, err
    }
    
    return db, nil
}