48 lines
1.0 KiB
Go
48 lines
1.0 KiB
Go
|
package agent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"forge.cadoles.com/Cadoles/emissary/internal/client"
|
||
|
|
"forge.cadoles.com/Cadoles/emissary/internal/command/client/apierr"
|
||
|
|
clientFlag "forge.cadoles.com/Cadoles/emissary/internal/command/client/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)
|
||
|
|
client := client.New(baseFlags.ServerURL)
|
||
|
|
|
||
|
|
_, 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
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|