Skip to content

Commit d8fe110

Browse files
DunqingDunqing
committed
feat(formatter): complete printing ForStatement, ForInStatement and ForOfStatement
1 parent 5a56cc4 commit d8fe110

File tree

11 files changed

+314
-95
lines changed

11 files changed

+314
-95
lines changed

‎crates/oxc_formatter/src/formatter/comments/mod.rs‎

Lines changed: 102 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,25 @@ impl<'a> Comments<'a> {
6060
&comments[..index]
6161
}
6262

63+
/// Returns own line comments that are before the given `pos`.
64+
pub fn own_line_comments_before(&self, pos: u32) -> &'a [Comment] {
65+
let mut index = 0;
66+
67+
let comments = self.unprinted_comments();
68+
for comment in comments {
69+
if !is_own_line_comment(comment, self.source_text) {
70+
break;
71+
}
72+
73+
if comment.span.end > pos {
74+
break;
75+
}
76+
index += 1;
77+
}
78+
79+
&comments[..index]
80+
}
81+
6382
/// Returns the comments that after the given `start` position, even if they were already printed.
6483
pub fn comments_after(&self, pos: u32) -> &'a [Comment] {
6584
let mut index = self.printed_count;
@@ -279,15 +298,39 @@ impl<'a> Comments<'a> {
279298
&self,
280299
enclosing_node: &SiblingNode<'a>,
281300
preceding_node: &SiblingNode<'a>,
282-
following_node: Option<&SiblingNode<'a>>,
301+
mut following_node: Option<&SiblingNode<'a>>,
283302
) -> &'a [Comment] {
303+
if !matches!(
304+
enclosing_node,
305+
SiblingNode::Program(_)
306+
| SiblingNode::BlockStatement(_)
307+
| SiblingNode::FunctionBody(_)
308+
| SiblingNode::TSModuleBlock(_)
309+
| SiblingNode::SwitchStatement(_)
310+
| SiblingNode::StaticBlock(_)
311+
) && matches!(following_node, Some(SiblingNode::EmptyStatement(_)))
312+
{
313+
let enclosing_span = enclosing_node.span();
314+
return self.comments_before(enclosing_span.end);
315+
}
316+
284317
// The preceding_node is the callee of the call expression or new expression, let following node to print it.
285318
// Based on https://github.com/prettier/prettier/blob/7584432401a47a26943dd7a9ca9a8e032ead7285/src/language-js/comments/handle-comments.js#L726-L741
286319
if matches!(enclosing_node, SiblingNode::CallExpression(CallExpression { callee, ..}) | SiblingNode::NewExpression(NewExpression { callee, ..}) if callee.span().contains_inclusive(preceding_node.span()))
287320
{
288321
return &[];
289322
}
290323

