-
Notifications
You must be signed in to change notification settings - Fork 394
/
claims.go
118 lines (98 loc) · 2.31 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
// 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/common/process"
"storj.io/storj/certificate/authorization"
)
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,
}
)
func cmdExportClaims(cmd *cobra.Command, args []string) (err error) {
ctx, _ := process.Ctx(cmd)
authDB, err := authorization.OpenDBFromCfg(ctx, claimsExportCfg.Config.AuthorizationDB)
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.Config.AuthorizationDB.URL)
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 := authorization.OpenDBFromCfg(ctx, claimsDeleteCfg.AuthorizationDB)
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 *authorization.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
}