186 lines
4.8 KiB
Go
186 lines
4.8 KiB
Go
![]() |
package azurerm
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"log"
|
||
|
"strings"
|
||
|
|
||
|
"github.com/hashicorp/terraform/helper/schema"
|
||
|
)
|
||
|
|
||
|
func resourceArmStorageBlob() *schema.Resource {
|
||
|
return &schema.Resource{
|
||
|
Create: resourceArmStorageBlobCreate,
|
||
|
Read: resourceArmStorageBlobRead,
|
||
|
Exists: resourceArmStorageBlobExists,
|
||
|
Delete: resourceArmStorageBlobDelete,
|
||
|
|
||
|
Schema: map[string]*schema.Schema{
|
||
|
"name": &schema.Schema{
|
||
|
Type: schema.TypeString,
|
||
|
Required: true,
|
||
|
ForceNew: true,
|
||
|
},
|
||
|
"resource_group_name": &schema.Schema{
|
||
|
Type: schema.TypeString,
|
||
|
Required: true,
|
||
|
ForceNew: true,
|
||
|
},
|
||
|
"storage_account_name": {
|
||
|
Type: schema.TypeString,
|
||
|
Required: true,
|
||
|
ForceNew: true,
|
||
|
},
|
||
|
"storage_container_name": &schema.Schema{
|
||
|
Type: schema.TypeString,
|
||
|
Required: true,
|
||
|
ForceNew: true,
|
||
|
},
|
||
|
"type": &schema.Schema{
|
||
|
Type: schema.TypeString,
|
||
|
Required: true,
|
||
|
ForceNew: true,
|
||
|
ValidateFunc: validateArmStorageBlobType,
|
||
|
},
|
||
|
"size": &schema.Schema{
|
||
|
Type: schema.TypeInt,
|
||
|
Optional: true,
|
||
|
ForceNew: true,
|
||
|
Default: 0,
|
||
|
},
|
||
|
"url": &schema.Schema{
|
||
|
Type: schema.TypeString,
|
||
|
Computed: true,
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func validateArmStorageBlobType(v interface{}, k string) (ws []string, errors []error) {
|
||
|
value := strings.ToLower(v.(string))
|
||
|
validTypes := map[string]struct{}{
|
||
|
"blob": struct{}{},
|
||
|
"page": struct{}{},
|
||
|
}
|
||
|
|
||
|
if _, ok := validTypes[value]; !ok {
|
||
|
errors = append(errors, fmt.Errorf("Blob type %q is invalid, must be %q or %q", value, "blob", "page"))
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
|
func resourceArmStorageBlobCreate(d *schema.ResourceData, meta interface{}) error {
|
||
|
armClient := meta.(*ArmClient)
|
||
|
|
||
|
resourceGroupName := d.Get("resource_group_name").(string)
|
||
|
storageAccountName := d.Get("storage_account_name").(string)
|
||
|
|
||
|
blobClient, err := armClient.getBlobStorageClientForStorageAccount(resourceGroupName, storageAccountName)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
name := d.Get("name").(string)
|
||
|
blobType := d.Get("type").(string)
|
||
|
cont := d.Get("storage_container_name").(string)
|
||
|
|
||
|
log.Printf("[INFO] Creating blob %q in storage account %q", name, storageAccountName)
|
||
|
switch strings.ToLower(blobType) {
|
||
|
case "block":
|
||
|
err = blobClient.CreateBlockBlob(cont, name)
|
||
|
case "page":
|
||
|
size := int64(d.Get("size").(int))
|
||
|
err = blobClient.PutPageBlob(cont, name, size, map[string]string{})
|
||
|
}
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Error creating storage blob on Azure: %s", err)
|
||
|
}
|
||
|
|
||
|
d.SetId(name)
|
||
|
return resourceArmStorageBlobRead(d, meta)
|
||
|
}
|
||
|
|
||
|
func resourceArmStorageBlobRead(d *schema.ResourceData, meta interface{}) error {
|
||
|
armClient := meta.(*ArmClient)
|
||
|
|
||
|
resourceGroupName := d.Get("resource_group_name").(string)
|
||
|
storageAccountName := d.Get("storage_account_name").(string)
|
||
|
|
||
|
blobClient, err := armClient.getBlobStorageClientForStorageAccount(resourceGroupName, storageAccountName)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
exists, err := resourceArmStorageBlobExists(d, meta)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if !exists {
|
||
|
// Exists already removed this from state
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
name := d.Get("name").(string)
|
||
|
storageContainerName := d.Get("storage_container_name").(string)
|
||
|
|
||
|
url := blobClient.GetBlobURL(storageContainerName, name)
|
||
|
if url == "" {
|
||
|
log.Printf("[INFO] URL for %q is empty", name)
|
||
|
}
|
||
|
d.Set("url", url)
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func resourceArmStorageBlobExists(d *schema.ResourceData, meta interface{}) (bool, error) {
|
||
|
armClient := meta.(*ArmClient)
|
||
|
|
||
|
resourceGroupName := d.Get("resource_group_name").(string)
|
||
|
storageAccountName := d.Get("storage_account_name").(string)
|
||
|
|
||
|
blobClient, err := armClient.getBlobStorageClientForStorageAccount(resourceGroupName, storageAccountName)
|
||
|
if err != nil {
|
||
|
return false, err
|
||
|
}
|
||
|
|
||
|
name := d.Get("name").(string)
|
||
|
storageContainerName := d.Get("storage_container_name").(string)
|
||
|
|
||
|
log.Printf("[INFO] Checking for existence of storage blob %q.", name)
|
||
|
exists, err := blobClient.BlobExists(storageContainerName, name)
|
||
|
if err != nil {
|
||
|
return false, fmt.Errorf("error testing existence of storage blob %q: %s", name, err)
|
||
|
}
|
||
|
|
||
|
if !exists {
|
||
|
log.Printf("[INFO] Storage blob %q no longer exists, removing from state...", name)
|
||
|
d.SetId("")
|
||
|
}
|
||
|
|
||
|
return exists, nil
|
||
|
}
|
||
|
|
||
|
func resourceArmStorageBlobDelete(d *schema.ResourceData, meta interface{}) error {
|
||
|
armClient := meta.(*ArmClient)
|
||
|
|
||
|
resourceGroupName := d.Get("resource_group_name").(string)
|
||
|
storageAccountName := d.Get("storage_account_name").(string)
|
||
|
|
||
|
blobClient, err := armClient.getBlobStorageClientForStorageAccount(resourceGroupName, storageAccountName)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
name := d.Get("name").(string)
|
||
|
storageContainerName := d.Get("storage_container_name").(string)
|
||
|
|
||
|
log.Printf("[INFO] Deleting storage blob %q", name)
|
||
|
if _, err = blobClient.DeleteBlobIfExists(storageContainerName, name); err != nil {
|
||
|
return fmt.Errorf("Error deleting storage blob %q: %s", name, err)
|
||
|
}
|
||
|
|
||
|
d.SetId("")
|
||
|
return nil
|
||
|
}
|