From efe1913c4f82762efd8ba14159f834a052a06d57 Mon Sep 17 00:00:00 2001
From: shun jiang <shun.jiang@newsbreak.com>
Date: Wed, 5 Mar 2025 21:49:29 -0800
Subject: [PATCH] mspai-config

---
 main.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index a83266665f0..23c110b7377 100644
--- a/main.go
+++ b/main.go
@@ -4,8 +4,10 @@ import (
 	"flag"
 	"math/rand"
 	"net/http"
+	"os"
 	"path/filepath"
 	"runtime"
+	"strings"
 	"time"
 
 	"github.com/prebid/prebid-server/config"
@@ -59,7 +61,17 @@ const infoDirectory = "./static/bidder-info"
 
 func loadConfig(bidderInfos config.BidderInfos) (*config.Configuration, error) {
 	v := viper.New()
-	config.SetupViper(v, configFileName, bidderInfos)
+	mspConfigFileName := configFileName
+	envVar := strings.ToLower(os.Getenv("MSP_RUN_MODE"))
+
+	switch envVar {
+	case "mspai_staging":
+		mspConfigFileName = "./msp_config/mspai-stage"
+	case "mspai_prod":
+		mspConfigFileName = "./msp_config/mspai-prod"
+	}
+
+	config.SetupViper(v, mspConfigFileName, bidderInfos)
 	return config.New(v, bidderInfos, openrtb_ext.NormalizeBidderName)
 }