From 523cd4d4c9e9dab4ceb4ee5108623864f67f157c Mon Sep 17 00:00:00 2001 From: RockYang Date: Fri, 28 Apr 2023 16:15:46 +0800 Subject: [PATCH] =?UTF-8?q?=E9=87=8D=E6=96=B0=E6=95=B4=E7=90=86=E5=90=8E?= =?UTF-8?q?=E5=8F=B0=E7=AE=A1=E7=90=86=20API=EF=BC=8C=E6=96=B0=E5=A2=9E?= =?UTF-8?q?=E7=B3=BB=E7=BB=9F=E7=AE=A1=E7=90=86=E9=A1=B5=E9=9D=A2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/handler_admin_apikey.go | 67 ++++ server/handler_admin_config.go | 121 ++++++ server/handler_admin_role.go | 140 +++++++ server/handler_admin_user.go | 235 ++++++++++++ server/handler_chat.go | 36 ++ server/handler_config.go | 588 ------------------------------ server/server.go | 51 ++- types/config.go | 21 +- utils/config.go | 11 +- web/src/views/admin/SysConfig.vue | 41 ++- 10 files changed, 684 insertions(+), 627 deletions(-) create mode 100644 server/handler_admin_apikey.go create mode 100644 server/handler_admin_config.go create mode 100644 server/handler_admin_role.go create mode 100644 server/handler_admin_user.go delete mode 100644 server/handler_config.go diff --git a/server/handler_admin_apikey.go b/server/handler_admin_apikey.go new file mode 100644 index 00000000..762b0f54 --- /dev/null +++ b/server/handler_admin_apikey.go @@ -0,0 +1,67 @@ +package server + +import ( + "encoding/json" + "github.com/gin-gonic/gin" + "net/http" + "openai/types" + "openai/utils" +) + +// AddApiKeyHandle 添加一个 API key +func (s *Server) AddApiKeyHandle(c *gin.Context) { + var data struct { + ApiKey string `json:"api_key"` + } + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + logger.Errorf("Error decode json data: %s", err.Error()) + c.JSON(http.StatusBadRequest, nil) + return + } + if len(data.ApiKey) > 20 { + s.Config.Chat.ApiKeys = append(s.Config.Chat.ApiKeys, data.ApiKey) + } + + // 保存配置文件 + err = utils.SaveConfig(s.Config, s.ConfigPath) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Chat.ApiKeys}) +} + +// RemoveApiKeyHandle 移除一个 API key +func (s *Server) RemoveApiKeyHandle(c *gin.Context) { + var data struct { + ApiKey string `json:"api_key"` + } + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + logger.Errorf("Error decode json data: %s", err.Error()) + c.JSON(http.StatusBadRequest, nil) + return + } + + for i, v := range s.Config.Chat.ApiKeys { + if v == data.ApiKey { + s.Config.Chat.ApiKeys = append(s.Config.Chat.ApiKeys[:i], s.Config.Chat.ApiKeys[i+1:]...) + } + } + + // 保存配置文件 + err = utils.SaveConfig(s.Config, s.ConfigPath) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Chat.ApiKeys}) +} + +// ListApiKeysHandle 获取 API key 列表 +func (s *Server) ListApiKeysHandle(c *gin.Context) { + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Chat.ApiKeys}) +} diff --git a/server/handler_admin_config.go b/server/handler_admin_config.go new file mode 100644 index 00000000..295a80be --- /dev/null +++ b/server/handler_admin_config.go @@ -0,0 +1,121 @@ +package server + +import ( + "encoding/json" + "github.com/gin-gonic/gin" + "net/http" + "openai/types" + "openai/utils" +) + +func (s *Server) TestHandle(c *gin.Context) { + roles := types.GetDefaultChatRole() + for _, v := range roles { + _ = PutChatRole(v) + } + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Data: GetChatRoles()}) + +} + +// ConfigSetHandle set configs +func (s *Server) ConfigSetHandle(c *gin.Context) { + var data map[string]interface{} + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + logger.Errorf("Error decode json data: %s", err.Error()) + c.JSON(http.StatusBadRequest, nil) + return + } + + if model, ok := data["model"]; ok { + s.Config.Chat.Model = model.(string) + } + // Temperature + if temperature, ok := data["temperature"]; ok { + s.Config.Chat.Temperature = temperature.(float32) + } + // max_users + if maxTokens, ok := data["max_tokens"]; ok { + s.Config.Chat.MaxTokens = int(maxTokens.(float64)) + } + // enable Context + if enableContext, ok := data["enable_context"]; ok { + s.Config.Chat.EnableContext = enableContext.(bool) + } + if expireTime, ok := data["chat_context_expire_time"]; ok { + s.Config.Chat.ChatContextExpireTime = int(expireTime.(float64)) + } + // enable auth + if enableAuth, ok := data["enable_auth"]; ok { + s.Config.EnableAuth = enableAuth.(bool) + } + + if debug, ok := data["debug"]; ok { + s.DebugMode = debug.(bool) + } + // 保存配置文件 + err = utils.SaveConfig(s.Config, s.ConfigPath) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config}) +} + +// AddProxyHandle 添加一个代理 +func (s *Server) AddProxyHandle(c *gin.Context) { + var data struct { + Proxy string `json:"proxy"` + } + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + logger.Errorf("Error decode json data: %s", err.Error()) + c.JSON(http.StatusBadRequest, nil) + return + } + + if data.Proxy != "" { + if !utils.ContainsStr(s.Config.ProxyURL, data.Proxy) { + s.Config.ProxyURL = append(s.Config.ProxyURL, data.Proxy) + } + } + + // 保存配置文件 + err = utils.SaveConfig(s.Config, s.ConfigPath) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.ProxyURL}) +} + +// RemoveProxyHandle 删除一个代理 +func (s *Server) RemoveProxyHandle(c *gin.Context) { + var data struct { + Proxy string `json:"proxy"` + } + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + logger.Errorf("Error decode json data: %s", err.Error()) + c.JSON(http.StatusBadRequest, nil) + return + } + + for i, v := range s.Config.ProxyURL { + if v == data.Proxy { + s.Config.ProxyURL = append(s.Config.ProxyURL[:i], s.Config.ProxyURL[i+1:]...) + break + } + } + + // 保存配置文件 + err = utils.SaveConfig(s.Config, s.ConfigPath) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.ProxyURL}) +} diff --git a/server/handler_admin_role.go b/server/handler_admin_role.go new file mode 100644 index 00000000..eaf9ab81 --- /dev/null +++ b/server/handler_admin_role.go @@ -0,0 +1,140 @@ +package server + +import ( + "encoding/json" + "github.com/gin-gonic/gin" + "net/http" + "openai/types" +) + +// 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 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"} + var res = make([]interface{}, 0) + var roles = GetChatRoles() + for _, k := range rolesOrder { + // 确认当前用户是否订阅了当前角色 + 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"` + Name string `json:"name"` + Icon string `json:"icon"` + }{ + Key: v.Key, + Name: v.Name, + Icon: v.Icon, + }) + } + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: res}) +} + +// AddChatRoleHandle 添加一个聊天角色 +func (s *Server) AddChatRoleHandle(c *gin.Context) { + var data types.ChatRole + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + logger.Errorf("Error decode json data: %s", err.Error()) + c.JSON(http.StatusBadRequest, nil) + return + } + + if data.Key == "" || data.Name == "" || data.Icon == "" { + c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid parameters"}) + return + } + + err = PutChatRole(data) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save levelDB"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: data}) +} + +// GetChatRoleHandle 获取指定的角色 +func (s *Server) GetChatRoleHandle(c *gin.Context) { + var data struct { + Key string `json:"key"` + } + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + logger.Errorf("Error decode json data: %s", err.Error()) + c.JSON(http.StatusBadRequest, nil) + return + } + + role, err := GetChatRole(data.Key) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "No role found"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Data: role}) + return +} + +// SetChatRoleHandle 更新某个聊天角色信息，这里只允许更改名称以及启用和禁用角色操作 +func (s *Server) SetChatRoleHandle(c *gin.Context) { + var data map[string]interface{} + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + logger.Errorf("Error decode json data: %s", err.Error()) + c.JSON(http.StatusBadRequest, nil) + return + } + + var key string + if v, ok := data["key"]; !ok { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Must specified the role key"}) + return + } else { + key = v.(string) + } + + role, err := GetChatRole(key) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Role key not exists"}) + return + } + + if v, ok := data["name"]; ok { + role.Name = v.(string) + } + if v, ok := data["hello_msg"]; ok { + role.HelloMsg = v.(string) + } + if v, ok := data["icon"]; ok { + role.Icon = v.(string) + } + if v, ok := data["enable"]; ok { + role.Enable = v.(bool) + } + if v, ok := data["context"]; ok { + bytes, _ := json.Marshal(v) + _ = json.Unmarshal(bytes, &role.Context) + } + + err = PutChatRole(*role) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save levelDB"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: data}) +} diff --git a/server/handler_admin_user.go b/server/handler_admin_user.go new file mode 100644 index 00000000..cbd4c668 --- /dev/null +++ b/server/handler_admin_user.go @@ -0,0 +1,235 @@ +package server + +import ( + "encoding/json" + "fmt" + "github.com/gin-gonic/gin" + "net/http" + "openai/types" + "openai/utils" + "strings" +) + +// AddUserHandle 添加 Username +func (s *Server) AddUserHandle(c *gin.Context) { + var data struct { + Name string `json:"name"` + MaxCalls int `json:"max_calls"` + EnableHistory bool `json:"enable_history"` + Term int `json:"term"` // 有效期 + ChatRoles []string `json:"chat_roles"` // 订阅角色 + } + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid args"}) + return + } + + // 参数处理 + if data.Name == "" || data.MaxCalls < 0 { + c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid args"}) + return + } + + // 检查当前要添加的 Username 是否已经存在 + _, err = GetUser(data.Name) + if err == nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Username " + data.Name + " already exists"}) + return + } + + var chatRoles = make(map[string]int) + if len(data.ChatRoles) > 0 { + if data.ChatRoles[0] == "all" { // 所有的角色 + roles := GetChatRoles() + for key := range roles { + chatRoles[key] = 1 + } + } else { + for _, key := range data.ChatRoles { + chatRoles[key] = 1 + } + } + + } + + user := types.User{ + Name: data.Name, + MaxCalls: data.MaxCalls, + RemainingCalls: data.MaxCalls, + EnableHistory: data.EnableHistory, + Term: data.Term, + ChatRoles: chatRoles, + Status: true} + err = PutUser(user) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: user}) +} + +// BatchAddUserHandle 批量生成 Username +func (s *Server) BatchAddUserHandle(c *gin.Context) { + var data struct { + Number int `json:"number"` + MaxCalls int `json:"max_calls"` + EnableHistory bool `json:"enable_history"` + Term int `json:"term"` + ChatRoles []string `json:"chat_roles"` + } + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil || data.MaxCalls <= 0 { + c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid args"}) + return + } + + var chatRoles = make(map[string]int) + if len(data.ChatRoles) > 0 { + if data.ChatRoles[0] == "all" { // 所有的角色 + roles := GetChatRoles() + for key := range roles { + chatRoles[key] = 1 + } + } else { + for _, key := range data.ChatRoles { + chatRoles[key] = 1 + } + } + + } + + var users = make([]types.User, 0) + for i := 0; i < data.Number; i++ { + name := utils.RandString(12) + _, err := GetUser(name) + for err == nil { + name = utils.RandString(12) + } + user := types.User{ + Name: name, + MaxCalls: data.MaxCalls, + RemainingCalls: data.MaxCalls, + EnableHistory: data.EnableHistory, + Term: data.Term, + ChatRoles: chatRoles, + Status: true} + err = PutUser(user) + if err == nil { + users = append(users, user) + } + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: users}) +} + +func (s *Server) SetUserHandle(c *gin.Context) { + var data map[string]interface{} + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid args"}) + return + } + + var user *types.User + if name, ok := data["name"]; ok { + user, err = GetUser(name.(string)) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "User not found"}) + return + } + } + var maxCalls int + if v, ok := data["max_calls"]; ok { + maxCalls = int(v.(float64)) + } + if maxCalls < 0 { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Invalid args"}) + return + } else if maxCalls > 0 { + user.MaxCalls = maxCalls + user.RemainingCalls += maxCalls - user.MaxCalls + if user.RemainingCalls < 0 { + user.RemainingCalls = 0 + } + } + + if v, ok := data["status"]; ok { + user.Status = v.(bool) + } + if v, ok := data["enable_history"]; ok { + user.EnableHistory = v.(bool) + } + if v, ok := data["remaining_calls"]; ok { + user.RemainingCalls = int(v.(float64)) + } + if v, ok := data["expired_time"]; ok { + user.ExpiredTime = int64(v.(float64)) + } + if v, ok := data["api_key"]; ok { + user.ApiKey = v.(string) + } + if v, ok := data["chat_roles"]; ok { + if roles, ok := v.([]interface{}); ok { + chatRoles := make(map[string]int) + if roles[0] == "all" { + roles := GetChatRoles() + for key := range roles { + chatRoles[key] = 1 + } + } else { + for _, key := range roles { + key := strings.TrimSpace(fmt.Sprintf("%v", key)) + chatRoles[key] = 1 + } + } + user.ChatRoles = chatRoles + } + + } + + err = PutUser(*user) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: user}) +} + +// RemoveUserHandle 删除 Username +func (s *Server) RemoveUserHandle(c *gin.Context) { + var data struct { + Name string `json:"name"` + } + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Invalid args"}) + return + } + + err = RemoveUser(data.Name) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg}) +} + +// GetUserListHandle 获取用户列表 +func (s *Server) GetUserListHandle(c *gin.Context) { + username := c.PostForm("username") + if username != "" { + user, err := GetUser(username) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "User not exists"}) + } else { + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: user}) + } + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: GetUsers()}) +} diff --git a/server/handler_chat.go b/server/handler_chat.go index e84da722..a4b78a1e 100644 --- a/server/handler_chat.go +++ b/server/handler_chat.go @@ -490,3 +490,39 @@ func (s *Server) GetHelloMsgHandle(c *gin.Context) { } c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Data: chatRole.HelloMsg}) } + +// SetImgURLHandle SetImgURL 设置图片地址集合 +func (s *Server) SetImgURLHandle(c *gin.Context) { + var data struct { + WechatCard string `json:"wechat_card"` // 个人微信二维码 + WechatGroup string `json:"wechat_group"` // 微信群聊二维码 + } + err := json.NewDecoder(c.Request.Body).Decode(&data) + if err != nil { + logger.Errorf("Error decode json data: %s", err.Error()) + c.JSON(http.StatusBadRequest, nil) + return + } + + if data.WechatCard != "" { + s.Config.ImgURL.WechatCard = data.WechatCard + } + if data.WechatGroup != "" { + s.Config.ImgURL.WechatGroup = data.WechatGroup + } + + // 保存配置文件 + err = utils.SaveConfig(s.Config, s.ConfigPath) + if err != nil { + c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) + return + } + + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.ImgURL}) + +} + +// GetImgURLHandle 获取图片地址集合 +func (s *Server) GetImgURLHandle(c *gin.Context) { + c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.ImgURL}) +} diff --git a/server/handler_config.go b/server/handler_config.go deleted file mode 100644 index e90fb657..00000000 --- a/server/handler_config.go +++ /dev/null @@ -1,588 +0,0 @@ -package server - -import ( - "encoding/json" - "fmt" - "github.com/gin-gonic/gin" - "net/http" - "openai/types" - "openai/utils" - "strings" -) - -func (s *Server) TestHandle(c *gin.Context) { - roles := types.GetDefaultChatRole() - for _, v := range roles { - _ = PutChatRole(v) - } - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Data: GetChatRoles()}) - -} - -// ConfigSetHandle set configs -func (s *Server) ConfigSetHandle(c *gin.Context) { - var data map[string]interface{} - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - logger.Errorf("Error decode json data: %s", err.Error()) - c.JSON(http.StatusBadRequest, nil) - return - } - - if model, ok := data["model"]; ok { - s.Config.Chat.Model = model.(string) - } - if accessKey, ok := data["access_key"]; ok { - s.Config.AccessKey = accessKey.(string) - } - // Temperature - if temperature, ok := data["temperature"]; ok { - s.Config.Chat.Temperature = temperature.(float32) - } - // max_users - if maxTokens, ok := data["max_tokens"]; ok { - s.Config.Chat.MaxTokens = int(maxTokens.(float64)) - } - // enable Context - if enableContext, ok := data["enable_context"]; ok { - s.Config.Chat.EnableContext = enableContext.(bool) - } - if expireTime, ok := data["chat_context_expire_time"]; ok { - s.Config.Chat.ChatContextExpireTime = int(expireTime.(float64)) - } - // enable auth - if enableAuth, ok := data["enable_auth"]; ok { - s.Config.EnableAuth = enableAuth.(bool) - } - - // 保存配置文件 - err = utils.SaveConfig(s.Config, s.ConfigPath) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config}) -} - -// SetDebugHandle 开启/关闭调试模式 -func (s *Server) SetDebugHandle(c *gin.Context) { - var data struct { - Debug bool `json:"debug"` - } - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid args"}) - return - } - - s.DebugMode = data.Debug - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg}) -} - -// AddUserHandle 添加 Username -func (s *Server) AddUserHandle(c *gin.Context) { - var data struct { - Name string `json:"name"` - MaxCalls int `json:"max_calls"` - EnableHistory bool `json:"enable_history"` - Term int `json:"term"` // 有效期 - ChatRoles []string `json:"chat_roles"` // 订阅角色 - } - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid args"}) - return - } - - // 参数处理 - if data.Name == "" || data.MaxCalls < 0 { - c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid args"}) - return - } - - // 检查当前要添加的 Username 是否已经存在 - _, err = GetUser(data.Name) - if err == nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Username " + data.Name + " already exists"}) - return - } - - var chatRoles = make(map[string]int) - if len(data.ChatRoles) > 0 { - if data.ChatRoles[0] == "all" { // 所有的角色 - roles := GetChatRoles() - for key := range roles { - chatRoles[key] = 1 - } - } else { - for _, key := range data.ChatRoles { - chatRoles[key] = 1 - } - } - - } - - user := types.User{ - Name: data.Name, - MaxCalls: data.MaxCalls, - RemainingCalls: data.MaxCalls, - EnableHistory: data.EnableHistory, - Term: data.Term, - ChatRoles: chatRoles, - Status: true} - err = PutUser(user) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: user}) -} - -// BatchAddUserHandle 批量生成 Username -func (s *Server) BatchAddUserHandle(c *gin.Context) { - var data struct { - Number int `json:"number"` - MaxCalls int `json:"max_calls"` - EnableHistory bool `json:"enable_history"` - Term int `json:"term"` - ChatRoles []string `json:"chat_roles"` - } - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil || data.MaxCalls <= 0 { - c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid args"}) - return - } - - var chatRoles = make(map[string]int) - if len(data.ChatRoles) > 0 { - if data.ChatRoles[0] == "all" { // 所有的角色 - roles := GetChatRoles() - for key := range roles { - chatRoles[key] = 1 - } - } else { - for _, key := range data.ChatRoles { - chatRoles[key] = 1 - } - } - - } - - var users = make([]types.User, 0) - for i := 0; i < data.Number; i++ { - name := utils.RandString(12) - _, err := GetUser(name) - for err == nil { - name = utils.RandString(12) - } - user := types.User{ - Name: name, - MaxCalls: data.MaxCalls, - RemainingCalls: data.MaxCalls, - EnableHistory: data.EnableHistory, - Term: data.Term, - ChatRoles: chatRoles, - Status: true} - err = PutUser(user) - if err == nil { - users = append(users, user) - } - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: users}) -} - -func (s *Server) SetUserHandle(c *gin.Context) { - var data map[string]interface{} - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid args"}) - return - } - - var user *types.User - if name, ok := data["name"]; ok { - user, err = GetUser(name.(string)) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "User not found"}) - return - } - } - var maxCalls int - if v, ok := data["max_calls"]; ok { - maxCalls = int(v.(float64)) - } - if maxCalls < 0 { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Invalid args"}) - return - } else if maxCalls > 0 { - user.MaxCalls = maxCalls - user.RemainingCalls += maxCalls - user.MaxCalls - if user.RemainingCalls < 0 { - user.RemainingCalls = 0 - } - } - - if v, ok := data["status"]; ok { - user.Status = v.(bool) - } - if v, ok := data["enable_history"]; ok { - user.EnableHistory = v.(bool) - } - if v, ok := data["remaining_calls"]; ok { - user.RemainingCalls = int(v.(float64)) - } - if v, ok := data["expired_time"]; ok { - user.ExpiredTime = int64(v.(float64)) - } - if v, ok := data["api_key"]; ok { - user.ApiKey = v.(string) - } - if v, ok := data["chat_roles"]; ok { - if roles, ok := v.([]interface{}); ok { - chatRoles := make(map[string]int) - if roles[0] == "all" { - roles := GetChatRoles() - for key := range roles { - chatRoles[key] = 1 - } - } else { - for _, key := range roles { - key := strings.TrimSpace(fmt.Sprintf("%v", key)) - chatRoles[key] = 1 - } - } - user.ChatRoles = chatRoles - } - - } - - err = PutUser(*user) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: user}) -} - -// RemoveUserHandle 删除 Username -func (s *Server) RemoveUserHandle(c *gin.Context) { - var data struct { - Name string `json:"name"` - } - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Invalid args"}) - return - } - - err = RemoveUser(data.Name) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg}) -} - -// GetUserListHandle 获取用户列表 -func (s *Server) GetUserListHandle(c *gin.Context) { - username := c.PostForm("username") - if username != "" { - user, err := GetUser(username) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "User not exists"}) - } else { - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: user}) - } - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: GetUsers()}) -} - -// AddApiKeyHandle 添加一个 API key -func (s *Server) AddApiKeyHandle(c *gin.Context) { - var data struct { - ApiKey string `json:"api_key"` - } - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - logger.Errorf("Error decode json data: %s", err.Error()) - c.JSON(http.StatusBadRequest, nil) - return - } - if len(data.ApiKey) > 20 { - s.Config.Chat.ApiKeys = append(s.Config.Chat.ApiKeys, data.ApiKey) - } - - // 保存配置文件 - err = utils.SaveConfig(s.Config, s.ConfigPath) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Chat.ApiKeys}) -} - -// RemoveApiKeyHandle 移除一个 API key -func (s *Server) RemoveApiKeyHandle(c *gin.Context) { - var data struct { - ApiKey string `json:"api_key"` - } - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - logger.Errorf("Error decode json data: %s", err.Error()) - c.JSON(http.StatusBadRequest, nil) - return - } - - for i, v := range s.Config.Chat.ApiKeys { - if v == data.ApiKey { - s.Config.Chat.ApiKeys = append(s.Config.Chat.ApiKeys[:i], s.Config.Chat.ApiKeys[i+1:]...) - } - } - - // 保存配置文件 - err = utils.SaveConfig(s.Config, s.ConfigPath) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Chat.ApiKeys}) -} - -// ListApiKeysHandle 获取 API key 列表 -func (s *Server) ListApiKeysHandle(c *gin.Context) { - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Chat.ApiKeys}) -} - -// 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 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"} - var res = make([]interface{}, 0) - var roles = GetChatRoles() - for _, k := range rolesOrder { - // 确认当前用户是否订阅了当前角色 - 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"` - Name string `json:"name"` - Icon string `json:"icon"` - }{ - Key: v.Key, - Name: v.Name, - Icon: v.Icon, - }) - } - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: res}) -} - -// AddChatRoleHandle 添加一个聊天角色 -func (s *Server) AddChatRoleHandle(c *gin.Context) { - var data types.ChatRole - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - logger.Errorf("Error decode json data: %s", err.Error()) - c.JSON(http.StatusBadRequest, nil) - return - } - - if data.Key == "" || data.Name == "" || data.Icon == "" { - c.JSON(http.StatusOK, types.BizVo{Code: types.InvalidParams, Message: "Invalid parameters"}) - return - } - - err = PutChatRole(data) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save levelDB"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: data}) -} - -// GetChatRoleHandle 获取指定的角色 -func (s *Server) GetChatRoleHandle(c *gin.Context) { - var data struct { - Key string `json:"key"` - } - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - logger.Errorf("Error decode json data: %s", err.Error()) - c.JSON(http.StatusBadRequest, nil) - return - } - - role, err := GetChatRole(data.Key) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "No role found"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Data: role}) - return -} - -// SetChatRoleHandle 更新某个聊天角色信息，这里只允许更改名称以及启用和禁用角色操作 -func (s *Server) SetChatRoleHandle(c *gin.Context) { - var data map[string]interface{} - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - logger.Errorf("Error decode json data: %s", err.Error()) - c.JSON(http.StatusBadRequest, nil) - return - } - - var key string - if v, ok := data["key"]; !ok { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Must specified the role key"}) - return - } else { - key = v.(string) - } - - role, err := GetChatRole(key) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Role key not exists"}) - return - } - - if v, ok := data["name"]; ok { - role.Name = v.(string) - } - if v, ok := data["hello_msg"]; ok { - role.HelloMsg = v.(string) - } - if v, ok := data["icon"]; ok { - role.Icon = v.(string) - } - if v, ok := data["enable"]; ok { - role.Enable = v.(bool) - } - if v, ok := data["context"]; ok { - bytes, _ := json.Marshal(v) - _ = json.Unmarshal(bytes, &role.Context) - } - - err = PutChatRole(*role) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save levelDB"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: data}) -} - -// AddProxyHandle 添加一个代理 -func (s *Server) AddProxyHandle(c *gin.Context) { - var data struct { - Proxy string `json:"proxy"` - } - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - logger.Errorf("Error decode json data: %s", err.Error()) - c.JSON(http.StatusBadRequest, nil) - return - } - - if data.Proxy != "" { - if !utils.ContainsStr(s.Config.ProxyURL, data.Proxy) { - s.Config.ProxyURL = append(s.Config.ProxyURL, data.Proxy) - } - } - - // 保存配置文件 - err = utils.SaveConfig(s.Config, s.ConfigPath) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.ProxyURL}) -} - -// RemoveProxyHandle 删除一个代理 -func (s *Server) RemoveProxyHandle(c *gin.Context) { - var data struct { - Proxy string `json:"proxy"` - } - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - logger.Errorf("Error decode json data: %s", err.Error()) - c.JSON(http.StatusBadRequest, nil) - return - } - - for i, v := range s.Config.ProxyURL { - if v == data.Proxy { - s.Config.ProxyURL = append(s.Config.ProxyURL[:i], s.Config.ProxyURL[i+1:]...) - break - } - } - - // 保存配置文件 - err = utils.SaveConfig(s.Config, s.ConfigPath) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.ProxyURL}) -} - -// SetImgURLHandle SetImgURL 设置图片地址集合 -func (s *Server) SetImgURLHandle(c *gin.Context) { - var data struct { - WechatCard string `json:"wechat_card"` // 个人微信二维码 - WechatGroup string `json:"wechat_group"` // 微信群聊二维码 - } - err := json.NewDecoder(c.Request.Body).Decode(&data) - if err != nil { - logger.Errorf("Error decode json data: %s", err.Error()) - c.JSON(http.StatusBadRequest, nil) - return - } - - if data.WechatCard != "" { - s.Config.ImgURL.WechatCard = data.WechatCard - } - if data.WechatGroup != "" { - s.Config.ImgURL.WechatGroup = data.WechatGroup - } - - // 保存配置文件 - err = utils.SaveConfig(s.Config, s.ConfigPath) - if err != nil { - c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"}) - return - } - - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.ImgURL}) - -} - -// GetImgURLHandle 获取图片地址集合 -func (s *Server) GetImgURLHandle(c *gin.Context) { - c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.ImgURL}) -} diff --git a/server/server.go b/server/server.go index 1134876e..900db3ba 100644 --- a/server/server.go +++ b/server/server.go @@ -98,22 +98,21 @@ func (s *Server) Run(webRoot embed.FS, path string, debug bool) { engine.POST("api/img/get", s.GetImgURLHandle) engine.POST("api/img/set", s.SetImgURLHandle) - engine.POST("api/config/set", s.ConfigSetHandle) - engine.GET("api/config/chat-roles/get", s.GetChatRoleListHandle) - engine.GET("api/config/chat-roles/add", s.AddChatRoleHandle) - engine.POST("api/config/user/add", s.AddUserHandle) - engine.POST("api/config/user/batch-add", s.BatchAddUserHandle) - engine.POST("api/config/user/set", s.SetUserHandle) - engine.POST("api/config/user/list", s.GetUserListHandle) - engine.POST("api/config/user/remove", s.RemoveUserHandle) - engine.POST("api/config/apikey/add", s.AddApiKeyHandle) - engine.POST("api/config/apikey/remove", s.RemoveApiKeyHandle) - engine.POST("api/config/apikey/list", s.ListApiKeysHandle) - engine.POST("api/config/role/set", s.SetChatRoleHandle) - engine.POST("api/config/role/get", s.GetChatRoleHandle) - engine.POST("api/config/proxy/add", s.AddProxyHandle) - engine.POST("api/config/proxy/remove", s.RemoveProxyHandle) - engine.POST("api/config/debug", s.SetDebugHandle) + engine.POST("api/admin/config", s.ConfigSetHandle) + engine.GET("api/admin/chat-roles/get", s.GetChatRoleListHandle) + engine.GET("api/admin/chat-roles/add", s.AddChatRoleHandle) + engine.POST("api/admin/user/add", s.AddUserHandle) + engine.POST("api/admin/user/batch-add", s.BatchAddUserHandle) + engine.POST("api/admin/user/set", s.SetUserHandle) + engine.POST("api/admin/user/list", s.GetUserListHandle) + engine.POST("api/admin/user/remove", s.RemoveUserHandle) + engine.POST("api/admin/apikey/add", s.AddApiKeyHandle) + engine.POST("api/admin/apikey/remove", s.RemoveApiKeyHandle) + engine.POST("api/admin/apikey/list", s.ListApiKeysHandle) + engine.POST("api/admin/role/set", s.SetChatRoleHandle) + engine.POST("api/admin/role/get", s.GetChatRoleHandle) + engine.POST("api/admin/proxy/add", s.AddProxyHandle) + engine.POST("api/admin/proxy/remove", s.RemoveProxyHandle) engine.NoRoute(func(c *gin.Context) { if c.Request.URL.Path == "/favicon.ico" { @@ -225,16 +224,16 @@ func AuthorizeMiddleware(s *Server) gin.HandlerFunc { return } - if strings.HasPrefix(c.Request.URL.Path, "/api/config") { - accessKey := c.GetHeader("ACCESS-KEY") - if accessKey != strings.TrimSpace(s.Config.AccessKey) { - c.Abort() - c.JSON(http.StatusOK, types.BizVo{Code: types.NotAuthorized, Message: "No Permissions"}) - } else { - c.Next() - } - return - } + //if strings.HasPrefix(c.Request.URL.Path, "/api/config") { + // accessKey := c.GetHeader("ACCESS-KEY") + // if accessKey != strings.TrimSpace(s.Config.AccessKey) { + // c.Abort() + // c.JSON(http.StatusOK, types.BizVo{Code: types.NotAuthorized, Message: "No Permissions"}) + // } else { + // c.Next() + // } + // return + //} // WebSocket 连接请求验证 if c.Request.URL.Path == "/api/chat" { diff --git a/types/config.go b/types/config.go index 0346deb1..6b7d642e 100644 --- a/types/config.go +++ b/types/config.go @@ -5,13 +5,14 @@ import ( ) type Config struct { - Listen string - Session Session - ProxyURL []string - EnableAuth bool // 是否开启鉴权 - AccessKey string // 管理员访问 AccessKey, 通过传入这个参数可以访问系统管理 API - Chat Chat - ImgURL ImgURL // 各种图片资源链接地址，比如微信二维码，群二维码 + Title string + ConsoleTitle string + Listen string + Session Session + ProxyURL []string + EnableAuth bool // 是否开启鉴权 + Chat Chat + ImgURL ImgURL // 各种图片资源链接地址，比如微信二维码，群二维码 } type User struct { @@ -27,6 +28,12 @@ type User struct { ChatRoles map[string]int `json:"chat_roles"` // 当前用户已订阅的聊天角色 map[role_key] => 0/1 } +// Manager 管理员 +type Manager struct { + Username string `json:"username"` + Password string `json:"password"` +} + // Chat configs struct type Chat struct { ApiURL string diff --git a/utils/config.go b/utils/config.go index 7747ba96..c8060425 100644 --- a/utils/config.go +++ b/utils/config.go @@ -11,11 +11,12 @@ import ( func NewDefaultConfig() *types.Config { return &types.Config{ - Listen: "0.0.0.0:5678", - ProxyURL: make([]string, 0), - EnableAuth: true, - AccessKey: "yangjian102621@gmail.com", - ImgURL: types.ImgURL{}, + Title: "Chat-Plus AI 助手", + ConsoleTitle: "Chat-Plus 控制台", + Listen: "0.0.0.0:5678", + ProxyURL: make([]string, 0), + EnableAuth: true, + ImgURL: types.ImgURL{}, Session: types.Session{ SecretKey: RandString(64), diff --git a/web/src/views/admin/SysConfig.vue b/web/src/views/admin/SysConfig.vue index e75d9601..72c2d1a0 100644 --- a/web/src/views/admin/SysConfig.vue +++ b/web/src/views/admin/SysConfig.vue @@ -1,6 +1,39 @@ @@ -12,8 +45,14 @@ export default defineComponent({ data() { return { title: "系统管理", + form: {} } }, + methods: { + save: function () { + + } + } })