Merge pull request #168 from tomvachon/feature/storagegateway

Add StorageGateway Coverage

Author: svenwltr

resources/storagegateway-fileshares.go

@@ -0,0 +1,61 @@
+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/storagegateway"
+)
+
+type StorageGatewayFileShare struct {
+	svc *storagegateway.StorageGateway
+	ARN *string
+}
+
+func init() {
+	register("StorageGatewayFileShare", ListStorageGatewayFileShares)
+}
+
+func ListStorageGatewayFileShares(sess *session.Session) ([]Resource, error) {
+	svc := storagegateway.New(sess)
+	resources := []Resource{}
+
+	params := &storagegateway.ListFileSharesInput{
+		Limit: aws.Int64(25),
+	}
+
+	for {
+		output, err := svc.ListFileShares(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, fileShareInfo := range output.FileShareInfoList {
+			resources = append(resources, &StorageGatewayFileShare{
+				svc: svc,
+				ARN: fileShareInfo.FileShareARN,
+			})
+		}
+
+		if output.Marker == nil {
+			break
+		}
+
+		params.Marker = output.Marker
+	}
+
+	return resources, nil
+}
+
+func (f *StorageGatewayFileShare) Remove() error {
+
+	_, err := f.svc.DeleteFileShare(&storagegateway.DeleteFileShareInput{
+		FileShareARN: f.ARN,
+		ForceDelete:  aws.Bool(true),
+	})
+
+	return err
+}
+
+func (f *StorageGatewayFileShare) String() string {
+	return *f.ARN
+}

resources/storagegateway-gateways.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/storagegateway"
+)
+
+type StorageGatewayGateway struct {
+	svc *storagegateway.StorageGateway
+	ARN *string
+}
+
+func init() {
+	register("StorageGatewayGateway", ListStorageGatewayGateways)
+}
+
+func ListStorageGatewayGateways(sess *session.Session) ([]Resource, error) {
+	svc := storagegateway.New(sess)
+	resources := []Resource{}
+
+	params := &storagegateway.ListGatewaysInput{
+		Limit: aws.Int64(25),
+	}
+
+	for {
+		output, err := svc.ListGateways(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, gateway := range output.Gateways {
+			resources = append(resources, &StorageGatewayGateway{
+				svc: svc,
+				ARN: gateway.GatewayARN,
+			})
+		}
+
+		if output.Marker == nil {
+			break
+		}
+
+		params.Marker = output.Marker
+	}
+
+	return resources, nil
+}
+
+func (f *StorageGatewayGateway) Remove() error {
+
+	_, err := f.svc.DeleteGateway(&storagegateway.DeleteGatewayInput{
+		GatewayARN: f.ARN,
+	})
+
+	return err
+}
+
+func (f *StorageGatewayGateway) String() string {
+	return *f.ARN
+}

resources/storagegateway-tapes.go

@@ -0,0 +1,63 @@
+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/storagegateway"
+)
+
+type StorageGatewayTape struct {
+	svc        *storagegateway.StorageGateway
+	tapeARN    *string
+	gatewayARN *string
+}
+
+func init() {
+	register("StorageGatewayTape", ListStorageGatewayTapes)
+}
+
+func ListStorageGatewayTapes(sess *session.Session) ([]Resource, error) {
+	svc := storagegateway.New(sess)
+	resources := []Resource{}
+
+	params := &storagegateway.ListTapesInput{
+		Limit: aws.Int64(25),
+	}
+
+	for {
+		output, err := svc.ListTapes(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, tapeInfo := range output.TapeInfos {
+			resources = append(resources, &StorageGatewayTape{
+				svc:        svc,
+				tapeARN:    tapeInfo.TapeARN,
+				gatewayARN: tapeInfo.GatewayARN,
+			})
+		}
+
+		if output.Marker == nil {
+			break
+		}
+
+		params.Marker = output.Marker
+	}
+
+	return resources, nil
+}
+
+func (f *StorageGatewayTape) Remove() error {
+
+	_, err := f.svc.DeleteTape(&storagegateway.DeleteTapeInput{
+		TapeARN:    f.tapeARN,
+		GatewayARN: f.gatewayARN,
+	})
+
+	return err
+}
+
+func (f *StorageGatewayTape) String() string {
+	return *f.tapeARN
+}

resources/storagegateway-volumes.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/storagegateway"
+)
+
+type StorageGatewayVolume struct {
+	svc *storagegateway.StorageGateway
+	ARN *string
+}
+
+func init() {
+	register("StorageGatewayVolume", ListStorageGatewayVolumes)
+}
+
+func ListStorageGatewayVolumes(sess *session.Session) ([]Resource, error) {
+	svc := storagegateway.New(sess)
+	resources := []Resource{}
+
+	params := &storagegateway.ListVolumesInput{
+		Limit: aws.Int64(25),
+	}
+
+	for {
+		output, err := svc.ListVolumes(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, volumeInfo := range output.VolumeInfos {
+			resources = append(resources, &StorageGatewayVolume{
+				svc: svc,
+				ARN: volumeInfo.VolumeARN,
+			})
+		}
+
+		if output.Marker == nil {
+			break
+		}
+
+		params.Marker = output.Marker
+	}
+
+	return resources, nil
+}
+
+func (f *StorageGatewayVolume) Remove() error {
+
+	_, err := f.svc.DeleteVolume(&storagegateway.DeleteVolumeInput{
+		VolumeARN: f.ARN,
+	})
+
+	return err
+}
+
+func (f *StorageGatewayVolume) String() string {
+	return *f.ARN
+}