Add with_use_tree

Author: matklad

crates/ra_syntax/src/ast/edit.rs

@@ -249,6 +249,16 @@ impl ast::PathSegment {
     }
 }
 
+impl ast::UseItem {
+    #[must_use]
+    pub fn with_use_tree(&self, use_tree: ast::UseTree) -> ast::UseItem {
+        if let Some(old) = self.use_tree() {
+            return replace_descendants(self, iter::once((old, use_tree)));
+        }
+        self.clone()
+    }
+}
+
 #[must_use]
 pub fn strip_attrs_and_docs<N: ast::AttrsOwner>(node: &N) -> N {
     N::cast(strip_attrs_and_docs_inner(node.syntax().clone())).unwrap()