func (p *SSHPlugin) Run(cli plugin.CliConnection, args []string) { p.OutputWriter = os.Stdout appFactory := app.NewAppFactory(cli, models.Curl) spaceFactory := space.NewSpaceFactory(cli, models.Curl) switch args[0] { case "CLI-MESSAGE-UNINSTALL": return case "enable-ssh": err := cmd.EnableSSH(args, appFactory) if err != nil { p.Fatal(err) } case "disable-ssh": err := cmd.DisableSSH(args, appFactory) if err != nil { p.Fatal(err) } case "ssh-enabled": err := cmd.SSHEnabled(args, appFactory, p.OutputWriter) if err != nil { p.Fatal(err) } case "allow-space-ssh": err := cmd.AllowSSH(args, spaceFactory) if err != nil { p.Fatal(err) } case "disallow-space-ssh": err := cmd.DisallowSSH(args, spaceFactory) if err != nil { p.Fatal(err) } case "space-ssh-allowed": err := cmd.SSHAllowed(args, spaceFactory, p.OutputWriter) if err != nil { p.Fatal(err) } case "ssh": opts := options.NewSSHOptions() err := opts.Parse(args) if err != nil { p.Fail(err.Error()) fmt.Fprintf(p.OutputWriter, options.SSHUsage()) return } secureShell := cmd.NewSecureShell( cmd.DefaultSecureDialer(), terminal.DefaultHelper(), cmd.DefaultListenerFactory(), 30*time.Second, appFactory, info.NewInfoFactory(cli), credential.NewCredentialFactory(cli), ) err = secureShell.Connect(opts) if err != nil { p.Fail(err.Error()) return } defer secureShell.Close() err = secureShell.LocalPortForward() if err != nil { return } if opts.SkipRemoteExecution { err = secureShell.Wait() } else { err = secureShell.InteractiveSession() } if err == nil { return } if exitError, ok := err.(*ssh.ExitError); ok { exitStatus := exitError.ExitStatus() if sig := exitError.Signal(); sig != "" { fmt.Printf("Process terminated by signal: %s. Exited with %d.\n", sig, exitStatus) } os.Exit(exitStatus) } else { p.Fail(err.Error()) } default: p.Fail("Invalid command") } }
var _ = Describe("Credential", func() { var ( fakeCliConnection *fakes.FakeCliConnection fakeInfoFactory *info_fakes.FakeInfoFactory credFactory credential.CredentialFactory ) BeforeEach(func() { fakeCliConnection = &fakes.FakeCliConnection{} fakeCliConnection.IsSSLDisabledReturns(true, nil) fakeInfoFactory = &info_fakes.FakeInfoFactory{} }) JustBeforeEach(func() { credFactory = credential.NewCredentialFactory(fakeCliConnection, fakeInfoFactory) }) Describe("AuthorizationCode", func() { var v2Info info.Info var fakeUAA *ghttp.Server BeforeEach(func() { fakeCliConnection.AccessTokenReturns("bearer client-bearer-token", nil) fakeUAA = ghttp.NewTLSServer() v2Info = info.Info{ SSHOAuthClient: "ssh-oauth-client-id", TokenEndpoint: fakeUAA.URL(), } fakeInfoFactory.GetReturns(v2Info, nil)