added format_commnt helper

Author: matryer

render/render.go

@@ -22,6 +22,7 @@ func Render(template string, def parser.Definition, params map[string]interface{
 	ctx.Set("def", def)
 	ctx.Set("params", params)
 	ctx.Set("json", toJSONHelper)
+	ctx.Set("formatComment", formatComment)
 	ctx.Set("format_comment_text", formatCommentText)
 	ctx.Set("format_comment_html", formatCommentHTML)
 	ctx.Set("format_tags", formatTags)
@@ -40,6 +41,12 @@ func toJSONHelper(v interface{}) (template.HTML, error) {
 	return template.HTML(b), nil
 }
 
+func formatComment(s string, linePrefix string) template.HTML {
+	var buf bytes.Buffer
+	doc.ToText(&buf, s, linePrefix, linePrefix, 80)
+	return template.HTML(buf.String())
+}
+
 func formatCommentText(s string) template.HTML {
 	var buf bytes.Buffer
 	doc.ToText(&buf, s, "// ", "", 80)

render/render_test.go

@@ -108,4 +108,19 @@ func TestFormatCommentText(t *testing.T) {
 	actual = strings.TrimSpace(string(formatCommentText(`What happens if I use "quotes"?`)))
 	is.Equal(actual, `// What happens if I use "quotes"?`)
 
+	actual = strings.TrimSpace(string(formatCommentText("What about\nnew lines?")))
+	is.Equal(actual, `// What about new lines?`)
+
+}
+
+func TestFormatComment(t *testing.T) {
+	is := is.New(t)
+
+	longComment := `This is a long comment that will end up spanning
+multiple lines so we get to test the indent string option
+in formatComment.`
+	actual := strings.TrimSpace(string(formatComment(longComment, "\t\t")))
+	is.Equal(actual, `This is a long comment that will end up spanning multiple lines so we get to
+		test the indent string option in formatComment.`)
+
 }