forked from rebuy-de/aws-nuke
/
macie2.go
48 lines (39 loc) · 880 Bytes
/
macie2.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
package resources
import (
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/macie2"
)
type Macie struct {
svc *macie2.Macie2
}
func init() {
register("Macie", CheckMacieStatus)
}
func CheckMacieStatus(sess *session.Session) ([]Resource, error) {
svc := macie2.New(sess)
status, err := svc.GetMacieSession(&macie2.GetMacieSessionInput{})
if err != nil {
if err.Error() == "AccessDeniedException: Macie is not enabled" {
return nil, nil
} else {
return nil, err
}
}
resources := make([]Resource, 0)
if *status.Status == macie2.AdminStatusEnabled {
resources = append(resources, &Macie{
svc: svc,
})
}
return resources, nil
}
func (b *Macie) Remove() error {
_, err := b.svc.DisableMacie(&macie2.DisableMacieInput{})
if err != nil {
return err
}
return nil
}
func (b *Macie) String() string {
return "macie"
}