package common

import (
	"github.com/hashicorp/packer/helper/multistep"
	"github.com/outscale/osc-go/oapi"
)

type BuildInfoTemplate struct {
	BuildRegion   string
	SourceOMI     string
	SourceOMIName string
	SourceOMITags map[string]string
}

func extractBuildInfo(region string, state multistep.StateBag) *BuildInfoTemplate {
	rawSourceOMI, hasSourceOMI := state.GetOk("source_image")
	if !hasSourceOMI {
		return &BuildInfoTemplate{
			BuildRegion: region,
		}
	}

	sourceOMI := rawSourceOMI.(oapi.Image)
	sourceOMITags := make(map[string]string, len(sourceOMI.Tags))
	for _, tag := range sourceOMI.Tags {
		sourceOMITags[tag.Key] = tag.Value
	}

	return &BuildInfoTemplate{
		BuildRegion:   region,
		SourceOMI:     sourceOMI.ImageId,
		SourceOMIName: sourceOMI.ImageName,
		SourceOMITags: sourceOMITags,
	}
}