terraform/builtin/providers/pagerduty/util.go

69 lines
1.4 KiB
Go

package pagerduty
import (
"fmt"
pagerduty "github.com/PagerDuty/go-pagerduty"
"github.com/hashicorp/terraform/helper/schema"
)
// Validate a value against a set of possible values
func validateValueFunc(values []string) schema.SchemaValidateFunc {
return func(v interface{}, k string) (we []string, errors []error) {
value := v.(string)
valid := false
for _, val := range values {
if value == val {
valid = true
break
}
}
if !valid {
errors = append(errors, fmt.Errorf("%#v is an invalid value for argument %s. Must be one of %#v", value, k, values))
}
return
}
}
// getVendors retrieves all PagerDuty vendors and returns a list of []pagerduty.Vendor
func getVendors(client *pagerduty.Client) ([]pagerduty.Vendor, error) {
var offset uint
var totalCount int
var vendors []pagerduty.Vendor
for {
o := &pagerduty.ListVendorOptions{
APIListObject: pagerduty.APIListObject{
Limit: 100,
Total: 1,
Offset: offset,
},
}
resp, err := client.ListVendors(*o)
if err != nil {
return nil, err
}
for _, v := range resp.Vendors {
totalCount++
vendors = append(vendors, v)
}
rOffset := uint(resp.Offset)
returnedCount := uint(len(resp.Vendors))
rTotal := uint(resp.Total)
if resp.More && uint(totalCount) != uint(rTotal) {
offset = returnedCount + rOffset
continue
}
break
}
return vendors, nil
}