Fix for lines-less remaining node

stateful · Oct 4, 2022 · ee0a340 · ee0a340
1 parent 1778dde
commit ee0a340
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/internal/renderer/json.go b/internal/renderer/json.go
@@ -78,6 +78,10 @@ func (r *renderer) GetDocument(source []byte, n ast.Node) (document, error) {
 		return document{}, err
 	}
 
+	if remainingNode != nil && remainingNode.Lines().Len() == 0 {
+		remainingNode = remainingNode.NextSibling()
+	}
+
 	if remainingNode != nil {
 		start := remainingNode.Lines().At(0).Start
 		stop := len(source) - 1

diff --git a/internal/renderer/json_test.go b/internal/renderer/json_test.go
@@ -13,7 +13,7 @@ import (
 	"github.com/yuin/goldmark/text"
 )
 
-var testCases = []string{"happy", "simple"}
+var testCases = []string{"happy", "simple", "linesless"}
 
 func TestParser_Renderer(t *testing.T) {
 	snapshotter := cupaloy.New(cupaloy.SnapshotSubdirectory("testdata/.snapshots"))

diff --git a/internal/renderer/testdata/linesless.md b/internal/renderer/testdata/linesless.md
@@ -0,0 +1,9 @@
+To deploy to production, run:
+
+```sh
+deployctl deploy --project=runme main.ts
+```
+
+---
+
+<p align="center"><small>Copyright 2022 © <a href="http://stateful.com/">Stateful</a> – Apache 2.0 License</small></p>