YAML: Fix comments in flowMapping and flowSequence (#9669)

changelog_unreleased/yaml/pr-9669.md

@@ -0,0 +1,60 @@
+#### Fix comments inside `flowMapping` and `flowSequence` (#9669 by @fisker)
+
+<!-- prettier-ignore -->
+```yaml
+# Input
+a:
+  [
+    a, b,
+    # comment
+  ]
+
+b:
+  # prettier-ignore
+  {
+    a: 1, b: 2,
+    # comment
+  }
+
+# Prettier stable
+a: [a, b]
+  # comment
+
+b:
+  # prettier-ignore
+  {
+    a: 1, b: 2,
+    # comment
+  }
+  # comment
+
+
+# Prettier stable (second format)
+a:
+  [a, b]
+  # comment
+
+b:
+  # prettier-ignore
+  {
+    a: 1, b: 2,
+    # comment
+  }
+  # comment
+  # comment
+
+
+# Prettier master
+a: [
+    a,
+    b,
+    # comment
+  ]
+
+b:
+  # prettier-ignore
+  {
+    a: 1, b: 2,
+    # comment
+  }
+```

src/language-yaml/printer-yaml.js

@@ -527,6 +527,12 @@ function _print(node, parentNode, path, options, print) {
               )
             ),
             ifBreak(",", ""),
+            hasEndComments(node)
+              ? concat([
+                  hardline,
+                  join(hardline, path.map(print, "endComments")),
+                ])
+              : "",
           ])
         ),
         isLastItemEmptyMappingItem ? "" : bracketSpacing,
@@ -703,7 +709,13 @@ function needsSpaceInFrontOfMappingValue(node) {
 
 function shouldPrintEndComments(node) {
   return (
-    hasEndComments(node) && !isNode(node, ["documentHead", "documentBody"])
+    hasEndComments(node) &&
+    !isNode(node, [
+      "documentHead",
+      "documentBody",
+      "flowMapping",
+      "flowSequence",
+    ])
   );
 }
 

tests/yaml/comment/__snapshots__/jsfmt.spec.js.snap

@@ -108,6 +108,84 @@ A:
 ================================================================================
 `;
 
+exports[`flow-sequence-mapping.yml format 1`] = `
+====================================options=====================================
+parsers: ["yaml"]
+printWidth: 80
+                                                                                | printWidth
+=====================================input======================================
+    a:
+      [
+        check-format, check-lint,
+        check-spelling,
+        # coverage,
+        # install-and-run-from-git,
+      ]
+
+    b:
+      {
+        a: check-format, b: check-lint,
+        c: check-spelling,
+        # d: coverage,
+        # e: install-and-run-from-git,
+      }
+
+    d:
+      # prettier-ignore
+      [
+        check-format, check-lint,
+        check-spelling,
+        # coverage,
+        # install-and-run-from-git,
+      ]
+
+    e:
+      # prettier-ignore
+      {
+        a: check-format, b: check-lint,
+        c: check-spelling,
+        # d: coverage,
+        # e: install-and-run-from-git,
+      }
+
+=====================================output=====================================
+a: [
+    check-format,
+    check-lint,
+    check-spelling,
+    # coverage,
+    # install-and-run-from-git,
+  ]
+
+b: {
+    a: check-format,
+    b: check-lint,
+    c: check-spelling,
+    # d: coverage,
+    # e: install-and-run-from-git,
+  }
+
+d:
+  # prettier-ignore
+  [
+        check-format, check-lint,
+        check-spelling,
+        # coverage,
+        # install-and-run-from-git,
+      ]
+
+e:
+  # prettier-ignore
+  {
+        a: check-format, b: check-lint,
+        c: check-spelling,
+        # d: coverage,
+        # e: install-and-run-from-git,
+      }
+
+================================================================================
+`;
+
 exports[`in-empty-item-without-newlline.yml format 1`] = `
 ====================================options=====================================
 parsers: ["yaml"]

tests/yaml/comment/flow-sequence-mapping.yml

@@ -0,0 +1,33 @@
+    a:
+      [
+        check-format, check-lint,
+        check-spelling,
+        # coverage,
+        # install-and-run-from-git,
+      ]
+
+    b:
+      {
+        a: check-format, b: check-lint,
+        c: check-spelling,
+        # d: coverage,
+        # e: install-and-run-from-git,
+      }
+
+    d:
+      # prettier-ignore
+      [
+        check-format, check-lint,
+        check-spelling,
+        # coverage,
+        # install-and-run-from-git,
+      ]
+
+    e:
+      # prettier-ignore
+      {
+        a: check-format, b: check-lint,
+        c: check-spelling,
+        # d: coverage,
+        # e: install-and-run-from-git,
+      }