Extract JSX related print functions (#9714)

prettier · Nov 19, 2020 · b12e869 · b12e869
1 parent 7acaa55
commit b12e869
Show file tree

Hide file tree

Showing 3 changed files with 781 additions and 702 deletions.
diff --git a/src/language-js/comments.js b/src/language-js/comments.js
@@ -15,6 +15,10 @@ const {
   isBlockComment,
   getFunctionParameters,
   isPrettierIgnoreComment,
+  isJSXNode,
+  hasFlowShorthandAnnotationComment,
+  hasFlowAnnotationComment,
+  hasIgnoreComment,
 } = require("./utils");
 const { locStart, locEnd } = require("./loc");
 
@@ -857,6 +861,33 @@ function isTypeCastComment(comment) {
   );
 }
 
+function willPrintOwnComments(path /*, options */) {
+  const node = path.getValue();
+  const parent = path.getParentNode();
+
+  return (
+    ((node &&
+      (isJSXNode(node) ||
+        hasFlowShorthandAnnotationComment(node) ||
+        (parent &&
+          (parent.type === "CallExpression" ||
+            parent.type === "OptionalCallExpression") &&
+          (hasFlowAnnotationComment(node.leadingComments) ||
+            hasFlowAnnotationComment(node.trailingComments))))) ||
+      (parent &&
+        (parent.type === "JSXSpreadAttribute" ||
+          parent.type === "JSXSpreadChild" ||
+          parent.type === "UnionTypeAnnotation" ||
+          parent.type === "TSUnionType" ||
+          ((parent.type === "ClassDeclaration" ||
+            parent.type === "ClassExpression") &&
+            parent.superClass === node)))) &&
+    (!hasIgnoreComment(path) ||
+      parent.type === "UnionTypeAnnotation" ||
+      parent.type === "TSUnionType")
+  );
+}
+
 module.exports = {
   handleOwnLineComment,
   handleEndOfLineComment,
@@ -865,4 +896,5 @@ module.exports = {
   isTypeCastComment,
   getGapRegex,
   getCommentChildNodes,
+  willPrintOwnComments,
 };