diff --git a/resources/inspector2.go b/resources/inspector2.go index 89818b88..b66299ba 100644 --- a/resources/inspector2.go +++ b/resources/inspector2.go @@ -8,6 +8,7 @@ import ( "github.com/ekristen/libnuke/pkg/registry" "github.com/ekristen/libnuke/pkg/resource" + "github.com/ekristen/libnuke/pkg/types" "github.com/ekristen/aws-nuke/v3/pkg/nuke" ) @@ -39,7 +40,13 @@ func (l *Inspector2Lister) List(_ context.Context, o interface{}) ([]resource.Re if *a.State.Status != inspector2.StatusDisabled { resources = append(resources, &Inspector2{ svc: svc, - accountID: a.AccountId, + AccountID: a.AccountId, + ResourceState: map[string]string{ + inspector2.ResourceScanTypeEc2: *a.ResourceState.Ec2.Status, + inspector2.ResourceScanTypeEcr: *a.ResourceState.Ecr.Status, + inspector2.ResourceScanTypeLambda: *a.ResourceState.Lambda.Status, + inspector2.ResourceScanTypeLambdaCode: *a.ResourceState.LambdaCode.Status, + }, }) } } @@ -48,14 +55,25 @@ func (l *Inspector2Lister) List(_ context.Context, o interface{}) ([]resource.Re } type Inspector2 struct { - svc *inspector2.Inspector2 - accountID *string + svc *inspector2.Inspector2 + AccountID *string + ResourceState map[string]string `property:"tagPrefix=resourceType"` +} + +func (e *Inspector2) GetEnabledResources() []string { + var resources = make([]string, 0) + for k, v := range e.ResourceState { + if v == inspector2.StatusEnabled { + resources = append(resources, k) + } + } + return resources } func (e *Inspector2) Remove(_ context.Context) error { _, err := e.svc.Disable(&inspector2.DisableInput{ - AccountIds: []*string{e.accountID}, - ResourceTypes: aws.StringSlice(inspector2.ResourceScanType_Values()), + AccountIds: []*string{e.AccountID}, + ResourceTypes: aws.StringSlice(e.GetEnabledResources()), }) if err != nil { return err @@ -65,5 +83,9 @@ func (e *Inspector2) Remove(_ context.Context) error { } func (e *Inspector2) String() string { - return *e.accountID + return *e.AccountID +} + +func (e *Inspector2) Properties() types.Properties { + return types.NewPropertiesFromStruct(e) }