feat: env manifest cdn.terminate_tls

internal/pkg/aws/elbv2/elbv2.go

@@ -5,10 +5,13 @@
 package elbv2
 
 import (
+	"context"
+	"errors"
 	"fmt"
 	"sort"
 
 	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/request"
 	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/elbv2"
 )
@@ -19,8 +22,9 @@ const (
 )
 
 type api interface {
-	DescribeTargetHealth(input *elbv2.DescribeTargetHealthInput) (*elbv2.DescribeTargetHealthOutput, error)
-	DescribeRules(input *elbv2.DescribeRulesInput) (*elbv2.DescribeRulesOutput, error)
+	DescribeTargetHealth(*elbv2.DescribeTargetHealthInput) (*elbv2.DescribeTargetHealthOutput, error)
+	DescribeRules(*elbv2.DescribeRulesInput) (*elbv2.DescribeRulesOutput, error)
+	DescribeRulesWithContext(context.Context, *elbv2.DescribeRulesInput, ...request.Option) (*elbv2.DescribeRulesOutput, error)
 }
 
 // ELBV2 wraps an AWS ELBV2 client.
@@ -73,6 +77,33 @@ func (e *ELBV2) ListenerRuleHostHeaders(ruleARN string) ([]string, error) {
 	return hostHeaders, nil
 }
 
+// Rule wraps an elbv2.Rule to add some nice functionality to it.
+type Rule elbv2.Rule
+
+// DescribeRule returns the Rule with ruleARN.
+func (e *ELBV2) DescribeRule(ctx context.Context, ruleARN string) (Rule, error) {
+	resp, err := e.client.DescribeRulesWithContext(ctx, &elbv2.DescribeRulesInput{
+		RuleArns: aws.StringSlice([]string{ruleARN}),
+	})
+	if err != nil {
+		return Rule{}, err
+	} else if len(resp.Rules) == 0 {
+		return Rule{}, errors.New("not found")
+	}
+
+	return Rule(*resp.Rules[0]), nil
+}
+
+// HasRedirectAction returns true if the rule has a redirect action.
+func (r *Rule) HasRedirectAction() bool {
+	for _, action := range r.Actions {
+		if aws.StringValue(action.Type) == elbv2.ActionTypeEnumRedirect {
+			return true
+		}
+	}
+	return false
+}
+
 // TargetHealth wraps up elbv2.TargetHealthDescription.
 type TargetHealth elbv2.TargetHealthDescription
 

internal/pkg/aws/elbv2/elbv2_test.go

@@ -4,6 +4,7 @@
 package elbv2
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"testing"
@@ -219,6 +220,103 @@ func TestELBV2_ListenerRuleHostHeaders(t *testing.T) {
 	}
 }
 
+func TestELBV2_DescribeRule(t *testing.T) {
+	mockARN := "mockListenerRuleARN"
+	testCases := map[string]struct {
+		setUpMock func(m *mocks.Mockapi)
+
+		expectedErr string
+		expected    Rule
+	}{
+		"fail to describe rules": {
+			setUpMock: func(m *mocks.Mockapi) {
+				m.EXPECT().DescribeRulesWithContext(gomock.Any(), &elbv2.DescribeRulesInput{
+					RuleArns: aws.StringSlice([]string{mockARN}),
+				}).Return(nil, errors.New("some error"))
+			},
+			expectedErr: "some error",
+		},
+		"cannot find listener rule": {
+			setUpMock: func(m *mocks.Mockapi) {
+				m.EXPECT().DescribeRulesWithContext(gomock.Any(), &elbv2.DescribeRulesInput{
+					RuleArns: aws.StringSlice([]string{mockARN}),
+				}).Return(&elbv2.DescribeRulesOutput{}, nil)
+			},
+			expectedErr: `not found`,
+		},
+		"success": {
+			setUpMock: func(m *mocks.Mockapi) {
+				m.EXPECT().DescribeRulesWithContext(gomock.Any(), &elbv2.DescribeRulesInput{
+					RuleArns: aws.StringSlice([]string{mockARN}),
+				}).Return(&elbv2.DescribeRulesOutput{
+					Rules: []*elbv2.Rule{
+						{
+							RuleArn: aws.String(mockARN),
+						},
+					},
+				}, nil)
+			},
+			expected: Rule(elbv2.Rule{
+				RuleArn: aws.String(mockARN),
+			}),
+		},
+	}
+
+	for name, tc := range testCases {
+		t.Run(name, func(t *testing.T) {
+			ctrl := gomock.NewController(t)
+			defer ctrl.Finish()
+
+			mockAPI := mocks.NewMockapi(ctrl)
+			tc.setUpMock(mockAPI)
+
+			elbv2Client := ELBV2{
+				client: mockAPI,
+			}
+
+			actual, err := elbv2Client.DescribeRule(context.Background(), mockARN)
+			if tc.expectedErr != "" {
+				require.EqualError(t, err, tc.expectedErr)
+			} else {
+				require.Equal(t, tc.expected, actual)
+			}
+		})
+	}
+}
+
+func TestELBV2Rule_HasRedirectAction(t *testing.T) {
+	testCases := map[string]struct {
+		rule     Rule
+		expected bool
+	}{
+		"rule doesn't have redirect": {
+			rule: Rule(elbv2.Rule{
+				Actions: []*elbv2.Action{
+					{
+						Type: aws.String(elbv2.ActionTypeEnumForward),
+					},
+				},
+			}),
+		},
+		"rule has redirect": {
+			rule: Rule(elbv2.Rule{
+				Actions: []*elbv2.Action{
+					{
+						Type: aws.String(elbv2.ActionTypeEnumRedirect),
+					},
+				},
+			}),
+			expected: true,
+		},
+	}
+
+	for name, tc := range testCases {
+		t.Run(name, func(t *testing.T) {
+			require.Equal(t, tc.expected, tc.rule.HasRedirectAction())
+		})
+	}
+}
+
 func TestTargetHealth_HealthStatus(t *testing.T) {
 	testCases := map[string]struct {
 		inTargetHealth *TargetHealth