forked from rebuy-de/aws-nuke
-
Notifications
You must be signed in to change notification settings - Fork 0
/
qldb_ledger.go
101 lines (83 loc) · 2.21 KB
/
qldb_ledger.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
package resources
import (
"time"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/qldb"
"github.com/dngferreira/aws-nuke/v2/pkg/config"
"github.com/dngferreira/aws-nuke/v2/pkg/types"
)
type QLDBLedger struct {
svc *qldb.QLDB
ledger *qldb.DescribeLedgerOutput
featureFlags config.FeatureFlags
}
func init() {
register("QLDBLedger", ListQLDBLedgers)
}
func ListQLDBLedgers(sess *session.Session) ([]Resource, error) {
svc := qldb.New(sess)
params := &qldb.ListLedgersInput{}
resources := make([]Resource, 0)
for {
resp, err := svc.ListLedgers(params)
if err != nil {
return nil, err
}
for _, ledger := range resp.Ledgers {
ledgerDescription, err := svc.DescribeLedger(
&qldb.DescribeLedgerInput{Name: ledger.Name},
)
if err != nil {
return nil, err
}
resources = append(resources, &QLDBLedger{
svc: svc,
ledger: ledgerDescription,
})
}
if resp.NextToken == nil {
break
}
params.NextToken = resp.NextToken
}
return resources, nil
}
func (l *QLDBLedger) FeatureFlags(ff config.FeatureFlags) {
l.featureFlags = ff
}
func (l *QLDBLedger) Remove() error {
if aws.BoolValue(l.ledger.DeletionProtection) &&
l.featureFlags.DisableDeletionProtection.QLDBLedger {
modifyParams := &qldb.UpdateLedgerInput{
DeletionProtection: aws.Bool(false),
Name: l.ledger.Name,
}
_, err := l.svc.UpdateLedger((modifyParams))
if err != nil {
return err
}
}
params := &qldb.DeleteLedgerInput{
Name: l.ledger.Name,
}
_, err := l.svc.DeleteLedger(params)
if err != nil {
return err
}
return nil
}
func (l *QLDBLedger) Properties() types.Properties {
properties := types.NewProperties()
properties.Set("Name", l.ledger.Name)
properties.Set("DeletionProtection", l.ledger.DeletionProtection)
properties.Set("Arn", l.ledger.Arn)
properties.Set("CreationDateTime", l.ledger.CreationDateTime.Format(time.RFC3339))
properties.Set("State", l.ledger.State)
properties.Set("PermissionsMode", l.ledger.PermissionsMode)
properties.Set("EncryptionDescription", l.ledger.EncryptionDescription)
return properties
}
func (l *QLDBLedger) String() string {
return aws.StringValue(l.ledger.Name)
}