2023-04-06 10:36:11 +05:00
|
|
|
package backend
|
|
|
|
|
|
|
|
import (
|
|
|
|
"gorm.io/driver/sqlite"
|
|
|
|
"gorm.io/gorm"
|
2023-04-09 12:23:58 +05:00
|
|
|
"os"
|
2023-04-06 10:36:11 +05:00
|
|
|
)
|
|
|
|
|
|
|
|
func GetDatabaseConnection() (*gorm.DB, error) {
|
2023-04-09 12:23:58 +05:00
|
|
|
dbPath := os.Getenv("PHOENIX_DB_PATH")
|
|
|
|
if dbPath == "" {
|
|
|
|
dbPath = "db.sqlite3"
|
|
|
|
}
|
2023-04-09 12:05:50 +05:00
|
|
|
|
|
|
|
db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
|
2023-04-06 10:36:11 +05:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
// Migrate the schema
|
|
|
|
db.AutoMigrate(&Admin{}, &AccessToken{}, &Group{}, &Link{})
|
|
|
|
|
|
|
|
return db, nil
|
|
|
|
}
|