Added support for appsync_graphql_api schema dn appsync_resolver

aws/provider.go

@@ -316,6 +316,7 @@ func Provider() terraform.ResourceProvider {
 			"aws_appsync_api_key":                              resourceAwsAppsyncApiKey(),
 			"aws_appsync_datasource":                           resourceAwsAppsyncDatasource(),
 			"aws_appsync_graphql_api":                          resourceAwsAppsyncGraphqlApi(),
+			"aws_appsync_resolver":                             resourceAwsAppsyncResolver(),
 			"aws_athena_database":                              resourceAwsAthenaDatabase(),
 			"aws_athena_named_query":                           resourceAwsAthenaNamedQuery(),
 			"aws_autoscaling_attachment":                       resourceAwsAutoscalingAttachment(),

aws/resource_aws_appsync_graphql_api.go

@@ -2,8 +2,10 @@ package aws
 
 import (
 	"fmt"
+	"github.com/hashicorp/terraform/helper/resource"
 	"log"
 	"regexp"
+	"time"
 
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/service/appsync"
@@ -130,6 +132,10 @@ func resourceAwsAppsyncGraphqlApi() *schema.Resource {
 				Computed: true,
 				Elem:     &schema.Schema{Type: schema.TypeString},
 			},
+			"schema": {
+				Type:     schema.TypeString,
+				Optional: true,
+			},
 		},
 	}
 }
@@ -161,9 +167,45 @@ func resourceAwsAppsyncGraphqlApiCreate(d *schema.ResourceData, meta interface{}
 
 	d.SetId(*resp.GraphqlApi.ApiId)
 
+	if v, ok := d.GetOk("schema"); ok {
+		if err := startSchemaCreationAndWaitForItToBeActive(d.Id(), v.(string), d.Timeout(schema.TimeoutCreate), conn); err != nil {
+			return err
+		}
+	}
+
 	return resourceAwsAppsyncGraphqlApiRead(d, meta)
 }
 
+func startSchemaCreationAndWaitForItToBeActive(apiID string, schema string, timeout time.Duration, conn *appsync.AppSync) error {
+	input := &appsync.StartSchemaCreationInput{
+		ApiId:      aws.String(apiID),
+		Definition: []byte(schema),
+	}
+
+	if _, err := conn.StartSchemaCreation(input); err != nil {
+		return err
+	}
+
+	stateConf := &resource.StateChangeConf{
+		Pending: []string{appsync.SchemaStatusProcessing},
+		Target:  []string{"SUCCESS"}, // should be appsync.SchemaStatusActive . I think this is a problem in documentation: https://docs.aws.amazon.com/appsync/latest/APIReference/API_GetSchemaCreationStatus.html
+		Timeout: timeout,
+		Refresh: func() (interface{}, string, error) {
+			result, err := conn.GetSchemaCreationStatus(&appsync.GetSchemaCreationStatusInput{
+				ApiId: aws.String(apiID),
+			})
+			if err != nil {
+				return 42, "", err
+			}
+
+			return result, *result.Status, nil
+		},
+	}
+	_, err := stateConf.WaitForState()
+
+	return err
+}
+
 func resourceAwsAppsyncGraphqlApiRead(d *schema.ResourceData, meta interface{}) error {
 	conn := meta.(*AWSClient).appsyncconn
 
@@ -230,6 +272,13 @@ func resourceAwsAppsyncGraphqlApiUpdate(d *schema.ResourceData, meta interface{}
 		return err
 	}
 
+	if d.HasChange("schema") {
+		if v, ok := d.GetOk("schema"); ok {
+			if err := startSchemaCreationAndWaitForItToBeActive(d.Id(), v.(string), d.Timeout(schema.TimeoutUpdate), conn); err != nil {
+				return err
+			}
+		}
+	}
 	return resourceAwsAppsyncGraphqlApiRead(d, meta)
 }
 

aws/resource_aws_appsync_graphql_api_test.go

@@ -521,6 +521,39 @@ func TestAccAWSAppsyncGraphqlApi_UserPoolConfig_DefaultAction(t *testing.T) {
 	})
 }
 
