/
ssh.go
108 lines (91 loc) · 2.61 KB
/
ssh.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package cmd
import (
. "github.com/cloudfoundry/bosh-cli/v6/cmd/opts"
boshdir "github.com/cloudfoundry/bosh-cli/v6/director"
boshssh "github.com/cloudfoundry/bosh-cli/v6/ssh"
boshui "github.com/cloudfoundry/bosh-cli/v6/ui"
bosherr "github.com/cloudfoundry/bosh-utils/errors"
boshuuid "github.com/cloudfoundry/bosh-utils/uuid"
)
type SSHCmd struct {
deployment boshdir.Deployment
uuidGen boshuuid.Generator
intSSHRunner boshssh.Runner
nonIntSSHRunner boshssh.Runner
resultsSSHRunner boshssh.Runner
ui boshui.UI
hostBuilder boshssh.HostBuilder
}
func NewSSHCmd(
uuidGen boshuuid.Generator,
intSSHRunner boshssh.Runner,
nonIntSSHRunner boshssh.Runner,
resultsSSHRunner boshssh.Runner,
ui boshui.UI,
hostBuilder boshssh.HostBuilder,
) SSHCmd {
return SSHCmd{
uuidGen: uuidGen,
intSSHRunner: intSSHRunner,
nonIntSSHRunner: nonIntSSHRunner,
resultsSSHRunner: resultsSSHRunner,
ui: ui,
hostBuilder: hostBuilder,
}
}
func (c SSHCmd) Run(opts SSHOpts, deploymentFetcher boshssh.DeploymentFetcher) error {
if opts.Results || !c.ui.IsInteractive() {
if len(opts.Command) == 0 {
return bosherr.Errorf("Non-interactive SSH requires non-empty command")
}
}
sshOpts, connOpts, err := opts.GatewayFlags.AsSSHOpts()
if err != nil {
return err
}
connOpts.RawOpts = opts.RawOpts.AsStrings()
var result boshdir.SSHResult
if opts.PrivateKey.Bytes == nil {
c.deployment, err = deploymentFetcher()
if err != nil {
return err
}
// host key will be returned by agent over NATS
connOpts.RawOpts = append(connOpts.RawOpts, "-o", "StrictHostKeyChecking=yes")
result, err = c.deployment.SetUpSSH(opts.Args.Slug, sshOpts)
if err != nil {
return err
}
defer func() {
_ = c.deployment.CleanUpSSH(opts.Args.Slug, sshOpts)
}()
} else {
// no automatic source of host key
connOpts.RawOpts = append(connOpts.RawOpts, "-o", "StrictHostKeyChecking=no")
connOpts.PrivateKey = string(opts.PrivateKey.Bytes)
host, err := c.hostBuilder.BuildHost(opts.Args.Slug, opts.Username, deploymentFetcher)
if err != nil {
return err
}
result = boshdir.SSHResult{
Hosts: []boshdir.Host{
host,
},
GatewayUsername: connOpts.GatewayUsername,
GatewayHost: connOpts.GatewayHost,
}
}
var runner boshssh.Runner
if opts.Results {
runner = c.resultsSSHRunner
} else if !c.ui.IsInteractive() || len(opts.Command) > 0 {
runner = c.nonIntSSHRunner
} else {
runner = c.intSSHRunner
}
err = runner.Run(connOpts, result, opts.Command)
if err != nil {
return bosherr.WrapErrorf(err, "Running SSH")
}
return nil
}