github.com/mook-as/cf-cli@v7.0.0-beta.28.0.20200120190804-b91c115fae48+incompatible/api/cloudcontroller/ccv3/service_broker.go

package ccv3

import (
	"bytes"
	"encoding/json"
	"errors"

	"code.cloudfoundry.org/cli/api/cloudcontroller"
	"code.cloudfoundry.org/cli/api/cloudcontroller/ccerror"
	"code.cloudfoundry.org/cli/api/cloudcontroller/ccv3/constant"
	"code.cloudfoundry.org/cli/api/cloudcontroller/ccv3/internal"
)

// ServiceBroker represents Service Broker data
type ServiceBroker struct {
	// GUID is a unique service broker identifier.
	GUID string
	// Name is the name of the service broker.
	Name string
	// URL is the url of the service broker.
	URL string
	// Status is the state of the service broker.
	Status string
}

type ServiceBrokerModel struct {
	// Name is the name of the service broker.
	Name string
	// URL is the url of the service broker.
	URL string
	// Username is the Basic Auth username for the service broker.
	Username string
	// Password is the Basic Auth password for the service broker.
	Password string
	// Space GUID for the space that the broker is in. Empty when not a space-scoped service broker.
	SpaceGUID string
}

// serviceBrokerRequest represents a Cloud Controller V3 Service Broker (when creating and updating).
type serviceBrokerRequest struct {
	// GUID is a unique service broker identifier.
	GUID string `json:"guid,omitempty"`
	// Name is the name of the service broker.
	Name string `json:"name,omitempty"`
	// URL is the url of the service broker.
	URL string `json:"url,omitempty"`
	// Authentication contains the authentication for authenticating with the service broker.
	Authentication *serviceBrokerAuthentication `json:"authentication,omitempty"`
	// This is the relationship for the space GUID
	Relationships *serviceBrokerRelationships `json:"relationships,omitempty"`
}

// serviceBrokerResponse represents a Cloud Controller V3 Service Broker (when reading).
type serviceBrokerResponse struct {
	// GUID is a unique service broker identifier.
	GUID string `json:"guid,omitempty"`
	// Name is the name of the service broker.
	Name string `json:"name"`
	// URL is the url of the service broker.
	URL string `json:"url"`
	// Status is the state of the service broker.
	Status string `json:"status,omitempty"`
	// Authentication contains the authentication for authenticating with the service broker.
	Authentication serviceBrokerAuthentication `json:"authentication"`
	// This is the relationship for the space GUID
	Relationships *serviceBrokerRelationships `json:"relationships,omitempty"`
}

// serviceBrokerAuthentication represents a data structure for the Credentials
// of V3 Service Broker.
type serviceBrokerAuthentication struct {
	// Type is the type of authentication for the service broker, e.g. "basic"
	Type constant.ServiceBrokerCredentialsType `json:"type"`
	// Data is the authentication data of the service broker of a particular type.
	Credentials serviceBrokerBasicAuthCredentials `json:"credentials"`
}

// serviceBrokerBasicAuthCredentials represents a data structure for the Credentials Data
// of V3 Service Broker Credentials.
type serviceBrokerBasicAuthCredentials struct {
	// Username is the Basic Auth username for the service broker.
	Username string `json:"username"`
	// Password is the Basic Auth password for the service broker.
	Password string `json:"password"`
}

// serviceBrokerRelationships represents a data structure for the relationships data
// of V3 Service Broker Relationships.
type serviceBrokerRelationships struct {
	// Space represents the space that a space-scoped broker is in
	Space serviceBrokerRelationshipsSpace `json:"space"`
}

// serviceBrokerRelationshipsSpace represents a data structure for the relationships space data
// of V3 Service Broker Relationships.
type serviceBrokerRelationshipsSpace struct {
	// Data holds the space GUID object
	Data serviceBrokerRelationshipsSpaceData `json:"data"`
}

// serviceBrokerRelationshipsSpaceData represents a data structure for the relationships space GUID data
// of V3 Service Broker Relationships.
type serviceBrokerRelationshipsSpaceData struct {
	// GUID is the space guid associated with a space-scoped broker
	GUID string `json:"guid"`
}

// CreateServiceBroker registers a new service broker.
func (client *Client) CreateServiceBroker(serviceBroker ServiceBrokerModel) (JobURL, Warnings, error) {
	bodyBytes, err := json.Marshal(newServiceBroker(serviceBroker))
	if err != nil {
		return "", nil, err
	}

	request, err := client.newHTTPRequest(requestOptions{
		RequestName: internal.PostServiceBrokerRequest,
		Body:        bytes.NewReader(bodyBytes),
	})
	if err != nil {
		return "", nil, err
	}

	response := cloudcontroller.Response{}
	err = client.connection.Make(request, &response)
	return JobURL(response.ResourceLocationURL), response.Warnings, err
}

