From 6a941748f86f190ea3664d8d522b07e5b6a7811e Mon Sep 17 00:00:00 2001
From: open source <peterwillcn@gmail.com>
Date: Sat, 6 Jul 2024 13:15:17 +0800
Subject: [PATCH] feat: add initial root access token (#1598)

Signed-off-by: xiaobo <peterwillcn@gmail.com>
---
 common/config/config.go | 2 ++
 model/main.go           | 6 +++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/common/config/config.go b/common/config/config.go
index 3f321c87..9b55e413 100644
--- a/common/config/config.go
+++ b/common/config/config.go
@@ -143,6 +143,8 @@ var MetricFailChanSize = env.Int("METRIC_FAIL_CHAN_SIZE", 128)
 
 var InitialRootToken = os.Getenv("INITIAL_ROOT_TOKEN")
 
+var InitialRootAccessToken = os.Getenv("INITIAL_ROOT_ACCESS_TOKEN")
+
 var GeminiVersion = env.String("GEMINI_VERSION", "v1")
 
 
diff --git a/model/main.go b/model/main.go
index 11752404..72e271a0 100644
--- a/model/main.go
+++ b/model/main.go
@@ -30,13 +30,17 @@ func CreateRootAccountIfNeed() error {
 		if err != nil {
 			return err
 		}
+		accessToken := random.GetUUID()
+		if config.InitialRootAccessToken != "" {
+			accessToken = config.InitialRootAccessToken
+		}
 		rootUser := User{
 			Username:    "root",
 			Password:    hashedPassword,
 			Role:        RoleRootUser,
 			Status:      UserStatusEnabled,
 			DisplayName: "Root User",
-			AccessToken: random.GetUUID(),
+			AccessToken: accessToken,
 			Quota:       500000000000000,
 		}
 		DB.Create(&rootUser)