60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
package docker
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/mitchellh/multistep"
|
|
"github.com/mitchellh/packer/packer"
|
|
)
|
|
|
|
type StepRun struct {
|
|
containerId string
|
|
}
|
|
|
|
func (s *StepRun) Run(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: map[string]string{
|
|
tempDir: "/packer-files",
|
|
},
|
|
}
|
|
|
|
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 = ""
|
|
}
|