Mirrors/rust-analyzer
2
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer synced 2024-12-26 13:03:31 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add LetStmt::set_ty

Browse source 
Way for setting and removing the type ascription of a let stmt
This commit is contained in:
DropDemBits 2023-07-16 16:22:29 -04:00
parent 92422f7488
commit 5fc8cc52e2
No known key found for this signature in database
GPG key ID: 7FE02A6C1EDFA075
1 changed files with 41 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
crates/syntax/src/ast/edit_in_place.rs
View file

@ -646,6 +646,47 @@ impl ast::MatchArmList {
}
}
impl ast::LetStmt {
pub fn set_ty(&self, ty: Option<ast::Type>) {
match ty {
None => {
if let Some(colon_token) = self.colon_token() {
ted::remove(colon_token);
}
if let Some(existing_ty) = self.ty() {
if let Some(sibling) = existing_ty.syntax().prev_sibling_or_token() {
if sibling.kind() == SyntaxKind::WHITESPACE {
ted::remove(sibling);
}
}
ted::remove(existing_ty.syntax());
}
}
Some(new_ty) => {
if self.colon_token().is_none() {
let mut to_insert: Vec<SyntaxElement> = vec![];
let position = match self.pat() {
Some(pat) => Position::after(pat.syntax()),
None => {
to_insert.push(make::tokens::single_space().into());
Position::after(self.let_token().unwrap())
}
};
to_insert.push(make::token(T![:]).into());
ted::insert_all_raw(position, to_insert);
}
ted::insert(Position::after(self.colon_token().unwrap()), new_ty.syntax());
}
}
}
}
impl ast::RecordExprFieldList {
pub fn add_field(&self, field: ast::RecordExprField) {
let is_multiline = self.syntax().text().contains_char('\n');