+func TestAccAWSAppsyncGraphqlApi_Schema(t *testing.T) {
+	rName := acctest.RandomWithPrefix("tf-acc-test")
+	resourceName := "aws_appsync_graphql_api.test"
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckAwsAppsyncGraphqlApiDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccAppsyncGraphqlApiConfig_Schema(rName),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckAwsAppsyncGraphqlApiExists(resourceName),
+					testAccCheckAwsAppsyncTypeExists(resourceName, "Post"),
+				),
+			},
+			{
+				ResourceName:            resourceName,
+				ImportState:             true,
+				ImportStateVerify:       true,
+				ImportStateVerifyIgnore: []string{"schema"},
+			},
+			{
+				Config: testAccAppsyncGraphqlApiConfig_SchemaUpdate(rName),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckAwsAppsyncGraphqlApiExists(resourceName),
+					testAccCheckAwsAppsyncTypeExists(resourceName, "PostV2"),
+				),
+			},
+		},
+	})
+}
+
 func testAccCheckAwsAppsyncGraphqlApiDestroy(s *terraform.State) error {
 	conn := testAccProvider.Meta().(*AWSClient).appsyncconn
 	for _, rs := range s.RootModule().Resources {
@@ -565,6 +598,30 @@ func testAccCheckAwsAppsyncGraphqlApiExists(name string) resource.TestCheckFunc
 	}
 }
 
+func testAccCheckAwsAppsyncTypeExists(name, typeName string) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+		rs, ok := s.RootModule().Resources[name]
+		if !ok {
+			return fmt.Errorf("Not found: %s", name)
+		}
+
+		conn := testAccProvider.Meta().(*AWSClient).appsyncconn
+
+		input := &appsync.GetTypeInput{
+			ApiId:    aws.String(rs.Primary.ID),
+			TypeName: aws.String(typeName),
+			Format:   aws.String(appsync.OutputTypeSdl),
+		}
+
+		_, err := conn.GetType(input)
+		if err != nil {
+			return err
+		}
+
+		return nil
+	}
+}
+
 func testAccAppsyncGraphqlApiConfig_AuthenticationType(rName, authenticationType string) string {
 	return fmt.Sprintf(`
 resource "aws_appsync_graphql_api" "test" {
@@ -705,3 +762,23 @@ resource "aws_appsync_graphql_api" "test" {
 }
 `, rName, rName, defaultAction)
 }
+
+func testAccAppsyncGraphqlApiConfig_Schema(rName string) string {
+	return fmt.Sprintf(`
+resource "aws_appsync_graphql_api" "test" {
+  authentication_type = "API_KEY"
+  name                = %q
+  schema              = "type Mutation {\n\tputPost(id: ID!, title: String!): Post\n}\n\ntype Post {\n\tid: ID!\n\ttitle: String!\n}\n\ntype Query {\n\tsinglePost(id: ID!): Post\n}\n\nschema {\n\tquery: Query\n\tmutation: Mutation\n\n}\n"
+}
+`, rName)
+}
+
+func testAccAppsyncGraphqlApiConfig_SchemaUpdate(rName string) string {
+	return fmt.Sprintf(`
+resource "aws_appsync_graphql_api" "test" {
+  authentication_type = "API_KEY"
+  name                = %q
+  schema              = "type Mutation {\n\tputPostV2(id: ID!, title: String!): PostV2\n}\n\ntype PostV2 {\n\tid: ID!\n\ttitle: String!\n}\n\ntype Query {\n\tsinglePostV2(id: ID!): PostV2\n}\n\nschema {\n\tquery: Query\n\tmutation: Mutation\n\n}\n"
+}
+`, rName)
+}

aws/resource_aws_appsync_resolver.go

