fix: infinite loop regression in literal (parse/ast) mode

.changeset/thick-wombats-dance.md

@@ -0,0 +1,5 @@
+---
+'@astrojs/compiler': patch
+---
+
+Fix regression introduced to `parse` handling in the last patch

internal/parser.go

@@ -2670,36 +2670,38 @@ func frontmatterIM(p *parser) bool {
 
 // Handle content very literally
 func inLiteralIM(p *parser) bool {
- if !p.exitLiteralIM() {
- switch p.tok.Type {
- case ErrorToken:
- p.oe.pop()
- case TextToken:
- p.addText(p.tok.Data)
- return true
- case StartTagToken:
- p.addElement()
- if p.hasSelfClosingToken {
- p.oe.pop()
- p.acknowledgeSelfClosingTag()
- }
- return true
- case EndTagToken:
- p.addLoc()
- p.oe.pop()
- return true
- case StartExpressionToken:
- p.addExpression()
- return true
- case EndExpressionToken:
- p.addLoc()
+ shouldExit := p.exitLiteralIM()
+ switch p.tok.Type {
+ case ErrorToken:
+ // Stop parsing.
+ case TextToken:
+ p.addText(p.tok.Data)
+ case StartTagToken:
+ p.addElement()
+ if p.hasSelfClosingToken {
  p.oe.pop()
- return true
+ p.acknowledgeSelfClosingTag()
  }
+ case CommentToken:
+ p.addChild(&Node{
+ Type: CommentNode,
+ Data: p.tok.Data,
+ Loc: p.generateLoc(),
+ })
+ case EndTagToken:
+ p.addLoc()
+ p.oe.pop()
+ case StartExpressionToken:
+ p.addExpression()
+ case EndExpressionToken:
+ p.addLoc()
+ p.oe.pop()
  }
- p.im = p.originalIM
- p.originalIM = nil
- return p.tok.Type == EndTagToken
+ if shouldExit {
+ p.im = p.originalIM
+ p.originalIM = nil
+ }
+ return true
 }
 
 func inExpressionIM(p *parser) bool {