Skip to content
This repository was archived by the owner on Oct 15, 2024. It is now read-only.

Commit 2d9db0e

Browse files
tomvachontomvachon
committed
Add SSM Support
1 parent d7697a9 commit 2d9db0e

7 files changed

+440
-0
lines changed

‎resources/ssm-activations.go

+60
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,60 @@
1+
package resources
2+
3+
import (
4+
"github.com/aws/aws-sdk-go/aws"
5+
"github.com/aws/aws-sdk-go/aws/session"
6+
"github.com/aws/aws-sdk-go/service/ssm"
7+
)
8+
9+
type SSMActivation struct {
10+
svc *ssm.SSM
11+
ID *string
12+
}
13+
14+
func init() {
15+
register("SSMActivation", ListSSMActivations)
16+
}
17+
18+
func ListSSMActivations(sess *session.Session) ([]Resource, error) {
19+
svc := ssm.New(sess)
20+
resources := []Resource{}
21+
22+
params := &ssm.DescribeActivationsInput{
23+
MaxResults: aws.Int64(50),
24+
}
25+
26+
for {
27+
output, err := svc.DescribeActivations(params)
28+
if err != nil {
29+
return nil, err
30+
}
31+
32+
for _, activation := range output.ActivationList {
33+
resources = append(resources, &SSMActivation{
34+
svc: svc,
35+
ID: activation.ActivationId,
36+
})
37+
}
38+
39+
if output.NextToken == nil {
40+
break
41+
}
42+
43+
params.NextToken = output.NextToken
44+
}
45+
46+
return resources, nil
47+
}
48+
49+
func (f *SSMActivation) Remove() error {
50+
51+
_, err := f.svc.DeleteActivation(&ssm.DeleteActivationInput{
52+
ActivationId: f.ID,
53+
})
54+
55+
return err
56+
}
57+
58+
func (f *SSMActivation) String() string {
59+
return *f.ID
60+
}

‎resources/ssm-associations.go

+63
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
package resources
2+
3+
import (
4+
"github.com/aws/aws-sdk-go/aws"
5+
"github.com/aws/aws-sdk-go/aws/session"
6+
"github.com/aws/aws-sdk-go/service/ssm"
7+
)
8+
9+
type SSMAssociation struct {
10+
svc *ssm.SSM
11+
associationID *string
12+
instanceID *string
13+
}
14+
15+
func init() {
16+
register("SSMAssociation", ListSSMAssociations)
17+
}
18+
19+
func ListSSMAssociations(sess *session.Session) ([]Resource, error) {
20+
svc := ssm.New(sess)
21+
resources := []Resource{}
22+
23+
params := &ssm.ListAssociationsInput{
24+
MaxResults: aws.Int64(50),
25+
}
26+
27+
for {
28+
output, err := svc.ListAssociations(params)
29+
if err != nil {
30+
return nil, err
31+
}
32+
33+
for _, association := range output.Associations {
34+
resources = append(resources, &SSMAssociation{
35+
svc: svc,
36+
associationID: association.AssociationId,
37+
instanceID: association.InstanceId,
38+
})
39+
}
40+
41+
if output.NextToken == nil {
42+
break
43+
}
44+
45+
params.NextToken = output.NextToken
46+
}
47+
48+
return resources, nil
49+
}
50+
51+
func (f *SSMAssociation) Remove() error {
52+
53+
_, err := f.svc.DeleteAssociation(&ssm.DeleteAssociationInput{
54+
AssociationId: f.associationID,
55+
//InstanceId: f.instanceID,
56+
})
57+
58+
return err
59+
}
60+
61+
func (f *SSMAssociation) String() string {
62+
return *f.associationID
63+
}

‎resources/ssm-documents.go

+68
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,68 @@
1+
package resources
2+
3+
import (
4+
"github.com/aws/aws-sdk-go/aws"
5+
"github.com/aws/aws-sdk-go/aws/session"
6+
"github.com/aws/aws-sdk-go/service/ssm"
7+
)
8+
9+
type SSMDocument struct {
10+
svc *ssm.SSM
11+
name *string
12+
}
13+
14+
func init() {
15+
register("SSMDocument", ListSSMDocuments)
16+
}
17+
18+
func ListSSMDocuments(sess *session.Session) ([]Resource, error) {
19+
svc := ssm.New(sess)
20+
resources := []Resource{}
21+
22+
documentKeyFilter := []*ssm.DocumentKeyValuesFilter{
23+
{
24+
Key: aws.String("Owner"),
25+
Values: []*string{aws.String("Self")},
26+
},
27+
}
28+
29+
params := &ssm.ListDocumentsInput{
30+
MaxResults: aws.Int64(50),
31+
Filters: documentKeyFilter,
32+
}
33+
34+
for {
35+
output, err := svc.ListDocuments(params)
36+
if err != nil {
37+
return nil, err
38+
}
39+
40+
for _, documentIdentifier := range output.DocumentIdentifiers {
41+
resources = append(resources, &SSMDocument{
42+
svc: svc,
43+
name: documentIdentifier.Name,
44+
})
45+
}
46+
47+
if output.NextToken == nil {
48+
break
49+
}
50+
51+
params.NextToken = output.NextToken
52+
}
53+
54+
return resources, nil
55+
}
56+
57+
func (f *SSMDocument) Remove() error {
58+
59+
_, err := f.svc.DeleteDocument(&ssm.DeleteDocumentInput{
60+
Name: f.name,
61+
})
62+
63+
return err
64+
}
65+
66+
func (f *SSMDocument) String() string {
67+
return *f.name
68+
}

‎resources/ssm-maintenancewindows.go

