renders receievers on custom types

mariotoffia · Oct 21, 2020 · dcf199b · dcf199b
1 parent e0b791a
commit dcf199b
Show file tree

Hide file tree

Showing 3 changed files with 62 additions and 1 deletion.
diff --git a/asciidoc/defaults.go b/asciidoc/defaults.go
@@ -111,7 +111,10 @@ var templateCustomTypeDefintion = `=== {{.TypeDefVar.Name}}
 ----
 {{.TypeDefVar.Decl}}
 ----
+
 {{.TypeDefVar.Doc}}
+
+{{if hasReceivers . .TypeDefVar.Name}}{{renderReceivers . .TypeDefVar.Name}}{{end}}
 `
 
 var templateCustomTypeDefintions = `== Variable Typedefinitions

diff --git a/asciidoc/template.go b/asciidoc/template.go
@@ -149,7 +149,19 @@ func NewTemplateWithOverrides(overrides map[string]string) *Template {
 					return buf.String()
 				},
 			}),
-			CustomVarTypeDefTemplate.String(): createTemplate(CustomVarTypeDefTemplate, templateCustomTypeDefintion, overrides, template.FuncMap{}),
+			CustomVarTypeDefTemplate.String(): createTemplate(CustomVarTypeDefTemplate, templateCustomTypeDefintion, overrides, template.FuncMap{
+				"renderReceivers": func(t *TemplateContext, receiver string) string {
+					var buf bytes.Buffer
+					t.RenderReceiverFunctions(&buf, receiver)
+					return buf.String()
+				},
+				"hasReceivers": func(t *TemplateContext, receiver string) bool {
+					if nil != t.Package {
+						return len(t.Package.FindMethodsByReceiver(receiver)) > 0
+					}
+					return len(t.File.FindMethodsByReceiver(receiver)) > 0
+				},
+			}),
 			VarDeclarationsTemplate.String(): createTemplate(VarDeclarationsTemplate, templateVarAssignments, overrides, template.FuncMap{
 				"render": func(t *TemplateContext, a *goparser.GoAssignment) string {
 					var buf bytes.Buffer

diff --git a/asciidoc/template_test.go b/asciidoc/template_test.go
@@ -971,3 +971,49 @@ func TestStructReceiverFunction(t *testing.T) {
 
 	fmt.Println(buf.String())
 }
+
+func TestCustomTypeReceiverFunction(t *testing.T) {
+	src := `	
+	package mypkg
+
+	// MyDataType is my and only...
+	type MyDataType int
+	// Bar is a public function receives a string and returns a string.
+	func (m MyDataType) Bar(msg string) string {
+		return "hello: " + msg
+	}`
+
+	m := dummyModule()
+	f, err := goparser.ParseInlineFile(m, m.Base+"/mypkg/file.go", src)
+	assert.NoError(t, err)
+
+	var buf bytes.Buffer
+
+	x := NewTemplateWithOverrides(map[string]string{
+		CustomVarTypeDefTemplate.String(): `=== {{.TypeDefVar.Name}}
+[source, go]
+----
+{{.TypeDefVar.Decl}}
+----
+
+{{.TypeDefVar.Doc}}
+
+{{if hasReceivers . .TypeDefVar.Name}}{{renderReceivers . .TypeDefVar.Name}}{{end}}
+`,
+		ReceiversTemplate.String(): `==== Receivers
+{{range .Receiver}}
+===== {{.Name}}
+[source, go]
+----
+{{ .Decl }}
+----
+
+{{.Doc}}
+{{- end}}
+`,
+	}).NewContextWithConfig(f, nil, &TemplateContextConfig{IncludeMethodCode: false})
+
+	x.RenderVarTypeDef(&buf, f.CustomTypes[0])
+
+	fmt.Println(buf.String())
+}