103 lines
2.7 KiB
Go
103 lines
2.7 KiB
Go
package chroot
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/antihax/optional"
|
|
osccommon "github.com/hashicorp/packer/builder/osc/common"
|
|
"github.com/hashicorp/packer/packer"
|
|
"github.com/hashicorp/packer/packer-plugin-sdk/multistep"
|
|
"github.com/outscale/osc-sdk-go/osc"
|
|
)
|
|
|
|
// StepLinkVolume attaches the previously created volume to an
|
|
// available device location.
|
|
//
|
|
// Produces:
|
|
// device string - The location where the volume was attached.
|
|
// attach_cleanup CleanupFunc
|
|
type StepLinkVolume struct {
|
|
attached bool
|
|
volumeId string
|
|
}
|
|
|
|
func (s *StepLinkVolume) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
|
oscconn := state.Get("osc").(*osc.APIClient)
|
|
device := state.Get("device").(string)
|
|
vm := state.Get("vm").(osc.Vm)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
volumeId := state.Get("volume_id").(string)
|
|
|
|
// For the API call, it expects "sd" prefixed devices.
|
|
//linkVolume := strings.Replace(device, "/xvd", "/sd", 1)
|
|
linkVolume := device
|
|
|
|
ui.Say(fmt.Sprintf("Attaching the root volume to %s", linkVolume))
|
|
_, _, err := oscconn.VolumeApi.LinkVolume(context.Background(), &osc.LinkVolumeOpts{
|
|
LinkVolumeRequest: optional.NewInterface(osc.LinkVolumeRequest{
|
|
VmId: vm.VmId,
|
|
VolumeId: volumeId,
|
|
DeviceName: linkVolume,
|
|
}),
|
|
})
|
|
|
|
if err != nil {
|
|
err := fmt.Errorf("Error attaching volume: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
// Mark that we attached it so we can detach it later
|
|
s.attached = true
|
|
s.volumeId = volumeId
|
|
|
|
// Wait for the volume to become attached
|
|
err = osccommon.WaitUntilOscVolumeIsLinked(oscconn, s.volumeId)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error waiting for volume: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
state.Put("attach_cleanup", s)
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *StepLinkVolume) Cleanup(state multistep.StateBag) {
|
|
ui := state.Get("ui").(packer.Ui)
|
|
if err := s.CleanupFunc(state); err != nil {
|
|
ui.Error(err.Error())
|
|
}
|
|
}
|
|
|
|
func (s *StepLinkVolume) CleanupFunc(state multistep.StateBag) error {
|
|
if !s.attached {
|
|
return nil
|
|
}
|
|
|
|
oscconn := state.Get("osc").(*osc.APIClient)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
ui.Say("Detaching BSU volume...")
|
|
_, _, err := oscconn.VolumeApi.UnlinkVolume(context.Background(), &osc.UnlinkVolumeOpts{
|
|
UnlinkVolumeRequest: optional.NewInterface(osc.UnlinkVolumeRequest{VolumeId: s.volumeId}),
|
|
})
|
|
|
|
if err != nil {
|
|
return fmt.Errorf("Error detaching BSU volume: %s", err)
|
|
}
|
|
|
|
s.attached = false
|
|
|
|
// Wait for the volume to detach
|
|
err = osccommon.WaitUntilOscVolumeIsUnlinked(oscconn, s.volumeId)
|
|
if err != nil {
|
|
return fmt.Errorf("Error waiting for volume: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|