fix(formatter): correct printing comments around the expression of ComputedMemberExpression (#14486)

Author: Dunqing

crates/oxc_formatter/src/generated/ast_nodes.rs

@@ -4189,7 +4189,7 @@ impl<'a> AstNode<'a, ComputedMemberExpression<'a>> {
 
     #[inline]
     pub fn expression(&self) -> &AstNode<'a, Expression<'a>> {
-        let following_node = self.following_node;
+        let following_node = None;
         self.allocator.alloc(AstNode {
             inner: &self.inner.expression,
             allocator: self.allocator,

crates/oxc_formatter/src/utils/member_chain/chain_member.rs

@@ -141,7 +141,9 @@ impl<'a> Format<'a> for FormatComputedMemberExpressionWithoutObject<'a, '_> {
             write!(f, [soft_line_break(), FormatLeadingComments::Comments(comments)])?;
         }
 
-        if matches!(self.expression, Expression::NumericLiteral(_)) {
+        if matches!(self.expression, Expression::NumericLiteral(_))
+            && !f.comments().has_comment_before(self.span.end)
+        {
             write!(f, [self.optional().then_some("?."), "[", self.expression(), "]"])
         } else {
             write!(

crates/oxc_formatter/tests/fixtures/js/comments/computed-member.js

@@ -0,0 +1,8 @@
+prop[
+  0
+  /* shouldCast */
+] = shouldCast;
+
+let handler =
+  props[handlerName = toHandlerKey(event)] || // also try camelCase event handler (#2249)
+  props[handlerName = toHandlerKey(camelize(event))];

crates/oxc_formatter/tests/fixtures/js/comments/computed-member.js.snap

@@ -0,0 +1,24 @@
+---
+source: crates/oxc_formatter/tests/fixtures/mod.rs
+---
+==================== Input ====================
+prop[
+  0
+  /* shouldCast */
+] = shouldCast;
+
+let handler =
+  props[handlerName = toHandlerKey(event)] || // also try camelCase event handler (#2249)
+  props[handlerName = toHandlerKey(camelize(event))];
+
+==================== Output ====================
+prop[
+  0
+  /* shouldCast */
+] = shouldCast;
+
+let handler =
+  props[handlerName = toHandlerKey(event)] || // also try camelCase event handler (#2249)
+  props[handlerName = toHandlerKey(camelize(event))];
+
+===================== End =====================

tasks/ast_tools/src/generators/formatter/ast_nodes.rs

@@ -20,8 +20,13 @@ const FORMATTER_CRATE_PATH: &str = "crates/oxc_formatter";
 
 /// Based on the printing comments algorithm, the last child of these AST nodes don't need to print comments.
 /// Without following nodes could lead to only print comments that before the end of the node, which is what we want.
-const AST_NODE_WITHOUT_FOLLOWING_NODE_LIST: &[&str] =
-    &["AssignmentExpression", "FormalParameters", "StaticMemberExpression", "ObjectProperty"];
+const AST_NODE_WITHOUT_FOLLOWING_NODE_LIST: &[&str] = &[
+    "AssignmentExpression",
+    "FormalParameters",
+    "StaticMemberExpression",
+    "ComputedMemberExpression",
+    "ObjectProperty",
+];
 
 const AST_NODE_WITH_FOLLOWING_NODE_LIST: &[&str] = &["Function", "Class"];