feat: init commit

dao/db.go

@@ -0,0 +1,33 @@
+package dao
+
+import (
+	"os"
+	"path/filepath"
+
+	"github.com/glebarez/sqlite"
+	"github.com/krau/SaveAny-Bot/config"
+	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/model"
+	"gorm.io/gorm"
+)
+
+var db *gorm.DB
+
+func Init() {
+	if err := os.MkdirAll(filepath.Dir(config.Cfg.DB.Path), 755); err != nil {
+		logger.L.Fatal("Failed to create data directory: ", err)
+		os.Exit(1)
+	}
+	var err error
+	db, err = gorm.Open(sqlite.Open(config.Cfg.DB.Path), &gorm.Config{})
+	if err != nil {
+		logger.L.Fatal("Failed to open database: ", err)
+		os.Exit(1)
+	}
+	logger.L.Debug("Database connected")
+	db.AutoMigrate(&model.ReceivedFile{}, &model.User{})
+
+	for _, admin := range config.Cfg.Telegram.Admins {
+		CreateUser(int64(admin))
+	}
+}