forked from spiffe/spire
-
Notifications
You must be signed in to change notification settings - Fork 4
/
list.go
65 lines (52 loc) · 1.84 KB
/
list.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
package federation
import (
"context"
"flag"
"fmt"
"github.com/accuknox/spire/cmd/spire-server/util"
commoncli "github.com/accuknox/spire/pkg/common/cli"
"github.com/accuknox/spire/pkg/common/cliprinter"
"github.com/mitchellh/cli"
trustdomainv1 "github.com/spiffe/spire-api-sdk/proto/spire/api/server/trustdomain/v1"
)
func NewListCommand() cli.Command {
return newListCommand(commoncli.DefaultEnv)
}
func newListCommand(env *commoncli.Env) cli.Command {
return util.AdaptCommand(env, &listCommand{env: env})
}
type listCommand struct {
env *commoncli.Env
printer cliprinter.Printer
}
func (c *listCommand) Name() string {
return "federation list"
}
func (c *listCommand) Synopsis() string {
return "Lists all dynamic federation relationships"
}
func (c *listCommand) AppendFlags(fs *flag.FlagSet) {
cliprinter.AppendFlagWithCustomPretty(&c.printer, fs, c.env, prettyPrintList)
}
func (c *listCommand) Run(ctx context.Context, env *commoncli.Env, serverClient util.ServerClient) error {
trustDomainClient := serverClient.NewTrustDomainClient()
resp, err := trustDomainClient.ListFederationRelationships(ctx, &trustdomainv1.ListFederationRelationshipsRequest{})
if err != nil {
return fmt.Errorf("error listing federation relationship: %w", err)
}
return c.printer.PrintProto(resp)
}
func prettyPrintList(env *commoncli.Env, results ...interface{}) error {
listResp, ok := results[0].(*trustdomainv1.ListFederationRelationshipsResponse)
if !ok {
return cliprinter.ErrInternalCustomPrettyFunc
}
msg := fmt.Sprintf("Found %v ", len(listResp.FederationRelationships))
msg = util.Pluralizer(msg, "federation relationship", "federation relationships", len(listResp.FederationRelationships))
env.Println(msg)
for _, fr := range listResp.FederationRelationships {
env.Println()
printFederationRelationship(fr, env.Printf)
}
return nil
}