fix(formatter): correct printing comments for for statements

Author: Dunqing

crates/oxc_formatter/src/write/mod.rs

@@ -659,7 +659,7 @@ impl<'a> FormatWrite<'a> for AstNode<'a, ForStatement<'a>> {
 
 impl<'a> FormatWrite<'a> for AstNode<'a, ForInStatement<'a>> {
     fn write(&self, f: &mut Formatter<'_, 'a>) -> FormatResult<()> {
-        let comments = f.context().comments().own_line_comments_before(self.body.span().start);
+        let comments = f.context().comments().own_line_comments_before(self.right.span().start);
         write!(
             f,
             [
@@ -683,7 +683,7 @@ impl<'a> FormatWrite<'a> for AstNode<'a, ForInStatement<'a>> {
 
 impl<'a> FormatWrite<'a> for AstNode<'a, ForOfStatement<'a>> {
     fn write(&self, f: &mut Formatter<'_, 'a>) -> FormatResult<()> {
-        let comments = f.context().comments().own_line_comments_before(self.body.span().start);
+        let comments = f.context().comments().own_line_comments_before(self.right.span().start);
 
         let r#await = self.r#await();
         let left = self.left();

crates/oxc_formatter/tests/fixtures/js/comments/for-statements.js

@@ -0,0 +1,11 @@
+for (let i in [
+  // comment1
+  1, 2, 3
+  // comment2
+]);
+
+for (let i of [
+  // comment1
+  1, 2, 3
+  // comment2
+]);

crates/oxc_formatter/tests/fixtures/js/comments/for-statements.js.snap

@@ -0,0 +1,31 @@
+---
+source: crates/oxc_formatter/tests/fixtures/mod.rs
+---
+==================== Input ====================
+for (let i in [
+  // comment1
+  1, 2, 3
+  // comment2
+]);
+
+for (let i of [
+  // comment1
+  1, 2, 3
+  // comment2
+]);
+==================== Output ====================
+for (let i in [
+  // comment1
+  1, 2,
+  3,
+  // comment2
+]);
+
+for (let i of [
+  // comment1
+  1, 2,
+  3,
+  // comment2
+]);
+
+===================== End =====================