45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
|
package openstack
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/gophercloud/gophercloud/openstack/imageservice/v2/members"
|
||
|
|
"github.com/mitchellh/multistep"
|
||
|
|
"github.com/mitchellh/packer/packer"
|
||
|
|
)
|
||
|
|
|
||
|
|
type stepAddImageMembers struct{}
|
||
|
|
|
||
|
|
func (s *stepAddImageMembers) Run(state multistep.StateBag) multistep.StepAction {
|
||
|
|
imageId := state.Get("image").(string)
|
||
|
|
ui := state.Get("ui").(packer.Ui)
|
||
|
|
config := state.Get("config").(Config)
|
||
|
|
|
||
|
|
if len(config.ImageMembers) == 0 {
|
||
|
|
return multistep.ActionContinue
|
||
|
|
}
|
||
|
|
|
||
|
|
imageClient, err := config.imageV2Client()
|
||
|
|
if err != nil {
|
||
|
|
err = fmt.Errorf("Error initializing image service client: %s", err)
|
||
|
|
state.Put("error", err)
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, member := range config.ImageMembers {
|
||
|
|
ui.Say(fmt.Sprintf("Adding member '%s' to image %s", member, imageId))
|
||
|
|
r := members.Create(imageClient, imageId, member)
|
||
|
|
if _, err = r.Extract(); err != nil {
|
||
|
|
err = fmt.Errorf("Error adding member to image: %s", err)
|
||
|
|
state.Put("error", err)
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return multistep.ActionContinue
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *stepAddImageMembers) Cleanup(multistep.StateBag) {
|
||
|
|
// No cleanup...
|
||
|
|
}
|