-
Notifications
You must be signed in to change notification settings - Fork 390
/
claims.go
126 lines (105 loc) · 2.45 KB
/
claims.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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package main
import (
"encoding/json"
"fmt"
"time"
"github.com/spf13/cobra"
"github.com/zeebo/errs"
"storj.io/storj/pkg/certificates"
"storj.io/storj/pkg/process"
)
var (
claimsCmd = &cobra.Command{
Use: "claims",
Short: "CSR authorization claim management",
}
claimsExportCmd = &cobra.Command{
Use: "export",
Short: "Export all claim data as JSON",
RunE: cmdExportClaims,
}
claimDeleteCmd = &cobra.Command{
Use: "delete",
Short: "Delete a claim on an authorization",
Args: cobra.ExactArgs(1),
RunE: cmdDeleteClaim,
}
claimsExportCfg struct {
Signer certificates.CertServerConfig
Raw bool `default:"false" help:"if true, the raw data structures will be printed"`
}
claimsDeleteCfg struct {
Signer certificates.CertServerConfig
}
)
func cmdExportClaims(cmd *cobra.Command, args []string) (err error) {
ctx := process.Ctx(cmd)
authDB, err := claimsExportCfg.Signer.NewAuthDB()
if err != nil {
return err
}
defer func() {
err = errs.Combine(err, authDB.Close())
}()
auths, err := authDB.List(ctx)
if err != nil {
return err
}
var toPrint []interface{}
for _, auth := range auths {
if claimsExportCfg.Raw {
toPrint = append(toPrint, auth)
} else {
toPrint = append(toPrint, toPrintableAuth(auth))
}
}
if len(toPrint) == 0 {
fmt.Printf("no claims in database: %s\n", claimsExportCfg.Signer.AuthorizationDBURL)
return nil
}
jsonBytes, err := json.MarshalIndent(toPrint, "", "\t")
if err != nil {
return err
}
fmt.Println(string(jsonBytes))
return err
}
func cmdDeleteClaim(cmd *cobra.Command, args []string) (err error) {
ctx := process.Ctx(cmd)
authDB, err := claimsDeleteCfg.Signer.NewAuthDB()
if err != nil {
return err
}
defer func() {
err = errs.Combine(err, authDB.Close())
}()
if err := authDB.Unclaim(ctx, args[0]); err != nil {
return err
}
return nil
}
type printableAuth struct {
UserID string
Token string
Claim *printableClaim
}
type printableClaim struct {
Addr string
Time string
NodeID string
}
func toPrintableAuth(auth *certificates.Authorization) *printableAuth {
pAuth := new(printableAuth)
pAuth.UserID = auth.Token.UserID
pAuth.Token = auth.Token.String()
if auth.Claim != nil {
pAuth.Claim = &printableClaim{
Time: time.Unix(auth.Claim.Timestamp, 0).String(),
Addr: auth.Claim.Addr,
NodeID: auth.Claim.Identity.ID.String(),
}
}
return pAuth
}