func MakeCliApp(
diegoVersion string,
latticeVersion string,
ltcConfigRoot string,
exitHandler exit_handler.ExitHandler,
config *config.Config,
logger lager.Logger,
receptorClientCreator receptor_client.Creator,
targetVerifier target_verifier.TargetVerifier,
cliStdout io.Writer,
) *cli.App {
config.Load()
app := cli.NewApp()
app.Name = AppName
app.Author = latticeCliAuthor
app.Version = defaultVersion(diegoVersion, latticeVersion)
app.Usage = LtcUsage
app.Email = "*****@*****.**"
ui := terminal.NewUI(os.Stdin, cliStdout, password_reader.NewPasswordReader(exitHandler))
app.Writer = ui
app.Before = func(context *cli.Context) error {
args := context.Args()
command := app.Command(args.First())
if command == nil {
return nil
}
if _, ok := nonTargetVerifiedCommandNames[command.Name]; ok || len(args) == 0 {
return nil
}
if receptorUp, authorized, err := targetVerifier.VerifyTarget(config.Receptor()); !receptorUp {
ui.SayLine(fmt.Sprintf("Error connecting to the receptor. Make sure your lattice target is set, and that lattice is up and running.\n\tUnderlying error: %s", err.Error()))
return err
} else if !authorized {
ui.SayLine("Could not authenticate with the receptor. Please run ltc target with the correct credentials.")
return errors.New("Could not authenticate with the receptor.")
}
return nil
}
app.Action = defaultAction
app.CommandNotFound = func(c *cli.Context, command string) {
ui.SayLine(fmt.Sprintf(unknownCommand, command))
exitHandler.Exit(1)
}
app.Commands = cliCommands(ltcConfigRoot, exitHandler, config, logger, receptorClientCreator, targetVerifier, ui)
return app
}
Describe("VerifyTarget", func() {
var (
fakeReceptorClient *fake_receptor.FakeClient
fakeReceptorClientCreator *fake_receptor_client_creator.FakeCreator
targetVerifier target_verifier.TargetVerifier
)
BeforeEach(func() {
fakeReceptorClient = &fake_receptor.FakeClient{}
fakeReceptorClientCreator = &fake_receptor_client_creator.FakeCreator{}
fakeReceptorClientCreator.CreateReceptorClientReturns(fakeReceptorClient)
targetVerifier = target_verifier.New(fakeReceptorClientCreator)
})
It("returns up=true, auth=true if the receptor does not return an error", func() {
up, auth, err := targetVerifier.VerifyTarget("http://receptor.mylattice.com")
Expect(err).NotTo(HaveOccurred())
Expect(up).To(BeTrue())
Expect(auth).To(BeTrue())
Expect(fakeReceptorClientCreator.CreateReceptorClientCallCount()).To(Equal(1))
Expect(fakeReceptorClientCreator.CreateReceptorClientArgsForCall(0)).To(Equal("http://receptor.mylattice.com"))
})
It("returns up=true, auth=false if the receptor returns an authorization error", func() {
fakeReceptorClient.DesiredLRPsReturns([]receptor.DesiredLRPResponse{}, receptor.Error{
Type: receptor.Unauthorized,
Message: "Go home. You're not welcome here.",
})
up, auth, err := targetVerifier.VerifyTarget("http://receptor.mylattice.com")
Expect(err).NotTo(HaveOccurred())