xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-11 03:33:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

使用 leveldb 存储用户 token 和聊天记录

Browse Source
This commit is contained in:
RockYang
2023-03-27 18:27:33 +08:00
parent 6a38de7eaa
commit a6bab7b12d
15 changed files with 370 additions and 175 deletions
Download Patch File Download Diff File Expand all files Collapse all files

165
server/config_handler.go
View File

@@ -77,7 +77,7 @@ func (s *Server) ConfigSetHandle(c *gin.Context) {
}
// 保存配置文件
err = types.SaveConfig(s.Config, s.ConfigPath)
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
@@ -86,6 +86,7 @@ func (s *Server) ConfigSetHandle(c *gin.Context) {
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg})
}
// AddToken 添加 Token
func (s *Server) AddToken(c *gin.Context) {
var data map[string]string
err := json.NewDecoder(c.Request.Body).Decode(&data)
@@ -95,22 +96,38 @@ func (s *Server) AddToken(c *gin.Context) {
return
}
if token, ok := data["token"]; ok {
if !utils.ContainsItem(s.Config.Tokens, token) {
s.Config.Tokens = append(s.Config.Tokens, token)
}
}
// 保存配置文件
err = types.SaveConfig(s.Config, s.ConfigPath)
if err != nil {
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"})
var name = data["name"]
var maxCalls = data["max_calls"]
if name == "" || maxCalls == "" {
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Invalid args"})
return
}
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Tokens})
n, err := strconv.Atoi(maxCalls)
if err != nil {
c.JSON(http.StatusOK, types.BizVo{
Code: types.InvalidParams,
Message: "enable_auth must be a int parameter",
})
return
}
var tokens = GetTokens()
if utils.ContainToken(tokens, name) {
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Token " + name + " already exists"})
return
}
err = PutToken(types.Token{Name: name, MaxCalls: n, RemainingCalls: n})
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: GetTokens()})
}
// RemoveToken 删除 Token
func (s *Server) RemoveToken(c *gin.Context) {
var data map[string]string
err := json.NewDecoder(c.Request.Body).Decode(&data)
@@ -121,22 +138,14 @@ func (s *Server) RemoveToken(c *gin.Context) {
}
if token, ok := data["token"]; ok {
for i, v := range s.Config.Tokens {
if v == token {
s.Config.Tokens = append(s.Config.Tokens[:i], s.Config.Tokens[i+1:]...)
break
}
err = RemoveToken(token)
if err != nil {
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"})
return
}
}
// 保存配置文件
err = types.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.Tokens})
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: GetTokens()})
}
// AddApiKey 添加一个 API key
@@ -153,7 +162,7 @@ func (s *Server) AddApiKey(c *gin.Context) {
}
// 保存配置文件
err = types.SaveConfig(s.Config, s.ConfigPath)
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
@@ -181,7 +190,7 @@ func (s *Server) RemoveApiKey(c *gin.Context) {
}
// 保存配置文件
err = types.SaveConfig(s.Config, s.ConfigPath)
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
@@ -196,22 +205,22 @@ func (s *Server) ListApiKeys(c *gin.Context) {
}
func (s *Server) GetChatRoles(c *gin.Context) {
var rolesOrder = []string{"gpt", "programmer", "teacher", "artist", "philosopher", "lu-xun", "english_trainer", "seller"}
var roles = make([]interface{}, 0)
for _, k := range rolesOrder {
if v, ok := s.Config.ChatRoles[k]; ok && v.Enable {
roles = append(roles, 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: roles})
//var rolesOrder = []string{"gpt", "programmer", "teacher", "artist", "philosopher", "lu-xun", "english_trainer", "seller"}
//var roles = make([]interface{}, 0)
//for _, k := range rolesOrder {
// if v, ok := s.Config.ChatRoles[k]; ok && v.Enable {
// roles = append(roles, 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: roles})
}
// UpdateChatRole 更新某个聊天角色信息，这里只允许更改名称以及启用和禁用角色操作
@@ -229,39 +238,39 @@ func (s *Server) UpdateChatRole(c *gin.Context) {
return
}
role := s.Config.ChatRoles[key]
if enable, ok := data["enable"]; ok {
v, err := strconv.ParseBool(enable)
if err != nil {
c.JSON(http.StatusOK, types.BizVo{
Code: types.InvalidParams,
Message: "enable must be a bool parameter",
})
return
}
role.Enable = v
}
//role := s.Config.ChatRoles[key]
//if enable, ok := data["enable"]; ok {
// v, err := strconv.ParseBool(enable)
// if err != nil {
// c.JSON(http.StatusOK, types.BizVo{
// Code: types.InvalidParams,
// Message: "enable must be a bool parameter",
// })
// return
// }
// role.Enable = v
//}
if name, ok := data["name"]; ok {
role.Name = name
}
if helloMsg, ok := data["hello_msg"]; ok {
role.HelloMsg = helloMsg
}
if icon, ok := data["icon"]; ok {
role.Icon = icon
}
s.Config.ChatRoles[key] = role
// 保存配置文件
err = types.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})
//if name, ok := data["name"]; ok {
// role.Name = name
//}
//if helloMsg, ok := data["hello_msg"]; ok {
// role.HelloMsg = helloMsg
//}
//if icon, ok := data["icon"]; ok {
// role.Icon = icon
//}
//
//s.Config.ChatRoles[key] = role
//
//// 保存配置文件
//err = types.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})
}
// AddProxy 添加一个代理
@@ -275,13 +284,13 @@ func (s *Server) AddProxy(c *gin.Context) {
}
if proxy, ok := data["proxy"]; ok {
if !utils.ContainsItem(s.Config.ProxyURL, proxy) {
if !utils.ContainsStr(s.Config.ProxyURL, proxy) {
s.Config.ProxyURL = append(s.Config.ProxyURL, proxy)
}
}
// 保存配置文件
err = types.SaveConfig(s.Config, s.ConfigPath)
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
@@ -309,7 +318,7 @@ func (s *Server) RemoveProxy(c *gin.Context) {
}
// 保存配置文件
err = types.SaveConfig(s.Config, s.ConfigPath)
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