73 lines
1.4 KiB
Go
73 lines
1.4 KiB
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
|
|
}
|