From d454efa152cf407ef4c1215efc975d585474378e Mon Sep 17 00:00:00 2001
From: Pierre Roullon <pierre.roullon@gmail.com>
Date: Thu, 22 Jun 2017 13:40:09 +0200
Subject: [PATCH] feat (helper): add haskellType helper

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

diff --git a/helpers/helpers.go b/helpers/helpers.go
index 161fdba..c5a75ec 100644
--- a/helpers/helpers.go
+++ b/helpers/helpers.go
@@ -84,6 +84,7 @@ var ProtoHelpersFuncMap = template.FuncMap{
 	"getEnumValue":          getEnumValue,
 	"isFieldMessage":        isFieldMessage,
 	"isFieldRepeated":       isFieldRepeated,
+	"haskellType":           haskellType,
 	"goType":                goType,
 	"goTypeWithPackage":     goTypeWithPackage,
 	"jsType":                jsType,
@@ -169,6 +170,68 @@ func goTypeWithPackage(f *descriptor.FieldDescriptorProto) string {
 	return goType(pkg, f)
 }
 
+func haskellType(pkg string, f *descriptor.FieldDescriptorProto) string {
+	switch *f.Type {
+	case descriptor.FieldDescriptorProto_TYPE_DOUBLE:
+		if *f.Label == descriptor.FieldDescriptorProto_LABEL_REPEATED {
+			return "[Float]"
+		}
+		return "Float"
+	case descriptor.FieldDescriptorProto_TYPE_FLOAT:
+		if *f.Label == descriptor.FieldDescriptorProto_LABEL_REPEATED {
+			return "[Float]"
+		}
+		return "Float"
+	case descriptor.FieldDescriptorProto_TYPE_INT64:
+		if *f.Label == descriptor.FieldDescriptorProto_LABEL_REPEATED {
+			return "[Int64]"
+		}
+		return "Int64"
+	case descriptor.FieldDescriptorProto_TYPE_UINT64:
+		if *f.Label == descriptor.FieldDescriptorProto_LABEL_REPEATED {
+			return "[Word]"
+		}
+		return "Word"
+	case descriptor.FieldDescriptorProto_TYPE_INT32:
+		if *f.Label == descriptor.FieldDescriptorProto_LABEL_REPEATED {
+			return "[Int]"
+		}
+		return "Int"
+	case descriptor.FieldDescriptorProto_TYPE_UINT32:
+		if *f.Label == descriptor.FieldDescriptorProto_LABEL_REPEATED {
+			return "[Word]"
+		}
+		return "Word"
+	case descriptor.FieldDescriptorProto_TYPE_BOOL:
+		if *f.Label == descriptor.FieldDescriptorProto_LABEL_REPEATED {
+			return "[Bool]"
+		}
+		return "Bool"
+	case descriptor.FieldDescriptorProto_TYPE_STRING:
+		if *f.Label == descriptor.FieldDescriptorProto_LABEL_REPEATED {
+			return "[Text]"
+		}
+		return "Text"
+	case descriptor.FieldDescriptorProto_TYPE_MESSAGE:
+		if pkg != "" {
+			pkg = pkg + "."
+		}
+		if *f.Label == descriptor.FieldDescriptorProto_LABEL_REPEATED {
+			return fmt.Sprintf("[%s%s]", pkg, shortType(*f.TypeName))
+		}
+		return fmt.Sprintf("%s%s", pkg, shortType(*f.TypeName))
+	case descriptor.FieldDescriptorProto_TYPE_BYTES:
+		if *f.Label == descriptor.FieldDescriptorProto_LABEL_REPEATED {
+			return "[Word8]"
+		}
+		return "Word8"
+	case descriptor.FieldDescriptorProto_TYPE_ENUM:
+		return fmt.Sprintf("%s%s", pkg, shortType(*f.TypeName))
+	default:
+		return "Generic"
+	}
+}
+
 func goType(pkg string, f *descriptor.FieldDescriptorProto) string {
 	switch *f.Type {
 	case descriptor.FieldDescriptorProto_TYPE_DOUBLE: