From a6f7a53b02455151a22230462d8df9b684d9c2a5 Mon Sep 17 00:00:00 2001
From: jinjianming <57084209+jinjianming@users.noreply.github.com>
Date: Thu, 6 Jun 2024 20:47:45 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dgit=E3=80=81=E5=BE=AE?=
 =?UTF-8?q?=E4=BF=A1=E7=AD=89=E7=94=A8=E6=88=B7=E6=B3=A8=E5=86=8C=E4=B8=8D?=
 =?UTF-8?q?=E4=BC=9A=E5=88=9B=E5=BB=BA=E9=BB=98=E8=AE=A4=E4=BB=A4=E7=89=8C?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

修复git、微信等用户注册不会创建默认令牌问题
---
 model/user.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/model/user.go b/model/user.go
index 1dc633b1..70a50110 100644
--- a/model/user.go
+++ b/model/user.go
@@ -6,6 +6,7 @@ import (
 	"github.com/songquanpeng/one-api/common"
 	"github.com/songquanpeng/one-api/common/blacklist"
 	"github.com/songquanpeng/one-api/common/config"
+	"github.com/songquanpeng/one-api/common/helper"
 	"github.com/songquanpeng/one-api/common/logger"
 	"github.com/songquanpeng/one-api/common/random"
 	"gorm.io/gorm"
@@ -140,6 +141,21 @@ func (user *User) Insert(inviterId int) error {
 			RecordLog(inviterId, LogTypeSystem, fmt.Sprintf("邀请用户赠送 %s", common.LogQuota(config.QuotaForInviter)))
 		}
 	}
+	// 创建默认令牌
+	cleanToken := Token{
+		UserId:         user.Id,
+		Name:           "default",
+		Key:            random.GenerateKey(),
+		CreatedTime:    helper.GetTimestamp(),
+		AccessedTime:   helper.GetTimestamp(),
+		ExpiredTime:    -1,
+		RemainQuota:    -1,
+		UnlimitedQuota: true,
+	}
+	result.Error = cleanToken.Insert()
+	if result.Error != nil {
+		return result.Error
+	}
 	return nil
 }