Prefer identifier tokens in expand_macro

This commit is contained in:
Lukas Wirth 2021-06-22 12:03:51 +02:00
parent c2aa7782d6
commit b423c61ce6

View File

@ -2,7 +2,10 @@
use hir::Semantics;
use ide_db::RootDatabase;
use syntax::{ast, ted, AstNode, NodeOrToken, SyntaxKind, SyntaxKind::*, SyntaxNode, WalkEvent, T};
use syntax::{
ast, ted, AstNode, NodeOrToken, SyntaxKind, SyntaxKind::*, SyntaxNode, SyntaxToken,
TokenAtOffset, WalkEvent, T,
};
use crate::FilePosition;
@ -26,7 +29,7 @@ pub(crate) fn expand_macro(db: &RootDatabase, position: FilePosition) -> Option<
let sema = Semantics::new(db);
let file = sema.parse(position.file_id);
let tok = file.syntax().token_at_offset(position.offset).left_biased()?;
let tok = pick_best(file.syntax().token_at_offset(position.offset))?;
let mut expanded = None;
let mut name = None;
for node in tok.ancestors() {
@ -54,6 +57,16 @@ pub(crate) fn expand_macro(db: &RootDatabase, position: FilePosition) -> Option<
Some(ExpandedMacro { name: name?, expansion })
}
fn pick_best(tokens: TokenAtOffset<SyntaxToken>) -> Option<SyntaxToken> {
return tokens.max_by_key(priority);
fn priority(n: &SyntaxToken) -> usize {
match n.kind() {
IDENT => 1,
_ => 0,
}
}
}
fn expand_macro_recur(
sema: &Semantics<RootDatabase>,
macro_call: &ast::MacroCall,