From 02944189500086f575f8c0ff96ae4fb119c2035f Mon Sep 17 00:00:00 2001
From: Mohammad Mahdi <mmm.6831.allah@gmail.com>
Date: Sun, 18 May 2025 12:46:44 +0330
Subject: [PATCH] Create data dir and move DB file into it for better
 organization

---
 internal/database/database.go | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/internal/database/database.go b/internal/database/database.go
index 4fc403b..c67abce 100644
--- a/internal/database/database.go
+++ b/internal/database/database.go
@@ -2,6 +2,7 @@ package database
 
 import (
 	"CatsOfMastodonBotGo/internal/domain"
+	"os"
 
 	"gorm.io/driver/sqlite"
 	"gorm.io/gorm"
@@ -10,7 +11,14 @@ import (
 var Gorm *gorm.DB
 
 func Connect() (*gorm.DB, error) {
-	db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
+	_, err := os.ReadDir("data")
+	if err != nil {
+		err := os.Mkdir("data", 0755)
+		if err != nil {
+			return nil, err
+		}
+	}
+	db, err := gorm.Open(sqlite.Open("data/caom.db"), &gorm.Config{})
 	if err != nil {
 		return nil, err
 	}
@@ -22,10 +30,10 @@ func Connect() (*gorm.DB, error) {
 }
 
 // IDK if this is how it works or not, leave it as is for now
-func Init () {
+func Init() {
 	var err error
 	Gorm, err = Connect()
 	if err != nil {
 		panic(err)
 	}
-}
\ No newline at end of file
+}