package docker import ( "context" "fmt" "github.com/hashicorp/packer/helper/multistep" "github.com/hashicorp/packer/packer" ) type StepRun struct { containerId string } func (s *StepRun) Run(_ context.Context, state multistep.StateBag) multistep.StepAction { config := state.Get("config").(*Config) driver := state.Get("driver").(Driver) tempDir := state.Get("temp_dir").(string) ui := state.Get("ui").(packer.Ui) runConfig := ContainerConfig{ Image: config.Image, RunCommand: config.RunCommand, Volumes: make(map[string]string), Privileged: config.Privileged, } for host, container := range config.Volumes { runConfig.Volumes[host] = container } runConfig.Volumes[tempDir] = config.ContainerDir ui.Say("Starting docker container...") containerId, err := driver.StartContainer(&runConfig) if err != nil { err := fmt.Errorf("Error running container: %s", err) state.Put("error", err) ui.Error(err.Error()) return multistep.ActionHalt } // Save the container ID s.containerId = containerId state.Put("container_id", s.containerId) ui.Message(fmt.Sprintf("Container ID: %s", s.containerId)) return multistep.ActionContinue } func (s *StepRun) Cleanup(state multistep.StateBag) { if s.containerId == "" { return } driver := state.Get("driver").(Driver) ui := state.Get("ui").(packer.Ui) // Kill the container. We don't handle errors because errors usually // just mean that the container doesn't exist anymore, which isn't a // big deal. ui.Say(fmt.Sprintf("Killing the container: %s", s.containerId)) driver.StopContainer(s.containerId) // Reset the container ID so that we're idempotent s.containerId = "" }