74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package profitbricks
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
"github.com/profitbricks/profitbricks-sdk-go"
|
|
"log"
|
|
"strings"
|
|
)
|
|
|
|
func dataSourceLocation() *schema.Resource {
|
|
return &schema.Resource{
|
|
Read: dataSourceLocationRead,
|
|
Schema: map[string]*schema.Schema{
|
|
"name": {
|
|
Type: schema.TypeString,
|
|
Optional: true,
|
|
},
|
|
"feature": {
|
|
Type: schema.TypeString,
|
|
Optional: true,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func dataSourceLocationRead(d *schema.ResourceData, meta interface{}) error {
|
|
locations := profitbricks.ListLocations()
|
|
|
|
if locations.StatusCode > 299 {
|
|
return fmt.Errorf("An error occured while fetching ProfitBricks locations %s", locations.Response)
|
|
}
|
|
|
|
name, nameOk := d.GetOk("name")
|
|
feature, featureOk := d.GetOk("features")
|
|
|
|
if !nameOk && !featureOk {
|
|
return fmt.Errorf("Either 'name' or 'feature' must be provided.")
|
|
}
|
|
results := []profitbricks.Location{}
|
|
|
|
for _, loc := range locations.Items {
|
|
if loc.Properties.Name == name.(string) || strings.Contains(loc.Properties.Name, name.(string)) {
|
|
results = append(results, loc)
|
|
}
|
|
}
|
|
|
|
if featureOk {
|
|
locationResults := []profitbricks.Location{}
|
|
for _, loc := range results {
|
|
for _, f := range loc.Properties.Features {
|
|
if f == feature.(string) {
|
|
locationResults = append(locationResults, loc)
|
|
}
|
|
}
|
|
}
|
|
results = locationResults
|
|
}
|
|
log.Printf("[INFO] Results length %d *************", len(results))
|
|
|
|
if len(results) > 1 {
|
|
log.Printf("[INFO] Results length greater than 1")
|
|
return fmt.Errorf("There is more than one location that match the search criteria")
|
|
}
|
|
|
|
if len(results) == 0 {
|
|
return fmt.Errorf("There are no locations that match the search criteria")
|
|
}
|
|
|
|
d.SetId(results[0].Id)
|
|
|
|
return nil
|
|
}
|