148 lines
3.6 KiB
Go
148 lines
3.6 KiB
Go
package cloudstack
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"regexp"
|
|
"time"
|
|
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
"github.com/xanzy/go-cloudstack/cloudstack"
|
|
)
|
|
|
|
// CloudStack uses a "special" ID of -1 to define an unlimited resource
|
|
const UnlimitedResourceID = "-1"
|
|
|
|
type retrieveError struct {
|
|
name string
|
|
value string
|
|
err error
|
|
}
|
|
|
|
func (e *retrieveError) Error() error {
|
|
return fmt.Errorf("Error retrieving ID of %s %s: %s", e.name, e.value, e.err)
|
|
}
|
|
|
|
func setValueOrID(d *schema.ResourceData, key string, value string, id string) {
|
|
if isID(d.Get(key).(string)) {
|
|
// If the given id is an empty string, check if the configured value matches
|
|
// the UnlimitedResourceID in which case we set id to UnlimitedResourceID
|
|
if id == "" && d.Get(key).(string) == UnlimitedResourceID {
|
|
id = UnlimitedResourceID
|
|
}
|
|
|
|
d.Set(key, id)
|
|
} else {
|
|
d.Set(key, value)
|
|
}
|
|
}
|
|
|
|
func retrieveID(cs *cloudstack.CloudStackClient, name, value string) (id string, e *retrieveError) {
|
|
// If the supplied value isn't a ID, try to retrieve the ID ourselves
|
|
if isID(value) {
|
|
return value, nil
|
|
}
|
|
|
|
log.Printf("[DEBUG] Retrieving ID of %s: %s", name, value)
|
|
|
|
var err error
|
|
switch name {
|
|
case "disk_offering":
|
|
id, err = cs.DiskOffering.GetDiskOfferingID(value)
|
|
case "virtual_machine":
|
|
id, err = cs.VirtualMachine.GetVirtualMachineID(value)
|
|
case "service_offering":
|
|
id, err = cs.ServiceOffering.GetServiceOfferingID(value)
|
|
case "network_offering":
|
|
id, err = cs.NetworkOffering.GetNetworkOfferingID(value)
|
|
case "project":
|
|
id, err = cs.Project.GetProjectID(value)
|
|
case "vpc_offering":
|
|
id, err = cs.VPC.GetVPCOfferingID(value)
|
|
case "vpc":
|
|
id, err = cs.VPC.GetVPCID(value)
|
|
case "network":
|
|
id, err = cs.Network.GetNetworkID(value)
|
|
case "zone":
|
|
id, err = cs.Zone.GetZoneID(value)
|
|
case "ipaddress":
|
|
p := cs.Address.NewListPublicIpAddressesParams()
|
|
p.SetIpaddress(value)
|
|
l, e := cs.Address.ListPublicIpAddresses(p)
|
|
if e != nil {
|
|
err = e
|
|
break
|
|
}
|
|
if l.Count == 1 {
|
|
id = l.PublicIpAddresses[0].Id
|
|
break
|
|
}
|
|
err = fmt.Errorf("Could not find ID of IP address: %s", value)
|
|
case "os_type":
|
|
p := cs.GuestOS.NewListOsTypesParams()
|
|
p.SetDescription(value)
|
|
l, e := cs.GuestOS.ListOsTypes(p)
|
|
if e != nil {
|
|
err = e
|
|
break
|
|
}
|
|
if l.Count == 1 {
|
|
id = l.OsTypes[0].Id
|
|
break
|
|
}
|
|
err = fmt.Errorf("Could not find ID of OS Type: %s", value)
|
|
default:
|
|
return id, &retrieveError{name: name, value: value,
|
|
err: fmt.Errorf("Unknown request: %s", name)}
|
|
}
|
|
|
|
if err != nil {
|
|
return id, &retrieveError{name: name, value: value, err: err}
|
|
}
|
|
|
|
return id, nil
|
|
}
|
|
|
|
func retrieveTemplateID(cs *cloudstack.CloudStackClient, zoneid, value string) (id string, e *retrieveError) {
|
|
// If the supplied value isn't a ID, try to retrieve the ID ourselves
|
|
if isID(value) {
|
|
return value, nil
|
|
}
|
|
|
|
log.Printf("[DEBUG] Retrieving ID of template: %s", value)
|
|
|
|
id, err := cs.Template.GetTemplateID(value, "executable", zoneid)
|
|
if err != nil {
|
|
return id, &retrieveError{name: "template", value: value, err: err}
|
|
}
|
|
|
|
return id, nil
|
|
}
|
|
|
|
// ID can be either a UUID or a UnlimitedResourceID
|
|
func isID(id string) bool {
|
|
re := regexp.MustCompile(`^([0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}|-1)$`)
|
|
return re.MatchString(id)
|
|
}
|
|
|
|
// RetryFunc is the function retried n times
|
|
type RetryFunc func() (interface{}, error)
|
|
|
|
// Retry is a wrapper around a RetryFunc that will retry a function
|
|
// n times or until it succeeds.
|
|
func Retry(n int, f RetryFunc) (interface{}, error) {
|
|
var lastErr error
|
|
|
|
for i := 0; i < n; i++ {
|
|
r, err := f()
|
|
if err == nil {
|
|
return r, nil
|
|
}
|
|
|
|
lastErr = err
|
|
time.Sleep(30 * time.Second)
|
|
}
|
|
|
|
return nil, lastErr
|
|
}
|