package api import ( "net/http" "forge.cadoles.com/Cadoles/emissary/internal/datastore" "github.com/pkg/errors" "gitlab.com/wpetit/goweb/api" "gitlab.com/wpetit/goweb/logger" ) func (m *Mount) queryAgents(w http.ResponseWriter, r *http.Request) { user, ok := assertRequestUser(w, r) if !ok { return } limit, ok := getIntQueryParam(w, r, "limit", 10) if !ok { return } offset, ok := getIntQueryParam(w, r, "offset", 0) if !ok { return } options := []datastore.AgentQueryOptionFunc{ datastore.WithAgentQueryLimit(int(limit)), datastore.WithAgentQueryOffset(int(offset)), datastore.WithAgentQueryTenantID(user.Tenant()), } ids, ok := getIntSliceValues(w, r, "ids", nil) if !ok { return } if ids != nil { agentIDs := func(ids []int64) []datastore.AgentID { agentIDs := make([]datastore.AgentID, 0, len(ids)) for _, id := range ids { agentIDs = append(agentIDs, datastore.AgentID(id)) } return agentIDs }(ids) options = append(options, datastore.WithAgentQueryID(agentIDs...)) } thumbprints, ok := getStringSliceValues(w, r, "thumbprints", nil) if !ok { return } if thumbprints != nil { options = append(options, datastore.WithAgentQueryThumbprints(thumbprints...)) } statuses, ok := getIntSliceValues(w, r, "statuses", nil) if !ok { return } if statuses != nil { agentStatuses := func(statuses []int64) []datastore.AgentStatus { agentStatuses := make([]datastore.AgentStatus, 0, len(statuses)) for _, status := range statuses { agentStatuses = append(agentStatuses, datastore.AgentStatus(status)) } return agentStatuses }(statuses) options = append(options, datastore.WithAgentQueryStatus(agentStatuses...)) } ctx := r.Context() agents, total, err := m.agentRepo.Query( ctx, options..., ) if err != nil { err = errors.WithStack(err) logger.Error(ctx, "could not list agents", logger.CapturedE(err)) api.ErrorResponse(w, http.StatusInternalServerError, ErrCodeUnknownError, nil) return } api.DataResponse(w, http.StatusOK, struct { Agents []*datastore.Agent `json:"agents"` Total int `json:"total"` }{ Agents: agents, Total: total, }) }