Fixed parser issue to support flow style mappings in sequences. (#6394)

Author: sudouser777

rewrite-yaml/src/main/java/org/openrewrite/yaml/YamlParser.java

@@ -748,11 +748,30 @@ public MappingWithPrefix(String prefix, @Nullable String startBracePrefix, List<
             this.prefix = prefix;
         }
 
+        public MappingWithPrefix(UUID id, Markers markers, @Nullable String openingBracePrefix, List<Entry> entries, @Nullable String closingBracePrefix, @Nullable Anchor anchor, @Nullable Tag tag, String prefix) {
+            super(id, markers, openingBracePrefix, entries, closingBracePrefix, anchor, tag);
+            this.prefix = prefix;
+        }
+
         @Override
         public Mapping withPrefix(String prefix) {
             this.prefix = prefix;
             return this;
         }
+
+        @Override
+        public MappingWithPrefix withClosingBracePrefix(@Nullable String closingBracePrefix) {
+            return new MappingWithPrefix(
+                    getId(),
+                    getMarkers(),
+                    getOpeningBracePrefix(),
+                    getEntries(),
+                    closingBracePrefix,
+                    getAnchor(),
+                    getTag(),
+                    this.prefix
+            );
+        }
     }
 
     @Getter
@@ -842,7 +861,7 @@ public Yaml.Mapping visitMapping(Yaml.Mapping mapping, Integer p) {
                 if (mapping instanceof MappingWithPrefix) {
                     MappingWithPrefix mappingWithPrefix = (MappingWithPrefix) mapping;
                     return super.visitMapping(new Yaml.Mapping(mappingWithPrefix.getId(),
-                            mappingWithPrefix.getMarkers(), mappingWithPrefix.getOpeningBracePrefix(), mappingWithPrefix.getEntries(), null, mappingWithPrefix.getAnchor(), mappingWithPrefix.getTag()), p);
+                            mappingWithPrefix.getMarkers(), mappingWithPrefix.getOpeningBracePrefix(), mappingWithPrefix.getEntries(), mappingWithPrefix.getClosingBracePrefix(), mappingWithPrefix.getAnchor(), mappingWithPrefix.getTag()), p);
                 }
                 return super.visitMapping(mapping, p);
             }

rewrite-yaml/src/test/java/org/openrewrite/yaml/YamlParserTest.java

@@ -559,4 +559,63 @@ void helmTemplateMatchingDocumentEndParsesCorrectly() {
           )
         );
     }
+
+    @Test
+    void flowStyleMappingsInSequences() {
+        rewriteRun(
+          yaml(
+            """
+              tasks:
+                - {"task_type": "Shell"}
+                - { "task_type": "Shell2"}
+              """
+          ),
+          yaml(
+            """
+              items:
+                - name: block-style
+                  type: mapping
+                - {"name": "flow-style", "type": "mapping"}
+                - key: another-block
+              """
+          ),
+          yaml(
+            """
+              items:
+                - {}
+                - {"key": "value"}
+              """
+          ),
+          yaml(
+            """
+              data:
+                - {"list": [1, 2, 3], "map": {"nested": "value"}}
+                - {"array": [{"inner": "map"}]}
+              """
+          ),
+          yaml(
+            """
+              items:
+                - {
+                    "key": "value",
+                    "another": "test"
+                  }
+              """
+          ),
+          yaml(
+            """
+              items:
+                - {"key": "value",}
+              """
+          ),
+          yaml(
+            """
+              defaults: &defaults {"type": "default", "enabled": true}
+              items:
+                - *defaults
+                - {"type": "custom"}
+              """
+          )
+        );
+    }
 }