From a4ba3f0121856f382ae366a6b9450472f7bd2ee0 Mon Sep 17 00:00:00 2001
From: Marvin Zhang <tikazyq@163.com>
Date: Sat, 16 Nov 2024 11:15:36 +0800
Subject: [PATCH] fix: compatibility issue

---
 core/user/service.go | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/core/user/service.go b/core/user/service.go
index 400a579a..10002a83 100644
--- a/core/user/service.go
+++ b/core/user/service.go
@@ -30,7 +30,7 @@ func (svc *Service) Init() (err error) {
 }
 
 func (svc *Service) init() (err error) {
-	_, err = service.NewModelService[models.User]().GetOne(bson.M{"username": constants.DefaultAdminUsername}, nil)
+	u, err := service.NewModelService[models.User]().GetOne(bson.M{"username": constants.DefaultAdminUsername}, nil)
 	if err != nil {
 		if !errors2.Is(err, mongo.ErrNoDocuments) {
 			return err
@@ -41,7 +41,7 @@ func (svc *Service) init() (err error) {
 	}
 
 	// add user
-	u := models.User{
+	u = &models.User{
 		Username:  constants.DefaultAdminUsername,
 		Password:  utils.EncryptMd5(constants.DefaultAdminPassword),
 		Role:      constants.RoleAdmin,
@@ -49,12 +49,12 @@ func (svc *Service) init() (err error) {
 	}
 	u.SetCreatedAt(time.Now())
 	u.SetUpdatedAt(time.Now())
-	_, err = service.NewModelService[models.User]().InsertOne(u)
+	_, err = service.NewModelService[models.User]().InsertOne(*u)
 	return err
 }
 
 func (svc *Service) initPro() (err error) {
-	_, err = service.NewModelService[models.User]().GetOne(bson.M{
+	u, err := service.NewModelService[models.User]().GetOne(bson.M{
 		"$or": []bson.M{
 			{"username": constants.DefaultAdminUsername},
 			{"root_admin": true},
@@ -65,19 +65,25 @@ func (svc *Service) initPro() (err error) {
 			return err
 		}
 	} else {
-		// exists
+		// exists, compatible with old versions
+		u.RootAdmin = true
+		u.SetUpdatedAt(time.Now())
+		err = service.NewModelService[models.User]().ReplaceById(u.Id, *u)
+		if err != nil {
+			log.Errorf("failed to update user: %v", err)
+		}
 		return
 	}
 
 	// add user
-	u := models.User{
+	u = &models.User{
 		Username:  constants.DefaultAdminUsername,
 		Password:  utils.EncryptMd5(constants.DefaultAdminPassword),
 		RootAdmin: true,
 	}
 	u.SetCreatedAt(time.Now())
 	u.SetUpdatedAt(time.Now())
-	_, err = service.NewModelService[models.User]().InsertOne(u)
+	_, err = service.NewModelService[models.User]().InsertOne(*u)
 	return err
 }