Method buildRegoRuleMap

app/cli/internal/policydevel/lint.go:389–422

Creates a mapping from line numbers to Rego rule names

(regoSrc string)

// Creates a mapping from line numbers to Rego rule names
func (p *PolicyToLint) buildRegoRuleMap(regoSrc string) map[int]string {
	ruleMap := make(map[int]string)

	// Parse the Rego source into AST
	module, err := opaAst.ParseModule("", regoSrc)
	if err != nil {
		// Return empty map if parsing fails
		return ruleMap
	}

	// Walk through the AST to find rule definitions
	for _, rule := range module.Rules {
		if rule.Location != nil {
			ruleName := string(rule.Head.Name)
			startLine := rule.Location.Row
			endLine := startLine

			// Try to find the end line of the rule
			if len(rule.Body) > 0 {
				lastExpr := rule.Body[len(rule.Body)-1]
				if lastExpr.Location != nil {
					endLine = lastExpr.Location.Row
				}
			}

			// Map all lines within this rule to the rule name
			for line := startLine; line <= endLine; line++ {
				ruleMap[line] = ruleName
			}
		}
	}

	return ruleMap
}