packer-cn/helper/communicator/step_connect.go

95 lines
2.5 KiB
Go
Raw Normal View History

2015-06-13 17:42:38 -04:00
package communicator
import (
2015-06-13 17:50:45 -04:00
"fmt"
"log"
2017-04-04 16:39:01 -04:00
"github.com/hashicorp/packer/communicator/none"
"github.com/hashicorp/packer/packer"
2015-06-13 17:42:38 -04:00
"github.com/mitchellh/multistep"
2016-02-08 20:34:06 -05:00
gossh "golang.org/x/crypto/ssh"
2015-06-13 17:42:38 -04:00
)
// StepConnect is a multistep Step implementation that connects to
// the proper communicator and stores it in the "communicator" key in the
// state bag.
type StepConnect struct {
// Config is the communicator config struct
Config *Config
// Host should return a host that can be connected to for communicator
// connections.
Host func(multistep.StateBag) (string, error)
2015-06-13 17:42:38 -04:00
// The fields below are callbacks to assist with connecting to SSH.
//
// SSHConfig should return the default configuration for
// connecting via SSH.
SSHConfig func(multistep.StateBag) (*gossh.ClientConfig, error)
SSHPort func(multistep.StateBag) (int, error)
2015-06-13 17:42:38 -04:00
2015-06-14 01:07:17 -04:00
// The fields below are callbacks to assist with connecting to WinRM.
//
// WinRMConfig should return the default configuration for
// connecting via WinRM.
WinRMConfig func(multistep.StateBag) (*WinRMConfig, error)
WinRMPort func(multistep.StateBag) (int, error)
2015-06-14 01:07:17 -04:00
// CustomConnect can be set to have custom connectors for specific
// types. These take highest precedence so you can also override
// existing types.
CustomConnect map[string]multistep.Step
2015-06-13 17:42:38 -04:00
substep multistep.Step
}
func (s *StepConnect) Run(state multistep.StateBag) multistep.StepAction {
2015-06-13 17:50:45 -04:00
typeMap := map[string]multistep.Step{
"none": nil,
"ssh": &StepConnectSSH{
Config: s.Config,
Host: s.Host,
SSHConfig: s.SSHConfig,
SSHPort: s.SSHPort,
2015-06-13 17:50:45 -04:00
},
2015-06-14 01:07:17 -04:00
"winrm": &StepConnectWinRM{
Config: s.Config,
Host: s.Host,
WinRMConfig: s.WinRMConfig,
2016-11-06 11:02:00 -05:00
WinRMPort: s.WinRMPort,
2015-06-14 01:07:17 -04:00
},
2015-06-13 17:50:45 -04:00
}
for k, v := range s.CustomConnect {
typeMap[k] = v
}
2015-06-13 17:50:45 -04:00
step, ok := typeMap[s.Config.Type]
if !ok {
state.Put("error", fmt.Errorf("unknown communicator type: %s", s.Config.Type))
return multistep.ActionHalt
}
if step == nil {
comm, err := none.New("none")
if err != nil {
err := fmt.Errorf("Failed to set communicator 'none': %s", err)
state.Put("error", err)
ui := state.Get("ui").(packer.Ui)
ui.Error(err.Error())
return multistep.ActionHalt
}
2016-02-08 20:34:06 -05:00
state.Put("communicator", comm)
2015-06-13 17:50:45 -04:00
log.Printf("[INFO] communicator disabled, will not connect")
return multistep.ActionContinue
2015-06-13 17:42:38 -04:00
}
2015-06-13 17:50:45 -04:00
s.substep = step
2015-06-13 17:42:38 -04:00
return s.substep.Run(state)
}
func (s *StepConnect) Cleanup(state multistep.StateBag) {
if s.substep != nil {
s.substep.Cleanup(state)
}
}