emissary/internal/command/api/agent/count.go

package agent

import (
	"os"

	"forge.cadoles.com/Cadoles/emissary/internal/client"
	"forge.cadoles.com/Cadoles/emissary/internal/command/api/apierr"
	clientFlag "forge.cadoles.com/Cadoles/emissary/internal/command/api/flag"
	"forge.cadoles.com/Cadoles/emissary/internal/format"
	"github.com/pkg/errors"
	"github.com/urfave/cli/v2"
)

func CountCommand() *cli.Command {
	return &cli.Command{
		Name:  "count",
		Usage: "Count agents",
		Flags: clientFlag.ComposeFlags(),
		Action: func(ctx *cli.Context) error {
			baseFlags := clientFlag.GetBaseFlags(ctx)

			token, err := clientFlag.GetToken(baseFlags)
			if err != nil {
				return errors.WithStack(apierr.Wrap(err))
			}

			client := client.New(baseFlags.ServerURL, client.WithToken(token))

			_, total, err := client.QueryAgents(ctx.Context)
			if err != nil {
				return errors.WithStack(apierr.Wrap(err))
			}

			hints := format.Hints{
				OutputMode: baseFlags.OutputMode,
			}

			results := []struct {
				Total int `json:"total"`
			}{
				{
					Total: total,
				},
			}

			if err := format.Write(baseFlags.Format, os.Stdout, hints, clientFlag.AsAnySlice(results)...); err != nil {
				return errors.WithStack(err)
			}

			return nil
		},
	}
}