package cloudstack import ( "crypto/sha1" "encoding/base64" "encoding/hex" "fmt" "log" "strings" "github.com/hashicorp/terraform/helper/schema" "github.com/xanzy/go-cloudstack/cloudstack" ) func resourceCloudStackInstance() *schema.Resource { return &schema.Resource{ Create: resourceCloudStackInstanceCreate, Read: resourceCloudStackInstanceRead, Update: resourceCloudStackInstanceUpdate, Delete: resourceCloudStackInstanceDelete, Schema: map[string]*schema.Schema{ "name": &schema.Schema{ Type: schema.TypeString, Required: true, ForceNew: true, }, "display_name": &schema.Schema{ Type: schema.TypeString, Optional: true, Computed: true, }, "service_offering": &schema.Schema{ Type: schema.TypeString, Required: true, }, "network": &schema.Schema{ Type: schema.TypeString, Optional: true, ForceNew: true, }, "ipaddress": &schema.Schema{ Type: schema.TypeString, Optional: true, Computed: true, ForceNew: true, }, "template": &schema.Schema{ Type: schema.TypeString, Required: true, ForceNew: true, }, "project": &schema.Schema{ Type: schema.TypeString, Optional: true, ForceNew: true, }, "zone": &schema.Schema{ Type: schema.TypeString, Required: true, ForceNew: true, }, "keypair": &schema.Schema{ Type: schema.TypeString, Optional: true, }, "user_data": &schema.Schema{ Type: schema.TypeString, Optional: true, ForceNew: true, StateFunc: func(v interface{}) string { switch v.(type) { case string: hash := sha1.Sum([]byte(v.(string))) return hex.EncodeToString(hash[:]) default: return "" } }, }, "expunge": &schema.Schema{ Type: schema.TypeBool, Optional: true, Default: false, }, }, } } func resourceCloudStackInstanceCreate(d *schema.ResourceData, meta interface{}) error { cs := meta.(*cloudstack.CloudStackClient) // Retrieve the service_offering UUID serviceofferingid, e := retrieveUUID(cs, "service_offering", d.Get("service_offering").(string)) if e != nil { return e.Error() } // Retrieve the zone UUID zoneid, e := retrieveUUID(cs, "zone", d.Get("zone").(string)) if e != nil { return e.Error() } // Retrieve the zone object zone, _, err := cs.Zone.GetZoneByID(zoneid) if err != nil { return err } // Retrieve the template UUID templateid, e := retrieveTemplateUUID(cs, zone.Id, d.Get("template").(string)) if e != nil { return e.Error() } // Create a new parameter struct p := cs.VirtualMachine.NewDeployVirtualMachineParams(serviceofferingid, templateid, zone.Id) // Set the name name := d.Get("name").(string) p.SetName(name) // Set the display name if displayname, ok := d.GetOk("display_name"); ok { p.SetDisplayname(displayname.(string)) } else { p.SetDisplayname(name) } if zone.Networktype == "Advanced" { // Retrieve the network UUID networkid, e := retrieveUUID(cs, "network", d.Get("network").(string)) if e != nil { return e.Error() } // Set the default network ID p.SetNetworkids([]string{networkid}) } // If there is a ipaddres supplied, add it to the parameter struct if ipaddres, ok := d.GetOk("ipaddress"); ok { p.SetIpaddress(ipaddres.(string)) } // If there is a project supplied, we retreive and set the project id if project, ok := d.GetOk("project"); ok { // Retrieve the project UUID projectid, e := retrieveUUID(cs, "project", project.(string)) if e != nil { return e.Error() } // Set the default project ID p.SetProjectid(projectid) } // If a keypair is supplied, add it to the parameter struct if keypair, ok := d.GetOk("keypair"); ok { p.SetKeypair(keypair.(string)) } // If the user data contains any info, it needs to be base64 encoded and // added to the parameter struct if userData, ok := d.GetOk("user_data"); ok { ud := base64.StdEncoding.EncodeToString([]byte(userData.(string))) // deployVirtualMachine uses POST by default, so max userdata is 32K maxUD := 32768 if cs.HTTPGETOnly { // deployVirtualMachine using GET instead, so max userdata is 2K maxUD = 2048 } if len(ud) > maxUD { return fmt.Errorf( "The supplied user_data contains %d bytes after encoding, "+ "this exeeds the limit of %d bytes", len(ud), maxUD) } p.SetUserdata(ud) } // Create the new instance r, err := cs.VirtualMachine.DeployVirtualMachine(p) if err != nil { return fmt.Errorf("Error creating the new instance %s: %s", name, err) } d.SetId(r.Id) // Set the connection info for any configured provisioners d.SetConnInfo(map[string]string{ "host": r.Nic[0].Ipaddress, "password": r.Password, }) return resourceCloudStackInstanceRead(d, meta) } func resourceCloudStackInstanceRead(d *schema.ResourceData, meta interface{}) error { cs := meta.(*cloudstack.CloudStackClient) // Get the virtual machine details vm, count, err := cs.VirtualMachine.GetVirtualMachineByID(d.Id()) if err != nil { if count == 0 { log.Printf("[DEBUG] Instance %s does no longer exist", d.Get("name").(string)) d.SetId("") return nil } return err } // Update the config d.Set("name", vm.Name) d.Set("display_name", vm.Displayname) d.Set("ipaddress", vm.Nic[0].Ipaddress) //NB cloudstack sometimes sends back the wrong keypair name, so dont update it setValueOrUUID(d, "network", vm.Nic[0].Networkname, vm.Nic[0].Networkid) setValueOrUUID(d, "service_offering", vm.Serviceofferingname, vm.Serviceofferingid) setValueOrUUID(d, "template", vm.Templatename, vm.Templateid) setValueOrUUID(d, "project", vm.Project, vm.Projectid) setValueOrUUID(d, "zone", vm.Zonename, vm.Zoneid) return nil } func resourceCloudStackInstanceUpdate(d *schema.ResourceData, meta interface{}) error { cs := meta.(*cloudstack.CloudStackClient) d.Partial(true) name := d.Get("name").(string) // Check if the display name is changed and if so, update the virtual machine if d.HasChange("display_name") { log.Printf("[DEBUG] Display name changed for %s, starting update", name) // Create a new parameter struct p := cs.VirtualMachine.NewUpdateVirtualMachineParams(d.Id()) // Set the new display name p.SetDisplayname(d.Get("display_name").(string)) // Update the display name _, err := cs.VirtualMachine.UpdateVirtualMachine(p) if err != nil { return fmt.Errorf( "Error updating the display name for instance %s: %s", name, err) } d.SetPartial("display_name") } // Attributes that require reboot to update if d.HasChange("service_offering") || d.HasChange("keypair") { // Before we can actually make these changes, the virtual machine must be stopped _, err := cs.VirtualMachine.StopVirtualMachine( cs.VirtualMachine.NewStopVirtualMachineParams(d.Id())) if err != nil { return fmt.Errorf( "Error stopping instance %s before making changes: %s", name, err) } // Check if the service offering is changed and if so, update the offering if d.HasChange("service_offering") { log.Printf("[DEBUG] Service offering changed for %s, starting update", name) // Retrieve the service_offering UUID serviceofferingid, e := retrieveUUID(cs, "service_offering", d.Get("service_offering").(string)) if e != nil { return e.Error() } // Create a new parameter struct p := cs.VirtualMachine.NewChangeServiceForVirtualMachineParams(d.Id(), serviceofferingid) // Change the service offering _, err = cs.VirtualMachine.ChangeServiceForVirtualMachine(p) if err != nil { return fmt.Errorf( "Error changing the service offering for instance %s: %s", name, err) } d.SetPartial("service_offering") } if d.HasChange("keypair") { log.Printf("[DEBUG] SSH keypair changed for %s, starting update", name) p := cs.SSH.NewResetSSHKeyForVirtualMachineParams(d.Id(), d.Get("keypair").(string)) // Change the ssh keypair _, err = cs.SSH.ResetSSHKeyForVirtualMachine(p) if err != nil { return fmt.Errorf( "Error changing the SSH keypair for instance %s: %s", name, err) } d.SetPartial("keypair") } // Start the virtual machine again _, err = cs.VirtualMachine.StartVirtualMachine( cs.VirtualMachine.NewStartVirtualMachineParams(d.Id())) if err != nil { return fmt.Errorf( "Error starting instance %s after making changes", name) } } d.Partial(false) return resourceCloudStackInstanceRead(d, meta) } func resourceCloudStackInstanceDelete(d *schema.ResourceData, meta interface{}) error { cs := meta.(*cloudstack.CloudStackClient) // Create a new parameter struct p := cs.VirtualMachine.NewDestroyVirtualMachineParams(d.Id()) if d.Get("expunge").(bool) { p.SetExpunge(true) } log.Printf("[INFO] Destroying instance: %s", d.Get("name").(string)) if _, err := cs.VirtualMachine.DestroyVirtualMachine(p); err != nil { // This is a very poor way to be told the UUID does no longer exist :( if strings.Contains(err.Error(), fmt.Sprintf( "Invalid parameter id value=%s due to incorrect long value format, "+ "or entity does not exist", d.Id())) { return nil } return fmt.Errorf("Error destroying instance: %s", err) } return nil }