package azurerm import ( "fmt" "log" "net/http" "regexp" "strings" "time" "github.com/Azure/azure-sdk-for-go/arm/storage" "github.com/hashicorp/terraform/helper/resource" "github.com/hashicorp/terraform/helper/schema" "github.com/hashicorp/terraform/helper/signalwrapper" ) func resourceArmStorageAccount() *schema.Resource { return &schema.Resource{ Create: resourceArmStorageAccountCreate, Read: resourceArmStorageAccountRead, Update: resourceArmStorageAccountUpdate, Delete: resourceArmStorageAccountDelete, Importer: &schema.ResourceImporter{ State: schema.ImportStatePassthrough, }, Schema: map[string]*schema.Schema{ "name": { Type: schema.TypeString, Required: true, ForceNew: true, ValidateFunc: validateArmStorageAccountName, }, "resource_group_name": { Type: schema.TypeString, Required: true, ForceNew: true, }, "location": { Type: schema.TypeString, Required: true, ForceNew: true, StateFunc: azureRMNormalizeLocation, }, "account_type": { Type: schema.TypeString, Required: true, ValidateFunc: validateArmStorageAccountType, }, "primary_location": { Type: schema.TypeString, Computed: true, }, "secondary_location": { Type: schema.TypeString, Computed: true, }, "primary_blob_endpoint": { Type: schema.TypeString, Computed: true, }, "secondary_blob_endpoint": { Type: schema.TypeString, Computed: true, }, "primary_queue_endpoint": { Type: schema.TypeString, Computed: true, }, "secondary_queue_endpoint": { Type: schema.TypeString, Computed: true, }, "primary_table_endpoint": { Type: schema.TypeString, Computed: true, }, "secondary_table_endpoint": { Type: schema.TypeString, Computed: true, }, // NOTE: The API does not appear to expose a secondary file endpoint "primary_file_endpoint": { Type: schema.TypeString, Computed: true, }, "primary_access_key": { Type: schema.TypeString, Computed: true, }, "secondary_access_key": { Type: schema.TypeString, Computed: true, }, "tags": tagsSchema(), }, } } func resourceArmStorageAccountCreate(d *schema.ResourceData, meta interface{}) error { client := meta.(*ArmClient) storageClient := client.storageServiceClient resourceGroupName := d.Get("resource_group_name").(string) storageAccountName := d.Get("name").(string) accountType := d.Get("account_type").(string) location := d.Get("location").(string) tags := d.Get("tags").(map[string]interface{}) sku := storage.Sku{ Name: storage.SkuName(accountType), } opts := storage.AccountCreateParameters{ Location: &location, Sku: &sku, Tags: expandTags(tags), } // Create the storage account. We wrap this so that it is cancellable // with a Ctrl-C since this can take a LONG time. wrap := signalwrapper.Run(func(cancelCh <-chan struct{}) error { _, err := storageClient.Create(resourceGroupName, storageAccountName, opts, cancelCh) return err }) // Check the result of the wrapped function. var createErr error select { case <-time.After(1 * time.Hour): // An hour is way above the expected P99 for this API call so // we premature cancel and error here. createErr = wrap.Cancel() case createErr = <-wrap.ErrCh: // Successfully ran (but perhaps not successfully completed) // the function. } // The only way to get the ID back apparently is to read the resource again read, err := storageClient.GetProperties(resourceGroupName, storageAccountName) // Set the ID right away if we have one if err == nil && read.ID != nil { log.Printf("[INFO] storage account %q ID: %q", storageAccountName, *read.ID) d.SetId(*read.ID) } // If we had a create error earlier then we return with that error now. // We do this later here so that we can grab the ID above is possible. if createErr != nil { return fmt.Errorf( "Error creating Azure Storage Account '%s': %s", storageAccountName, createErr) } // Check the read error now that we know it would exist without a create err if err != nil { return err } // If we got no ID then the resource group doesn't yet exist if read.ID == nil { return fmt.Errorf("Cannot read Storage Account %s (resource group %s) ID", storageAccountName, resourceGroupName) } log.Printf("[DEBUG] Waiting for Storage Account (%s) to become available", storageAccountName) stateConf := &resource.StateChangeConf{ Pending: []string{"Updating", "Creating"}, Target: []string{"Succeeded"}, Refresh: storageAccountStateRefreshFunc(client, resourceGroupName, storageAccountName), Timeout: 30 * time.Minute, MinTimeout: 15 * time.Second, } if _, err := stateConf.WaitForState(); err != nil { return fmt.Errorf("Error waiting for Storage Account (%s) to become available: %s", storageAccountName, err) } return resourceArmStorageAccountRead(d, meta) } // resourceArmStorageAccountUpdate is unusual in the ARM API where most resources have a combined // and idempotent operation for CreateOrUpdate. In particular updating all of the parameters // available requires a call to Update per parameter... func resourceArmStorageAccountUpdate(d *schema.ResourceData, meta interface{}) error { client := meta.(*ArmClient).storageServiceClient id, err := parseAzureResourceID(d.Id()) if err != nil { return err } storageAccountName := id.Path["storageAccounts"] resourceGroupName := id.ResourceGroup d.Partial(true) if d.HasChange("account_type") { accountType := d.Get("account_type").(string) sku := storage.Sku{ Name: storage.SkuName(accountType), } opts := storage.AccountUpdateParameters{ Sku: &sku, } _, err := client.Update(resourceGroupName, storageAccountName, opts) if err != nil { return fmt.Errorf("Error updating Azure Storage Account type %q: %s", storageAccountName, err) } d.SetPartial("account_type") } if d.HasChange("tags") { tags := d.Get("tags").(map[string]interface{}) opts := storage.AccountUpdateParameters{ Tags: expandTags(tags), } _, err := client.Update(resourceGroupName, storageAccountName, opts) if err != nil { return fmt.Errorf("Error updating Azure Storage Account tags %q: %s", storageAccountName, err) } d.SetPartial("tags") } d.Partial(false) return nil } func resourceArmStorageAccountRead(d *schema.ResourceData, meta interface{}) error { client := meta.(*ArmClient).storageServiceClient id, err := parseAzureResourceID(d.Id()) if err != nil { return err } name := id.Path["storageAccounts"] resGroup := id.ResourceGroup resp, err := client.GetProperties(resGroup, name) if err != nil { return fmt.Errorf("Error reading the state of AzureRM Storage Account %q: %s", name, err) } if resp.StatusCode == http.StatusNotFound { d.SetId("") return nil } keys, err := client.ListKeys(resGroup, name) if err != nil { return err } accessKeys := *keys.Keys d.Set("primary_access_key", accessKeys[0].Value) d.Set("secondary_access_key", accessKeys[1].Value) d.Set("location", resp.Location) d.Set("account_type", resp.Sku.Name) d.Set("primary_location", resp.Properties.PrimaryLocation) d.Set("secondary_location", resp.Properties.SecondaryLocation) if resp.Properties.PrimaryEndpoints != nil { d.Set("primary_blob_endpoint", resp.Properties.PrimaryEndpoints.Blob) d.Set("primary_queue_endpoint", resp.Properties.PrimaryEndpoints.Queue) d.Set("primary_table_endpoint", resp.Properties.PrimaryEndpoints.Table) d.Set("primary_file_endpoint", resp.Properties.PrimaryEndpoints.File) } if resp.Properties.SecondaryEndpoints != nil { if resp.Properties.SecondaryEndpoints.Blob != nil { d.Set("secondary_blob_endpoint", resp.Properties.SecondaryEndpoints.Blob) } else { d.Set("secondary_blob_endpoint", "") } if resp.Properties.SecondaryEndpoints.Queue != nil { d.Set("secondary_queue_endpoint", resp.Properties.SecondaryEndpoints.Queue) } else { d.Set("secondary_queue_endpoint", "") } if resp.Properties.SecondaryEndpoints.Table != nil { d.Set("secondary_table_endpoint", resp.Properties.SecondaryEndpoints.Table) } else { d.Set("secondary_table_endpoint", "") } } d.Set("name", resp.Name) flattenAndSetTags(d, resp.Tags) return nil } func resourceArmStorageAccountDelete(d *schema.ResourceData, meta interface{}) error { client := meta.(*ArmClient).storageServiceClient id, err := parseAzureResourceID(d.Id()) if err != nil { return err } name := id.Path["storageAccounts"] resGroup := id.ResourceGroup _, err = client.Delete(resGroup, name) if err != nil { return fmt.Errorf("Error issuing AzureRM delete request for storage account %q: %s", name, err) } return nil } func validateArmStorageAccountName(v interface{}, k string) (ws []string, es []error) { input := v.(string) if !regexp.MustCompile(`\A([a-z0-9]{3,24})\z`).MatchString(input) { es = append(es, fmt.Errorf("name can only consist of lowercase letters and numbers, and must be between 3 and 24 characters long")) } return } func validateArmStorageAccountType(v interface{}, k string) (ws []string, es []error) { validAccountTypes := []string{"standard_lrs", "standard_zrs", "standard_grs", "standard_ragrs", "premium_lrs"} input := strings.ToLower(v.(string)) for _, valid := range validAccountTypes { if valid == input { return } } es = append(es, fmt.Errorf("Invalid storage account type %q", input)) return } func storageAccountStateRefreshFunc(client *ArmClient, resourceGroupName string, storageAccountName string) resource.StateRefreshFunc { return func() (interface{}, string, error) { res, err := client.storageServiceClient.GetProperties(resourceGroupName, storageAccountName) if err != nil { return nil, "", fmt.Errorf("Error issuing read request in storageAccountStateRefreshFunc to Azure ARM for Storage Account '%s' (RG: '%s'): %s", storageAccountName, resourceGroupName, err) } return res, string(res.Properties.ProvisioningState), nil } }