datasource_digitalocean_spaces_bucket.go

package digitalocean

import (
	"fmt"
	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/awserr"
	"github.com/aws/aws-sdk-go/service/s3"
	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
)

func dataSourceDigitalOceanSpacesBucket() *schema.Resource {
	recordSchema := spacesBucketSchema()

	for _, f := range recordSchema {
		f.Computed = true
	}

	recordSchema["region"].Required = true
	recordSchema["region"].Computed = false
	recordSchema["name"].Required = true
	recordSchema["name"].Computed = false

	return &schema.Resource{
		Read:   dataSourceDigitalOceanSpacesBucketRead,
		Schema: recordSchema,
	}
}

func dataSourceDigitalOceanSpacesBucketRead(d *schema.ResourceData, meta interface{}) error {
	region := d.Get("region").(string)
	name := d.Get("name").(string)

	client, err := meta.(*CombinedConfig).spacesClient(region)
	if err != nil {
		return fmt.Errorf("Error reading bucket: %s", err)
	}

	svc := s3.New(client)

	_, err = retryOnAwsCode("NoSuchBucket", func() (interface{}, error) {
		return svc.HeadBucket(&s3.HeadBucketInput{
			Bucket: aws.String(name),
		})
	})
	if err != nil {
		if awsError, ok := err.(awserr.RequestFailure); ok && awsError.StatusCode() == 404 {
			d.SetId("")
			return fmt.Errorf("Spaces Bucket (%s) not found", name)
		} else {
			// some of the AWS SDK's errors can be empty strings, so let's add
			// some additional context.
			return fmt.Errorf("error reading Spaces bucket \"%s\": %s", d.Id(), err)
		}
	}

	metadata := bucketMetadataStruct{
		name:   name,
		region: region,
	}

	flattenedBucket, err := flattenSpacesBucket(&metadata, meta)
	if err != nil {
		return err
	}

	if err := setResourceDataFromMap(d, flattenedBucket); err != nil {
		return err
	}

	d.SetId(name)
	return nil
}