Add Transfer Server & User resources

resources/transfer-server-user.go

@@ -0,0 +1,108 @@
+package resources
+
+import (
+	"fmt"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/transfer"
+	"github.com/rebuy-de/aws-nuke/pkg/types"
+)
+
+type TransferServerUser struct {
+	svc      *transfer.Transfer
+	username *string
+	serverID *string
+	tags     []*transfer.Tag
+}
+
+func init() {
+	register("TransferServerUser", ListTransferServerUsers)
+}
+
+func ListTransferServerUsers(sess *session.Session) ([]Resource, error) {
+	svc := transfer.New(sess)
+	resources := []Resource{}
+
+	params := &transfer.ListServersInput{
+		MaxResults: aws.Int64(50),
+	}
+
+	for {
+		output, err := svc.ListServers(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, item := range output.Servers {
+			userParams := &transfer.ListUsersInput{
+				MaxResults: aws.Int64(100),
+				ServerId:   item.ServerId,
+			}
+
+			for {
+				userOutput, err := svc.ListUsers(userParams)
+				if err != nil {
+					return nil, err
+				}
+
+				for _, user := range userOutput.Users {
+					descOutput, err := svc.DescribeUser(&transfer.DescribeUserInput{
+						ServerId: item.ServerId,
+						UserName: user.UserName,
+					})
+					if err != nil {
+						return nil, err
+					}
+
+					resources = append(resources, &TransferServerUser{
+						svc:      svc,
+						username: user.UserName,
+						serverID: item.ServerId,
+						tags:     descOutput.User.Tags,
+					})
+
+				}
+
+				if userOutput.NextToken == nil {
+					break
+				}
+
+				userParams.NextToken = userOutput.NextToken
+			}
+		}
+
+		if output.NextToken == nil {
+			break
+		}
+
+		params.NextToken = output.NextToken
+	}
+
+	return resources, nil
+}
+
+func (ts *TransferServerUser) Remove() error {
+
+	_, err := ts.svc.DeleteUser(&transfer.DeleteUserInput{
+		ServerId: ts.serverID,
+		UserName: ts.username,
+	})
+
+	return err
+}
+
+func (ts *TransferServerUser) String() string {
+	return fmt.Sprintf("%s -> %s", *ts.serverID, *ts.username)
+}
+
+func (ts *TransferServerUser) Properties() types.Properties {
+	properties := types.NewProperties()
+	for _, tag := range ts.tags {
+		properties.SetTag(tag.Key, tag.Value)
+	}
+	properties.
+		Set("Username", ts.username).
+		Set("ServerID", ts.serverID)
+	return properties
+}

resources/transfer-server.go

@@ -0,0 +1,93 @@
+package resources
+
+import (
+	"strings"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/transfer"
+	"github.com/rebuy-de/aws-nuke/pkg/types"
+)
+
+type TransferServer struct {
+	svc          *transfer.Transfer
+	serverID     *string
+	endpointType *string
+	protocols    []string
+	tags         []*transfer.Tag
+}
+
+func init() {
+	register("TransferServer", ListTransferServers)
+}
+
+func ListTransferServers(sess *session.Session) ([]Resource, error) {
+	svc := transfer.New(sess)
+	resources := []Resource{}
+
+	params := &transfer.ListServersInput{
+		MaxResults: aws.Int64(50),
+	}
+
+	for {
+		output, err := svc.ListServers(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, item := range output.Servers {
+			descOutput, err := svc.DescribeServer(&transfer.DescribeServerInput{
+				ServerId: item.ServerId,
+			})
+			if err != nil {
+				return nil, err
+			}
+
+			protocols := []string{}
+			for _, protocol := range descOutput.Server.Protocols {
+				protocols = append(protocols, *protocol)
+			}
+
+			resources = append(resources, &TransferServer{
+				svc:          svc,
+				serverID:     item.ServerId,
+				endpointType: item.EndpointType,
+				protocols:    protocols,
+				tags:         descOutput.Server.Tags,
+			})
+		}
+
+		if output.NextToken == nil {
+			break
+		}
+
+		params.NextToken = output.NextToken
+	}
+
+	return resources, nil
+}
+
+func (ts *TransferServer) Remove() error {
+
+	_, err := ts.svc.DeleteServer(&transfer.DeleteServerInput{
+		ServerId: ts.serverID,
+	})
+
+	return err
+}
+
+func (ts *TransferServer) String() string {
+	return *ts.serverID
+}
+
+func (ts *TransferServer) Properties() types.Properties {
+	properties := types.NewProperties()
+	for _, tag := range ts.tags {
+		properties.SetTag(tag.Key, tag.Value)
+	}
+	properties.
+		Set("ServerID", ts.serverID).
+		Set("EndpointType", ts.endpointType).
+		Set("Protocols", strings.Join(ts.protocols, ", "))
+	return properties
+}