package client import ( "context" "fmt" "net/url" "forge.cadoles.com/Cadoles/emissary/internal/datastore" "github.com/pkg/errors" ) type QueryAgentsOptionFunc func(*QueryAgentsOptions) type QueryAgentsOptions struct { Limit *int Offset *int Thumbprints []string IDs []datastore.AgentID Statuses []datastore.AgentStatus } func WithQueryAgentsLimit(limit int) QueryAgentsOptionFunc { return func(opts *QueryAgentsOptions) { opts.Limit = &limit } } func WithQueryAgentsOffset(offset int) QueryAgentsOptionFunc { return func(opts *QueryAgentsOptions) { opts.Offset = &offset } } func WithQueryAgentsThumbprints(thumbprints ...string) QueryAgentsOptionFunc { return func(opts *QueryAgentsOptions) { opts.Thumbprints = thumbprints } } func WithQueryAgentsID(ids ...datastore.AgentID) QueryAgentsOptionFunc { return func(opts *QueryAgentsOptions) { opts.IDs = ids } } func WithQueryAgentsStatus(statuses ...datastore.AgentStatus) QueryAgentsOptionFunc { return func(opts *QueryAgentsOptions) { opts.Statuses = statuses } } func (c *Client) QueryAgents(ctx context.Context, funcs ...QueryAgentsOptionFunc) ([]*datastore.Agent, int, error) { options := &QueryAgentsOptions{} for _, fn := range funcs { fn(options) } query := url.Values{} if options.IDs != nil && len(options.IDs) > 0 { query.Set("ids", joinSlice(options.IDs)) } if options.Thumbprints != nil && len(options.Thumbprints) > 0 { query.Set("thumbprints", joinSlice(options.Thumbprints)) } if options.Statuses != nil && len(options.Statuses) > 0 { query.Set("statuses", joinSlice(options.Statuses)) } path := fmt.Sprintf("/api/v1/agents?%s", query.Encode()) response := withResponse[struct { Agents []*datastore.Agent `json:"agents"` Total int `json:"total"` }]() if err := c.apiGet(ctx, path, &response); err != nil { return nil, 0, errors.WithStack(err) } if response.Error != nil { return nil, 0, errors.WithStack(response.Error) } return response.Data.Agents, response.Data.Total, nil }