template.go

package template

import (
	_ "embed"
	"fmt"
	"text/template"

	"google.golang.org/protobuf/compiler/protogen"
	"google.golang.org/protobuf/types/pluginpb"
)

//go:embed template.go.tmpl
var tmpl string

func GenerateTemplate(v *pluginpb.Version, g *protogen.GeneratedFile) (*template.Template, error) {
	return template.New("").Funcs(template.FuncMap{
		"PluginVersion": func() string { return "v0.0.0" },
		"ProtocVersion": func() string {
			return fmt.Sprintf("v%d.%d.%d", *v.Major, *v.Minor, *v.Patch)
		},
	}).Funcs(template.FuncMap{
		"QualifiedGoIdent": func(imp protogen.GoImportPath, s string) string {
			return g.QualifiedGoIdent(imp.Ident(s))
		},
		"proto": func(s string) string {
			return g.QualifiedGoIdent(protogen.GoImportPath("google.golang.org/protobuf/proto").Ident(s))
		},
		"cfg": func(s string) string {
			return g.QualifiedGoIdent(protogen.GoImportPath("github.com/Neakxs/protoc-gen-authz/cfg").Ident(s))
		},
		"authorize": func(s string) string {
			return g.QualifiedGoIdent(protogen.GoImportPath("github.com/Neakxs/protoc-gen-authz/authorize").Ident(s))
		},
		"cel": func(s string) string {
			return g.QualifiedGoIdent(protogen.GoImportPath("github.com/google/cel-go/cel").Ident(s))
		},
		"functions": func(s string) string {
			return g.QualifiedGoIdent(protogen.GoImportPath("github.com/google/cel-go/interpreter/functions").Ident(s))
		},
	}).Parse(tmpl)
}