added smartPrefix function

render/render.go

@@ -26,6 +26,7 @@ func Render(template string, def parser.Definition, params map[string]interface{
 	ctx.Set("format_comment_text", formatCommentText)
 	ctx.Set("format_comment_html", formatCommentHTML)
 	ctx.Set("format_tags", formatTags)
+	ctx.Set("smart_prefix", smartPrefix)
 	s, err := plush.Render(string(template), ctx)
 	if err != nil {
 		return "", err
@@ -80,3 +81,13 @@ func formatTags(tags ...string) (template.HTML, error) {
 	tagsStr = "`" + tagsStr + "`"
 	return template.HTML(tagsStr), nil
 }
+
+// smartPrefix prepends a string before s, allowing for the specific use
+// case of pointers to objects. If the s begins with * (as in, *Object), the
+// result will be *prefixObject to preserve its original meaning.
+func smartPrefix(prefix, s string) string {
+	if strings.HasPrefix(s, "*") {
+		return "*" + prefix + s[1:]
+	}
+	return prefix + s
+}

render/render_test.go

@@ -112,3 +112,14 @@ func TestFormatCommentText(t *testing.T) {
 	is.Equal(actual, `// What about new lines?`)
 
 }
+
+func TestSmartPrefix(t *testing.T) {
+	is := is.New(t)
+
+	actual := smartPrefix("public", "*Object")
+	is.Equal(actual, "*publicObject")
+
+	actual = smartPrefix("public", "Object")
+	is.Equal(actual, "publicObject")
+
+}