@@ -0,0 +1,160 @@
+package aws
+
+import (
+	"fmt"
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/service/appsync"
+	"github.com/hashicorp/terraform/helper/schema"
+	"strings"
+)
+
+func resourceAwsAppsyncResolver() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceAwsAppsyncResolverCreate,
+		Read:   resourceAwsAppsyncResolverRead,
+		Update: resourceAwsAppsyncResolverUpdate,
+		Delete: resourceAwsAppsyncResolverDelete,
+
+		Importer: &schema.ResourceImporter{
+			State: schema.ImportStatePassthrough,
+		},
+
+		Schema: map[string]*schema.Schema{
+			"api_id": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: true,
+			},
+			"type": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: true,
+			},
+			"field": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: true,
+			},
+			"data_source": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"request_template": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"response_template": {
+				Type:     schema.TypeString,
+				Required: true, // documentation bug, the api returns 400 if this is not specified.
+			},
+			"arn": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+		},
+	}
+}
+
+func resourceAwsAppsyncResolverCreate(d *schema.ResourceData, meta interface{}) error {
+	conn := meta.(*AWSClient).appsyncconn
+
+	input := &appsync.CreateResolverInput{
+		ApiId:                   aws.String(d.Get("api_id").(string)),
+		DataSourceName:          aws.String(d.Get("data_source").(string)),
+		TypeName:                aws.String(d.Get("type").(string)),
+		FieldName:               aws.String(d.Get("field").(string)),
+		RequestMappingTemplate:  aws.String(d.Get("request_template").(string)),
+		ResponseMappingTemplate: aws.String(d.Get("response_template").(string)),
+	}
+
+	_, err := conn.CreateResolver(input)
+	if err != nil {
+		return err
+	}
+
+	d.SetId(d.Get("api_id").(string) + "-" + d.Get("type").(string) + "-" + d.Get("field").(string))
+
+	return resourceAwsAppsyncResolverRead(d, meta)
+}
+
+func resourceAwsAppsyncResolverRead(d *schema.ResourceData, meta interface{}) error {
+	conn := meta.(*AWSClient).appsyncconn
+
+	apiID, typeName, fieldName, err := decodeAppsyncResolverID(d.Id())
+
+	if err != nil {
+		return err
+	}
+
+	input := &appsync.GetResolverInput{
+		ApiId:     aws.String(apiID),
+		TypeName:  aws.String(typeName),
+		FieldName: aws.String(fieldName),
+	}
+
+	resp, err := conn.GetResolver(input)
+	if err != nil {
+		return err
+	}
+
+	d.Set("api_id", apiID)
+	d.Set("arn", resp.Resolver.ResolverArn)
+	d.Set("type", resp.Resolver.TypeName)
+	d.Set("field", resp.Resolver.FieldName)
+	d.Set("data_source", resp.Resolver.DataSourceName)
+	d.Set("request_template", resp.Resolver.RequestMappingTemplate)
+	d.Set("response_template", resp.Resolver.ResponseMappingTemplate)
+
+	return nil
+}
+
+func resourceAwsAppsyncResolverUpdate(d *schema.ResourceData, meta interface{}) error {
+	conn := meta.(*AWSClient).appsyncconn
+
+	input := &appsync.UpdateResolverInput{
+		ApiId:                   aws.String(d.Get("api_id").(string)),
+		DataSourceName:          aws.String(d.Get("data_source").(string)),
+		FieldName:               aws.String(d.Get("field").(string)),
+		TypeName:                aws.String(d.Get("type").(string)),
+		RequestMappingTemplate:  aws.String(d.Get("request_template").(string)),
+		ResponseMappingTemplate: aws.String(d.Get("response_template").(string)),
+	}
+
+	_, err := conn.UpdateResolver(input)
+	if err != nil {
+		return err
+	}
+
+	return resourceAwsAppsyncResolverRead(d, meta)
+}
+
+func resourceAwsAppsyncResolverDelete(d *schema.ResourceData, meta interface{}) error {
+	conn := meta.(*AWSClient).appsyncconn
+
+	apiID, typeName, fieldName, err := decodeAppsyncResolverID(d.Id())
+
+	if err != nil {
+		return err
+	}
+
+	input := &appsync.DeleteResolverInput{
+		ApiId:     aws.String(apiID),
+		TypeName:  aws.String(typeName),
+		FieldName: aws.String(fieldName),
+	}
+
+	_, err = conn.DeleteResolver(input)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func decodeAppsyncResolverID(id string) (string, string, string, error) {
+	idParts := strings.SplitN(id, "-", 3)
+	if len(idParts) != 3 {
+		return "", "", "", fmt.Errorf("expected ID in format ApiID-TypeName-FieldName, received: %s", id)
+	}
+	return idParts[0], idParts[1], idParts[2], nil
+}