query-orchestration/api/queryAPI/collector.go

package queryapi

import (
	"fmt"
	"net/http"

	collectorset "queryorchestration/internal/collector/set"

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

// GetCollectorByClientId retrieves a client's collector configuration.
// Query fields have been removed - only document processing settings remain.
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,
		MinimumTextVersion:    coll.MinTextVersion,
		ActiveVersion:         coll.ActiveVersion,
		LatestVersion:         coll.LatestVersion,
	})
}

// SetCollectorByClientId updates a client's collector configuration.
// Query fields have been removed - only document processing settings remain.
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,
		MinTextVersion:  req.MinimumTextVersion,
	})
	if err != nil {
		return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Unable to update collector: %s", err))
	}

	return ctx.NoContent(http.StatusOK)
}