From 70d2ff3d8a80df7507ee5cb07e9d2f6ae91bd289 Mon Sep 17 00:00:00 2001
From: Marin Salinas <marinssalinas@gmail.com>
Date: Mon, 4 Mar 2019 10:02:55 -0600
Subject: [PATCH] feature: add create omi step in chroot builder

---
 builder/osc/chroot/builder.go         |  10 +++
 builder/osc/chroot/step_create_omi.go | 112 ++++++++++++++++++++++++++
 2 files changed, 122 insertions(+)
 create mode 100644 builder/osc/chroot/step_create_omi.go

diff --git a/builder/osc/chroot/builder.go b/builder/osc/chroot/builder.go
index d4340ce44..2cf0e4ece 100644
--- a/builder/osc/chroot/builder.go
+++ b/builder/osc/chroot/builder.go
@@ -264,6 +264,16 @@ func (b *Builder) Run(ui packer.Ui, hook packer.Hook, cache packer.Cache) (packe
 		&StepChrootProvision{},
 		&StepEarlyCleanup{},
 		&StepSnapshot{},
+		&osccommon.StepDeregisterOMI{
+			AccessConfig:        &b.config.AccessConfig,
+			ForceDeregister:     b.config.OMIForceDeregister,
+			ForceDeleteSnapshot: b.config.OMIForceDeleteSnapshot,
+			OMIName:             b.config.OMIName,
+			Regions:             b.config.OMIRegions,
+		},
+		&StepCreateOMI{
+			RootVolumeSize: b.config.RootVolumeSize,
+		},
 	)
 
 	// Run!
diff --git a/builder/osc/chroot/step_create_omi.go b/builder/osc/chroot/step_create_omi.go
new file mode 100644
index 000000000..1dd69979c
--- /dev/null
+++ b/builder/osc/chroot/step_create_omi.go
@@ -0,0 +1,112 @@
+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
+}