From e12b0416d4d2247ab2db035a65e0f49bb29f6c22 Mon Sep 17 00:00:00 2001
From: Siwoo Jeon <me@migan.co.kr>
Date: Tue, 8 Apr 2025 20:13:01 +0900
Subject: [PATCH] fix: Removing hard coding

---
 configs/config.go  | 2 ++
 databases/Learn.go | 3 ++-
 databases/Text.go  | 3 ++-
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/configs/config.go b/configs/config.go
index 204e71c..445e617 100644
--- a/configs/config.go
+++ b/configs/config.go
@@ -23,6 +23,7 @@ type MuffinConfig struct {
 	Bot         botConfig
 	Train       trainConfig
 	DatabaseURL string
+	DBName      string
 }
 
 func loadConfig() *MuffinConfig {
@@ -57,6 +58,7 @@ func setConfig(config *MuffinConfig) {
 	config.Train.UserID = getValue("TRAIN_USER_ID")
 
 	config.DatabaseURL = getRequiredValue("DATABASE_URL")
+	config.DBName = getRequiredValue("DATABASE_NAME")
 }
 
 var Config *MuffinConfig = loadConfig()
diff --git a/databases/Learn.go b/databases/Learn.go
index 8aeddde..f0d25aa 100644
--- a/databases/Learn.go
+++ b/databases/Learn.go
@@ -3,6 +3,7 @@ package databases
 import (
 	"time"
 
+	"git.wh64.net/muffin/goMuffin/configs"
 	"go.mongodb.org/mongo-driver/v2/bson"
 	"go.mongodb.org/mongo-driver/v2/mongo"
 )
@@ -22,4 +23,4 @@ type Learn struct {
 	CreatedAt time.Time `bson:"created_at"`
 }
 
-var Learns *mongo.Collection = Client.Database("muffin_ai_test").Collection("learn")
+var Learns *mongo.Collection = Client.Database(configs.Config.DBName).Collection("learn")
diff --git a/databases/Text.go b/databases/Text.go
index 295d688..06dc147 100644
--- a/databases/Text.go
+++ b/databases/Text.go
@@ -3,6 +3,7 @@ package databases
 import (
 	"time"
 
+	"git.wh64.net/muffin/goMuffin/configs"
 	"go.mongodb.org/mongo-driver/v2/bson"
 	"go.mongodb.org/mongo-driver/v2/mongo"
 )
@@ -20,4 +21,4 @@ type Text struct {
 	CreatedAt time.Time `bson:"created_at"`
 }
 
-var Texts *mongo.Collection = Client.Database("muffin_ai_test").Collection("text")
+var Texts *mongo.Collection = Client.Database(configs.Config.DBName).Collection("text")