From bf3ca674ae5f80076b2ccd7a087f653a7c31d443 Mon Sep 17 00:00:00 2001
From: hys <hysyeah@gmail.com>
Date: Wed, 6 Nov 2024 13:38:32 +0800
Subject: [PATCH] fix: cloud url

---
 pkg/controller/cloud.go | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/pkg/controller/cloud.go b/pkg/controller/cloud.go
index defe7aa..776b15a 100644
--- a/pkg/controller/cloud.go
+++ b/pkg/controller/cloud.go
@@ -5,6 +5,8 @@ import (
 	"encoding/json"
 	"fmt"
 	"net/http"
+	"os"
+	"strings"
 	"time"
 
 	"github.com/emicklei/go-restful/v3"
@@ -151,7 +153,12 @@ func getAccountFromSettings(admin string) (userid, token string, err error) {
 
 func GetAwsAccountFromCloud(ctx context.Context, client dynamic.Interface, bucket string) (*AWSAccount, error) {
 	// cloudUrl := "https://cloud-dev-api.bttcdn.com/v1/resource/stsToken"
-	cloudUrl := "https://cloud-api.bttcdn.com/v1/resource/stsToken/setup"
+	cloudHost := "https://cloud-api.bttcdn.com"
+	cloudHostFromEnv := os.Getenv("OLARES_SPACE_URL")
+	if cloudHostFromEnv != "" {
+		cloudHost = cloudHostFromEnv
+	}
+	cloudURL := fmt.Sprintf("%s/v1/resource/stsToken/setup", strings.TrimSuffix(cloudHost, "/"))
 
 	clusterId, ak, sk, st, err := getClusterId(ctx, client)
 	if err != nil {
@@ -175,10 +182,10 @@ func GetAwsAccountFromCloud(ctx context.Context, client dynamic.Interface, bucke
 			"durationSeconds": fmt.Sprintf("%.0f", duration.Seconds()),
 		}).
 		SetResult(&AWSAccountResponse{}).
-		Post(cloudUrl)
+		Post(cloudURL)
 
 	if err != nil {
-		klog.Error("fetch data from cloud error, ", err, ", ", cloudUrl)
+		klog.Error("fetch data from cloud error, ", err, ", ", cloudURL)
 		return nil, err
 	}