Merge branch 'main' of https://github.com/pacedotdev/oto

render/render.go

@@ -27,6 +27,7 @@ func Render(template string, def parser.Definition, params map[string]interface{
 	ctx.Set("format_comment_html", formatCommentHTML)
 	ctx.Set("format_tags", formatTags)
 	ctx.Set("object_golang", ObjectGolang)
+	ctx.Set("smart_prefix", smartPrefix)
 	s, err := plush.Render(string(template), ctx)
 	if err != nil {
 		return "", err
@@ -81,3 +82,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")
+
+}