Integrated "aws_ssm" plugin for checkenv

Author: zomglings

aws_ssm.go

@@ -0,0 +1,37 @@
+// checkenv plugin that provides the environment variables defined in the checkenv process.
+
+package main
+
+import (
+	"context"
+
+	"github.com/bugout-dev/checkenv/aws_ssm"
+)
+
+func AWSSystemsManagerParameterStoreProvider(filter string) (map[string]string, error) {
+	environment := make(map[string]string)
+
+	AWSSystemsManagerFlags := aws_ssm.Flags{
+		Export:     false,
+		MaxResults: 10,
+		Outfile:    "",
+		Update:     false,
+	}
+	AWSSystemsManagerFlags.FilterTags = aws_ssm.ParseFilterTags(filter)
+
+	ctx := context.Background()
+	api := aws_ssm.InitAWSClient(ctx)
+	keys := aws_ssm.FetchKeysOfParameters(ctx, api, AWSSystemsManagerFlags)
+	keyChunks := aws_ssm.GenerateChunks(keys, 10)
+	parameters := aws_ssm.FetchParameters(ctx, api, keyChunks, AWSSystemsManagerFlags)
+	for _, parameter := range parameters {
+		environment[parameter.Name] = parameter.Value
+	}
+
+	return environment, nil
+}
+
+func init() {
+	helpString := "Provides environment variables defined in AWS Systems Manager Parameter Store."
+	RegisterPlugin("aws_ssm", helpString, noop, AWSSystemsManagerParameterStoreProvider)
+}

aws_ssm/aws_ssm.go

@@ -11,25 +11,6 @@ import (
 	"github.com/aws/aws-sdk-go-v2/service/ssm"
 )
 
-// SSMGetParametersAPI and SSMDescribeParametersAPI defines the interface
-// for the GetParameters and DescribeParameters function.
-// We use this interface to test the function using a mocked service
-type SSMGetParametersAPI interface {
-	GetParameters(
-		ctx context.Context,
-		params *ssm.GetParametersInput,
-		optFns ...func(*ssm.Options),
-	) (*ssm.GetParametersOutput, error)
-}
-
-type SSMDescribeParametersAPI interface {
-	DescribeParameters(
-		ctx context.Context,
-		params *ssm.DescribeParametersInput,
-		optFns ...func(*ssm.Options),
-	) (*ssm.DescribeParametersOutput, error)
-}
-
 type AWSSystemsManagerParameterStore interface {
 	GetParameters(
 		ctx context.Context,
@@ -52,11 +33,11 @@ type AWSSystemsManagerParameterStore interface {
 // Output:
 // 		If success, a GetParametersOutput object containing the result of the service call and nil
 // 		Otherwise, nil and an error from the call to GetParameters
-func ExecGetParameters(c context.Context, api SSMGetParametersAPI, input *ssm.GetParametersInput) (*ssm.GetParametersOutput, error) {
+func ExecGetParameters(c context.Context, api AWSSystemsManagerParameterStore, input *ssm.GetParametersInput) (*ssm.GetParametersOutput, error) {
 	return api.GetParameters(c, input)
 }
 
-func ExecDescribeParameters(c context.Context, api SSMDescribeParametersAPI, input *ssm.DescribeParametersInput) (*ssm.DescribeParametersOutput, error) {
+func ExecDescribeParameters(c context.Context, api AWSSystemsManagerParameterStore, input *ssm.DescribeParametersInput) (*ssm.DescribeParametersOutput, error) {
 	return api.DescribeParameters(c, input)
 }
 

aws_ssm/aws_ssm_test.go

@@ -13,10 +13,9 @@ import (
 )
 
 // Fill fake output data
-type SSMGetParametersImpl struct{}
-type SSMDescribeParametersImpl struct{}
+type MockAWSSystemManagerParameterStore struct{}
 
-func (dt SSMGetParametersImpl) GetParameters(
+func (dt MockAWSSystemManagerParameterStore) GetParameters(
 	ctx context.Context,
 	params *ssm.GetParametersInput,
 	optFns ...func(*ssm.Options),
@@ -34,7 +33,7 @@ func (dt SSMGetParametersImpl) GetParameters(
 
 	return output, nil
 }
-func (dt SSMDescribeParametersImpl) DescribeParameters(
+func (dt MockAWSSystemManagerParameterStore) DescribeParameters(
 	ctx context.Context,
 	params *ssm.DescribeParametersInput,
 	optFns ...func(*ssm.Options),
@@ -87,7 +86,7 @@ func TestDescribeParameters(t *testing.T) {
 		t.Fatal("Failed to populate data")
 	}
 
-	api := &SSMDescribeParametersImpl{}
+	api := &MockAWSSystemManagerParameterStore{}
 
 	flags := Flags{FilterTags: []FilterTag{{Name: "Product", Value: "test"}}}
 
@@ -111,7 +110,7 @@ func TestDescribeParameters(t *testing.T) {
 func TestGetParameters(t *testing.T) {
 	parameterKeyChunks := GenerateChunks(globalParameterKeys, 10)
 
-	api := &SSMGetParametersImpl{}
+	api := &MockAWSSystemManagerParameterStore{}
 
 	flags := Flags{Export: false}
 

aws_ssm/parameters.go

@@ -16,7 +16,7 @@ import (
 // Fetch values for parameters
 // Inputs:
 //		chunks: list of lists with parameter key values
-func FetchParameters(ctx context.Context, api SSMGetParametersAPI, chunks [][]string, flags Flags) []Parameter {
+func FetchParameters(ctx context.Context, api AWSSystemsManagerParameterStore, chunks [][]string, flags Flags) []Parameter {
 	var parameters []Parameter
 
 	for _, chunk := range chunks {
@@ -46,7 +46,7 @@ func FetchParameters(ctx context.Context, api SSMGetParametersAPI, chunks [][]st
 // Fetch list of parameter keys from AWS with defined filters
 func FetchKeysOfParameters(
 	ctx context.Context,
-	api SSMDescribeParametersAPI,
+	api AWSSystemsManagerParameterStore,
 	flags Flags,
 ) []string {
 	var parameters []string