command_application.go

package apm

import (
	"strconv"

	log "github.com/sirupsen/logrus"
	"github.com/spf13/cobra"

	"github.com/newrelic/newrelic-client-go/pkg/entities"

	"github.com/newrelic/newrelic-cli/internal/client"
	configAPI "github.com/newrelic/newrelic-cli/internal/config/api"
	"github.com/newrelic/newrelic-cli/internal/output"
	"github.com/newrelic/newrelic-cli/internal/utils"
)

var (
	appName string
	appGUID string
)

// Command represents the apm command
var cmdApp = &cobra.Command{
	Use:     "application",
	Short:   "Interact with New Relic APM applications",
	Example: "newrelic apm application --help",
	Long:    "Interact with New Relic APM applications",
}

var cmdAppSearch = &cobra.Command{
	Use:   "search",
	Short: "Search for a New Relic application",
	Long: `Search for a New Relic application

The search command performs a query for an APM application name and/or account ID.
`,
	Example: "newrelic apm application search --name <appName>",
	PreRun:  client.RequireClient,
	Run: func(cmd *cobra.Command, args []string) {
		accountID := configAPI.GetActiveProfileAccountID()

		if appGUID == "" && appName == "" && accountID == 0 {
			utils.LogIfError(cmd.Help())
			log.Fatal("one of --accountId, --guid, --name are required")
		}

		var entityResults []entities.EntityOutlineInterface
		var err error

		// Look for just the GUID if passed in
		if appGUID != "" {
			if appName != "" || accountID != 0 {
				log.Warnf("Searching for --guid only, ignoring --accountId and --name")
			}

			var singleResult *entities.EntityInterface
			singleResult, err = client.NRClient.Entities.GetEntity(entities.EntityGUID(appGUID))
			utils.LogIfFatal(err)
			utils.LogIfFatal(output.Print(*singleResult))
		} else {
			params := entities.EntitySearchQueryBuilder{
				Domain: entities.EntitySearchQueryBuilderDomain("APM"),
				Type:   entities.EntitySearchQueryBuilderType("APPLICATION"),
			}

			if appName != "" {
				params.Name = appName
			}

			if accountID != 0 {
				params.Tags = []entities.EntitySearchQueryBuilderTag{{Key: "accountId", Value: strconv.Itoa(accountID)}}
			}

			results, err := client.NRClient.Entities.GetEntitySearch(
				entities.EntitySearchOptions{},
				"",
				params,
				[]entities.EntitySearchSortCriteria{},
			)

			entityResults = results.Results.Entities
			utils.LogIfFatal(err)
		}

		utils.LogIfFatal(output.Print(entityResults))
	},
}

//
var cmdAppGet = &cobra.Command{
	Use:   "get",
	Short: "Get a New Relic application",
	Long: `Get a New Relic application

The get command performs a query for an APM application by GUID.
`,
	Example: "newrelic apm application get --guid <entityGUID>",
	PreRun:  client.RequireClient,
	Run: func(cmd *cobra.Command, args []string) {
		var results *entities.EntityInterface
		var err error

		if appGUID != "" {
			results, err = client.NRClient.Entities.GetEntity(entities.EntityGUID(appGUID))
			utils.LogIfFatal(err)
		} else {
			utils.LogIfError(cmd.Help())
			log.Fatal(" --guid <entityGUID> is required")
		}

		utils.LogIfFatal(output.Print(results))
	},
}

func init() {
	Command.AddCommand(cmdApp)

	cmdApp.PersistentFlags().StringVarP(&appGUID, "guid", "g", "", "search for results matching the given APM application GUID")

	cmdApp.AddCommand(cmdAppGet)

	cmdApp.AddCommand(cmdAppSearch)
	cmdAppSearch.Flags().StringVarP(&appName, "name", "n", "", "search for results matching the given APM application name")
}