Merge remote-tracking branch 'origin/upstream/main'

controller/user.go

@@ -7,13 +7,15 @@ import (
 	"strconv"
 	"time"
 
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
 	"github.com/songquanpeng/one-api/common"
 	"github.com/songquanpeng/one-api/common/config"
 	"github.com/songquanpeng/one-api/common/ctxkey"
+	"github.com/songquanpeng/one-api/common/helper"
+	"github.com/songquanpeng/one-api/common/logger"
 	"github.com/songquanpeng/one-api/common/random"
 	"github.com/songquanpeng/one-api/model"
-	"github.com/gin-contrib/sessions"
-	"github.com/gin-gonic/gin"
 )
 
 type LoginRequest struct {
@@ -115,6 +117,7 @@ func Logout(c *gin.Context) {
 }
 
 func Register(c *gin.Context) {
+	ctx := c.Request.Context()
 	if !config.RegisterEnabled {
 		c.JSON(http.StatusOK, gin.H{
 			"message": "管理员关闭了新用户注册",
@@ -179,6 +182,28 @@ func Register(c *gin.Context) {
 		})
 		return
 	}
+	go func() {
+		err := user.ValidateAndFill()
+		if err != nil {
+			logger.Errorf(ctx, "user.ValidateAndFill failed: %v", err)
+			return
+		}
+		cleanToken := model.Token{
+			UserId:         user.Id,
+			Name:           "default",
+			Key:            random.GenerateKey(),
+			CreatedTime:    helper.GetTimestamp(),
+			AccessedTime:   helper.GetTimestamp(),
+			ExpiredTime:    -1,
+			RemainQuota:    -1,
+			UnlimitedQuota: true,
+		}
+		err = cleanToken.Insert()
+		if err != nil {
+			logger.Errorf(ctx, "cleanToken.Insert failed: %v", err)
+			return
+		}
+	}()
 	c.JSON(http.StatusOK, gin.H{
 		"success": true,
 		"message": "",