-
Notifications
You must be signed in to change notification settings - Fork 23
/
instances.go
68 lines (55 loc) · 1.81 KB
/
instances.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
package ec2
import (
"fmt"
"strings"
"github.com/aws/aws-sdk-go/aws"
awsec2 "github.com/aws/aws-sdk-go/service/ec2"
"github.com/genevieve/leftovers/common"
)
type instancesClient interface {
DescribeInstances(*awsec2.DescribeInstancesInput) (*awsec2.DescribeInstancesOutput, error)
TerminateInstances(*awsec2.TerminateInstancesInput) (*awsec2.TerminateInstancesOutput, error)
DescribeAddresses(*awsec2.DescribeAddressesInput) (*awsec2.DescribeAddressesOutput, error)
ReleaseAddress(*awsec2.ReleaseAddressInput) (*awsec2.ReleaseAddressOutput, error)
}
type Instances struct {
client instancesClient
logger logger
resourceTags resourceTags
}
func NewInstances(client instancesClient, logger logger, resourceTags resourceTags) Instances {
return Instances{
client: client,
logger: logger,
resourceTags: resourceTags,
}
}
func (i Instances) List(filter string) ([]common.Deletable, error) {
instances, err := i.client.DescribeInstances(&awsec2.DescribeInstancesInput{
Filters: []*awsec2.Filter{{
Name: aws.String("instance-state-name"),
Values: []*string{aws.String("pending"), aws.String("running"), aws.String("shutting-down"), aws.String("stopping"), aws.String("stopped")},
}},
})
if err != nil {
return nil, fmt.Errorf("Describing EC2 Instances: %s", err)
}
var resources []common.Deletable
for _, r := range instances.Reservations {
for _, instance := range r.Instances {
r := NewInstance(i.client, i.logger, i.resourceTags, instance.InstanceId, instance.KeyName, instance.Tags)
if !strings.Contains(r.Name(), filter) {
continue
}
proceed := i.logger.PromptWithDetails(r.Type(), r.Name())
if !proceed {
continue
}
resources = append(resources, r)
}
}
return resources, nil
}
func (i Instances) Type() string {
return "ec2-instance"
}