From ccc98b429840a4a5a18cab9962616f7344644bd2 Mon Sep 17 00:00:00 2001
From: Akulij <aakulij@gmail.com>
Date: Thu, 20 Mar 2025 03:05:18 +0700
Subject: [PATCH] move out db migtration to separate function

---
 cmd/app/main.go | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/cmd/app/main.go b/cmd/app/main.go
index 7d85ab3..86a764b 100644
--- a/cmd/app/main.go
+++ b/cmd/app/main.go
@@ -47,19 +47,32 @@ type BotController struct {
     updates tgbotapi.UpdatesChannel
 }
 
+func DBMigrate() (*gorm.DB, error) {
+    db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
+    if err != nil {
+        return db, err
+    }
+
+    db.AutoMigrate(&User{})
+    db.AutoMigrate(&BotContent{})
+
+    return db, err
+}
+
 func GetBotController() BotController {
     cfg := config.GetConfig()
     fmt.Printf("Token value: '%v'\n", cfg.BotToken)
     fmt.Printf("Admin password: '%v'\n", cfg.AdminPass)
+
     bot, err := tgbotapi.NewBotAPI(cfg.BotToken)
     if err != nil {
          log.Panic(err)
     }
 
-    db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
-    db.AutoMigrate(&User{})
-    db.AutoMigrate(&BotContent{})
-
+    db, err := DBMigrate()
+    if err != nil {
+         log.Panic(err)
+    }
 
     bot.Debug = true