+60
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,60 @@
1+
package resources
2+
3+
import (
4+
"github.com/aws/aws-sdk-go/aws"
5+
"github.com/aws/aws-sdk-go/aws/session"
6+
"github.com/aws/aws-sdk-go/service/ssm"
7+
)
8+
9+
type SSMMaintenanceWindow struct {
10+
svc *ssm.SSM
11+
ID *string
12+
}
13+
14+
func init() {
15+
register("SSMMaintenanceWindow", ListSSMMaintenanceWindows)
16+
}
17+
18+
func ListSSMMaintenanceWindows(sess *session.Session) ([]Resource, error) {
19+
svc := ssm.New(sess)
20+
resources := []Resource{}
21+
22+
params := &ssm.DescribeMaintenanceWindowsInput{
23+
MaxResults: aws.Int64(50),
24+
}
25+
26+
for {
27+
output, err := svc.DescribeMaintenanceWindows(params)
28+
if err != nil {
29+
return nil, err
30+
}
31+
32+
for _, windowIdentity := range output.WindowIdentities {
33+
resources = append(resources, &SSMMaintenanceWindow{
34+
svc: svc,
35+
ID: windowIdentity.WindowId,
36+
})
37+
}
38+
39+
if output.NextToken == nil {
40+
break
41+
}
42+
43+
params.NextToken = output.NextToken
44+
}
45+
46+
return resources, nil
47+
}
48+
49+
func (f *SSMMaintenanceWindow) Remove() error {
50+
51+
_, err := f.svc.DeleteMaintenanceWindow(&ssm.DeleteMaintenanceWindowInput{
52+
WindowId: f.ID,
53+
})
54+
55+
return err
56+
}
57+
58+
func (f *SSMMaintenanceWindow) String() string {
59+
return *f.ID
60+
}

‎resources/ssm-parameters.go

+60
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,60 @@
1+
package resources
2+
3+
import (
4+
"github.com/aws/aws-sdk-go/aws"
5+
"github.com/aws/aws-sdk-go/aws/session"
6+
"github.com/aws/aws-sdk-go/service/ssm"
7+
)
8+
9+
type SSMParameter struct {
10+
svc *ssm.SSM
11+
name *string
12+
}
13+
14+
func init() {
15+
register("SSMParameter", ListSSMParameters)
16+
}
17+
18+
func ListSSMParameters(sess *session.Session) ([]Resource, error) {
19+
svc := ssm.New(sess)
20+
resources := []Resource{}
21+
22+
params := &ssm.DescribeParametersInput{
23+
MaxResults: aws.Int64(50),
24+
}
25+
26+
for {
27+
output, err := svc.DescribeParameters(params)
28+
if err != nil {
29+
return nil, err
30+
}
31+
32+
for _, parameter := range output.Parameters {
33+
resources = append(resources, &SSMParameter{
34+
svc: svc,
35+
name: parameter.Name,
36+
})
37+
}
38+
39+
if output.NextToken == nil {
40+
break
41+
}
42+
43+
params.NextToken = output.NextToken
44+
}
45+
46+
return resources, nil
47+
}
48+
49+
func (f *SSMParameter) Remove() error {
50+
51+
_, err := f.svc.DeleteParameter(&ssm.DeleteParameterInput{
52+
Name: f.name,
53+
})
54+
55+
return err
56+
}
57+
58+
func (f *SSMParameter) String() string {
59+
return *f.name
60+
}

‎resources/ssm-patchbaselines.go

+69
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,69 @@
1+
package resources
2+
3+
import (
4+
"strings"
5+
6+
"github.com/aws/aws-sdk-go/aws"
7+
"github.com/aws/aws-sdk-go/aws/session"
8+
"github.com/aws/aws-sdk-go/service/ssm"
9+
)
10+
11+
type SSMPatchBaseline struct {
12+
svc *ssm.SSM
13+
ID *string
14+
}
15+
16+
func init() {
17+
register("SSMPatchBaseline", ListSSMPatchBaselines)
18+
}
19+
20+
func ListSSMPatchBaselines(sess *session.Session) ([]Resource, error) {
21+
svc := ssm.New(sess)
22+
resources := []Resource{}
23+
24+
params := &ssm.DescribePatchBaselinesInput{
25+
MaxResults: aws.Int64(50),
26+
}
27+
28+
for {
29+
output, err := svc.DescribePatchBaselines(params)
30+
if err != nil {
31+
return nil, err
32+
}
33+
34+
for _, baselineIdentity := range output.BaselineIdentities {
35+
if baselineIdentity.BaselineDescription == nil {
36+
resources = append(resources, &SSMPatchBaseline{
37+
svc: svc,
38+
ID: baselineIdentity.BaselineId,
39+
})
40+
} else if !strings.Contains(*baselineIdentity.BaselineDescription, "Provided by AWS") {
41+
resources = append(resources, &SSMPatchBaseline{
42+
svc: svc,
43+
ID: baselineIdentity.BaselineId,
44+
})
45+
}
46+
}
47+
48+
if output.NextToken == nil {
49+
break
50+
}
51+
52+
params.NextToken = output.NextToken
53+
}
54+
55+
return resources, nil
56+
}
57+
58+
func (f *SSMPatchBaseline) Remove() error {
59+
60+
_, err := f.svc.DeletePatchBaseline(&ssm.DeletePatchBaselineInput{
61+
BaselineId: f.ID,
62+
})
63+
64+
return err
65+
}
66+
67+
func (f *SSMPatchBaseline) String() string {
68+
return *f.ID
69+
}

0 commit comments

Comments
 (0)