Delete add Add admin's tenant & user when initializing Backend

fix linter
free5gc · Jul 13, 2023 · e7d2ed3 · e7d2ed3
1 parent eb081ac
commit e7d2ed3
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 25 deletions.
diff --git a/backend/WebUI/api_webui.go b/backend/WebUI/api_webui.go
@@ -617,6 +617,7 @@ func PostTenant(c *gin.Context) {
 		return
 	}
 
+	// fmt.Println("in Post Tenant")
 	if tenantData.TenantId == "" {
 		tenantData.TenantId = uuid.Must(uuid.NewRandom()).String()
 	}
@@ -1674,3 +1675,53 @@ func GetUEPDUSessionInfo(c *gin.Context) {
 		})
 	}
 }
+
+// Create Admin's Tenant & Account
+func SetAdmin() {
+	err := mongoapi.RestfulAPIDeleteOne("tenantData", bson.M{"tenantName": "admin"})
+	if err != nil {
+		logger.InitLog.Errorf("RestfulAPIDeleteOne err: %+v", err)
+	}
+	err = mongoapi.RestfulAPIDeleteOne("userData", bson.M{"email": "admin"})
+	if err != nil {
+		logger.InitLog.Errorf("RestfulAPIDeleteOne err: %+v", err)
+	}
+
+	// Create Admin tenant
+	logger.InitLog.Infoln("Create tenant: admin")
+
+	adminTenantData := bson.M{
+		"tenantId":   uuid.Must(uuid.NewRandom()).String(),
+		"tenantName": "admin",
+	}
+
+	_, err = mongoapi.RestfulAPIPutOne("tenantData", bson.M{"tenantName": "admin"}, adminTenantData)
+	if err != nil {
+		logger.InitLog.Errorf("RestfulAPIPutOne err: %+v", err)
+	}
+
+	AmdinTenant, err := mongoapi.RestfulAPIGetOne("tenantData", bson.M{"tenantName": "admin"})
+	if err != nil {
+		logger.InitLog.Errorf("RestfulAPIGetOne err: %+v", err)
+	}
+
+	// Create Admin user
+	logger.InitLog.Infoln("Create user: admin")
+
+	hash, err := bcrypt.GenerateFromPassword([]byte("free5gc"), 12)
+	if err != nil {
+		logger.InitLog.Errorf("GenerateFromPassword err: %+v", err)
+	}
+
+	adminUserData := bson.M{
+		"userId":            uuid.Must(uuid.NewRandom()).String(),
+		"tenantId":          AmdinTenant["tenantId"],
+		"email":             "admin",
+		"encryptedPassword": string(hash),
+	}
+
+	_, err = mongoapi.RestfulAPIPutOne("userData", bson.M{"email": "admin"}, adminUserData)
+	if err != nil {
+		logger.InitLog.Errorf("RestfulAPIPutOne err: %+v", err)
+	}
+}
diff --git a/backend/webui_service/webui_init.go b/backend/webui_service/webui_init.go
@@ -6,8 +6,6 @@ import (
 
 	"github.com/gin-contrib/cors"
 	"github.com/sirupsen/logrus"
-	"go.mongodb.org/mongo-driver/bson"
-	"golang.org/x/crypto/bcrypt"
 
 	"github.com/free5gc/util/mongoapi"
 	"github.com/free5gc/webconsole/backend/WebUI"
@@ -79,32 +77,10 @@ func (a *WebuiApp) Start(tlsKeyLogPath string) {
 		return
 	}
 
-	// Create admin account
-	filter := bson.M{"email": "admin"}
-	hash, err := bcrypt.GenerateFromPassword([]byte("free5gc"), 12)
-	if err != nil {
-		logger.InitLog.Errorf("GenerateFromPassword err: %+v", err)
-	}
-
-	data := bson.M{
-		"userId":            "1",
-		"tenantId":          "1",
-		"email":             "admin",
-		"encryptedPassword": string(hash),
-	}
-
-	existed, err := mongoapi.RestfulAPIPutOne("userData", filter, data)
-	if err != nil {
-		logger.InitLog.Errorf("RestfulAPIPutOne err: %+v", err)
-	}
-
-	if existed {
-		logger.InitLog.Infof("Admin existed.")
-	}
-
 	logger.InitLog.Infoln("Server started")
 
 	router := WebUI.NewRouter()
+	WebUI.SetAdmin()
 
 	router.Use(cors.New(cors.Config{
 		AllowMethods: []string{"GET", "POST", "OPTIONS", "PUT", "PATCH", "DELETE"},