d13d81baba
- Changed the structure of usableGroups in GetUserGroups to store additional information (ratio and description) for each group. - Introduced a new renderRatio function to visually represent group ratios with color coding. - Updated the Playground and EditToken components to utilize the new group structure and rendering options. - Enhanced the renderGroupOption function for better UI representation of group options. - Fixed minor comments and improved code readability.
43 lines
1004 B
Go
43 lines
1004 B
Go
package controller
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"net/http"
|
|
"one-api/model"
|
|
"one-api/setting"
|
|
)
|
|
|
|
func GetGroups(c *gin.Context) {
|
|
groupNames := make([]string, 0)
|
|
for groupName, _ := range setting.GetGroupRatioCopy() {
|
|
groupNames = append(groupNames, groupName)
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"success": true,
|
|
"message": "",
|
|
"data": groupNames,
|
|
})
|
|
}
|
|
|
|
func GetUserGroups(c *gin.Context) {
|
|
usableGroups := make(map[string]map[string]interface{})
|
|
userGroup := ""
|
|
userId := c.GetInt("id")
|
|
userGroup, _ = model.GetUserGroup(userId, false)
|
|
for groupName, ratio := range setting.GetGroupRatioCopy() {
|
|
// UserUsableGroups contains the groups that the user can use
|
|
userUsableGroups := setting.GetUserUsableGroups(userGroup)
|
|
if desc, ok := userUsableGroups[groupName]; ok {
|
|
usableGroups[groupName] = map[string]interface{}{
|
|
"ratio": ratio,
|
|
"desc": desc,
|
|
}
|
|
}
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"success": true,
|
|
"message": "",
|
|
"data": usableGroups,
|
|
})
|
|
}
|