From c17a3d02d01340607230987ec2253405baed9add Mon Sep 17 00:00:00 2001
From: Egor Kovetskiy <e.kovetskiy@gmail.com>
Date: Thu, 4 Feb 2021 12:18:28 +0300
Subject: [PATCH] fix regression after #60

---
 pkg/confluence/api.go        |  2 +-
 pkg/mark/markdown.go         |  8 +++---
 pkg/mark/stdlib/stdlib.go    | 22 +++++++-------
 pkg/mark/testdata/codes.html | 56 ++++++++----------------------------
 pkg/mark/testdata/codes.md   |  5 ++++
 5 files changed, 34 insertions(+), 59 deletions(-)

diff --git a/pkg/confluence/api.go b/pkg/confluence/api.go
index 188bfbd..e96f8c9 100644
--- a/pkg/confluence/api.go
+++ b/pkg/confluence/api.go
@@ -647,7 +647,7 @@ func newErrorStatusNotOK(request *gopencils.Resource) error {
 
 	return fmt.Errorf(
 		"Confluence API returned unexpected status: %v, "+
-			"output: %s",
+			"output: %q",
 		request.Raw.Status, output,
 	)
 }
diff --git a/pkg/mark/markdown.go b/pkg/mark/markdown.go
index cd55094..b4f51fa 100644
--- a/pkg/mark/markdown.go
+++ b/pkg/mark/markdown.go
@@ -3,7 +3,6 @@ package mark
 import (
 	"io"
 	"regexp"
-	"strconv"
 	"strings"
 
 	"github.com/kovetskiy/mark/pkg/mark/stdlib"
@@ -55,19 +54,20 @@ func (renderer ConfluenceRenderer) RenderNode(
 ) bf.WalkStatus {
 	if node.Type == bf.CodeBlock {
 		lang := string(node.Info)
+
 		renderer.Stdlib.Templates.ExecuteTemplate(
 			writer,
 			"ac:code",
 			struct {
 				Language string
-				Collapse string
+				Collapse bool
 				Title    string
 				Text     string
 			}{
 				ParseLanguage(lang),
-				strconv.FormatBool(strings.Contains(lang, "collapse")),
+				strings.Contains(lang, "collapse"),
 				ParseTitle(lang),
-				string(node.Literal),
+				strings.TrimSuffix(string(node.Literal), "\n"),
 			},
 		)
 
diff --git a/pkg/mark/stdlib/stdlib.go b/pkg/mark/stdlib/stdlib.go
index 15c1d45..053b020 100644
--- a/pkg/mark/stdlib/stdlib.go
+++ b/pkg/mark/stdlib/stdlib.go
@@ -105,17 +105,19 @@ func templates(api *confluence.API) (*template.Template, error) {
 
 		// This template is used for rendering code in ```
 		`ac:code`: text(
-			`<ac:structured-macro ac:name="expand">{{printf "\n"}}`,
-			`<ac:parameter ac:name="title">{{ .Title }}</ac:parameter>{{printf "\n"}}`,
-			`<ac:rich-text-body>{{printf "\n"}}`,
-				`<ac:structured-macro ac:name="code">{{printf "\n"}}`,
-				`<ac:parameter ac:name="language">{{ .Language }}</ac:parameter>{{printf "\n"}}`,
-				`<ac:parameter ac:name="collapse">{{ .Collapse }}</ac:parameter>{{printf "\n"}}`,
-				`<ac:parameter ac:name="title">{{ .Title }}</ac:parameter>{{printf "\n"}}`,
-				`<ac:plain-text-body><![CDATA[{{ .Text | cdata }}]]></ac:plain-text-body>{{printf "\n"}}`,
-				`</ac:structured-macro>{{printf "\n"}}`,
-			`</ac:rich-text-body>{{printf "\n"}}`,
+			`{{ if .Collapse }}<ac:structured-macro ac:name="expand">{{printf "\n"}}`,
+			`{{ if .Title }}<ac:parameter ac:name="title">{{ .Title }}</ac:parameter>{{printf "\n"}}{{ end }}`,
+			`<ac:rich-text-body>{{printf "\n"}}{{ end }}`,
+
+			`<ac:structured-macro ac:name="code">{{printf "\n"}}`,
+			/**/ `<ac:parameter ac:name="language">{{ .Language }}</ac:parameter>{{printf "\n"}}`,
+			/**/ `<ac:parameter ac:name="collapse">{{ .Collapse }}</ac:parameter>{{printf "\n"}}`,
+			/**/ `{{ if .Title }}<ac:parameter ac:name="title">{{ .Title }}</ac:parameter>{{printf "\n"}}{{ end }}`,
+			/**/ `<ac:plain-text-body><![CDATA[{{ .Text | cdata }}]]></ac:plain-text-body>{{printf "\n"}}`,
 			`</ac:structured-macro>{{printf "\n"}}`,
+
+			`{{ if .Collapse }}</ac:rich-text-body>{{printf "\n"}}`,
+			`</ac:structured-macro>{{printf "\n"}}{{ end }}`,
 		),
 
 		`ac:status`: text(
diff --git a/pkg/mark/testdata/codes.html b/pkg/mark/testdata/codes.html
index fe11812..c2f8f80 100644
--- a/pkg/mark/testdata/codes.html
+++ b/pkg/mark/testdata/codes.html
@@ -1,66 +1,38 @@
 <p><code>inline</code></p>
-<ac:structured-macro ac:name="expand">
-<ac:parameter ac:name="title"></ac:parameter>
-<ac:rich-text-body>
 <ac:structured-macro ac:name="code">
 <ac:parameter ac:name="language"></ac:parameter>
 <ac:parameter ac:name="collapse">false</ac:parameter>
-<ac:parameter ac:name="title"></ac:parameter>
-<ac:plain-text-body><![CDATA[some code
-]]></ac:plain-text-body>
+<ac:plain-text-body><![CDATA[some code]]></ac:plain-text-body>
 </ac:structured-macro>
-</ac:rich-text-body>
-</ac:structured-macro>
-<ac:structured-macro ac:name="expand">
-<ac:parameter ac:name="title"></ac:parameter>
-<ac:rich-text-body>
 <ac:structured-macro ac:name="code">
 <ac:parameter ac:name="language">bash</ac:parameter>
 <ac:parameter ac:name="collapse">false</ac:parameter>
-<ac:parameter ac:name="title"></ac:parameter>
-<ac:plain-text-body><![CDATA[code bash
+<ac:plain-text-body><![CDATA[code bash]]></ac:plain-text-body>
+</ac:structured-macro>
+<ac:structured-macro ac:name="code">
+<ac:parameter ac:name="language">bash</ac:parameter>
+<ac:parameter ac:name="collapse">false</ac:parameter>
+<ac:plain-text-body><![CDATA[with a newline
 ]]></ac:plain-text-body>
 </ac:structured-macro>
-</ac:rich-text-body>
-</ac:structured-macro>
-<ac:structured-macro ac:name="expand">
-<ac:parameter ac:name="title"></ac:parameter>
-<ac:rich-text-body>
 <ac:structured-macro ac:name="code">
 <ac:parameter ac:name="language">unknown</ac:parameter>
 <ac:parameter ac:name="collapse">false</ac:parameter>
-<ac:parameter ac:name="title"></ac:parameter>
-<ac:plain-text-body><![CDATA[unknown code
-]]></ac:plain-text-body>
-</ac:structured-macro>
-</ac:rich-text-body>
+<ac:plain-text-body><![CDATA[unknown code]]></ac:plain-text-body>
 </ac:structured-macro>
 
 <p>text
 text 2</p>
-<ac:structured-macro ac:name="expand">
-<ac:parameter ac:name="title"></ac:parameter>
-<ac:rich-text-body>
 <ac:structured-macro ac:name="code">
 <ac:parameter ac:name="language">unknown</ac:parameter>
 <ac:parameter ac:name="collapse">false</ac:parameter>
-<ac:parameter ac:name="title"></ac:parameter>
-<ac:plain-text-body><![CDATA[unknown code 2
-]]></ac:plain-text-body>
+<ac:plain-text-body><![CDATA[unknown code 2]]></ac:plain-text-body>
 </ac:structured-macro>
-</ac:rich-text-body>
-</ac:structured-macro>
-<ac:structured-macro ac:name="expand">
-<ac:parameter ac:name="title">A b c</ac:parameter>
-<ac:rich-text-body>
 <ac:structured-macro ac:name="code">
 <ac:parameter ac:name="language">sh</ac:parameter>
 <ac:parameter ac:name="collapse">false</ac:parameter>
 <ac:parameter ac:name="title">A b c</ac:parameter>
-<ac:plain-text-body><![CDATA[no-collapse-title
-]]></ac:plain-text-body>
-</ac:structured-macro>
-</ac:rich-text-body>
+<ac:plain-text-body><![CDATA[no-collapse-title]]></ac:plain-text-body>
 </ac:structured-macro>
 <ac:structured-macro ac:name="expand">
 <ac:parameter ac:name="title">A b c</ac:parameter>
@@ -69,20 +41,16 @@ text 2</p>
 <ac:parameter ac:name="language">bash</ac:parameter>
 <ac:parameter ac:name="collapse">true</ac:parameter>
 <ac:parameter ac:name="title">A b c</ac:parameter>
-<ac:plain-text-body><![CDATA[collapse-and-title
-]]></ac:plain-text-body>
+<ac:plain-text-body><![CDATA[collapse-and-title]]></ac:plain-text-body>
 </ac:structured-macro>
 </ac:rich-text-body>
 </ac:structured-macro>
 <ac:structured-macro ac:name="expand">
-<ac:parameter ac:name="title"></ac:parameter>
 <ac:rich-text-body>
 <ac:structured-macro ac:name="code">
 <ac:parameter ac:name="language">c</ac:parameter>
 <ac:parameter ac:name="collapse">true</ac:parameter>
-<ac:parameter ac:name="title"></ac:parameter>
-<ac:plain-text-body><![CDATA[collapse-no-title
-]]></ac:plain-text-body>
+<ac:plain-text-body><![CDATA[collapse-no-title]]></ac:plain-text-body>
 </ac:structured-macro>
 </ac:rich-text-body>
 </ac:structured-macro>
diff --git a/pkg/mark/testdata/codes.md b/pkg/mark/testdata/codes.md
index dd6497e..efedcbc 100644
--- a/pkg/mark/testdata/codes.md
+++ b/pkg/mark/testdata/codes.md
@@ -7,6 +7,11 @@ some code
 code bash
 ```
 
+```bash
+with a newline
+
+```
+
 ```unknown
 unknown code
 ```