Add CloudHSMv2 support

Author: tomvachon

resources/cloudhsmv2-cluster.go

@@ -0,0 +1,60 @@
+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/cloudhsmv2"
+)
+
+type CloudHSMV2Cluster struct {
+	svc       *cloudhsmv2.CloudHSMV2
+	clusterID *string
+}
+
+func init() {
+	register("CloudHSMV2Cluster", ListCloudHSMV2Clusters)
+}
+
+func ListCloudHSMV2Clusters(sess *session.Session) ([]Resource, error) {
+	svc := cloudhsmv2.New(sess)
+	resources := []Resource{}
+
+	params := &cloudhsmv2.DescribeClustersInput{
+		MaxResults: aws.Int64(25),
+	}
+
+	for {
+		resp, err := svc.DescribeClusters(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, cluster := range resp.Clusters {
+			resources = append(resources, &CloudHSMV2Cluster{
+				svc:       svc,
+				clusterID: cluster.ClusterId,
+			})
+		}
+
+		if resp.NextToken == nil {
+			break
+		}
+
+		params.NextToken = resp.NextToken
+	}
+
+	return resources, nil
+}
+
+func (f *CloudHSMV2Cluster) Remove() error {
+
+	_, err := f.svc.DeleteCluster(&cloudhsmv2.DeleteClusterInput{
+		ClusterId: f.clusterID,
+	})
+
+	return err
+}
+
+func (f *CloudHSMV2Cluster) String() string {
+	return *f.clusterID
+}

resources/cloudhsmv2-clusterhsms.go

@@ -0,0 +1,72 @@
+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/cloudhsmv2"
+)
+
+type CloudHSMV2ClusterHSM struct {
+	svc       *cloudhsmv2.CloudHSMV2
+	clusterID *string
+	hsmIDs    []*string
+}
+
+func init() {
+	register("CloudHSMV2ClusterHSM", ListCloudHSMV2ClusterHSMs)
+}
+
+func ListCloudHSMV2ClusterHSMs(sess *session.Session) ([]Resource, error) {
+	svc := cloudhsmv2.New(sess)
+	resources := []Resource{}
+
+	params := &cloudhsmv2.DescribeClustersInput{
+		MaxResults: aws.Int64(25),
+	}
+
+	for {
+		resp, err := svc.DescribeClusters(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, cluster := range resp.Clusters {
+			var hsmIds []*string
+			for _, hsm := range cluster.Hsms {
+				hsmIds = append(hsmIds, hsm.HsmId)
+			}
+			resources = append(resources, &CloudHSMV2ClusterHSM{
+				svc:       svc,
+				clusterID: cluster.ClusterId,
+				hsmIDs:    hsmIds,
+			})
+		}
+
+		if resp.NextToken == nil {
+			break
+		}
+
+		params.NextToken = resp.NextToken
+	}
+
+	return resources, nil
+}
+
+func (f *CloudHSMV2ClusterHSM) Remove() error {
+	var err error
+	for _, hsm := range f.hsmIDs {
+		_, err := f.svc.DeleteHsm(&cloudhsmv2.DeleteHsmInput{
+			ClusterId: f.clusterID,
+			HsmId:     hsm,
+		})
+		if err != nil {
+			return err
+		}
+	}
+
+	return err
+}
+
+func (f *CloudHSMV2ClusterHSM) String() string {
+	return *f.clusterID
+}