feature_flag.go

package ccv3

import (
	"bytes"

	"encoding/json"

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

// FeatureFlag represents a Cloud Controller V3 Feature Flag.
type FeatureFlag struct {
	Name    string `json:"name"`
	Enabled bool   `json:"enabled"`
}

func (f FeatureFlag) MarshalJSON() ([]byte, error) {
	var ccBodyFlag struct {
		Enabled bool `json:"enabled"`
	}

	ccBodyFlag.Enabled = f.Enabled

	return json.Marshal(ccBodyFlag)
}

func (client *Client) GetFeatureFlag(flagName string) (FeatureFlag, Warnings, error) {
	request, err := client.newHTTPRequest(requestOptions{
		RequestName: internal.GetFeatureFlagRequest,
		URIParams:   map[string]string{"name": flagName},
	})
	if err != nil {
		return FeatureFlag{}, nil, err
	}
	var ccFlag FeatureFlag
	response := cloudcontroller.Response{
		DecodeJSONResponseInto: &ccFlag,
	}

	err = client.connection.Make(request, &response)

	if err != nil {
		return FeatureFlag{}, response.Warnings, err
	}
	return ccFlag, response.Warnings, nil
}

// GetFeatureFlags lists feature flags.
func (client *Client) GetFeatureFlags() ([]FeatureFlag, Warnings, error) {
	request, err := client.newHTTPRequest(requestOptions{
		RequestName: internal.GetFeatureFlagsRequest,
	})

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

	var fullFeatureFlagList []FeatureFlag
	warnings, err := client.paginate(request, FeatureFlag{}, func(item interface{}) error {
		if featureFlag, ok := item.(FeatureFlag); ok {
			fullFeatureFlagList = append(fullFeatureFlagList, featureFlag)
		} else {
			return ccerror.UnknownObjectInListError{
				Expected:   FeatureFlag{},
				Unexpected: item,
			}
		}
		return nil
	})

	return fullFeatureFlagList, warnings, err

}

func (client *Client) UpdateFeatureFlag(flag FeatureFlag) (FeatureFlag, Warnings, error) {
	bodyBytes, err := json.Marshal(flag)
	if err != nil {
		return FeatureFlag{}, nil, err
	}

	request, err := client.newHTTPRequest(requestOptions{
		RequestName: internal.PatchFeatureFlagRequest,
		URIParams:   map[string]string{"name": flag.Name},
		Body:        bytes.NewReader(bodyBytes),
	})

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

	var ccFlag FeatureFlag
	response := cloudcontroller.Response{
		DecodeJSONResponseInto: &ccFlag,
	}

	err = client.connection.Make(request, &response)

	return ccFlag, response.Warnings, err
}