fix: mergeAllOf at the root of the document

closes: #32

Author: ivangsa

src/main/java/io/zenwave360/jsonrefparser/$RefParser.java

@@ -177,20 +177,31 @@ private void mergeAllOf(Object value, String[] paths, URI currentFileURL) {
                 }
 
                 try {
+                    var isRoot = "$".equals(jsonPath);
                     var mergedAllOfObject = allOfObject.buildAllOfObject();
-                    $Ref originalRef = refs.getOriginalRef(originalAllOfRoot);
+                    $Ref originalRef = isRoot? $Ref.of("", uri) : refs.getOriginalRef(originalAllOfRoot);
                     if (originalRef != null) {
                         refs.saveOriginalRef(originalRef, mergedAllOfObject);
                     }
-                    refs.jsonContext.set(jsonPath, mergedAllOfObject);
+                    if (isRoot) {
+                        var root = refs.jsonContext.json();
+                        if(root instanceof Map) {
+                            ((Map) root).remove("allOf");
+                            ((Map) root).putAll(mergedAllOfObject);
+                        } else {
+                            throw new RuntimeException("Could not understand root: " + root);
+                        }
+                    } else {
+                        refs.jsonContext.set(jsonPath, mergedAllOfObject);
+                    }
                     refs.saveOriginalAllOf(mergedAllOfObject, allOf);
                 } catch (Exception e){
                     log.error("Error setting jsonPath:{} in file:{}", jsonPath, currentFileURL, e);
                     throw e;
                 }
             } else if(value instanceof Map) {
-                // visit
-                ((Map<String, Object>) value).entrySet().forEach(e -> {
+                // visit - use ArrayList to avoid ConcurrentModificationException
+                new ArrayList<>(((Map<String, Object>) value).entrySet()).forEach(e -> {
                     mergeAllOf(e.getValue(), ArrayUtils.add(paths, e.getKey()), currentFileURL);
                 });
             } else if(value instanceof List) {
@@ -257,6 +268,7 @@ private String indent() {
         return StringUtils.join(indent, "");
     }
     private void dereference(ExtendedJsonContext jsonContext, Object value, String[] paths, URI currentFileURL) {
+        // var $id = jsonContext.read("$['$id']");
         var visitedNodeRef = String.format("%s%s", currentFileURL, jsonPointer(paths));
         log.trace("{}visiting {}", indent(), visitedNodeRef);
 

src/test/java/io/zenwave360/jsonrefparser/Gh32MergeAllOfAtRootTest.java

@@ -0,0 +1,24 @@
+package io.zenwave360.jsonrefparser;
+
+import org.junit.Test;
+
+import java.io.File;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
+
+import static org.junit.Assert.assertTrue;
+
+public class Gh32MergeAllOfAtRootTest {
+
+    private ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
+
+    @Test
+    public void testMergeAllOfAtRoot() throws Exception {
+        File file = new File("src/test/resources/gh-32-mergeAllOf-root.yml");
+        $RefParser parser = new $RefParser(file).parse();
+        $Refs refs = parser.dereference().mergeAllOf().getRefs();
+        assertTrue(refs.schema().containsKey("properties"));
+//        System.out.println(mapper.writerWithDefaultPrettyPrinter().writeValueAsString(refs.schema()));
+    }
+}

src/test/resources/gh-32-mergeAllOf-root.yml

@@ -0,0 +1,18 @@
+$schema: https://json-schema.org/draft-07/schema
+$id: urn:schema:example:1.0.0
+title: Example
+type: object
+allOf:
+  - $ref: "#/definitions/Core"
+  - $ref: "#/definitions/Extension"
+definitions:
+  Core:
+    type: object
+    properties:
+      prop1:
+        type: string
+  Extension:
+    type: object
+    properties:
+      prop2:
+        type: string