用户列表功能完成

server/handler_admin_role.go

@@ -2,20 +2,25 @@ package server
 
 import (
 	"encoding/json"
-	"github.com/gin-gonic/gin"
 	"net/http"
 	"openai/types"
+
+	"github.com/gin-gonic/gin"
 )
 
 // GetChatRoleListHandle 获取聊天角色列表
 func (s *Server) GetChatRoleListHandle(c *gin.Context) {
-	sessionId := c.GetHeader(types.TokenName)
-	session := s.ChatSession[sessionId]
-	user, err := GetUser(session.Username)
-	if err != nil {
-		c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Hacker Access!!!"})
-		return
+	var user *types.User
+	username := c.Query("username")
+	if username != "" {
+		u, err := GetUser(username)
+		if err != nil {
+			c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Hacker Access!!!"})
+			return
+		}
+		user = u
 	}
+
 	var rolesOrder = []string{"gpt", "teacher", "translator", "english_trainer", "weekly_report", "girl_friend",
 		"kong_zi", "lu_xun", "steve_jobs", "elon_musk", "red_book", "dou_yin", "programmer",
 		"seller", "good_comment", "psychiatrist", "artist"}
@@ -23,10 +28,11 @@ func (s *Server) GetChatRoleListHandle(c *gin.Context) {
 	var roles = GetChatRoles()
 	for _, k := range rolesOrder {
 		// 确认当前用户是否订阅了当前角色
-		if v, ok := user.ChatRoles[k]; !ok || v != 1 {
-			continue
+		if user != nil {
+			if v, ok := user.ChatRoles[k]; !ok || v != 1 {
+				continue
+			}
 		}
-
 		if v, ok := roles[k]; ok && v.Enable {
 			res = append(res, struct {
 				Key  string `json:"key"`