73 lines
1.7 KiB
Go
73 lines
1.7 KiB
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/v2/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
|
|
}
|