62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package metal
|
|
|
|
import (
|
|
"go.uber.org/zap"
|
|
|
|
"go.equinixmetal.net/infra9-metal-bridge/internal/metal/providers"
|
|
"go.equinixmetal.net/infra9-metal-bridge/internal/metal/providers/emapi"
|
|
"go.equinixmetal.net/infra9-metal-bridge/internal/metal/providers/emgql"
|
|
)
|
|
|
|
// Option is a Client configuration Option definition.
|
|
type Option func(c *Client) error
|
|
|
|
// WithProvider sets the provider on the client.
|
|
func WithProvider(provider providers.Provider) Option {
|
|
return func(c *Client) error {
|
|
c.provider = provider
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithLogger sets the logger for the client.
|
|
func WithLogger(logger *zap.Logger) Option {
|
|
return func(c *Client) error {
|
|
c.logger = logger
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithConfig applies all configurations defined in the config.
|
|
func WithConfig(config Config) Option {
|
|
return func(c *Client) error {
|
|
var options []Option
|
|
|
|
if config.EMGQL.Populated() {
|
|
client, err := emgql.New(emgql.WithConfig(config.EMGQL))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
options = append(options, WithProvider(client))
|
|
} else if config.EMAPI.Populated() {
|
|
client, err := emapi.New(emapi.WithConfig(config.EMAPI))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
options = append(options, WithProvider(client))
|
|
}
|
|
|
|
for _, opt := range options {
|
|
if err := opt(c); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|