125 lines
3.3 KiB
Go
125 lines
3.3 KiB
Go
package openstack
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/gophercloud/gophercloud/openstack/imageservice/v2/images"
|
|
"github.com/gophercloud/gophercloud/pagination"
|
|
"github.com/hashicorp/packer/helper/multistep"
|
|
"github.com/hashicorp/packer/packer"
|
|
)
|
|
|
|
type StepSourceImageInfo struct {
|
|
SourceImage string
|
|
SourceImageName string
|
|
SourceImageOpts images.ListOpts
|
|
SourceMostRecent bool
|
|
SourceProperties map[string]string
|
|
}
|
|
|
|
func PropertiesSatisfied(image *images.Image, props *map[string]string) bool {
|
|
for key, value := range *props {
|
|
if image.Properties[key] != value {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func (s *StepSourceImageInfo) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
|
config := state.Get("config").(*Config)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
if s.SourceImage != "" {
|
|
state.Put("source_image", s.SourceImage)
|
|
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
client, err := config.imageV2Client()
|
|
|
|
if s.SourceImageName != "" {
|
|
s.SourceImageOpts = images.ListOpts{
|
|
Name: s.SourceImageName,
|
|
}
|
|
}
|
|
|
|
log.Printf("Using Image Filters %+v", s.SourceImageOpts)
|
|
image := &images.Image{}
|
|
count := 0
|
|
err = images.List(client, s.SourceImageOpts).EachPage(func(page pagination.Page) (bool, error) {
|
|
imgs, err := images.ExtractImages(page)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
ui.Message(fmt.Sprintf("Resulting images: %d", len(imgs)))
|
|
|
|
for index, img := range imgs {
|
|
ui.Message(fmt.Sprintf("index +%v, image %+v", index, img))
|
|
ui.Message(fmt.Sprintf("Metadata %+v", img.Metadata))
|
|
ui.Message(fmt.Sprintf("Properties %+v", img.Properties))
|
|
|
|
// Check if all Properties are satisfied
|
|
if PropertiesSatisfied(&img, &s.SourceProperties) {
|
|
ui.Message(fmt.Sprintf("Matched properties %+v", s.SourceProperties))
|
|
count++
|
|
if count == 1 {
|
|
// Tentatively return this result.
|
|
*image = img
|
|
}
|
|
// Don't iterate over entries we will never use.
|
|
if count > 1 {
|
|
break
|
|
}
|
|
} else {
|
|
ui.Message(fmt.Sprintf("FAILED to match properties %+v", s.SourceProperties))
|
|
}
|
|
}
|
|
|
|
ui.Message(fmt.Sprintf("Count = %v", count))
|
|
switch count {
|
|
case 0: // Continue looking at next page.
|
|
return true, nil
|
|
case 1: // Maybe we're done, maybe there is another result in a later page and it is an error.
|
|
if s.SourceMostRecent {
|
|
return false, nil
|
|
}
|
|
return true, nil
|
|
default: // By now we should know if getting 2+ results is an error or not.
|
|
if s.SourceMostRecent {
|
|
return false, nil
|
|
}
|
|
return false, fmt.Errorf(
|
|
"Your query returned more than one result. Please try a more specific search, or set most_recent to true. Search filters: %+v properties %+v",
|
|
s.SourceImageOpts, s.SourceProperties)
|
|
}
|
|
})
|
|
|
|
if err != nil {
|
|
err := fmt.Errorf("Error querying image: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
if image.ID == "" {
|
|
err := fmt.Errorf("No image was found matching filters: %+v properties %+v",
|
|
s.SourceImageOpts, s.SourceProperties)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
ui.Message(fmt.Sprintf("Found Image ID: %s", image.ID))
|
|
|
|
state.Put("source_image", image.ID)
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *StepSourceImageInfo) Cleanup(state multistep.StateBag) {
|
|
// No cleanup required for backout
|
|
}
|