aiproxy/controller/option.go

package controller

import (
	"net/http"

	"github.com/gin-gonic/gin"
	"github.com/labring/aiproxy/middleware"
	"github.com/labring/aiproxy/model"
)

// GetOptions godoc
//
//	@Summary		Get options
//	@Description	Returns a list of options
//	@Tags			option
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Success		200	{object}	middleware.APIResponse{data=map[string]string}
//	@Router			/api/options [get]
func GetOptions(c *gin.Context) {
	dbOptions, err := model.GetAllOption()
	if err != nil {
		middleware.ErrorResponse(c, http.StatusOK, err.Error())
		return
	}
	options := make(map[string]string, len(dbOptions))
	for _, option := range dbOptions {
		options[option.Key] = option.Value
	}
	middleware.SuccessResponse(c, options)
}

// GetOption godoc
//
//	@Summary		Get option
//	@Description	Returns a single option
//	@Tags			option
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Param			key	path		string	true	"Option key"
//	@Success		200	{object}	middleware.APIResponse{data=model.Option}
//	@Router			/api/options/{key} [get]
func GetOption(c *gin.Context) {
	key := c.Param("key")
	if key == "" {
		middleware.ErrorResponse(c, http.StatusOK, "key is required")
		return
	}
	option, err := model.GetOption(key)
	if err != nil {
		middleware.ErrorResponse(c, http.StatusOK, err.Error())
		return
	}
	middleware.SuccessResponse(c, option)
}

// UpdateOption godoc
//
//	@Summary		Update option
//	@Description	Updates a single option
//	@Tags			option
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Param			key		path		string	true	"Option key"
//	@Param			value	body		string	true	"Option value"
//	@Success		200		{object}	middleware.APIResponse
//	@Router			/api/options/{key} [put]
func UpdateOption(c *gin.Context) {
	var option model.Option
	err := c.BindJSON(&option)
	if err != nil {
		middleware.ErrorResponse(c, http.StatusOK, err.Error())
		return
	}
	err = model.UpdateOption(option.Key, option.Value)
	if err != nil {
		middleware.ErrorResponse(c, http.StatusOK, err.Error())
		return
	}
	middleware.SuccessResponse(c, nil)
}

// UpdateOptions godoc
//
//	@Summary		Update options
//	@Description	Updates multiple options
//	@Tags			option
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Param			options	body		map[string]string	true	"Options"
//	@Success		200		{object}	middleware.APIResponse
//	@Router			/api/options [put]
func UpdateOptions(c *gin.Context) {
	var options map[string]string
	err := c.BindJSON(&options)
	if err != nil {
		middleware.ErrorResponse(c, http.StatusOK, err.Error())
		return
	}
	err = model.UpdateOptions(options)
	if err != nil {
		middleware.ErrorResponse(c, http.StatusOK, err.Error())
		return
	}
	middleware.SuccessResponse(c, nil)
}