95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package linode
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/hashicorp/packer/helper/multistep"
|
|
"github.com/hashicorp/packer/packer"
|
|
"github.com/linode/linodego"
|
|
)
|
|
|
|
type stepCreateLinode struct {
|
|
client linodego.Client
|
|
}
|
|
|
|
func (s *stepCreateLinode) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
|
c := state.Get("config").(*Config)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
ui.Say("Creating Linode...")
|
|
|
|
createOpts := linodego.InstanceCreateOptions{
|
|
RootPass: c.Comm.Password(),
|
|
AuthorizedKeys: []string{string(c.Comm.SSHPublicKey)},
|
|
Region: c.Region,
|
|
Type: c.InstanceType,
|
|
Label: c.Label,
|
|
Image: c.Image,
|
|
SwapSize: &c.SwapSize,
|
|
}
|
|
|
|
instance, err := s.client.CreateInstance(ctx, createOpts)
|
|
if err != nil {
|
|
err = errors.New("Error creating Linode: " + err.Error())
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
state.Put("instance", instance)
|
|
|
|
// wait until instance is running
|
|
for instance.Status != linodego.InstanceRunning {
|
|
time.Sleep(2 * time.Second)
|
|
if instance, err = s.client.GetInstance(ctx, instance.ID); err != nil {
|
|
err = errors.New("Error creating Linode: " + err.Error())
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
state.Put("instance", instance)
|
|
}
|
|
|
|
disk, err := s.findDisk(ctx, instance.ID)
|
|
if err != nil {
|
|
err = errors.New("Error creating Linode: " + err.Error())
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
} else if disk == nil {
|
|
err := errors.New("Error creating Linode: no suitable disk was found")
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
state.Put("disk", disk)
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *stepCreateLinode) findDisk(ctx context.Context, instanceID int) (*linodego.InstanceDisk, error) {
|
|
disks, err := s.client.ListInstanceDisks(ctx, instanceID, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, disk := range disks {
|
|
if disk.Filesystem != linodego.FilesystemSwap {
|
|
return &disk, nil
|
|
}
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (s *stepCreateLinode) Cleanup(state multistep.StateBag) {
|
|
instance, ok := state.GetOk("instance")
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
if err := s.client.DeleteInstance(context.Background(), instance.(*linodego.Instance).ID); err != nil {
|
|
ui.Error("Error cleaning up Linode: " + err.Error())
|
|
}
|
|
}
|