Added deep_clone for nodes

Author: Quotique

src/lib.rs

@@ -290,6 +290,26 @@ impl<T> Tree<T> {
     }
 }
 
+impl<T: Clone> Tree<T> {
+    /// Clones the node deeply and creates a new tree.
+    /// Skips orphaned nodes.
+    pub fn deep_clone(&self, id: NodeId) -> Self {
+        let start_node = unsafe { self.get_unchecked(id) };
+        let mut result = Self::new(start_node.value().clone());
+        let mut id_map = std::collections::HashMap::new();
+        id_map.insert(start_node.id, result.root().id);
+
+        for orig in start_node.descendants() {
+            let copy_id = id_map.get(&orig.id).unwrap();
+            let mut copy = unsafe { result.get_unchecked_mut(*copy_id) };
+            for child in orig.children() {
+                id_map.insert(child.id, copy.append(child.value().clone()).id);
+            }
+        }
+        result
+    }
+}
+
 impl<'a, T: 'a> NodeRef<'a, T> {
     /// Returns the ID of this node.
     pub fn id(&self) -> NodeId {
@@ -349,6 +369,14 @@ impl<'a, T: 'a> NodeRef<'a, T> {
     }
 }
 
+impl<'a, T: 'a + Clone> NodeRef<'a, T> {
+    /// Clones the node deeply and creates a new tree.
+    /// Skips orphaned nodes.
+    pub fn deep_clone(&self) -> Tree<T> {
+        self.tree.deep_clone(self.id)
+    }
+}
+
 impl<'a, T: 'a> NodeMut<'a, T> {
     /// Returns the ID of this node.
     pub fn id(&self) -> NodeId {

tests/tree.rs

@@ -238,3 +238,50 @@ fn test_display() {
 
     assert_eq!(repr, expected);
 }
+
+#[test]
+fn clone_subtree() {
+    let mut tree = tree! {
+        "root" => {
+            "a" => {
+                "child 1",
+                "child 2",
+                "child 3",
+            },
+            "b" => {
+                "child 4",
+            },
+        }
+    };
+
+    let test_node_id = tree.root().first_child().unwrap().id();
+    let test_subtree = tree.deep_clone(test_node_id);
+
+    assert_eq!(
+        test_subtree,
+        tree! {
+            "a" => {
+                "child 1",
+                "child 2",
+                "child 3",
+            }
+        }
+    );
+
+    tree.root_mut()
+        .first_child()
+        .unwrap()
+        .last_child()
+        .unwrap()
+        .detach();
+    let test_subtree = tree.deep_clone(test_node_id);
+    assert_eq!(
+        test_subtree,
+        tree! {
+            "a" => {
+                "child 1",
+                "child 2",
+            }
+        }
+    );
+}