Add CloudFormation resource type configuration support

Author: omkhegde

README.md

@@ -45,8 +45,6 @@ pre-commit run --all-files
 pre-commit run pytest-local
 ```
 
-Use `./generate-examples.sh` to run install `cloudformation-cli-go-plugin` locally and run `cfn generate` in each example.
-
 Getting started
 ---------------
 

cfn/cfn.go

@@ -130,6 +130,7 @@ func makeEventFunc(h Handler) eventFunc {
 			credentials.SessionFromCredentialsProvider(&event.RequestData.CallerCredentials),
 			event.RequestData.PreviousResourceProperties,
 			event.RequestData.ResourceProperties,
+			event.RequestData.TypeConfiguration,
 		)
 		p := invoke(handlerFn, request, m, event.Action)
 		r, err := newResponse(&p, event.BearerToken)

cfn/event.go

@@ -36,6 +36,7 @@ type requestData struct {
 	ProviderLogGroupName       string                                        `json:"providerLogGroupName"`
 	StackTags                  tags                                          `json:"stackTags"`
 	SystemTags                 tags                                          `json:"systemTags"`
+	TypeConfiguration          json.RawMessage                               `json:"typeConfiguration"`
 }
 
 // validateEvent ensures the event struct generated from the Lambda SDK is correct

cfn/handler/event.go

@@ -34,11 +34,8 @@ type ProgressEvent struct {
 	// and by CREATE/UPDATE/DELETE for final response validation/confirmation
 	ResourceModel interface{} `json:"resourceModel,omitempty"`
 
-	// ResourceModels is the output resource instances populated by a LIST for
-	// synchronous results. ResourceModels must be returned by LIST so it's
-	// always included in the response. When ResourceModels is not set, null is
-	// returned.
-	ResourceModels []interface{} `json:"resourceModels"`
+	// ResourceModels is the output resource instances populated by a LIST for synchronous results
+	ResourceModels []interface{} `json:"resourceModels,omitempty"`
 
 	// NextToken is the token used to request additional pages of resources for a LIST operation
 	NextToken string `json:"nextToken,omitempty"`

cfn/handler/event_test.go

@@ -17,7 +17,7 @@ func TestNewRequest(t *testing.T) {
 		prev := Props{}
 		curr := Props{}
 
-		req := NewRequest("foo", nil, rctx, nil, []byte(`{"color": "red"}`), []byte(`{"color": "green"}`))
+		req := NewRequest("foo", nil, rctx, nil, []byte(`{"color": "red"}`), []byte(`{"color": "green"}`), []byte(``))
 
 		if err := req.UnmarshalPrevious(&prev); err != nil {
 			t.Fatalf("Unable to unmarshal props: %v", err)
@@ -43,7 +43,7 @@ func TestNewRequest(t *testing.T) {
 
 	t.Run("ResourceProps", func(t *testing.T) {
 		t.Run("Invalid Body", func(t *testing.T) {
-			req := NewRequest("foo", nil, rctx, nil, []byte(``), []byte(``))
+			req := NewRequest("foo", nil, rctx, nil, []byte(``), []byte(``), []byte(``))
 
 			invalid := struct {
 				Color *int `json:"color"`
@@ -61,7 +61,7 @@ func TestNewRequest(t *testing.T) {
 		})
 
 		t.Run("Invalid Marshal", func(t *testing.T) {
-			req := NewRequest("foo", nil, rctx, nil, []byte(`{"color": "ref"}`), []byte(`---BAD JSON---`))
+			req := NewRequest("foo", nil, rctx, nil, []byte(`{"color": "ref"}`), []byte(`---BAD JSON---`), []byte(``))
 
 			var invalid Props
 
@@ -79,7 +79,7 @@ func TestNewRequest(t *testing.T) {
 
 	t.Run("PreviousResourceProps", func(t *testing.T) {
 		t.Run("Invalid Marshal", func(t *testing.T) {
-			req := NewRequest("foo", nil, rctx, nil, []byte(`---BAD JSON---`), []byte(`{"color": "green"}`))
+			req := NewRequest("foo", nil, rctx, nil, []byte(`---BAD JSON---`), []byte(`{"color": "green"}`), []byte(``))
 
 			var invalid Props
 

cfn/handler/handler_test.go

@@ -36,6 +36,7 @@ type Request struct {
 
 	previousResourcePropertiesBody []byte
 	resourcePropertiesBody         []byte
+	typeConfiguration              []byte
 }
 
 // RequestContext represents information about the current
@@ -61,14 +62,15 @@ type RequestContext struct {
 }
 
 // NewRequest returns a new Request based on the provided parameters
-func NewRequest(id string, ctx map[string]interface{}, requestCTX RequestContext, sess *session.Session, previousBody, body []byte) Request {
+func NewRequest(id string, ctx map[string]interface{}, requestCTX RequestContext, sess *session.Session, previousBody, body []byte, config []byte) Request {
 	return Request{
 		LogicalResourceID:              id,
 		CallbackContext:                ctx,
 		Session:                        sess,
 		previousResourcePropertiesBody: previousBody,
 		resourcePropertiesBody:         body,
 		RequestContext:                 requestCTX,
+		typeConfiguration:              config,
 	}
 }
 
@@ -99,3 +101,17 @@ func (r *Request) Unmarshal(v interface{}) error {
 
 	return nil
 }
+
+// UnmarshalType populates the provided interface
+// with the current resource type configuration
+func (r *Request) UnmarshalType(v interface{}) error {
+	if len(r.resourcePropertiesBody) == 0 {
+		return cfnerr.New(bodyEmptyError, "Body is empty", nil)
+	}
+
+	if err := encoding.Unmarshal(r.typeConfiguration, v); err != nil {
+		return cfnerr.New(marshalingError, "Unable to convert type", err)
+	}
+
+	return nil
+}

cfn/handler/request.go

@@ -31,11 +31,8 @@ type response struct {
 	//passed back to CloudFormation
 	BearerToken string `json:"bearerToken,omitempty"`
 
-	// ResourceModels is the output resource instances populated by a LIST for
-	// synchronous results. ResourceModels must be returned by LIST so it's
-	// always included in the response. When ResourceModels is not set, null is
-	// returned.
-	ResourceModels []interface{} `json:"resourceModels"`
+	// ResourceModels is the output resource instances populated by a LIST for synchronous results
+	ResourceModels []interface{} `json:"resourceModels,omitempty"`
 
 	// NextToken the token used to request additional pages of resources for a LIST operation
 	NextToken string `json:"nextToken,omitempty"`

cfn/response.go

@@ -12,66 +12,31 @@ import (
 	"github.com/aws-cloudformation/cloudformation-cli-go-plugin/cfn/handler"
 )
 
-func TestResponseMarshalJSON(t *testing.T) {
+func TestMarshalJSON(t *testing.T) {
 	type Model struct {
 		Name    *encoding.String
 		Version *encoding.Float
 	}
 
-	for _, tt := range []struct {
-		name     string
-		response response
-		expected string
-	}{
-		{
-			name: "updated failed",
-			response: response{
-				Message:         "foo",
-				OperationStatus: handler.Failed,
-				ResourceModel: Model{
-					Name:    encoding.NewString("Douglas"),
-					Version: encoding.NewFloat(42.1),
-				},
-				ErrorCode:   cloudformation.HandlerErrorCodeNotUpdatable,
-				BearerToken: "xyzzy",
-			},
-			expected: `{"message":"foo","status":"FAILED","resourceModel":{"Name":"Douglas","Version":"42.1"},"errorCode":"NotUpdatable","bearerToken":"xyzzy","resourceModels":null}`,
+	r := response{
+		Message:         "foo",
+		OperationStatus: handler.Success,
+		ResourceModel: Model{
+			Name:    encoding.NewString("Douglas"),
+			Version: encoding.NewFloat(42.1),
 		},
-		{
-			name: "list with 1 result",
-			response: response{
-				OperationStatus: handler.Success,
-				ResourceModels: []interface{}{
-					Model{
-						Name:    encoding.NewString("Douglas"),
-						Version: encoding.NewFloat(42.1),
-					},
-				},
-				BearerToken: "xyzzy",
-			},
-			expected: `{"status":"SUCCESS","bearerToken":"xyzzy","resourceModels":[{"Name":"Douglas","Version":"42.1"}]}`,
-		},
-		{
-			name: "list with empty array",
-			response: response{
-				OperationStatus: handler.Success,
-				ResourceModels:  []interface{}{},
-				BearerToken:     "xyzzy",
-			},
-			expected: `{"status":"SUCCESS","bearerToken":"xyzzy","resourceModels":[]}`,
-		},
-	} {
-		t.Run(tt.name, func(t *testing.T) {
+		ErrorCode:   cloudformation.HandlerErrorCodeNotUpdatable,
+		BearerToken: "xyzzy",
+	}
 
-			actual, err := json.Marshal(tt.response)
-			if err != nil {
-				t.Errorf("Unexpected error marshaling response JSON: %s", err)
-			}
+	expected := `{"message":"foo","status":"SUCCESS","resourceModel":{"Name":"Douglas","Version":"42.1"},"errorCode":"NotUpdatable","bearerToken":"xyzzy"}`
 
-			if diff := cmp.Diff(string(actual), tt.expected); diff != "" {
-				t.Errorf(diff)
-			}
-		})
+	actual, err := json.Marshal(r)
+	if err != nil {
+		t.Errorf("Unexpected error marshaling response JSON: %s", err)
 	}
 
+	if diff := cmp.Diff(string(actual), expected); diff != "" {
+		t.Errorf(diff)
+	}
 }

cfn/response_test.go

@@ -3,7 +3,6 @@ module github.com/aws-cloudformation/cloudformation-cli-go-plugin
 go 1.13
 
 require (
-	github.com/avast/retry-go v2.6.0+incompatible
 	github.com/aws/aws-lambda-go v1.13.3
 	github.com/aws/aws-sdk-go v1.25.37
 	github.com/google/go-cmp v0.3.1