first try

This commit is contained in:
Sotirios Pupakis
2025-09-05 17:44:49 +02:00
parent f0aadeb2c7
commit 72731aeb10
10 changed files with 360 additions and 0 deletions

View File

@@ -0,0 +1,55 @@
package initializers
import (
"log"
"os"
"gitea.tbdevent.eu/TBD/reforger_crawler_main/models"
"gopkg.in/yaml.v3"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
)
var DB *gorm.DB
var PORT string
var IP string
var SECRET string
var DB_NAME string
func ConnectToDB() {
db, err := gorm.Open(sqlite.Open(DB_NAME), &gorm.Config{})
if err != nil {
log.Fatal("Failed to connect to database")
}
DB = db
DB.AutoMigrate(&models.Addon{}, &models.AddonFile{})
}
type Configuration struct {
Port string `yaml:"port"`
IP string `yaml:"ip"`
Secret string `yaml:"secret"`
DB string `yaml:"db"`
}
func Load() {
file, err := os.ReadFile("config.yaml")
if err != nil {
log.Fatal("Failed to open config file")
}
configuration := Configuration{
DB: "register.db",
}
err = yaml.Unmarshal(file, &configuration)
if err != nil {
log.Fatal("Failed to read yaml file")
}
PORT = configuration.Port
IP = configuration.IP
SECRET = configuration.Secret
DB_NAME = configuration.DB
}