diff --git a/helpers/helpers.go b/helpers/helpers.go index 3c408fa..2be399c 100644 --- a/helpers/helpers.go +++ b/helpers/helpers.go @@ -160,6 +160,8 @@ var ProtoHelpersFuncMap = template.FuncMap{ "replaceDict": replaceDict, "setStore": setStore, "getStore": getStore, + "goPkg": goPkg, + "goPkgLastElement": goPkgLastElement, } var pathMap map[interface{}]*descriptor.SourceCodeInfo_Location @@ -1272,3 +1274,13 @@ func replaceDict(src string, dict map[string]interface{}) string { } return src } + +func goPkg(f *descriptor.FileDescriptorProto) string { + return f.Options.GetGoPackage() +} + +func goPkgLastElement(f *descriptor.FileDescriptorProto) string { + pkg := goPkg(f) + pkgSplitted := strings.Split(pkg, "/") + return pkgSplitted[len(pkgSplitted)-1] +}