fix(#531): handle slots before <head>

withastro · Oct 4, 2022 · f6390ce · f6390ce
1 parent c27a7b5
commit f6390ce
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/internal/parser.go b/internal/parser.go
@@ -702,6 +702,14 @@ func beforeHeadIM(p *parser) bool {
 		return true
 	case StartTagToken:
 		switch p.tok.DataAtom {
+		case a.Slot:
+			p.addElement()
+			if p.hasSelfClosingToken {
+				p.addLoc()
+				p.oe.pop()
+				p.acknowledgeSelfClosingTag()
+			}
+			return true
 		case a.Head:
 			p.addElement()
 			p.head = p.top()

diff --git a/internal/printer/printer_test.go b/internal/printer/printer_test.go
@@ -109,6 +109,13 @@ func TestPrinter(t *testing.T) {
 				code: `<html><head><title>Ah</title>` + RENDER_HEAD_RESULT + `</head></html>`,
 			},
 		},
+		{
+			name:   "head inside slot",
+			source: `<html><slot><head></head></html>`,
+			want: want{
+				code: `<html>${$$renderSlot($$result,$$slots["default"],$$render` + BACKTICK + `<head>` + RENDER_HEAD_RESULT + `</head>` + BACKTICK + `)}</html>`,
+			},
+		},
 		{
 			name:   "head slot",
 			source: `<html><head><slot /></html>`,