package db

import (
	"log"
	"os"

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

func NewDB() *gorm.DB {
	if _, err := os.Stat("db"); os.IsNotExist(err) {
		errDir := os.MkdirAll("db", 0755)
		if errDir != nil {
			log.Fatalln("Error creating directory:", err)
		}
	}
	var err error
	db, err = gorm.Open(sqlite.Open("./db/cat.db"), &gorm.Config{})
	if err != nil {
		panic("failed to connect database")
	}

}