feat(pretty): Preserve the frontmatter

rust-lang · Mar 10, 2025 · 078a6f6 · 078a6f6
1 parent c002b34
commit 078a6f6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/compiler/rustc_ast_pretty/src/pprust/state.rs b/compiler/rustc_ast_pretty/src/pprust/state.rs
@@ -121,6 +121,15 @@ fn gather_comments(sm: &SourceMap, path: FileName, src: String) -> Vec<Comment>
         pos += shebang_len;
     }
 
+    if let Some(frontmatter_len) = rustc_lexer::strip_frontmatter(&text[pos..]) {
+        comments.push(Comment {
+            style: CommentStyle::Isolated,
+            lines: vec![text[pos..pos + frontmatter_len].to_string()],
+            pos: start_bpos + BytePos(pos as u32),
+        });
+        pos += frontmatter_len;
+    }
+
     for token in rustc_lexer::tokenize(&text[pos..]) {
         let token_text = &text[pos..pos + token.len as usize];
         match token.kind {

diff --git a/tests/ui/unpretty/frontmatter.stdout b/tests/ui/unpretty/frontmatter.stdout
@@ -1,3 +1,5 @@
+---
+---
 
 //@ compile-flags: -Zunpretty=normal
 //@ check-pass