улучшена безопасность переменных окружения

cmd/bot/main.go

@@ -10,7 +10,11 @@ import (
 )
 
 func main() {
-	if os.Getenv("ENV") != "production" {
+	env := os.Getenv("ENV")
+
+	defer os.Unsetenv("ENV")
+
+	if env != "production" {
 		godotenv.Load()
 	}
 

cmd/server/main.go

@@ -10,7 +10,13 @@ import (
 )
 
 func main() {
-	if os.Getenv("ENV") != "production" {
+	env := os.Getenv("ENV")
+	adr := os.Getenv("SERVER_ADR")
+
+	defer os.Unsetenv("ENV")
+	defer os.Unsetenv("SERVER_ADR")
+
+	if env != "production" {
 		godotenv.Load()
 	}
 
@@ -19,7 +25,7 @@ func main() {
 		log.Fatalf(" ERR %v", err)
 	}
 
-	err = app.Listen(os.Getenv("SERVER_ADR"))
+	err = app.Listen(adr)
 	if err != nil {
 		log.Fatalf(" ERR %v", err)
 	}

internal/bot/app/app.go

@@ -16,7 +16,13 @@ import (
 
 // Инициализирует бота, возвращает структуру *telebot.Bot или одну из возможных ошибок.
 func InitApp() (*telebot.Bot, error) {
-	db, err := db.InitDB()
+	token := os.Getenv("BOT_TOKEN")
+	dsn := os.Getenv("DB_DSN")
+
+	defer os.Unsetenv("BOT_TOKEN")
+	defer os.Unsetenv("DB_DSN")
+
+	db, err := db.InitDB(dsn)
 	if err != nil {
 		return nil, err
 	}
@@ -30,7 +36,7 @@ func InitApp() (*telebot.Bot, error) {
 	}
 
 	bot, err := telebot.NewBot(telebot.Settings{
-		Token: os.Getenv("BOT_TOKEN"),
+		Token: token,
 		Poller: &telebot.LongPoller{
 			Limit:   50,
 			Timeout: time.Second * 15,

internal/db/db.go

@@ -1,7 +1,6 @@
 package db
 
 import (
-	"os"
 	"time"
 
 	"gorm.io/driver/postgres"
@@ -33,8 +32,8 @@ type Shorties struct {
 }
 
 // Инициализирует БД, возвращает структуру *DB или одну из возможных ошибок.
-func InitDB() (*DB, error) {
-	db, err := gorm.Open(postgres.Open(os.Getenv("DB_DSN")), &gorm.Config{
+func InitDB(dsn string) (*DB, error) {
+	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
 		Logger: logger.Default.LogMode(logger.Silent),
 	})
 	if err != nil {

internal/server/app/app.go

@@ -1,6 +1,7 @@
 package app
 
 import (
+	"os"
 	"time"
 
 	"github.com/gofiber/fiber/v2"
@@ -12,7 +13,11 @@ import (
 
 // Инициализирует сервер, возвращает структуру *fiber.App или одну из возможных ошибок.
 func InitApp() (*fiber.App, error) {
-	db, err := db.InitDB()
+	dsn := os.Getenv("DB_DSN")
+
+	defer os.Unsetenv("DB_DSN")
+
+	db, err := db.InitDB(dsn)
 	if err != nil {
 		return nil, err
 	}