From 5dd30f38f3dfbb3ce2ddc623f797ac1686f31b7d Mon Sep 17 00:00:00 2001
From: Pierre Roullon <pierre.roullon@gmail.com>
Date: Wed, 21 Dec 2016 09:26:26 +0000
Subject: [PATCH] feat (helper): add field reflection helpers getMessageType
 and isFieldMessage

---
 helpers.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/helpers.go b/helpers.go
index f4817cb..877bd05 100644
--- a/helpers.go
+++ b/helpers.go
@@ -7,6 +7,8 @@ import (
 
 	"github.com/Masterminds/sprig"
 	"github.com/huandu/xstrings"
+
+	"github.com/golang/protobuf/protoc-gen-go/descriptor"
 )
 
 var ProtoHelpersFuncMap = template.FuncMap{
@@ -51,6 +53,8 @@ var ProtoHelpersFuncMap = template.FuncMap{
 	"kebabCase": func(s string) string {
 		return strings.Replace(xstrings.ToSnakeCase(s), "_", "-", -1)
 	},
+	"getMessageType": getMessageType,
+	"isFieldMessage": isFieldMessage,
 }
 
 func init() {
@@ -58,3 +62,25 @@ func init() {
 		ProtoHelpersFuncMap[k] = v
 	}
 }
+
+func getMessageType(f *descriptor.FileDescriptorProto, name string) *descriptor.DescriptorProto {
+	for _, m := range f.MessageType {
+		// name usually contains the package name
+		if strings.HasSuffix(name, *m.Name) {
+			return m
+		}
+	}
+
+	return nil
+}
+
+func isFieldMessage(f *descriptor.FieldDescriptorProto) bool {
+	if f.Type != nil && *f.Type == descriptor.FieldDescriptorProto_TYPE_MESSAGE {
+		return true
+	}
+	if f.TypeName != nil && (*f.TypeName)[0] == '.' {
+		return true
+	}
+
+	return false
+}