package configs

import (
	"log"
	"os"

	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

func init() {
	// 設置日誌顯示文件名和行號
	log.SetFlags(log.Lshortfile | log.LstdFlags)

	// 原生golang 創建 data 目錄不存在則創建
	if _, err := os.Stat("data"); os.IsNotExist(err) {
		os.Mkdir("data", os.ModePerm)
	}
}

// ORMDB 使用 GORM
func ORMDB() (db *gorm.DB) {
	db, err := gorm.Open(sqlite.Open("data/sqlite3.db"), &gorm.Config{})
	if err != nil {
		log.Println(err)
	}
	return db
}