query-orchestration/api/queryAPI/collector.go

// Package queryapi provides HTTP handlers for the Query API.
// Note: Text extraction has been removed. MinimumTextVersion is no longer supported.
// See remove_texttract_and_mocks_plan.md for details.
package queryapi

import (
	"fmt"
	"net/http"

	collectorset "queryorchestration/internal/collector/set"

	"github.com/labstack/echo/v4"
)

// GetCollectorByClientId retrieves a client's collector configuration.
// Note: MinTextVersion has been removed since text extraction is deprecated.
func (s *Controllers) GetCollectorByClientId(ctx echo.Context, clientId ClientID) error {
	coll, err := s.svc.Collector.GetByClientID(ctx.Request().Context(), clientId)
	if err != nil {
		return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Unable to get collector: %s", err))
	}

	return ctx.JSON(http.StatusOK, Collector{
		ClientId:              clientId,
		MinimumCleanerVersion: coll.MinCleanVersion,
		ActiveVersion:         coll.ActiveVersion,
		LatestVersion:         coll.LatestVersion,
	})
}

// SetCollectorByClientId updates a client's collector configuration.
// Note: MinTextVersion has been removed since text extraction is deprecated.
func (s *Controllers) SetCollectorByClientId(ctx echo.Context, clientId ClientID) error {
	req := CollectorSet{}
	if err := ctx.Bind(&req); err != nil {
		return echo.NewHTTPError(http.StatusBadRequest, err)
	}

	err := s.svc.CollectorSet.SetByClientId(ctx.Request().Context(), clientId, &collectorset.SetParams{
		ActiveVersion:   req.ActiveVersion,
		MinCleanVersion: req.MinimumCleanerVersion,
	})
	if err != nil {
		return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Unable to update collector: %s", err))
	}

	return ctx.NoContent(http.StatusOK)
}