clusterctl fix public library

Author: fabriziopandini

cmd/clusterctl/pkg/client/client.go

@@ -231,13 +231,13 @@ func newClusterctlClient(path string, options ...Option) (*clusterctlClient, err
 // defaultClusterFactory is a ClusterClientFactory func the uses the default client provided by the cluster low level library.
 func defaultClusterFactory() func(kubeconfig string) (cluster.Client, error) {
 	return func(kubeconfig string) (cluster.Client, error) {
-		return cluster.New(kubeconfig, cluster.Options{}), nil
+		return cluster.New(kubeconfig), nil
 	}
 }
 
 // defaultRepositoryFactory is a RepositoryClientFactory func the uses the default client provided by the repository low level library.
 func defaultRepositoryFactory(configClient config.Client) func(providerConfig config.Provider) (repository.Client, error) {
 	return func(providerConfig config.Provider) (repository.Client, error) {
-		return repository.New(providerConfig, configClient.Variables(), repository.Options{})
+		return repository.New(providerConfig, configClient.Variables())
 	}
 }

cmd/clusterctl/pkg/client/client_test.go

@@ -140,11 +140,7 @@ func (f *fakeClient) WithRepository(repositoryClient repository.Client) *fakeCli
 func newFakeCluster(kubeconfig string) *fakeClusterClient {
 	fakeProxy := test.NewFakeProxy()
 
-	options := cluster.Options{
-		InjectProxy: fakeProxy,
-	}
-
-	client := cluster.New("", options)
+	client := cluster.New("", cluster.InjectProxy(fakeProxy))
 
 	return &fakeClusterClient{
 		kubeconfig:     kubeconfig,
@@ -257,15 +253,12 @@ func (f *fakeConfigClient) WithProvider(provider config.Provider) *fakeConfigCli
 // the WithPaths or WithDefaultVersion methods to configure the repository and WithFile to set the map values.
 func newFakeRepository(provider config.Provider, configVariablesClient config.VariablesClient) *fakeRepositoryClient {
 	fakeRepository := test.NewFakeRepository()
-	options := repository.Options{
-		InjectRepository: fakeRepository,
-	}
 
 	if configVariablesClient == nil {
 		configVariablesClient = newFakeConfig().Variables()
 	}
 
-	client, _ := repository.New(provider, configVariablesClient, options)
+	client, _ := repository.New(provider, configVariablesClient, repository.InjectRepository(fakeRepository))
 
 	return &fakeRepositoryClient{
 		Provider:       provider,

cmd/clusterctl/pkg/client/cluster/client.go

@@ -110,20 +110,48 @@ func (c *clusterClient) ObjectMover() ObjectMover {
 	return newObjectMover(c.proxy, c.machineWaiter, log)
 }
 
+// NewOptions carries the options supported by New
+type NewOptions struct {
+	injectProxy         Proxy
+	injectMachineWaiter MachineWaiter
+}
+
+// Option is a configuration option supplied to New
+type Option func(*NewOptions)
+
+// InjectProxy implements a New Option that allows to override the default proxy used by clusterctl.
+func InjectProxy(proxy Proxy) Option {
+	return func(c *NewOptions) {
+		c.injectProxy = proxy
+	}
+}
+
+// InjectMachineWaiter implements a New Option that allows to override the default machine waiter used by clusterctl.
+func InjectMachineWaiter(machineWaiter MachineWaiter) Option {
+	return func(c *NewOptions) {
+		c.injectMachineWaiter = machineWaiter
+	}
+}
+
 // New returns a cluster.Client.
-func New(kubeconfig string, options Options) Client {
-	return newClusterClient(kubeconfig, options)
+func New(kubeconfig string, options ...Option) Client {
+	return newClusterClient(kubeconfig, options...)
 }
 
-func newClusterClient(kubeconfig string, options Options) *clusterClient {
-	// if there is an injected proxy, use it, otherwise use the default one
-	proxy := options.InjectProxy
+func newClusterClient(kubeconfig string, options ...Option) *clusterClient {
+	cfg := &NewOptions{}
+	for _, o := range options {
+		o(cfg)
+	}
+
+	// if there is an injected proxy, use it, otherwise use a default one
+	proxy := cfg.injectProxy
 	if proxy == nil {
 		proxy = newProxy(kubeconfig)
 	}
 
 	// if there is an injected machineWaiter, use it, otherwise use the default one
-	machineWaiter := options.InjectMachineWaiter
+	machineWaiter := cfg.injectMachineWaiter
 	if machineWaiter == nil {
 		machineWaiter = waitForMachineReady
 	}
@@ -135,12 +163,6 @@ func newClusterClient(kubeconfig string, options Options) *clusterClient {
 	}
 }
 
-// Options allow to set ConfigClient options
-type Options struct {
-	InjectProxy         Proxy
-	InjectMachineWaiter MachineWaiter
-}
-
 type Proxy interface {
 	// CurrentNamespace returns the namespace from the current context in the kubeconfig file
 	CurrentNamespace() (string, error)

cmd/clusterctl/pkg/client/config/client.go

@@ -75,6 +75,7 @@ func newConfigClient(path string, options ...Option) (*configClient, error) {
 		o(cfg)
 	}
 
+	// if there is an injected reader, use it, otherwise use a default one
 	reader := cfg.injectReader
 	if reader == nil {
 		reader = newViperReader()

cmd/clusterctl/pkg/client/repository/client.go

@@ -93,12 +93,18 @@ func InjectRepository(repository Repository) Option {
 }
 
 // New returns a Client.
-func New(provider config.Provider, configVariablesClient config.VariablesClient, options Options) (Client, error) {
-	return newRepositoryClient(provider, configVariablesClient, options)
+func New(provider config.Provider, configVariablesClient config.VariablesClient, options ...Option) (Client, error) {
+	return newRepositoryClient(provider, configVariablesClient, options...)
 }
 
-func newRepositoryClient(provider config.Provider, configVariablesClient config.VariablesClient, options Options) (*repositoryClient, error) {
-	repository := options.InjectRepository
+func newRepositoryClient(provider config.Provider, configVariablesClient config.VariablesClient, options ...Option) (*repositoryClient, error) {
+	cfg := &NewOptions{}
+	for _, o := range options {
+		o(cfg)
+	}
+
+	// if there is an injected repository, use it, otherwise use a default one
+	repository := cfg.injectRepository
 	if repository == nil {
 		r, err := repositoryFactory(provider, configVariablesClient)
 		if err != nil {
@@ -114,11 +120,6 @@ func newRepositoryClient(provider config.Provider, configVariablesClient config.
 	}, nil
 }
 
-// Options allow to set Client options
-type Options struct {
-	InjectRepository Repository
-}
-
 // Repository defines the behavior of a repository implementation.
 // clusterctl is designed to support different repository types; each repository implementation should be aware of
 // the provider version they are hosting, and possibly to host more than one version.