mirror of
https://github.com/ralsina/tartrazine.git
synced 2024-09-20 15:31:21 +00:00
59 lines
1.2 KiB
Cheetah
59 lines
1.2 KiB
Cheetah
package data
|
|
|
|
import (
|
|
"github.com/go-enry/go-enry/v2/data/rule"
|
|
"github.com/go-enry/go-enry/v2/regex"
|
|
)
|
|
|
|
var ContentHeuristics = map[string]*Heuristics{
|
|
{{ range $ext, $rules := . -}}
|
|
{{ printf "%q" $ext }}: &Heuristics{
|
|
{{ range $rule := $rules -}}
|
|
{{template "Rule" $rule}}
|
|
{{ end -}}
|
|
},
|
|
{{ end -}}
|
|
}
|
|
|
|
{{ define "Rule" -}}
|
|
{{ if eq .Op "And" -}}
|
|
rule.And(
|
|
{{ template "Languages" .Langs -}}
|
|
{{ range $rule := .Rules -}}
|
|
{{template "Rule" $rule}}
|
|
{{ end -}}
|
|
),
|
|
{{- else if eq .Op "Or" -}}
|
|
rule.Or(
|
|
{{ template "Languages" .Langs -}}
|
|
{{ template "mustCompile" . }}
|
|
),
|
|
{{- else if eq .Op "Not" -}}
|
|
rule.Not(
|
|
{{ template "Languages" .Langs -}}
|
|
{{ template "mustCompile" . }}
|
|
),
|
|
{{- else if eq .Op "Always" -}}
|
|
rule.Always(
|
|
{{ template "Languages" .Langs -}}
|
|
),
|
|
{{ end -}}
|
|
{{ end -}}
|
|
|
|
|
|
{{define "Languages" -}}
|
|
{{with . -}}
|
|
rule.MatchingLanguages( {{range .}} {{printf "\"%s\"" .}}, {{end}} ),
|
|
{{ else -}}
|
|
rule.MatchingLanguages(""),
|
|
{{end -}}
|
|
{{end}}
|
|
|
|
{{define "mustCompile" -}}
|
|
{{ if .IsRE2 -}}
|
|
regex.MustCompileMultiline({{ .Pattern | stringVal }}),
|
|
{{- else -}}
|
|
regex.MustCompileRuby({{ .Pattern | stringVal }}),
|
|
{{ end -}}
|
|
{{end}}
|