mirror of
https://github.com/rust-lang/rust.git
synced 2024-12-04 20:54:13 +00:00
internal: remove one more usage of old editing API.
This commit is contained in:
parent
a26b1c3923
commit
e300f58d2c
@ -130,7 +130,8 @@ pub fn add_trait_assoc_items_to_impl(
|
|||||||
let items = items.into_iter().map(|assoc_item| {
|
let items = items.into_iter().map(|assoc_item| {
|
||||||
let assoc_item = assoc_item.clone_for_update();
|
let assoc_item = assoc_item.clone_for_update();
|
||||||
transform.apply(assoc_item.syntax());
|
transform.apply(assoc_item.syntax());
|
||||||
edit::remove_attrs_and_docs(&assoc_item).clone_subtree().clone_for_update()
|
edit::remove_attrs_and_docs(&assoc_item);
|
||||||
|
assoc_item
|
||||||
});
|
});
|
||||||
|
|
||||||
let res = impl_.clone_for_update();
|
let res = impl_.clone_for_update();
|
||||||
|
@ -194,10 +194,10 @@ fn get_transformed_assoc_item(
|
|||||||
);
|
);
|
||||||
|
|
||||||
transform.apply(assoc_item.syntax());
|
transform.apply(assoc_item.syntax());
|
||||||
Some(match assoc_item {
|
if let ast::AssocItem::Fn(func) = &assoc_item {
|
||||||
ast::AssocItem::Fn(func) => ast::AssocItem::Fn(edit::remove_attrs_and_docs(&func)),
|
edit::remove_attrs_and_docs(func)
|
||||||
_ => assoc_item,
|
}
|
||||||
})
|
Some(assoc_item)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn add_type_alias_impl(
|
fn add_type_alias_impl(
|
||||||
@ -253,7 +253,7 @@ fn add_const_impl(
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn make_const_compl_syntax(const_: &ast::Const) -> String {
|
fn make_const_compl_syntax(const_: &ast::Const) -> String {
|
||||||
let const_ = edit::remove_attrs_and_docs(const_);
|
edit::remove_attrs_and_docs(const_);
|
||||||
|
|
||||||
let const_start = const_.syntax().text_range().start();
|
let const_start = const_.syntax().text_range().start();
|
||||||
let const_end = const_.syntax().text_range().end();
|
let const_end = const_.syntax().text_range().end();
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
//! Collection of assorted algorithms for syntax trees.
|
//! Collection of assorted algorithms for syntax trees.
|
||||||
|
|
||||||
use std::{hash::BuildHasherDefault, ops::RangeInclusive};
|
use std::hash::BuildHasherDefault;
|
||||||
|
|
||||||
use indexmap::IndexMap;
|
use indexmap::IndexMap;
|
||||||
use itertools::Itertools;
|
use itertools::Itertools;
|
||||||
@ -295,41 +295,6 @@ fn _insert_children(
|
|||||||
with_children(parent, new_children)
|
with_children(parent, new_children)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Replaces all nodes in `to_delete` with nodes from `to_insert`
|
|
||||||
///
|
|
||||||
/// This is a type-unsafe low-level editing API, if you need to use it,
|
|
||||||
/// prefer to create a type-safe abstraction on top of it instead.
|
|
||||||
pub fn replace_children(
|
|
||||||
parent: &SyntaxNode,
|
|
||||||
to_delete: RangeInclusive<SyntaxElement>,
|
|
||||||
to_insert: impl IntoIterator<Item = SyntaxElement>,
|
|
||||||
) -> SyntaxNode {
|
|
||||||
let mut to_insert = to_insert.into_iter();
|
|
||||||
_replace_children(parent, to_delete, &mut to_insert)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn _replace_children(
|
|
||||||
parent: &SyntaxNode,
|
|
||||||
to_delete: RangeInclusive<SyntaxElement>,
|
|
||||||
to_insert: &mut dyn Iterator<Item = SyntaxElement>,
|
|
||||||
) -> SyntaxNode {
|
|
||||||
let start = position_of_child(parent, to_delete.start().clone());
|
|
||||||
let end = position_of_child(parent, to_delete.end().clone());
|
|
||||||
let parent_green = parent.green();
|
|
||||||
let mut old_children = parent_green.children().map(|it| match it {
|
|
||||||
NodeOrToken::Token(it) => NodeOrToken::Token(it.to_owned()),
|
|
||||||
NodeOrToken::Node(it) => NodeOrToken::Node(it.to_owned()),
|
|
||||||
});
|
|
||||||
|
|
||||||
let before = old_children.by_ref().take(start).collect::<Vec<_>>();
|
|
||||||
let new_children = before
|
|
||||||
.into_iter()
|
|
||||||
.chain(to_insert.map(to_green_element))
|
|
||||||
.chain(old_children.skip(end + 1 - start))
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
with_children(parent, new_children)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn with_children(
|
fn with_children(
|
||||||
parent: &SyntaxNode,
|
parent: &SyntaxNode,
|
||||||
new_children: Vec<NodeOrToken<rowan::GreenNode, rowan::GreenToken>>,
|
new_children: Vec<NodeOrToken<rowan::GreenNode, rowan::GreenToken>>,
|
||||||
|
@ -48,22 +48,26 @@ impl ast::UseTree {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[must_use]
|
pub fn remove_attrs_and_docs<N: ast::AttrsOwner>(node: &N) {
|
||||||
pub fn remove_attrs_and_docs<N: ast::AttrsOwner>(node: &N) -> N {
|
remove_attrs_and_docs_inner(node.syntax())
|
||||||
N::cast(remove_attrs_and_docs_inner(node.syntax().clone())).unwrap()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn remove_attrs_and_docs_inner(mut node: SyntaxNode) -> SyntaxNode {
|
fn remove_attrs_and_docs_inner(node: &SyntaxNode) {
|
||||||
while let Some(start) =
|
let mut remove_next_ws = false;
|
||||||
node.children_with_tokens().find(|it| it.kind() == ATTR || it.kind() == COMMENT)
|
for child in node.children_with_tokens() {
|
||||||
{
|
match child.kind() {
|
||||||
let end = match &start.next_sibling_or_token() {
|
ATTR | COMMENT => {
|
||||||
Some(el) if el.kind() == WHITESPACE => el.clone(),
|
remove_next_ws = true;
|
||||||
Some(_) | None => start.clone(),
|
child.detach();
|
||||||
};
|
continue;
|
||||||
node = algo::replace_children(&node, start..=end, &mut iter::empty());
|
}
|
||||||
|
WHITESPACE if remove_next_ws => {
|
||||||
|
child.detach();
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
remove_next_ws = false;
|
||||||
}
|
}
|
||||||
node
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy)]
|
#[derive(Debug, Clone, Copy)]
|
||||||
|
Loading…
Reference in New Issue
Block a user