github.com/rahart/packer@v0.12.2-0.20161229105310-282bb6ad370f/builder/openstack/step_add_image_members.go (about)

     1  package openstack
     2  
     3  import (
     4  	"fmt"
     5  
     6  	"github.com/gophercloud/gophercloud/openstack/imageservice/v2/members"
     7  	"github.com/mitchellh/multistep"
     8  	"github.com/mitchellh/packer/packer"
     9  )
    10  
    11  type stepAddImageMembers struct{}
    12  
    13  func (s *stepAddImageMembers) Run(state multistep.StateBag) multistep.StepAction {
    14  	imageId := state.Get("image").(string)
    15  	ui := state.Get("ui").(packer.Ui)
    16  	config := state.Get("config").(Config)
    17  
    18  	if len(config.ImageMembers) == 0 {
    19  		return multistep.ActionContinue
    20  	}
    21  
    22  	imageClient, err := config.imageV2Client()
    23  	if err != nil {
    24  		err = fmt.Errorf("Error initializing image service client: %s", err)
    25  		state.Put("error", err)
    26  		return multistep.ActionHalt
    27  	}
    28  
    29  	for _, member := range config.ImageMembers {
    30  		ui.Say(fmt.Sprintf("Adding member '%s' to image %s", member, imageId))
    31  		r := members.Create(imageClient, imageId, member)
    32  		if _, err = r.Extract(); err != nil {
    33  			err = fmt.Errorf("Error adding member to image: %s", err)
    34  			state.Put("error", err)
    35  			return multistep.ActionHalt
    36  		}
    37  	}
    38  
    39  	return multistep.ActionContinue
    40  }
    41  
    42  func (s *stepAddImageMembers) Cleanup(multistep.StateBag) {
    43  	// No cleanup...
    44  }