From a24bd3700bc5a92630b60d0d5fd0658522047077 Mon Sep 17 00:00:00 2001
From: Zexi Li <zexi.li@icloud.com>
Date: Thu, 16 Feb 2023 14:28:59 +0800
Subject: [PATCH] feat(mcclient): allow set default authenticate source

---
 pkg/mcclient/auth/auth.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/pkg/mcclient/auth/auth.go b/pkg/mcclient/auth/auth.go
index b0f300cd73..9a37c0978a 100644
--- a/pkg/mcclient/auth/auth.go
+++ b/pkg/mcclient/auth/auth.go
@@ -181,12 +181,24 @@ func (a *authManager) verify(ctx context.Context, token string) (mcclient.TokenC
 	return cred, nil
 }
 
+var (
+	defaultAuthSource = mcclient.AuthSourceSrv
+)
+
+func SetDefaultAuthSource(src string) {
+	defaultAuthSource = src
+}
+
+func GetDefaultAuthSource() string {
+	return defaultAuthSource
+}
+
 func (a *authManager) authAdmin() error {
 	var token mcclient.TokenCredential
 	var err error
 	token, err = a.client.AuthenticateWithSource(
 		a.info.Username, a.info.Passwd, a.info.Domain,
-		a.info.Project, a.info.ProjectDomain, mcclient.AuthSourceSrv)
+		a.info.Project, a.info.ProjectDomain, GetDefaultAuthSource())
 	if err != nil {
 		log.Errorf("Admin auth failed: %s", err)
 		return err