forked from rebuy-de/aws-nuke
/
appstream-stack-fleet-attachments.go
76 lines (59 loc) · 1.55 KB
/
appstream-stack-fleet-attachments.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package resources
import (
"fmt"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/appstream"
)
type AppStreamStackFleetAttachment struct {
svc *appstream.AppStream
stackName *string
fleetName *string
}
func init() {
register("AppStreamStackFleetAttachment", ListAppStreamStackFleetAttachments)
}
func ListAppStreamStackFleetAttachments(sess *session.Session) ([]Resource, error) {
svc := appstream.New(sess)
resources := []Resource{}
stacks := []*appstream.Stack{}
params := &appstream.DescribeStacksInput{}
for {
output, err := svc.DescribeStacks(params)
if err != nil {
return nil, err
}
for _, stack := range output.Stacks {
stacks = append(stacks, stack)
}
if output.NextToken == nil {
break
}
params.NextToken = output.NextToken
}
stackAssocParams := &appstream.ListAssociatedFleetsInput{}
for _, stack := range stacks {
stackAssocParams.StackName = stack.Name
output, err := svc.ListAssociatedFleets(stackAssocParams)
if err != nil {
return nil, err
}
for _, name := range output.Names {
resources = append(resources, &AppStreamStackFleetAttachment{
svc: svc,
stackName: stack.Name,
fleetName: name,
})
}
}
return resources, nil
}
func (f *AppStreamStackFleetAttachment) Remove() error {
_, err := f.svc.DisassociateFleet(&appstream.DisassociateFleetInput{
StackName: f.stackName,
FleetName: f.fleetName,
})
return err
}
func (f *AppStreamStackFleetAttachment) String() string {
return fmt.Sprintf("%s -> %s", *f.stackName, *f.fleetName)
}