Add Firehose Delivery Streams

resources/firehose-deliverystreams.go

@@ -0,0 +1,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/firehose"
+)
+
+type FirehoseDeliveryStream struct {
+	svc                *firehose.Firehose
+	deliveryStreamName *string
+}
+
+func init() {
+	register("FirehoseDeliveryStream", ListFirehoseDeliveryStreams)
+}
+
+func ListFirehoseDeliveryStreams(sess *session.Session) ([]Resource, error) {
+	svc := firehose.New(sess)
+	resources := []Resource{}
+	var lastDeliveryStreamName *string
+
+	params := &firehose.ListDeliveryStreamsInput{
+		Limit: aws.Int64(25),
+	}
+
+	for {
+		output, err := svc.ListDeliveryStreams(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, deliveryStreamName := range output.DeliveryStreamNames {
+			resources = append(resources, &FirehoseDeliveryStream{
+				svc:                svc,
+				deliveryStreamName: deliveryStreamName,
+			})
+			lastDeliveryStreamName = deliveryStreamName
+		}
+
+		if *output.HasMoreDeliveryStreams == false {
+			break
+		}
+
+		params.ExclusiveStartDeliveryStreamName = lastDeliveryStreamName
+	}
+
+	return resources, nil
+}
+
+func (f *FirehoseDeliveryStream) Remove() error {
+
+	_, err := f.svc.DeleteDeliveryStream(&firehose.DeleteDeliveryStreamInput{
+		DeliveryStreamName: f.deliveryStreamName,
+	})
+
+	return err
+}
+
+func (f *FirehoseDeliveryStream) String() string {
+	return *f.deliveryStreamName
+}