packer-cn/builder/docker/step_connect_docker.go

77 lines
2.0 KiB
Go
Raw Normal View History

package docker
import (
"context"
"fmt"
"os/exec"
"strings"
2018-01-22 20:21:10 -05:00
"github.com/hashicorp/packer/helper/multistep"
)
type StepConnectDocker struct{}
func (s *StepConnectDocker) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
2015-07-16 22:07:39 -04:00
config := state.Get("config").(*Config)
containerId := state.Get("container_id").(string)
2015-05-29 12:29:59 -04:00
driver := state.Get("driver").(Driver)
tempDir := state.Get("temp_dir").(string)
2015-05-29 12:29:59 -04:00
// Get the version so we can pass it to the communicator
version, err := driver.Version()
if err != nil {
state.Put("error", err)
return multistep.ActionHalt
}
containerUser, err := getContainerUser(containerId)
if err != nil {
state.Put("error", err)
return multistep.ActionHalt
}
// Create the communicator that talks to Docker via various
// os/exec tricks.
if config.WindowsContainer {
comm := &WindowsContainerCommunicator{Communicator{
ContainerID: containerId,
HostDir: tempDir,
ContainerDir: config.ContainerDir,
Version: version,
Config: config,
ContainerUser: containerUser,
2019-03-29 14:14:01 -04:00
EntryPoint: []string{"powershell"},
},
}
state.Put("communicator", comm)
} else {
comm := &Communicator{
ContainerID: containerId,
HostDir: tempDir,
ContainerDir: config.ContainerDir,
Version: version,
Config: config,
ContainerUser: containerUser,
2019-03-29 14:14:01 -04:00
EntryPoint: []string{"/bin/sh", "-c"},
}
state.Put("communicator", comm)
}
return multistep.ActionContinue
}
func (s *StepConnectDocker) Cleanup(state multistep.StateBag) {}
func getContainerUser(containerId string) (string, error) {
inspectArgs := []string{"docker", "inspect", "--format", "{{.Config.User}}", containerId}
stdout, err := exec.Command(inspectArgs[0], inspectArgs[1:]...).Output()
if err != nil {
errStr := fmt.Sprintf("Failed to inspect the container: %s", err)
if ee, ok := err.(*exec.ExitError); ok {
errStr = fmt.Sprintf("%s, %s", errStr, ee.Stderr)
}
return "", fmt.Errorf(errStr)
}
return strings.TrimSpace(string(stdout)), nil
}