intel-aws-sagemaker-endpoint-configuration-check-optimal-instance-types.sentinel

import "tfplan/v2" as tfplan
import "approved"
import "policy_summary"

param valid_actions default [
	["no-op"],
	["create"],
	["update"],
]

doc = {
	"allowed":   approved.aws.aws_sagemaker_endpoint_configuration.optimal,
	"category":  approved.aws.aws_sagemaker_endpoint_configuration.category,
	"error":     approved.aws.aws_sagemaker_endpoint_configuration.error,
	"file_name": approved.aws.aws_sagemaker_endpoint_configuration.check_file_name,
	"md_url":    approved.aws.policies_url,
	"parameter": approved.aws.aws_sagemaker_endpoint_configuration.parameter,
	"provider":  approved.aws.aws_sagemaker_endpoint_configuration.provider,
	"resource":  approved.aws.aws_sagemaker_endpoint_configuration.resource,
	"violation": approved.aws.aws_sagemaker_endpoint_configuration.violation,
}

// Filter resources by type
all_resources = filter tfplan.resource_changes as _, rc {
	rc.type is doc.resource and
		rc.mode is "managed" and
		rc.change.actions in valid_actions
}

// Filter resources that violate a given condition
violators = filter all_resources as _, r {
	all r.change.after.production_variants as _, production_variants {
		production_variants.instance_type not in doc.allowed
	}
}

// Build a summary report
summaryreport = {
	"name": doc.name,
	"violations": map violators as _, violation {
		{
			"name":    violation.name,
			"address": violation.address,
			"type":    violation.type,
			"message": violation.name + doc.error,
		}
	},
}

# A non-optimal instance type was chosen
main = rule {
	policy_summary.summary(summaryreport, doc)
}