Merge pull request #6 from oreillymedia/Add-Elasticache-User-and-Group-Support

Add elasticache user and group support

Author: swhite-oreilly

resources/elasticache-subnetgroups.go

@@ -1,6 +1,9 @@
 package resources
 
 import (
+	"fmt"
+	"strings"
+
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/elasticache"
@@ -35,6 +38,13 @@ func ListElasticacheSubnetGroups(sess *session.Session) ([]Resource, error) {
 	return resources, nil
 }
 
+func (i *ElasticacheSubnetGroup) Filter() error {
+	if strings.HasPrefix(*i.name, "default") {
+		return fmt.Errorf("Cannot delete default subnet group")
+	}
+	return nil
+}
+
 func (i *ElasticacheSubnetGroup) Remove() error {
 	params := &elasticache.DeleteCacheSubnetGroupInput{
 		CacheSubnetGroupName: i.name,

resources/elasticache-usergroups.go

@@ -0,0 +1,53 @@
+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/elasticache"
+)
+
+type ElasticacheUserGroup struct {
+	svc     *elasticache.ElastiCache
+	groupId *string
+}
+
+func init() {
+	register("ElasticacheUserGroup", ListElasticacheUserGroups)
+}
+
+func ListElasticacheUserGroups(sess *session.Session) ([]Resource, error) {
+	svc := elasticache.New(sess)
+
+	params := &elasticache.DescribeUserGroupsInput{MaxRecords: aws.Int64(100)}
+	resp, err := svc.DescribeUserGroups(params)
+	if err != nil {
+		return nil, err
+	}
+	var resources []Resource
+	for _, userGroup := range resp.UserGroups {
+		resources = append(resources, &ElasticacheUserGroup{
+			svc:     svc,
+			groupId: userGroup.UserGroupId,
+		})
+
+	}
+
+	return resources, nil
+}
+
+func (i *ElasticacheUserGroup) Remove() error {
+	params := &elasticache.DeleteUserGroupInput{
+		UserGroupId: i.groupId,
+	}
+
+	_, err := i.svc.DeleteUserGroup(params)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (i *ElasticacheUserGroup) String() string {
+	return *i.groupId
+}

resources/elasticache-users.go

@@ -0,0 +1,65 @@
+package resources
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/elasticache"
+)
+
+type ElasticacheUser struct {
+	svc      *elasticache.ElastiCache
+	userId   *string
+	userName *string
+}
+
+func init() {
+	register("ElasticacheUser", ListElasticacheUsers)
+}
+
+func ListElasticacheUsers(sess *session.Session) ([]Resource, error) {
+	svc := elasticache.New(sess)
+
+	params := &elasticache.DescribeUsersInput{MaxRecords: aws.Int64(100)}
+	resp, err := svc.DescribeUsers(params)
+	if err != nil {
+		return nil, err
+	}
+	var resources []Resource
+	for _, user := range resp.Users {
+		resources = append(resources, &ElasticacheUser{
+			svc:      svc,
+			userId:   user.UserId,
+			userName: user.UserName,
+		})
+
+	}
+
+	return resources, nil
+}
+
+func (i *ElasticacheUser) Filter() error {
+	if strings.HasPrefix(*i.userName, "default") {
+		return fmt.Errorf("Cannot delete default user")
+	}
+	return nil
+}
+
+func (i *ElasticacheUser) Remove() error {
+	params := &elasticache.DeleteUserInput{
+		UserId: i.userId,
+	}
+
+	_, err := i.svc.DeleteUser(params)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (i *ElasticacheUser) String() string {
+	return *i.userId
+}