aws_dynamodb_table_invalid_stream_view_type.go

package rules

import (
	"fmt"

	"github.com/terraform-linters/tflint-plugin-sdk/hclext"
	"github.com/terraform-linters/tflint-plugin-sdk/tflint"
)

// AwsDynamoDBTableInvalidStreamViewTypeRule checks the pattern is valid
type AwsDynamoDBTableInvalidStreamViewTypeRule struct {
	tflint.DefaultRule

	resourceType  string
	attributeName string
	enum          []string
}

// NewAwsDynamoDBTableInvalidStreamViewTypeRule returns new rule with default attributes
func NewAwsDynamoDBTableInvalidStreamViewTypeRule() *AwsDynamoDBTableInvalidStreamViewTypeRule {
	return &AwsDynamoDBTableInvalidStreamViewTypeRule{
		resourceType:  "aws_dynamodb_table",
		attributeName: "stream_view_type",
		enum: []string{
			"",
			"NEW_IMAGE",
			"OLD_IMAGE",
			"NEW_AND_OLD_IMAGES",
			"KEYS_ONLY",
		},
	}
}

// Name returns the rule name
func (r *AwsDynamoDBTableInvalidStreamViewTypeRule) Name() string {
	return "aws_dynamodb_table_invalid_stream_view_type"
}

// Enabled returns whether the rule is enabled by default
func (r *AwsDynamoDBTableInvalidStreamViewTypeRule) Enabled() bool {
	return true
}

// Severity returns the rule severity
func (r *AwsDynamoDBTableInvalidStreamViewTypeRule) Severity() tflint.Severity {
	return tflint.ERROR
}

// Link returns the rule reference link
func (r *AwsDynamoDBTableInvalidStreamViewTypeRule) Link() string {
	return ""
}

// Check checks the pattern is valid
func (r *AwsDynamoDBTableInvalidStreamViewTypeRule) Check(runner tflint.Runner) error {
	resources, err := runner.GetResourceContent(r.resourceType, &hclext.BodySchema{
		Attributes: []hclext.AttributeSchema{{Name: r.attributeName}},
	}, nil)
	if err != nil {
		return err
	}

	for _, resource := range resources.Blocks {
		attribute, exists := resource.Body.Attributes[r.attributeName]
		if !exists {
			continue
		}

		err := runner.EvaluateExpr(attribute.Expr, func(val string) error {
			found := false
			for _, item := range r.enum {
				if item == val {
					found = true
				}
			}
			if !found {
				runner.EmitIssue(
					r,
					fmt.Sprintf(`"%s" is an invalid value as stream_view_type`, val),
					attribute.Expr.Range(),
				)
			}
			return nil
		}, nil)
		if err != nil {
			return err
		}
	}

	return nil
}