diff options
Diffstat (limited to 'crates/syntax/src/ast/edit.rs')
-rw-r--r-- | crates/syntax/src/ast/edit.rs | 78 |
1 files changed, 73 insertions, 5 deletions
diff --git a/crates/syntax/src/ast/edit.rs b/crates/syntax/src/ast/edit.rs index 060b20966..8b1c65dd6 100644 --- a/crates/syntax/src/ast/edit.rs +++ b/crates/syntax/src/ast/edit.rs | |||
@@ -13,7 +13,7 @@ use crate::{ | |||
13 | ast::{ | 13 | ast::{ |
14 | self, | 14 | self, |
15 | make::{self, tokens}, | 15 | make::{self, tokens}, |
16 | AstNode, TypeBoundsOwner, | 16 | AstNode, GenericParamsOwner, NameOwner, TypeBoundsOwner, |
17 | }, | 17 | }, |
18 | AstToken, Direction, InsertPosition, SmolStr, SyntaxElement, SyntaxKind, | 18 | AstToken, Direction, InsertPosition, SmolStr, SyntaxElement, SyntaxKind, |
19 | SyntaxKind::{ATTR, COMMENT, WHITESPACE}, | 19 | SyntaxKind::{ATTR, COMMENT, WHITESPACE}, |
@@ -46,6 +46,19 @@ impl ast::Fn { | |||
46 | to_insert.push(body.syntax().clone().into()); | 46 | to_insert.push(body.syntax().clone().into()); |
47 | self.replace_children(single_node(old_body_or_semi), to_insert) | 47 | self.replace_children(single_node(old_body_or_semi), to_insert) |
48 | } | 48 | } |
49 | |||
50 | #[must_use] | ||
51 | pub fn with_generic_param_list(&self, generic_args: ast::GenericParamList) -> ast::Fn { | ||
52 | if let Some(old) = self.generic_param_list() { | ||
53 | return self.replace_descendant(old, generic_args); | ||
54 | } | ||
55 | |||
56 | let anchor = self.name().expect("The function must have a name").syntax().clone(); | ||
57 | |||
58 | let mut to_insert: ArrayVec<[SyntaxElement; 1]> = ArrayVec::new(); | ||
59 | to_insert.push(generic_args.syntax().clone().into()); | ||
60 | self.insert_children(InsertPosition::After(anchor.into()), to_insert) | ||
61 | } | ||
49 | } | 62 | } |
50 | 63 | ||
51 | fn make_multiline<N>(node: N) -> N | 64 | fn make_multiline<N>(node: N) -> N |
@@ -260,16 +273,16 @@ impl ast::Path { | |||
260 | 273 | ||
261 | impl ast::PathSegment { | 274 | impl ast::PathSegment { |
262 | #[must_use] | 275 | #[must_use] |
263 | pub fn with_type_args(&self, type_args: ast::GenericArgList) -> ast::PathSegment { | 276 | pub fn with_generic_args(&self, type_args: ast::GenericArgList) -> ast::PathSegment { |
264 | self._with_type_args(type_args, false) | 277 | self._with_generic_args(type_args, false) |
265 | } | 278 | } |
266 | 279 | ||
267 | #[must_use] | 280 | #[must_use] |
268 | pub fn with_turbo_fish(&self, type_args: ast::GenericArgList) -> ast::PathSegment { | 281 | pub fn with_turbo_fish(&self, type_args: ast::GenericArgList) -> ast::PathSegment { |
269 | self._with_type_args(type_args, true) | 282 | self._with_generic_args(type_args, true) |
270 | } | 283 | } |
271 | 284 | ||
272 | fn _with_type_args(&self, type_args: ast::GenericArgList, turbo: bool) -> ast::PathSegment { | 285 | fn _with_generic_args(&self, type_args: ast::GenericArgList, turbo: bool) -> ast::PathSegment { |
273 | if let Some(old) = self.generic_arg_list() { | 286 | if let Some(old) = self.generic_arg_list() { |
274 | return self.replace_children( | 287 | return self.replace_children( |
275 | single_node(old.syntax().clone()), | 288 | single_node(old.syntax().clone()), |
@@ -459,6 +472,61 @@ impl ast::MatchArmList { | |||
459 | } | 472 | } |
460 | } | 473 | } |
461 | 474 | ||
475 | impl ast::GenericParamList { | ||
476 | #[must_use] | ||
477 | pub fn append_params( | ||
478 | &self, | ||
479 | params: impl IntoIterator<Item = ast::GenericParam>, | ||
480 | ) -> ast::GenericParamList { | ||
481 | let mut res = self.clone(); | ||
482 | params.into_iter().for_each(|it| res = res.append_param(it)); | ||
483 | res | ||
484 | } | ||
485 | |||
486 | #[must_use] | ||
487 | pub fn append_param(&self, item: ast::GenericParam) -> ast::GenericParamList { | ||
488 | let space = tokens::single_space(); | ||
489 | |||
490 | let mut to_insert: ArrayVec<[SyntaxElement; 4]> = ArrayVec::new(); | ||
491 | if self.generic_params().next().is_some() { | ||
492 | to_insert.push(space.into()); | ||
493 | } | ||
494 | to_insert.push(item.syntax().clone().into()); | ||
495 | |||
496 | macro_rules! after_l_angle { | ||
497 | () => {{ | ||
498 | let anchor = match self.l_angle_token() { | ||
499 | Some(it) => it.into(), | ||
500 | None => return self.clone(), | ||
501 | }; | ||
502 | InsertPosition::After(anchor) | ||
503 | }}; | ||
504 | } | ||
505 | |||
506 | macro_rules! after_field { | ||
507 | ($anchor:expr) => { | ||
508 | if let Some(comma) = $anchor | ||
509 | .syntax() | ||
510 | .siblings_with_tokens(Direction::Next) | ||
511 | .find(|it| it.kind() == T![,]) | ||
512 | { | ||
513 | InsertPosition::After(comma) | ||
514 | } else { | ||
515 | to_insert.insert(0, make::token(T![,]).into()); | ||
516 | InsertPosition::After($anchor.syntax().clone().into()) | ||
517 | } | ||
518 | }; | ||
519 | }; | ||
520 | |||
521 | let position = match self.generic_params().last() { | ||
522 | Some(it) => after_field!(it), | ||
523 | None => after_l_angle!(), | ||
524 | }; | ||
525 | |||
526 | self.insert_children(position, to_insert) | ||
527 | } | ||
528 | } | ||
529 | |||
462 | #[must_use] | 530 | #[must_use] |
463 | pub fn remove_attrs_and_docs<N: ast::AttrsOwner>(node: &N) -> N { | 531 | pub fn remove_attrs_and_docs<N: ast::AttrsOwner>(node: &N) -> N { |
464 | N::cast(remove_attrs_and_docs_inner(node.syntax().clone())).unwrap() | 532 | N::cast(remove_attrs_and_docs_inner(node.syntax().clone())).unwrap() |