36 lines
891 B
Go
36 lines
891 B
Go
package random
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
type ruleConstructor func(map[string]interface{}) (Rule, error)
|
|
|
|
var (
|
|
// defaultRuleNameMapping is the default mapping of HCL rule names to the appropriate rule constructor.
|
|
// Add to this map when adding a new Rule type to be recognized in HCL.
|
|
defaultRuleNameMapping = map[string]ruleConstructor{
|
|
"charset": ParseCharset,
|
|
}
|
|
|
|
defaultRegistry = Registry{
|
|
Rules: defaultRuleNameMapping,
|
|
}
|
|
)
|
|
|
|
// Registry of HCL rule names to rule constructors.
|
|
type Registry struct {
|
|
// Rules maps names of rules to a constructor for the rule
|
|
Rules map[string]ruleConstructor
|
|
}
|
|
|
|
func (r Registry) parseRule(ruleType string, ruleData map[string]interface{}) (rule Rule, err error) {
|
|
constructor, exists := r.Rules[ruleType]
|
|
if !exists {
|
|
return nil, fmt.Errorf("unrecognized rule type %s", ruleType)
|
|
}
|
|
|
|
rule, err = constructor(ruleData)
|
|
return rule, err
|
|
}
|