add API to remove API key, set chat roles

server/config_handler.go

@@ -106,6 +106,13 @@ func (s *Server) AddToken(c *gin.Context) {
 		}
 	}
 
+	// 保存配置文件
+	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})
 }
 
@@ -127,11 +134,18 @@ func (s *Server) RemoveToken(c *gin.Context) {
 		}
 	}
 
+	// 保存配置文件
+	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})
 }
 
+// AddApiKey 添加一个 API key
 func (s *Server) AddApiKey(c *gin.Context) {
-
 	var data map[string]string
 	err := json.NewDecoder(c.Request.Body).Decode(&data)
 	if err != nil {
@@ -143,9 +157,45 @@ func (s *Server) AddApiKey(c *gin.Context) {
 		s.Config.Chat.ApiKeys = append(s.Config.Chat.ApiKeys, key)
 	}
 
+	// 保存配置文件
+	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.Chat.ApiKeys})
 }
 
+// RemoveApiKey 移除一个 API key
+func (s *Server) RemoveApiKey(c *gin.Context) {
+	var data map[string]string
+	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 key, ok := data["api_key"]; ok {
+		for i, v := range s.Config.Chat.ApiKeys {
+			if v == key {
+				s.Config.Chat.ApiKeys = append(s.Config.Chat.ApiKeys[:i], s.Config.Chat.ApiKeys[i+1:]...)
+			}
+		}
+	}
+
+	// 保存配置文件
+	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.Chat.ApiKeys})
+}
+
+// ListApiKeys 获取 API key 列表
 func (s *Server) ListApiKeys(c *gin.Context) {
 	c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Chat.ApiKeys})
 }
@@ -154,7 +204,7 @@ 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 {
+		if v, ok := s.Config.ChatRoles[k]; ok && v.Enable {
 			roles = append(roles, struct {
 				Key  string `json:"key"`
 				Name string `json:"name"`
@@ -168,3 +218,53 @@ func (s *Server) GetChatRoles(c *gin.Context) {
 	}
 	c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: roles})
 }
+
+// UpdateChatRole 更新某个聊天角色信息，这里只允许更改名称以及启用和禁用角色操作
+func (s *Server) UpdateChatRole(c *gin.Context) {
+	var data map[string]string
+	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
+	}
+	key := data["key"]
+	if key == "" {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Must specified the role key"})
+		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
+	}
+
+	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})
+}