YamlSequenceBuilder.add(JsonStructure)

decorators-squad · May 1, 2024 · bd36e10 · bd36e10
1 parent dfac7f0
commit bd36e10
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 1 deletion.
diff --git a/src/main/java/com/amihaiemil/eoyaml/YamlSequenceBuilder.java b/src/main/java/com/amihaiemil/eoyaml/YamlSequenceBuilder.java
@@ -27,6 +27,9 @@
  */
 package com.amihaiemil.eoyaml;
 
+import javax.json.JsonArray;
+import javax.json.JsonObject;
+import javax.json.JsonStructure;
 import java.util.Collection;
 
 /**
@@ -51,6 +54,23 @@ public interface YamlSequenceBuilder {
      */
     YamlSequenceBuilder add(final YamlNode node);
 
+    /**
+     * Add a value to the sequence - this will be printed in Flow style
+     * (JSON-like).
+     * @param value JsonStructure ({@link javax.json.JsonObject}
+     *              or {@link javax.json.JsonArray})
+     * @return Builder
+     */
+    default YamlSequenceBuilder add(final JsonStructure value) {
+        final YamlNode node;
+        if(value instanceof JsonObject) {
+            node = new JsonYamlMapping((JsonObject) value);
+        } else {
+            node = new JsonYamlSequence((JsonArray) value);
+        }
+        return this.add(node);
+    }
+
     /**
      * Build the YamlSequence.
      * @return Built YamlSequence

diff --git a/src/test/java/com/amihaiemil/eoyaml/YamlMappingPrintTest.java b/src/test/java/com/amihaiemil/eoyaml/YamlMappingPrintTest.java
@@ -179,6 +179,18 @@ public void printsBuiltYamlMappingWithAllNodes() throws Exception {
                 Yaml.createYamlInput("[a, b, c]").readYamlSequence(),
                 "scalarToFlowSequence"
             )
+            .add(
+                "someFlowSeq",
+                Yaml.createYamlSequenceBuilder()
+                    .add(
+                        Json.createArrayBuilder()
+                            .add("a")
+                            .add("b")
+                            .add("c")
+                            .build()
+                    )
+                    .build()
+            )
             .build();
         MatcherAssert.assertThat(
             built.toString(),

diff --git a/src/test/resources/printing_tests/yamlMappingAllNodes.txt b/src/test/resources/printing_tests/yamlMappingAllNodes.txt
@@ -32,4 +32,6 @@ key10: {a: b, c: d, e: f}
 : scalarToFlowMapping
 ?
   [a, b, c]
-: scalarToFlowSequence
+: scalarToFlowSequence
+someFlowSeq:
+  - [a, b, c]