Added condition for '+' case. (prettier#13115)

Co-authored-by: Sosuke Suzuki <aosukeke@gmail.com>
nthachus · Aug 7, 2022 · 00ec91b · 00ec91b
1 parent 67efa96
commit 00ec91b
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 1 deletion.
diff --git a/src/language-js/utils/index.js b/src/language-js/utils/index.js
@@ -883,9 +883,26 @@ function isSimpleCallArgument(node, depth) {
     );
   }
 
+  const targetUnaryExpressionOperators = {
+    "!": true,
+    "-": true,
+    "+": true,
+    "~": true,
+  };
   if (
     node.type === "UnaryExpression" &&
-    (node.operator === "!" || node.operator === "-")
+    targetUnaryExpressionOperators[node.operator]
+  ) {
+    return isSimpleCallArgument(node.argument, depth);
+  }
+
+  const targetUpdateExpressionOperators = {
+    "++": true,
+    "--": true,
+  };
+  if (
+    node.type === "UpdateExpression" &&
+    targetUpdateExpressionOperators[node.operator]
   ) {
     return isSimpleCallArgument(node.argument, depth);
   }

diff --git a/tests/format/js/method-chain/13018.js b/tests/format/js/method-chain/13018.js
@@ -0,0 +1,8 @@
+foo(_a).bar().leet();
+foo(-a).bar().leet();
+foo(+a).bar().leet();
+foo(~a).bar().leet();
+foo(++a).bar().leet();
+foo(--a).bar().leet();
+foo(a++).bar().leet();
+foo(a--).bar().leet();
diff --git a/tests/format/js/method-chain/__snapshots__/jsfmt.spec.js.snap b/tests/format/js/method-chain/__snapshots__/jsfmt.spec.js.snap
@@ -1,5 +1,33 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
+exports[`13018.js format 1`] = `
+====================================options=====================================
+parsers: ["babel", "flow", "typescript"]
+printWidth: 80
+                                                                                | printWidth
+=====================================input======================================
+foo(_a).bar().leet();
+foo(-a).bar().leet();
+foo(+a).bar().leet();
+foo(~a).bar().leet();
+foo(++a).bar().leet();
+foo(--a).bar().leet();
+foo(a++).bar().leet();
+foo(a--).bar().leet();
+
+=====================================output=====================================
+foo(_a).bar().leet();
+foo(-a).bar().leet();
+foo(+a).bar().leet();
+foo(~a).bar().leet();
+foo(++a).bar().leet();
+foo(--a).bar().leet();
+foo(a++).bar().leet();
+foo(a--).bar().leet();
+
+================================================================================
+`;
+
 exports[`bracket_0.js format 1`] = `
 ====================================options=====================================
 parsers: ["babel", "flow", "typescript"]