/
ctl_access_token_list_cmd.go
47 lines (39 loc) · 1.43 KB
/
ctl_access_token_list_cmd.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
package main
import (
"context"
"fmt"
adminv1 "github.com/nanzhong/tstr/api/admin/v1"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"google.golang.org/protobuf/encoding/protojson"
)
var (
ctlAccessTokenListIncludeExpired bool
ctlAccessTokenListIncludeRevoked bool
ctlAccessTokenListCmd = &cobra.Command{
Use: "list",
Short: "Retrieve information for all access tokens",
Args: cobra.ExactArgs(0),
RunE: func(cmd *cobra.Command, args []string) error {
ctx, cancel := context.WithTimeout(context.Background(), viper.GetDuration("ctl.timeout"))
defer cancel()
req := &adminv1.ListAccessTokensRequest{
IncludeExpired: ctlAccessTokenListIncludeRevoked,
IncludeRevoked: ctlAccessTokenListIncludeRevoked,
}
return withAdminClient(ctx, viper.GetString("ctl.grpc-addr"), !viper.GetBool("ctl.insecure"), viper.GetString("ctl.access-token"), func(ctx context.Context, client adminv1.AdminServiceClient) error {
res, err := client.ListAccessTokens(ctx, req)
if err != nil {
return err
}
fmt.Println(protojson.Format(res))
return nil
})
},
}
)
func init() {
ctlAccessTokenListCmd.Flags().BoolVar(&ctlAccessTokenListIncludeExpired, "include-expired", false, "Include expired tokens in result.")
ctlAccessTokenListCmd.Flags().BoolVar(&ctlAccessTokenListIncludeRevoked, "include-revoked", false, "Include revoked tokens in result.")
accessTokenCmd.AddCommand(ctlAccessTokenListCmd)
}