79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
|
package pagerduty
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"log"
|
||
|
"regexp"
|
||
|
|
||
|
pagerduty "github.com/PagerDuty/go-pagerduty"
|
||
|
"github.com/hashicorp/terraform/helper/schema"
|
||
|
)
|
||
|
|
||
|
func dataSourcePagerDutyVendor() *schema.Resource {
|
||
|
return &schema.Resource{
|
||
|
Read: dataSourcePagerDutyVendorRead,
|
||
|
|
||
|
Schema: map[string]*schema.Schema{
|
||
|
"name_regex": &schema.Schema{
|
||
|
Type: schema.TypeString,
|
||
|
Required: true,
|
||
|
},
|
||
|
"name": &schema.Schema{
|
||
|
Type: schema.TypeString,
|
||
|
Computed: true,
|
||
|
},
|
||
|
"type": &schema.Schema{
|
||
|
Type: schema.TypeString,
|
||
|
Computed: true,
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func dataSourcePagerDutyVendorRead(d *schema.ResourceData, meta interface{}) error {
|
||
|
client := meta.(*pagerduty.Client)
|
||
|
|
||
|
log.Printf("[INFO] Reading PagerDuty vendors")
|
||
|
|
||
|
resp, err := getVendors(client)
|
||
|
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
r := regexp.MustCompile("(?i)" + d.Get("name_regex").(string))
|
||
|
|
||
|
var vendors []pagerduty.Vendor
|
||
|
var vendorNames []string
|
||
|
|
||
|
for _, v := range resp {
|
||
|
if r.MatchString(v.Name) {
|
||
|
vendors = append(vendors, v)
|
||
|
vendorNames = append(vendorNames, v.Name)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if len(vendors) == 0 {
|
||
|
return fmt.Errorf("Unable to locate any vendor using the regex string: %s", r.String())
|
||
|
} else if len(vendors) > 1 {
|
||
|
return fmt.Errorf("Your query returned more than one result using the regex string: %#v. Found vendors: %#v", r.String(), vendorNames)
|
||
|
}
|
||
|
|
||
|
vendor := vendors[0]
|
||
|
|
||
|
genericServiceType := vendor.GenericServiceType
|
||
|
|
||
|
switch {
|
||
|
case genericServiceType == "email":
|
||
|
genericServiceType = "generic_email_inbound_integration"
|
||
|
case genericServiceType == "api":
|
||
|
genericServiceType = "generic_events_api_inbound_integration"
|
||
|
}
|
||
|
|
||
|
d.SetId(vendor.ID)
|
||
|
d.Set("name", vendor.Name)
|
||
|
d.Set("type", genericServiceType)
|
||
|
|
||
|
return nil
|
||
|
}
|