Validate type earlier for aws_directory_service_directory

Also DRY it up a little

builtin/providers/aws/resource_aws_directory_service_directory.go

@@ -12,6 +12,11 @@ import (
 	"github.com/hashicorp/terraform/helper/resource"
 )
 
+var directoryCreationFuncs = map[string]func(*directoryservice.DirectoryService, *schema.ResourceData) (string, error){
+	"SimpleAD":    createSimpleDirectoryService,
+	"MicrosoftAD": createActiveDirectoryService,
+}
+
 func resourceAwsDirectoryServiceDirectory() *schema.Resource {
 	return &schema.Resource{
 		Create: resourceAwsDirectoryServiceDirectoryCreate,
@@ -92,6 +97,17 @@ func resourceAwsDirectoryServiceDirectory() *schema.Resource {
 				Optional: true,
 				Default:  "SimpleAD",
 				ForceNew: true,
+				ValidateFunc: func(v interface{}, k string) (ws []string, es []error) {
+					validTypes := []string{"SimpleAD", "MicrosoftAD"}
+					value := v.(string)
+					for validType, _ := range directoryCreationFuncs {
+						if validType == value {
+							return
+						}
+					}
+					es = append(es, fmt.Errorf("%q must be one of %q", k, validTypes))
+					return
+				},
 			},
 		},
 	}
@@ -184,19 +200,13 @@ func createActiveDirectoryService(dsconn *directoryservice.DirectoryService, d *
 func resourceAwsDirectoryServiceDirectoryCreate(d *schema.ResourceData, meta interface{}) error {
 	dsconn := meta.(*AWSClient).dsconn
 
-	var (
+	creationFunc, ok := directoryCreationFuncs[d.Get("type").(string)]
-		directoryId string
+	if !ok {
-		err         error
+		// Shouldn't happen as this is validated above
-	)
-
-	switch d.Get("type").(string) {
-	case "SimpleAD":
-		directoryId, err = createSimpleDirectoryService(dsconn, d)
-	case "MicrosoftAD":
-		directoryId, err = createActiveDirectoryService(dsconn, d)
-	default:
 		return fmt.Errorf("Unsupported directory type: %s", d.Get("type"))
 	}
+
+	directoryId, err := creationFunc(dsconn, d)
 	if err != nil {
 		return err
 	}