75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
package digitalocean
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/digitalocean/godo"
|
|
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
|
|
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
|
|
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
|
|
)
|
|
|
|
func dataSourceDigitalOceanRegion() *schema.Resource {
|
|
return &schema.Resource{
|
|
Read: dataSourceDigitalOceanRegionRead,
|
|
Schema: map[string]*schema.Schema{
|
|
"slug": {
|
|
Type: schema.TypeString,
|
|
Required: true,
|
|
ValidateFunc: validation.StringIsNotEmpty,
|
|
},
|
|
"name": {
|
|
Type: schema.TypeString,
|
|
Computed: true,
|
|
},
|
|
"sizes": {
|
|
Type: schema.TypeSet,
|
|
Elem: &schema.Schema{Type: schema.TypeString},
|
|
Computed: true,
|
|
},
|
|
"features": {
|
|
Type: schema.TypeSet,
|
|
Elem: &schema.Schema{Type: schema.TypeString},
|
|
Computed: true,
|
|
},
|
|
"available": {
|
|
Type: schema.TypeBool,
|
|
Computed: true,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func dataSourceDigitalOceanRegionRead(d *schema.ResourceData, meta interface{}) error {
|
|
regions, err := getDigitalOceanRegions(meta)
|
|
if err != nil {
|
|
return fmt.Errorf("Unable to load regions: %s", err)
|
|
}
|
|
|
|
slug := d.Get("slug").(string)
|
|
|
|
var regionForSlug *interface{}
|
|
for _, region := range regions {
|
|
if region.(godo.Region).Slug == slug {
|
|
regionForSlug = ®ion
|
|
break
|
|
}
|
|
}
|
|
|
|
if regionForSlug == nil {
|
|
return fmt.Errorf("Region does not exist: %s", slug)
|
|
}
|
|
|
|
flattenedRegion, err := flattenRegion(*regionForSlug, meta)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
if err := setResourceDataFromMap(d, flattenedRegion); err != nil {
|
|
return err
|
|
}
|
|
|
|
d.SetId(resource.UniqueId())
|
|
return nil
|
|
}
|