chore: add more log types

883901 · Jun 10, 2023 · d29c273 · d29c273
1 parent 74f508e
commit d29c273
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 2 deletions.
diff --git a/controller/user.go b/controller/user.go
@@ -2,6 +2,7 @@ package controller
 
 import (
 	"encoding/json"
+	"fmt"
 	"github.com/gin-contrib/sessions"
 	"github.com/gin-gonic/gin"
 	"net/http"
@@ -351,6 +352,9 @@ func UpdateUser(c *gin.Context) {
 		})
 		return
 	}
+	if originUser.Quota != updatedUser.Quota {
+		model.RecordLog(originUser.Id, model.LogTypeManage, fmt.Sprintf("管理员将用户额度从 %d 点修改为 %d 点", originUser.Quota, updatedUser.Quota))
+	}
 	c.JSON(http.StatusOK, gin.H{
 		"success": true,
 		"message": "",

diff --git a/model/log.go b/model/log.go
@@ -17,6 +17,8 @@ const (
 	LogTypeUnknown = iota
 	LogTypeTopup
 	LogTypeConsume
+	LogTypeManage
+	LogTypeSystem
 )
 
 func RecordLog(userId int, logType int, content string) {

diff --git a/model/user.go b/model/user.go
@@ -2,6 +2,7 @@ package model
 
 import (
 	"errors"
+	"fmt"
 	"gorm.io/gorm"
 	"one-api/common"
 	"strings"
@@ -73,8 +74,14 @@ func (user *User) Insert() error {
 	}
 	user.Quota = common.QuotaForNewUser
 	user.AccessToken = common.GetUUID()
-	err = DB.Create(user).Error
-	return err
+	result := DB.Create(user)
+	if result.Error != nil {
+		return result.Error
+	}
+	if common.QuotaForNewUser > 0 {
+		RecordLog(user.Id, LogTypeSystem, fmt.Sprintf("新用户注册赠送 %d 点额度", common.QuotaForNewUser))
+	}
+	return nil
 }
 
 func (user *User) Update(updatePassword bool) error {

diff --git a/web/src/components/LogsTable.js b/web/src/components/LogsTable.js
@@ -18,6 +18,10 @@ function renderType(type) {
       return <Label basic color='green'> 充值 </Label>;
     case 2:
       return <Label basic color='olive'> 消费 </Label>;
+    case 3:
+      return <Label basic color='orange'> 管理 </Label>;
+    case 4:
+      return <Label basic color='red'> 系统 </Label>;
     default:
       return <Label basic color='black'> 未知 </Label>;
   }