// DeleteServiceBroker deletes a named service broker
func (client *Client) DeleteServiceBroker(serviceBrokerGUID string) (JobURL, Warnings, error) {
	request, err := client.newHTTPRequest(requestOptions{
		RequestName: internal.DeleteServiceBrokerRequest,
		URIParams: map[string]string{
			"service_broker_guid": serviceBrokerGUID,
		},
	})

	if err != nil {
		return "", nil, err
	}

	response := cloudcontroller.Response{}
	err = client.connection.Make(request, &response)
	return JobURL(response.ResourceLocationURL), response.Warnings, err
}

// GetServiceBrokers lists service brokers.
func (client *Client) GetServiceBrokers() ([]ServiceBroker, Warnings, error) {
	request, err := client.newHTTPRequest(requestOptions{
		RequestName: internal.GetServiceBrokersRequest,
	})
	if err != nil {
		return nil, nil, err
	}

	var fullList []ServiceBroker
	warnings, err := client.paginate(request, serviceBrokerResponse{}, func(item interface{}) error {
		if serviceBroker, ok := item.(serviceBrokerResponse); ok {
			fullList = append(fullList, extractServiceBrokerData(serviceBroker))
		} else {
			return ccerror.UnknownObjectInListError{
				Expected:   serviceBrokerResponse{},
				Unexpected: item,
			}
		}
		return nil
	})

	return fullList, warnings, err
}

// UpdateServiceBroker updates an existing service broker.
func (client *Client) UpdateServiceBroker(serviceBrokerGUID string, serviceBroker ServiceBrokerModel) (JobURL, Warnings, error) {
	brokerUpdateRequest, err := newUpdateServiceBroker(serviceBroker)
	if err != nil {
		return "", nil, err
	}

	bodyBytes, err := json.Marshal(brokerUpdateRequest)
	if err != nil {
		return "", nil, err
	}

	request, err := client.newHTTPRequest(requestOptions{
		RequestName: internal.PatchServiceBrokerRequest,
		URIParams: map[string]string{
			"service_broker_guid": serviceBrokerGUID,
		},
		Body: bytes.NewReader(bodyBytes),
	})
	if err != nil {
		return "", nil, err
	}

	response := cloudcontroller.Response{}
	err = client.connection.Make(request, &response)
	return JobURL(response.ResourceLocationURL), response.Warnings, err
}

func newServiceBroker(serviceBroker ServiceBrokerModel) serviceBrokerRequest {
	serviceBrokerRequest := serviceBrokerRequest{
		Name: serviceBroker.Name,
		URL:  serviceBroker.URL,
		Authentication: &serviceBrokerAuthentication{
			Type: constant.BasicCredentials,
			Credentials: serviceBrokerBasicAuthCredentials{
				Username: serviceBroker.Username,
				Password: serviceBroker.Password,
			},
		},
	}

	if serviceBroker.SpaceGUID != "" {
		serviceBrokerRequest.Relationships = &serviceBrokerRelationships{
			Space: serviceBrokerRelationshipsSpace{
				Data: serviceBrokerRelationshipsSpaceData{
					GUID: serviceBroker.SpaceGUID,
				},
			},
		}
	}

	return serviceBrokerRequest
}

func newUpdateServiceBroker(serviceBroker ServiceBrokerModel) (serviceBrokerRequest, error) {
	name := serviceBroker.Name
	username := serviceBroker.Username
	password := serviceBroker.Password
	brokerURL := serviceBroker.URL
	if (username == "" && password != "") || (username != "" && password == "") {
		return serviceBrokerRequest{}, errors.New("boom!") // TODO: fix this
	}
	request := serviceBrokerRequest{
		Name: name,
		URL:  brokerURL,
	}

	if username != "" && password != "" {
		request.Authentication = &serviceBrokerAuthentication{
			Type: constant.BasicCredentials,
			Credentials: serviceBrokerBasicAuthCredentials{
				Username: username,
				Password: password,
			},
		}
	}

	return request, nil
}

func extractServiceBrokerData(response serviceBrokerResponse) ServiceBroker {
	return ServiceBroker{
		Name:   response.Name,
		URL:    response.URL,
		GUID:   response.GUID,
		Status: response.Status,
	}
}