feat(edit): Add ImDocument::into_table (#928)

epage · web-flow · commit f3e8c6893c38 · 2025-06-06T09:53:40.000-05:00
Fixes #926
diff --git a/crates/toml_edit/src/document.rs b/crates/toml_edit/src/document.rs
@@ -43,11 +43,23 @@ impl<S> ImDocument<S> {
         &self.root
     }
 
+    /// Returns the root item.
+    pub fn into_item(self) -> Item {
+        self.root
+    }
+
     /// Returns a reference to the root table.
     pub fn as_table(&self) -> &Table {
         self.root.as_table().expect("root should always be a table")
     }
 
+    /// Returns the root table.
+    pub fn into_table(self) -> Table {
+        self.root
+            .into_table()
+            .expect("root should always be a table")
+    }
+
     /// Returns an iterator over the root table.
     pub fn iter(&self) -> Iter<'_> {
         self.as_table().iter()
@@ -129,6 +141,11 @@ impl DocumentMut {
         &mut self.root
     }
 
+    /// Returns the root item.
+    pub fn into_item(self) -> Item {
+        self.root
+    }
+
     /// Returns a reference to the root table.
     pub fn as_table(&self) -> &Table {
         self.root.as_table().expect("root should always be a table")
@@ -141,6 +158,13 @@ impl DocumentMut {
             .expect("root should always be a table")
     }
 
+    /// Returns the root table.
+    pub fn into_table(self) -> Table {
+        self.root
+            .into_table()
+            .expect("root should always be a table")
+    }
+
     /// Returns an iterator over the root table.
     pub fn iter(&self) -> Iter<'_> {
         self.as_table().iter()
