emissary/pkg/client/update_agent.go

package client

import (
	"context"
	"fmt"

	"forge.cadoles.com/Cadoles/emissary/internal/datastore"
	"github.com/pkg/errors"
)

type UpdateAgentOptions struct {
	Status  *int
	Label   *string
	Options []OptionFunc
}

type UpdateAgentOptionFunc func(*UpdateAgentOptions)

func WithAgentStatus(status int) UpdateAgentOptionFunc {
	return func(opts *UpdateAgentOptions) {
		opts.Status = &status
	}
}

func WithAgentLabel(label string) UpdateAgentOptionFunc {
	return func(opts *UpdateAgentOptions) {
		opts.Label = &label
	}
}

func WithUpdateAgentsOptions(funcs ...OptionFunc) UpdateAgentOptionFunc {
	return func(opts *UpdateAgentOptions) {
		opts.Options = funcs
	}
}

func (c *Client) UpdateAgent(ctx context.Context, agentID datastore.AgentID, funcs ...UpdateAgentOptionFunc) (*datastore.Agent, error) {
	opts := &UpdateAgentOptions{}
	for _, fn := range funcs {
		fn(opts)
	}

	payload := map[string]any{}

	if opts.Status != nil {
		payload["status"] = *opts.Status
	}

	if opts.Label != nil {
		payload["label"] = *opts.Label
	}

	response := withResponse[struct {
		Agent *datastore.Agent `json:"agent"`
	}]()

	path := fmt.Sprintf("/api/v1/agents/%d", agentID)

	if opts.Options == nil {
		opts.Options = make([]OptionFunc, 0)
	}

	if err := c.apiPut(ctx, path, payload, &response, opts.Options...); err != nil {
		return nil, errors.WithStack(err)
	}

	if response.Error != nil {
		return nil, errors.WithStack(response.Error)
	}

	return response.Data.Agent, nil
}