Merge pull request #23 from rebuy-de/extend-aws-nuke

Extend aws nuke for Lambda functions and IAM Policy Versions

Author: Florian Zeidler

resources/iam-policies.go

@@ -30,7 +30,25 @@ func (n *IamNuke) ListPolicies() ([]Resource, error) {
 }
 
 func (e *IamPolicy) Remove() error {
-	_, err := e.svc.DeletePolicy(&iam.DeletePolicyInput{
+	resp, err := e.svc.ListPolicyVersions(&iam.ListPolicyVersionsInput{
+		PolicyArn: &e.arn,
+	})
+	if err != nil {
+		return err
+	}
+	for _, version := range resp.Versions {
+		if !*version.IsDefaultVersion {
+			_, err = e.svc.DeletePolicyVersion(&iam.DeletePolicyVersionInput{
+				PolicyArn: &e.arn,
+				VersionId: version.VersionId,
+			})
+			if err != nil {
+				return err
+			}
+
+		}
+	}
+	_, err = e.svc.DeletePolicy(&iam.DeletePolicyInput{
 		PolicyArn: &e.arn,
 	})
 	if err != nil {

resources/lambda-functions.go

@@ -0,0 +1,39 @@
+package resources
+
+import "github.com/aws/aws-sdk-go/service/lambda"
+
+type LambdaFunction struct {
+	svc          *lambda.Lambda
+	functionName *string
+}
+
+func (n *LambdaNuke) ListFunctions() ([]Resource, error) {
+	params := &lambda.ListFunctionsInput{}
+	resp, err := n.Service.ListFunctions(params)
+	if err != nil {
+		return nil, err
+	}
+
+	resources := make([]Resource, 0)
+	for _, function := range resp.Functions {
+		resources = append(resources, &LambdaFunction{
+			svc:          n.Service,
+			functionName: function.FunctionName,
+		})
+	}
+
+	return resources, nil
+}
+
+func (f *LambdaFunction) Remove() error {
+
+	_, err := f.svc.DeleteFunction(&lambda.DeleteFunctionInput{
+		FunctionName: f.functionName,
+	})
+
+	return err
+}
+
+func (f *LambdaFunction) String() string {
+	return *f.functionName
+}

resources/listers.go

@@ -12,6 +12,7 @@ import (
 	"github.com/aws/aws-sdk-go/service/elasticache"
 	"github.com/aws/aws-sdk-go/service/elb"
 	"github.com/aws/aws-sdk-go/service/iam"
+	"github.com/aws/aws-sdk-go/service/lambda"
 	"github.com/aws/aws-sdk-go/service/rds"
 	"github.com/aws/aws-sdk-go/service/route53"
 	"github.com/aws/aws-sdk-go/service/s3"
@@ -30,6 +31,7 @@ func GetListers(sess *session.Session) []ResourceLister {
 		elasticache      = ElasticacheNuke{elasticache.New(sess)}
 		elb              = ElbNuke{elb.New(sess)}
 		iam              = IamNuke{iam.New(sess)}
+		lambda           = LambdaNuke{lambda.New(sess)}
 		rds              = RDSNuke{rds.New(sess)}
 		route53          = Route53Nuke{route53.New(sess)}
 		s3               = S3Nuke{s3.New(sess)}
@@ -79,6 +81,7 @@ func GetListers(sess *session.Session) []ResourceLister {
 		iam.ListUserGroupAttachements,
 		iam.ListUserPolicyAttachements,
 		iam.ListUsers,
+		lambda.ListFunctions,
 		rds.ListInstances,
 		rds.ListParameterGroups,
 		rds.ListSnapshots,

resources/types.go

@@ -11,6 +11,7 @@ import (
 	"github.com/aws/aws-sdk-go/service/elasticache"
 	"github.com/aws/aws-sdk-go/service/elb"
 	"github.com/aws/aws-sdk-go/service/iam"
+	"github.com/aws/aws-sdk-go/service/lambda"
 	"github.com/aws/aws-sdk-go/service/rds"
 	"github.com/aws/aws-sdk-go/service/route53"
 	"github.com/aws/aws-sdk-go/service/s3"
@@ -57,6 +58,10 @@ type IamNuke struct {
 	Service *iam.IAM
 }
 
+type LambdaNuke struct {
+	Service *lambda.Lambda
+}
+
 type RDSNuke struct {
 	Service *rds.RDS
 }