nordic-dev.net/rust
nordic-dev.net/rust
2
0
Fork 0
mirror of https://github.com/rust-lang/rust.git synced 2024-11-26 00:34:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

added eq_token() method to LetStmt

Browse Source
This commit is contained in:
TomasKralCZ 2020-01-19 18:55:56 +01:00
parent 514df15d9e
commit f67ee69777
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
crates/ra_assists/src/assists/add_explicit_type.rs
View File

@ -1,7 +1,6 @@
use hir::{db::HirDatabase, HirDisplay};
use ra_syntax::{
ast::{self, AstNode, LetStmt, NameOwner},
SyntaxKind::EQ,
TextRange, T,
};
@ -37,7 +36,7 @@ pub(crate) fn add_explicit_type(ctx: AssistCtx<impl HirDatabase>) -> Option<Assi
let name_range = name.syntax().text_range();
// Assist should only be applicable if cursor is between 'let' and '='
let stmt_range = stmt.syntax().text_range();
let eq_range = stmt.syntax().descendants_with_tokens().find(|t| t.kind() == EQ)?.text_range();
let eq_range = stmt.eq_token()?.text_range();
let let_range = TextRange::from_to(stmt_range.start(), eq_range.start());
let cursor_in_range = ctx.frange.range.is_subrange(&let_range);
if !cursor_in_range {

7
crates/ra_syntax/src/ast/extensions.rs
View File

@ -234,6 +234,13 @@ impl ast::LetStmt {
Some(node) => node.kind() == T![;],
}
}
pub fn eq_token(&self) -> Option<SyntaxToken> {
self.syntax()
.descendants_with_tokens()
.find(|t| t.kind() == EQ)
.and_then(|it| it.into_token())
}
}
impl ast::ExprStmt {