app-api/internal/handler/model_list.go

package handler

import (
	"net/http"
	"strconv"

	"github.com/gin-gonic/gin"
	"github.com/luxsin/app-api/internal/response"
	"github.com/luxsin/app-api/internal/search"
	"github.com/luxsin/app-api/pkg/encode"
	"go.uber.org/zap"
)

type ModelListHandler struct {
	search *search.Client
	log    *zap.Logger
}

func NewModelListHandler(searchClient *search.Client, log *zap.Logger) *ModelListHandler {
	return &ModelListHandler{
		search: searchClient,
		log:    log,
	}
}

// ModelList 搜索型号列表
//
//	@Summary	搜索型号列表（基于 Meilisearch）
//	@Tags		Model
//	@Produce	json
//	@Param		key			query		string	false	"搜索关键词"
//	@Param		count		query		int		false	"返回数量上限"default(100)
//	@Param		base64Resp	query		string	false	"是否返回 base64 编码响应"
//	@Success	200			{array}		string
//	@Failure	500			{object}	object
//	@Router		/audio/modelList [get]
func (h *ModelListHandler) ModelList(c *gin.Context) {
	key := c.Query("key")
	base64Resp := encode.ParseBase64Param(c)

	count := 100
	if v := c.Query("count"); v != "" {
		if n, err := strconv.Atoi(v); err == nil && n > 0 {
			count = n
		}
	}

	list, err := h.search.ModelList(c.Request.Context(), key, count)
	if err != nil {
		h.log.Error("model list search failed", zap.Error(err))
		response.InternalError(c, "failed to search models")
		return
	}

	if base64Resp {
		encoded, err := encode.EncodeJSON(list)
		if err != nil {
			h.log.Error("encode response failed", zap.Error(err))
			response.InternalError(c, "failed to encode response")
			return
		}
		c.String(http.StatusOK, encoded)
		return
	}

	c.JSON(http.StatusOK, list)
}