credhub_shims.go

package credhub_shims

import (
	"code.cloudfoundry.org/credhub-cli/credhub"
	"code.cloudfoundry.org/credhub-cli/credhub/auth"
	"code.cloudfoundry.org/credhub-cli/credhub/credentials"
	"code.cloudfoundry.org/credhub-cli/credhub/credentials/values"
)

//go:generate counterfeiter -o ./credhub_fakes/credhub_auth_fake.go . CredhubAuth
type CredhubAuth interface {
	UaaClientCredentials(clientId, clientSecret string) auth.Builder
}

type CredhubAuthShim struct {
}

func (c *CredhubAuthShim) UaaClientCredentials(clientId, clientSecret string) auth.Builder {
	return auth.UaaClientCredentials(clientId, clientSecret)
}

//go:generate counterfeiter -o ./credhub_fakes/credhub_fake.go . Credhub
type Credhub interface {
	SetJSON(name string, value values.JSON) (credentials.JSON, error)
	GetLatestJSON(name string) (credentials.JSON, error)
	SetValue(name string, value values.Value) (credentials.Value, error)
	GetLatestValue(name string) (credentials.Value, error)
	FindByPath(path string) (credentials.FindResults, error)
	Delete(name string) error
}

type CredhubShim struct {
	delegate *credhub.CredHub
}

func NewCredhubShim(
	url string,
	caCert string,
	clientID string,
	clientSecret string,
	uaaCACert string,
	authShim CredhubAuth,
) (Credhub, error) {
	var delegate *credhub.CredHub
	var err error
	auth := credhub.Auth(authShim.UaaClientCredentials(clientID, clientSecret))

	caCerts := []string{}
	if caCert != "" {
		caCerts = append(caCerts, caCert)
	}

	if uaaCACert != "" {
		caCerts = append(caCerts, uaaCACert)
	}

	if len(caCerts) > 0 {
		delegate, err = credhub.New(url, credhub.CaCerts(caCerts...), auth)
	} else {
		delegate, err = credhub.New(url, auth)
	}

	if err != nil {
		return nil, err
	}

	return &CredhubShim{
		delegate: delegate,
	}, nil
}

func (ch *CredhubShim) SetJSON(name string, value values.JSON) (credentials.JSON, error) {
	return ch.delegate.SetJSON(name, value)
}

func (ch *CredhubShim) GetLatestJSON(name string) (credentials.JSON, error) {
	return ch.delegate.GetLatestJSON(name)
}

func (ch *CredhubShim) SetValue(name string, value values.Value) (credentials.Value, error) {
	return ch.delegate.SetValue(name, value)
}

func (ch *CredhubShim) GetLatestValue(name string) (credentials.Value, error) {
	return ch.delegate.GetLatestValue(name)
}

func (ch *CredhubShim) FindByPath(path string) (credentials.FindResults, error) {
	return ch.delegate.FindByPath(path)
}

func (ch *CredhubShim) Delete(name string) error {
	return ch.delegate.Delete(name)
}