fix(Staging env): Fix staging env

flagship-io · Aug 22, 2024 · ae39009 · ae39009
1 parent 6c5d674
commit ae39009
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 6 deletions.
diff --git a/utils/const.go b/utils/const.go
@@ -1,42 +1,59 @@
 package utils
 
 import (
+	"fmt"
 	"log"
 	"os"
 )
 
 func GetFeatureExperimentationHost() string {
-	if os.Getenv("FS_STAGING") == "true" {
+	if os.Getenv("ABT_STAGING") == "true" {
 		return "https://staging-api.flagship.io"
 	}
 
 	return "https://api.flagship.io"
 }
 
 func GetWebExperimentationHost() string {
-	if os.Getenv("FS_STAGING") == "true" {
+	if os.Getenv("ABT_STAGING") == "true" {
 		return "https://staging-api.abtasty.com/api"
 	}
 
 	return "https://api.abtasty.com/api"
 }
 
 func GetHostFeatureExperimentationAuth() string {
-	if os.Getenv("FS_STAGING") == "true" {
+	if os.Getenv("ABT_STAGING") == "true" {
 		return "https://staging-auth.flagship.io"
 	}
 
 	return "https://auth.flagship.io"
 }
 
 func GetHostWebExperimentationAuth() string {
-	if os.Getenv("FS_STAGING") == "true" {
+	if os.Getenv("ABT_STAGING") == "true" {
 		return "https://staging-api-auth.abtasty.com"
 	}
 
 	return "https://api-auth.abtasty.com"
 }
 
+func GetWebExperimentationBrowserAuth(clientId, clientSecret string) string {
+	if os.Getenv("ABT_STAGING") == "true" {
+		return fmt.Sprintf(`https://staging-auth.abtasty.com/authorize?client_id=%s&client_secret=%s&redirect_uri=http://localhost:8010/auth/callback`, clientId, clientSecret)
+	}
+
+	return fmt.Sprintf(`https://auth.abtasty.com/authorize?client_id=%s&client_secret=%s&redirect_uri=http://localhost:8010/auth/callback`, clientId, clientSecret)
+}
+
+func GetWebExperimentationBrowserAuthSuccess() string {
+	if os.Getenv("ABT_STAGING") == "true" {
+		return "https://staging-auth.abtasty.com/authorization-granted"
+	}
+
+	return "https://auth.abtasty.com/authorization-granted"
+}
+
 func DefaultGlobalCodeWorkingDir() string {
 	wdDir, err := os.Getwd()
 	if err != nil {

diff --git a/utils/http_request/common/token.go b/utils/http_request/common/token.go
@@ -41,7 +41,7 @@ func handleCallback(w http.ResponseWriter, r *http.Request, codeChan chan<- stri
 
 	codeChan <- code
 
-	http.Redirect(w, r, "https://auth.abtasty.com/authorization-granted", http.StatusSeeOther)
+	http.Redirect(w, r, utils.GetWebExperimentationBrowserAuthSuccess(), http.StatusSeeOther)
 
 	go func() {
 		time.Sleep(5 * time.Second)
@@ -112,7 +112,7 @@ func InitiateBrowserAuth(username, clientID, clientSecret string) (models.TokenR
 	}
 
 	codeChan := make(chan string)
-	var url = fmt.Sprintf("https://auth.abtasty.com/authorize?client_id=%s&client_secret=%s&redirect_uri=http://localhost:8010/auth/callback", clientID, clientSecret)
+	var url = utils.GetWebExperimentationBrowserAuth(clientID, clientSecret)
 
 	if err := openLink(url); err != nil {
 		log.Fatalf("Error opening link: %s", err)