324+
// No need to print trailing comments for the most right side for `BinaryExpression` and `LogicalExpression`,
325+
// instead, print trailing comments for expression itself.
326+
if matches!(
327+
enclosing_node,
328+
SiblingNode::BinaryExpression(_) | SiblingNode::LogicalExpression(_)
329+
) && matches!(following_node, Some(SiblingNode::ExpressionStatement(_)))
330+
{
331+
return &[];
332+
}
333+
291334
let comments = self.unprinted_comments();
292335
if comments.is_empty() {
293336
return &[];
@@ -303,15 +346,11 @@ impl<'a> Comments<'a> {
303346

304347
let Some(following_node) = following_node else {
305348
let enclosing_span = enclosing_node.span();
306-
return comments
307-
.iter()
308-
.enumerate()
309-
.take_while(|(_, comment)| comment.span.end <= enclosing_span.end)
310-
.last()
311-
.map_or(&[], |(index, _)| &comments[..=index]);
349+
return self.comments_before(enclosing_span.end);
312350
};
313351

314352
let following_span = following_node.span();
353+
315354
let mut comment_index = 0;
316355
while let Some(comment) = comments.get(comment_index) {
317356
// Check if the comment is before the following node's span
@@ -322,21 +361,42 @@ impl<'a> Comments<'a> {
322361
if is_own_line_comment(comment, source_text) {
323362
// TODO: describe the logic here
324363
// Reached an own line comment, which means it is the leading comment for the next node.
364+
365+
if matches!(enclosing_node, SiblingNode::IfStatement(stmt) if stmt.test.span() == preceding_span)
366+
|| matches!(enclosing_node, SiblingNode::WhileStatement(stmt) if stmt.test.span() == preceding_span)
367+
{
368+
return handle_if_and_while_statement_comments(
369+
following_span.start,
370+
comment_index,
371+
comments,
372+
source_text,
373+
);
374+
}
375+
325376
break;
326377
} else if is_end_of_line_comment(comment, source_text) {
327-
let first_comment_start = comments[0].span.start;
328-
// TODO: Maybe we can remove this check once we complete the logic of handling comments.
329-
if preceding_span.end < first_comment_start {
330-
let is_break = source_text.as_bytes()
331-
[(preceding_span.end as usize)..(first_comment_start as usize)]
332-
.iter()
333-
.any(|&b| matches!(b, b'\n' | b'\r' | b')'));
334-
335-
if is_break {
336-
return &[];
378+
if let SiblingNode::IfStatement(if_stmt) = enclosing_node {
379+
if if_stmt.consequent.span() == preceding_span {
380+
// If comment is after the `else` keyword, it is not a trailing comment of consequent.
381+
if source_text[preceding_span.end as usize..comment.span.start as usize]
382+
.contains("else")
383+
{
384+
return &[];
385+
}
337386
}
338387
}
339388

389+
if matches!(enclosing_node, SiblingNode::IfStatement(stmt) if stmt.test.span() == preceding_span)
390+
|| matches!(enclosing_node, SiblingNode::WhileStatement(stmt) if stmt.test.span() == preceding_span)
391+
{
392+
return handle_if_and_while_statement_comments(
393+
following_span.start,
394+
comment_index,
395+
comments,
396+
source_text,
397+
);
398+
}
399+
340400
// Should be a leading comment of following node.
341401
// Based on https://github.com/prettier/prettier/blob/7584432401a47a26943dd7a9ca9a8e032ead7285/src/language-js/comments/handle-comments.js#L852-L883
342402
if matches!(
@@ -383,6 +443,31 @@ impl<'a> Comments<'a> {
383443
}
384444
}
385445

446+
fn handle_if_and_while_statement_comments<'a>(
447+
mut end: u32,
448+
mut comment_index: usize,
449+
comments: &'a [Comment],
450+
source_text: &'a str,
451+
) -> &'a [Comment] {
452+
// `if (a /* comment before paren */) // comment after paren`
453+
loop {
454+
let cur_comment_span = comments[comment_index].span;
455+
if source_text.as_bytes()[cur_comment_span.end as usize..end as usize].contains(&b')') {
456+
return &comments[..=comment_index];
457+
}
458+
459+
end = cur_comment_span.start;
460+
461+
if comment_index == 0 {
462+
return &[];
463+
}
464+
465+
comment_index -= 1;
466+
}
467+
468+
unreachable!()
469+
}
470+
386471
#[inline]
387472
pub fn is_new_line(char: char) -> ControlFlow<bool> {
388473
if char == ' ' || char == '\t' {

‎crates/oxc_formatter/src/formatter/trivia.rs‎

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -208,14 +208,13 @@ impl<'a> Format<'a> for FormatTrailingComments<'a, '_> {
208208

209209
match self {
210210
Self::Node((enclosing_node, preceding_node, following_node)) => {
211-
format_trailing_comments_impl(
212-
f.context().comments().get_trailing_comments(
213-
enclosing_node,
214-
preceding_node,
215-
*following_node,
216-
),
217-
f,
218-
)
211+
let comments = f.context().comments().get_trailing_comments(
212+
enclosing_node,
213+
preceding_node,
214+
*following_node,
215+
);
216+
217+
format_trailing_comments_impl(comments, f)
219218
}
220219
Self::Comments(comments) => format_trailing_comments_impl(*comments, f),
221220
}

‎crates/oxc_formatter/src/generated/ast_nodes.rs‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -5818,7 +5818,7 @@ impl<'a> GetSpan for AstNode<'a, Hashbang<'a>> {
58185818
impl<'a> AstNode<'a, BlockStatement<'a>> {
58195819
#[inline]
58205820
pub fn body(&self) -> &AstNode<'a, Vec<'a, Statement<'a>>> {
5821-
let following_node = self.following_node;
5821+
let following_node = None;
58225822
self.allocator.alloc(AstNode {
58235823
inner: &self.inner.body,
58245824
allocator: self.allocator,
@@ -6154,7 +6154,7 @@ impl<'a> AstNode<'a, DoWhileStatement<'a>> {
61546154

61556155
#[inline]
61566156
pub fn test(&self) -> &AstNode<'a, Expression<'a>> {
6157-
let following_node = self.following_node;
6157+
let following_node = None;
61586158
self.allocator.alloc(AstNode {
61596159
inner: &self.inner.test,
61606160
allocator: self.allocator,
@@ -6198,7 +6198,7 @@ impl<'a> AstNode<'a, WhileStatement<'a>> {
61986198

61996199
#[inline]
62006200
pub fn body(&self) -> &AstNode<'a, Statement<'a>> {
6201-
let following_node = self.following_node;
6201+
let following_node = None;
62026202
self.allocator.alloc(AstNode {
62036203
inner: &self.inner.body,
62046204
allocator: self.allocator,
@@ -6281,7 +6281,7 @@ impl<'a> AstNode<'a, ForStatement<'a>> {
62816281

62826282
#[inline]
62836283
pub fn body(&self) -> &AstNode<'a, Statement<'a>> {
6284-
let following_node = self.following_node;
6284+
let following_node = None;
62856285
self.allocator.alloc(AstNode {
62866286
inner: &self.inner.body,
62876287
allocator: self.allocator,
@@ -6372,7 +6372,7 @@ impl<'a> AstNode<'a, ForInStatement<'a>> {
63726372

63736373
#[inline]
63746374
pub fn body(&self) -> &AstNode<'a, Statement<'a>> {
6375-
let following_node = self.following_node;
6375+
let following_node = None;
63766376
self.allocator.alloc(AstNode {
63776377
inner: &self.inner.body,
63786378
allocator: self.allocator,
@@ -6468,7 +6468,7 @@ impl<'a> AstNode<'a, ForOfStatement<'a>> {
64686468

64696469
#[inline]
64706470
pub fn body(&self) -> &AstNode<'a, Statement<'a>> {
6471-
let following_node = self.following_node;
6471+
let following_node = None;
64726472
self.allocator.alloc(AstNode {
64736473
inner: &self.inner.body,
64746474
allocator: self.allocator,
@@ -6501,7 +6501,7 @@ impl<'a> GetSpan for AstNode<'a, ForOfStatement<'a>> {
65016501
impl<'a> AstNode<'a, ContinueStatement<'a>> {
65026502
#[inline]
65036503
pub fn label(&self) -> Option<&AstNode<'a, LabelIdentifier<'a>>> {
6504-
let following_node = self.following_node;
6504+
let following_node = None;
65056505
self.allocator
65066506
.alloc(self.inner.label.as_ref().map(|inner| AstNode {
65076507
inner,
@@ -6536,7 +6536,7 @@ impl<'a> GetSpan for AstNode<'a, ContinueStatement<'a>> {
65366536
impl<'a> AstNode<'a, BreakStatement<'a>> {
65376537
#[inline]
65386538
pub fn label(&self) -> Option<&AstNode<'a, LabelIdentifier<'a>>> {
6539-
let following_node = self.following_node;
6539+
let following_node = None;
65406540
self.allocator
65416541
.alloc(self.inner.label.as_ref().map(|inner| AstNode {
65426542
inner,
@@ -6617,7 +6617,7 @@ impl<'a> AstNode<'a, WithStatement<'a>> {
66176617

66186618
#[inline]
66196619
pub fn body(&self) -> &AstNode<'a, Statement<'a>> {
6620-
let following_node = self.following_node;
6620+
let following_node = None;
66216621
self.allocator.alloc(AstNode {
66226622
inner: &self.inner.body,
66236623
allocator: self.allocator,
@@ -6668,7 +6668,7 @@ impl<'a> AstNode<'a, SwitchStatement<'a>> {
66686668

66696669
#[inline]
66706670
pub fn cases(&self) -> &AstNode<'a, Vec<'a, SwitchCase<'a>>> {
6671-
let following_node = self.following_node;
6671+
let following_node = None;
66726672
self.allocator.alloc(AstNode {
66736673
inner: &self.inner.cases,
66746674
allocator: self.allocator,
@@ -6765,7 +6765,7 @@ impl<'a> AstNode<'a, LabeledStatement<'a>> {
67656765

67666766
#[inline]
67676767
pub fn body(&self) -> &AstNode<'a, Statement<'a>> {
6768-
let following_node = self.following_node;
6768+
let following_node = None;
67696769
self.allocator.alloc(AstNode {
67706770
inner: &self.inner.body,
67716771
allocator: self.allocator,
@@ -6862,7 +6862,7 @@ impl<'a> AstNode<'a, TryStatement<'a>> {
68626862

68636863
#[inline]
68646864
pub fn finalizer(&self) -> Option<&AstNode<'a, BlockStatement<'a>>> {
6865-
let following_node = self.following_node;
6865+
let following_node = None;
68666866
self.allocator
68676867
.alloc(self.inner.finalizer.as_ref().map(|inner| AstNode {
68686868
inner: inner.as_ref(),

‎crates/oxc_formatter/src/generated/format_write.rs‎

Lines changed: 27 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,11 @@ impl<'a> FormatWrite<'a> for AstNode<'a, AssignmentTarget<'a>> {
523523
fn write(&self, f: &mut Formatter<'_, 'a>) -> FormatResult<()> {
524524
let allocator = self.allocator;
525525
let parent = self.parent;
526-
match self.inner {
526+
let needs_parentheses = self.needs_parentheses(f);
527+
if needs_parentheses {
528+
"(".fmt(f)?;
529+
}
530+
let result = match self.inner {
527531
it @ match_simple_assignment_target!(AssignmentTarget) => {
528532
let inner = it.to_simple_assignment_target();
529533
allocator
@@ -546,7 +550,11 @@ impl<'a> FormatWrite<'a> for AstNode<'a, AssignmentTarget<'a>> {
546550
})
547551
.fmt(f)
548552
}
553+
};
554+
if needs_parentheses {
555+
")".fmt(f)?;
549556
}
557+
result
550558
}
551559
}
552560

@@ -555,7 +563,11 @@ impl<'a> FormatWrite<'a> for AstNode<'a, SimpleAssignmentTarget<'a>> {
555563
fn write(&self, f: &mut Formatter<'_, 'a>) -> FormatResult<()> {
556564
let allocator = self.allocator;
557565
let parent = self.parent;
558-
match self.inner {
566+
let needs_parentheses = self.needs_parentheses(f);
567+
if needs_parentheses {
568+
"(".fmt(f)?;
569+
}
570+
let result = match self.inner {
559571
SimpleAssignmentTarget::AssignmentTargetIdentifier(inner) => allocator
560572
.alloc(AstNode::<IdentifierReference> {
561573
inner,
@@ -607,7 +619,11 @@ impl<'a> FormatWrite<'a> for AstNode<'a, SimpleAssignmentTarget<'a>> {
607619
})
608620
.fmt(f)
609621
}
622+
};
623+
if needs_parentheses {
624+
")".fmt(f)?;
610625
}
626+
result
611627
}
612628
}
613629

@@ -616,7 +632,11 @@ impl<'a> FormatWrite<'a> for AstNode<'a, AssignmentTargetPattern<'a>> {
616632
fn write(&self, f: &mut Formatter<'_, 'a>) -> FormatResult<()> {
617633
let allocator = self.allocator;
618634
let parent = self.parent;
619-
match self.inner {
635+
let needs_parentheses = self.needs_parentheses(f);
636+
if needs_parentheses {
637+
"(".fmt(f)?;
638+
}
639+
let result = match self.inner {
620640
AssignmentTargetPattern::ArrayAssignmentTarget(inner) => allocator
621641
.alloc(AstNode::<ArrayAssignmentTarget> {
622642
inner,
@@ -633,7 +653,11 @@ impl<'a> FormatWrite<'a> for AstNode<'a, AssignmentTargetPattern<'a>> {
633653
following_node: self.following_node,
634654
})
635655
.fmt(f),
656+
};
657+
if needs_parentheses {
658+
")".fmt(f)?;
636659
}
660+
result
637661
}
638662
}
639663

0 commit comments

Comments
 (0)