// SSHCommandFactory is the factory method for SSHCommand.
func SSHCommandFactory(c *cli.Context, log logging.Logger, _ string) int {
if len(c.Args()) != 1 {
cli.ShowCommandHelp(c, "ssh")
return 1
}
if c.Bool("debug") {
log.SetLevel(logging.DEBUG)
}
opts := ssh.SSHCommandOpts{
Debug: c.Bool("debug") || config.Konfig.Debug,
RemoteUsername: c.String("username"),
Ask: true,
}
cmd, err := ssh.NewSSHCommand(log, opts)
mountName := c.Args()[0]
// TODO: Refactor SSHCommand instance to require no initialization,
// and thus avoid needing to log an error in a weird place.
if err != nil {
log.Error("Error initializing ssh: %s", err)
switch err {
case ssh.ErrLocalDialingFailed:
fmt.Println(
defaultHealthChecker.CheckAllFailureOrMessagef(KlientIsntRunning),
)
default:
fmt.Println(GenericInternalError)
}
metrics.TrackSSHFailed(mountName, err.Error(), config.VersionNum())
return 1
}
now := time.Now()
// track metrics
go func() {
metrics.TrackSSH(mountName, config.VersionNum())
}()
err = cmd.Run(mountName)
switch err {
case nil:
metrics.TrackSSHEnd(mountName, "", -now.Sub(now).Minutes(), config.VersionNum())
return 0
case ssh.ErrMachineNotFound:
fmt.Println(MachineNotFound)
case ssh.ErrCannotFindUser:
fmt.Println(CannotFindSSHUser)
metrics.TrackSSHFailed(mountName, err.Error(), config.VersionNum())
case ssh.ErrFailedToGetSSHKey:
fmt.Println(FailedGetSSHKey)
metrics.TrackSSHFailed(mountName, err.Error(), config.VersionNum())
case ssh.ErrMachineNotValidYet:
fmt.Println(defaultHealthChecker.CheckAllFailureOrMessagef(MachineNotValidYet))
metrics.TrackSSHFailed(mountName, err.Error(), config.VersionNum())
case ssh.ErrRemoteDialingFailed:
fmt.Println(defaultHealthChecker.CheckAllFailureOrMessagef(FailedDialingRemote))
metrics.TrackSSHFailed(mountName, err.Error(), config.VersionNum())
case shortcut.ErrMachineNotFound:
fmt.Println(MachineNotFound)
metrics.TrackSSHFailed(mountName, err.Error(), config.VersionNum())
}
log.Error("SSHCommand.Run returned err:%s", err)
// ssh returns `exit status 255` on disconnection; so we also send how long
// session has been running for to indicate if ssh was successful at least
// once and the failed due to disconnection
metrics.TrackSSHEnd(mountName, err.Error(), -now.Sub(now).Minutes(), config.VersionNum())
return 1
}