Add HasChildren::replaceChildren()

1 files changed, 7 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/documents/Document.java b/src/main/java/org/zwobble/mammoth/internal/documents/Document.java
index 5adc50d..b1807d5 100644
--- a/src/main/java/org/zwobble/mammoth/internal/documents/Document.java
+++ b/src/main/java/org/zwobble/mammoth/internal/documents/Document.java
@@ -2,7 +2,7 @@ package org.zwobble.mammoth.internal.documents;
 
 import java.util.List;
 
-public class Document implements HasChildren {
+public class Document implements HasChildren<Document> {
     private final List<DocumentElement> children;
     private final Notes notes;
     private final List<Comment> comments;
@@ -13,10 +13,16 @@ public class Document implements HasChildren {
         this.comments = comments;
     }
 
+    @Override
     public List<DocumentElement> getChildren() {
         return children;
     }
 
+    @Override
+    public Document replaceChildren(List<DocumentElement> newChildren) {
+        return new Document(newChildren, this.notes, this.comments);
+    }
+
     public Notes getNotes() {
         return notes;
     }