kairos-agent/internal/agent/agent_options.go

package agent

// Options yields the options for the running agent
type Options struct {
	ApiAddress string
	Dir        []string
	Force      bool
	Restart    bool
}

// Apply applies option to the options struct
func (o *Options) Apply(opts ...Option) error {
	for _, oo := range opts {
		if err := oo(o); err != nil {
			return err
		}
	}
	return nil
}

// Option is a generic option for the Agent
type Option func(o *Options) error

// ForceAgent forces the agent to run
var ForceAgent Option = func(o *Options) error {
	o.Force = true
	return nil
}

// RestartAgent makes the agent restart on error
var RestartAgent Option = func(o *Options) error {
	o.Restart = true
	return nil
}

// WithAPI sets the API address
func WithAPI(address string) Option {
	return func(o *Options) error {
		o.ApiAddress = address
		return nil
	}
}

// WithDirectory sets the Agent config directories
func WithDirectory(dirs ...string) Option {
	return func(o *Options) error {
		o.Dir = dirs
		return nil
	}
}