/
inspector2.go
62 lines (49 loc) · 1.22 KB
/
inspector2.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
package resources
import (
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/inspector2"
"github.com/rebuy-de/aws-nuke/v2/pkg/types"
)
type Inspector2 struct {
svc *inspector2.Inspector2
accountId *string
}
func init() {
register("Inspector2", ListInspector2)
}
func ListInspector2(sess *session.Session) ([]Resource, error) {
svc := inspector2.New(sess)
resources := make([]Resource, 0)
resp, err := svc.BatchGetAccountStatus(nil)
if err != nil {
return resources, err
}
for _, a := range resp.Accounts {
if *a.State.Status != inspector2.StatusDisabled {
resources = append(resources, &Inspector2{
svc: svc,
accountId: a.AccountId,
})
}
}
return resources, nil
}
func (e *Inspector2) Remove() error {
_, err := e.svc.Disable(&inspector2.DisableInput{
AccountIds: []*string{e.accountId},
ResourceTypes: aws.StringSlice(inspector2.ResourceScanType_Values()),
})
if err != nil {
return err
}
return nil
}
func (e *Inspector2) Properties() types.Properties {
properties := types.NewProperties()
properties.Set("AccountID", e.accountId)
return properties
}
func (e *Inspector2) String() string {
return *e.accountId
}