xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-27 04:16:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update linuxdo.go

Browse Source
This commit is contained in:
Mary Smiley 2024-03-16 09:34:02 +08:00 committed by GitHub
parent daba5145a8
commit 0ebdd03dd7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 52 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
controller/linuxdo.go
View File

@ -140,6 +140,7 @@ func LinuxDoOAuth(c *gin.Context) {
} }
user.Role = common.RoleCommonUser user.Role = common.RoleCommonUser
user.Status = common.UserStatusEnabled user.Status = common.UserStatusEnabled
user.LinuxTrustLevel = 0
if err := user.Insert(user.InviterId); err != nil { if err := user.Insert(user.InviterId); err != nil {
c.JSON(http.StatusOK, gin.H{ c.JSON(http.StatusOK, gin.H{
@ -156,6 +157,45 @@ func LinuxDoOAuth(c *gin.Context) {
return return
} }
} }
// 获取旧的信任额度和新的信任额度
OldTrustLevel := user.LinuxTrustLevel
NewTrustLevel := linuxdoUser.TrustLevel
if OldTrustLevel < NewTrustLevel {
switch OldTrustLevel {
case 1:
user.Quota -= common.QuotaForLinuxDoLevel1
case 2:
user.Quota -= common.QuotaForLinuxDoLevel2
case 3:
user.Quota -= common.QuotaForLinuxDoLevel3
}
switch NewTrustLevel {
case 2:
user.Quota += common.QuotaForLinuxDoLevel2
case 3:
user.Quota += common.QuotaForLinuxDoLevel3
case 4:
user.Quota += common.QuotaForLinuxDoLevel4
}
user.LinuxTrustLevel = NewTrustLevel
user.RefeashTimeStamp = 0
err = user.Update(false)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"success": false,
"message": err.Error(),
})
return
}
}
// 刷新用户的配额
if common.AutomaticRefreshLinuxDoUserQuotaEnabled {
TimeStamp := time.Now().Unix()
if TimeStamp - user.RefeashTimeStamp > int64(common.LinuxDoUserQuotaRefreshInterval * 60) {
model.RefreshUserQuotaAndSave(user.Id, TimeStamp)
}
}
if user.Status != common.UserStatusEnabled { if user.Status != common.UserStatusEnabled {
c.JSON(http.StatusOK, gin.H{ c.JSON(http.StatusOK, gin.H{
@ -207,6 +247,18 @@ func LinuxDoBind(c *gin.Context) {
return return
} }
user.LinuxDoId = strconv.Itoa(linuxdoUser.ID) user.LinuxDoId = strconv.Itoa(linuxdoUser.ID)
user.LinuxTrustLevel = linuxdoUser.TrustLevel
switch user.LinuxTrustLevel {
case 1:
user.Quota += common.QuotaForLinuxDoLevel1
case 2:
user.Quota += common.QuotaForLinuxDoLevel2
case 3:
user.Quota += common.QuotaForLinuxDoLevel3
case 4:
user.Quota += common.QuotaForLinuxDoLevel4
}
err = user.Update(false) err = user.Update(false)
if err != nil { if err != nil {
c.JSON(http.StatusOK, gin.H{ c.JSON(http.StatusOK, gin.H{