113 lines
3.0 KiB
Go
113 lines
3.0 KiB
Go
|
package chroot
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
osccommon "github.com/hashicorp/packer/builder/osc/common"
|
||
|
|
"github.com/hashicorp/packer/helper/multistep"
|
||
|
|
"github.com/hashicorp/packer/packer"
|
||
|
|
"github.com/outscale/osc-go/oapi"
|
||
|
|
)
|
||
|
|
|
||
|
|
// StepCreateOMI creates the OMI.
|
||
|
|
type StepCreateOMI struct {
|
||
|
|
RootVolumeSize int64
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *StepCreateOMI) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
||
|
|
config := state.Get("config").(*Config)
|
||
|
|
oapiconn := state.Get("oapi").(*oapi.Client)
|
||
|
|
snapshotId := state.Get("snapshot_id").(string)
|
||
|
|
ui := state.Get("ui").(packer.Ui)
|
||
|
|
|
||
|
|
ui.Say("Creating the OMI...")
|
||
|
|
|
||
|
|
var (
|
||
|
|
registerOpts oapi.CreateImageRequest
|
||
|
|
mappings []oapi.BlockDeviceMappingImage
|
||
|
|
image oapi.Image
|
||
|
|
rootDeviceName string
|
||
|
|
)
|
||
|
|
|
||
|
|
if config.FromScratch {
|
||
|
|
mappings = config.OMIBlockDevices.BuildOMIDevices()
|
||
|
|
rootDeviceName = config.RootDeviceName
|
||
|
|
} else {
|
||
|
|
image = state.Get("source_image").(oapi.Image)
|
||
|
|
mappings = image.BlockDeviceMappings
|
||
|
|
rootDeviceName = image.RootDeviceName
|
||
|
|
}
|
||
|
|
|
||
|
|
newMappings := make([]oapi.BlockDeviceMappingImage, len(mappings))
|
||
|
|
for i, device := range mappings {
|
||
|
|
newDevice := device
|
||
|
|
|
||
|
|
//FIX: Temporary fix
|
||
|
|
gibSize := newDevice.Bsu.VolumeSize / (1024 * 1024 * 1024)
|
||
|
|
newDevice.Bsu.VolumeSize = gibSize
|
||
|
|
|
||
|
|
if newDevice.DeviceName == rootDeviceName {
|
||
|
|
if &newDevice.Bsu != nil {
|
||
|
|
newDevice.Bsu.SnapshotId = snapshotId
|
||
|
|
} else {
|
||
|
|
newDevice.Bsu = oapi.BsuToCreate{SnapshotId: snapshotId}
|
||
|
|
}
|
||
|
|
|
||
|
|
if config.FromScratch || s.RootVolumeSize > newDevice.Bsu.VolumeSize {
|
||
|
|
newDevice.Bsu.VolumeSize = s.RootVolumeSize
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
newMappings[i] = newDevice
|
||
|
|
}
|
||
|
|
|
||
|
|
if config.FromScratch {
|
||
|
|
registerOpts = oapi.CreateImageRequest{
|
||
|
|
ImageName: config.OMIName,
|
||
|
|
Architecture: "x86_64",
|
||
|
|
RootDeviceName: rootDeviceName,
|
||
|
|
BlockDeviceMappings: newMappings,
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
registerOpts = buildRegisterOpts(config, image, newMappings)
|
||
|
|
}
|
||
|
|
|
||
|
|
registerResp, err := oapiconn.POST_CreateImage(registerOpts)
|
||
|
|
if err != nil {
|
||
|
|
state.Put("error", fmt.Errorf("Error registering OMI: %s", err))
|
||
|
|
ui.Error(state.Get("error").(error).Error())
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
|
||
|
|
imageID := registerResp.OK.Image.ImageId
|
||
|
|
|
||
|
|
// Set the OMI ID in the state
|
||
|
|
ui.Say(fmt.Sprintf("OMI: %s", imageID))
|
||
|
|
omis := make(map[string]string)
|
||
|
|
omis[oapiconn.GetConfig().Region] = imageID
|
||
|
|
state.Put("omis", omis)
|
||
|
|
|
||
|
|
ui.Say("Waiting for OMI to become ready...")
|
||
|
|
if err := osccommon.WaitUntilImageAvailable(oapiconn, imageID); err != nil {
|
||
|
|
err := fmt.Errorf("Error waiting for OMI: %s", err)
|
||
|
|
state.Put("error", err)
|
||
|
|
ui.Error(err.Error())
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
|
||
|
|
return multistep.ActionContinue
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *StepCreateOMI) Cleanup(state multistep.StateBag) {}
|
||
|
|
|
||
|
|
func buildRegisterOpts(config *Config, image oapi.Image, mappings []oapi.BlockDeviceMappingImage) oapi.CreateImageRequest {
|
||
|
|
registerOpts := oapi.CreateImageRequest{
|
||
|
|
ImageName: config.OMIName,
|
||
|
|
Architecture: image.Architecture,
|
||
|
|
RootDeviceName: image.RootDeviceName,
|
||
|
|
BlockDeviceMappings: mappings,
|
||
|
|
}
|
||
|
|
return registerOpts
|
||
|
|
}
|