Unnamed repository; edit this file 'description' to name the repository.
-rw-r--r--crates/ide/src/syntax_highlighting.rs18
-rw-r--r--crates/ide/src/syntax_highlighting/highlight.rs322
-rw-r--r--crates/ide/src/syntax_highlighting/test_data/highlight_doctest.html8
-rw-r--r--crates/ide/src/syntax_highlighting/test_data/highlighting.html2
-rw-r--r--crates/ide_db/src/defs.rs94
5 files changed, 209 insertions, 235 deletions
diff --git a/crates/ide/src/syntax_highlighting.rs b/crates/ide/src/syntax_highlighting.rs
index a908cdb718..28c7c546ff 100644
--- a/crates/ide/src/syntax_highlighting.rs
+++ b/crates/ide/src/syntax_highlighting.rs
@@ -265,6 +265,7 @@ fn traverse(
}
let element = match event {
+ WalkEvent::Enter(NodeOrToken::Token(tok)) if tok.kind() == WHITESPACE => continue,
WalkEvent::Enter(it) => it,
WalkEvent::Leave(NodeOrToken::Token(_)) => continue,
WalkEvent::Leave(NodeOrToken::Node(node)) => {
@@ -347,13 +348,16 @@ fn traverse(
}
// do the normal highlighting
- let element = highlight::element(
- sema,
- krate,
- &mut bindings_shadow_count,
- syntactic_name_ref_highlighting,
- element_to_highlight,
- );
+ let element = match element_to_highlight {
+ NodeOrToken::Node(node) => highlight::node(
+ sema,
+ krate,
+ &mut bindings_shadow_count,
+ syntactic_name_ref_highlighting,
+ node,
+ ),
+ NodeOrToken::Token(token) => highlight::token(sema, krate, token).zip(Some(None)),
+ };
if let Some((mut highlight, binding_hash)) = element {
if inside_attribute {
highlight |= HlMod::Attribute
diff --git a/crates/ide/src/syntax_highlighting/highlight.rs b/crates/ide/src/syntax_highlighting/highlight.rs
index 3b322bd525..4a09d341f7 100644
--- a/crates/ide/src/syntax_highlighting/highlight.rs
+++ b/crates/ide/src/syntax_highlighting/highlight.rs
@@ -8,7 +8,7 @@ use ide_db::{
};
use rustc_hash::FxHashMap;
use syntax::{
- ast, match_ast, AstNode, AstToken, NodeOrToken, SyntaxElement,
+ ast, match_ast, AstNode, AstToken, NodeOrToken,
SyntaxKind::{self, *},
SyntaxNode, SyntaxToken, T,
};
@@ -18,174 +18,46 @@ use crate::{
Highlight, HlMod, HlTag,
};
-pub(super) fn element(
- sema: &Semantics<RootDatabase>,
- krate: Option<hir::Crate>,
- bindings_shadow_count: &mut FxHashMap<hir::Name, u32>,
- syntactic_name_ref_highlighting: bool,
- element: SyntaxElement,
-) -> Option<(Highlight, Option<u64>)> {
- match element {
- NodeOrToken::Node(it) => {
- node(sema, krate, bindings_shadow_count, syntactic_name_ref_highlighting, it)
- }
- NodeOrToken::Token(it) => Some((token(sema, krate, it)?, None)),
- }
-}
-
-fn token(
+pub(super) fn token(
sema: &Semantics<RootDatabase>,
krate: Option<hir::Crate>,
token: SyntaxToken,
) -> Option<Highlight> {
- let highlight: Highlight = if let Some(comment) = ast::Comment::cast(token.clone()) {
+ if let Some(comment) = ast::Comment::cast(token.clone()) {
let h = HlTag::Comment;
- match comment.kind().doc {
+ return Some(match comment.kind().doc {
Some(_) => h | HlMod::Documentation,
None => h.into(),
+ });
+ }
+
+ let highlight: Highlight = match token.kind() {
+ STRING | BYTE_STRING => HlTag::StringLiteral.into(),
+ INT_NUMBER if token.ancestors().nth(1).map(|it| it.kind()) == Some(FIELD_EXPR) => {
+ SymbolKind::Field.into()
}
- } else {
- match token.kind() {
- STRING | BYTE_STRING => HlTag::StringLiteral.into(),
- INT_NUMBER if token.ancestors().nth(1).map_or(false, |it| it.kind() == FIELD_EXPR) => {
- SymbolKind::Field.into()
- }
- INT_NUMBER | FLOAT_NUMBER => HlTag::NumericLiteral.into(),
- BYTE => HlTag::ByteLiteral.into(),
- CHAR => HlTag::CharLiteral.into(),
- T![?] => HlTag::Operator(HlOperator::Other) | HlMod::ControlFlow,
- IDENT if parent_matches::<ast::TokenTree>(&token) => {
- if let Some(attr) = token.ancestors().nth(2).and_then(ast::Attr::cast) {
+ INT_NUMBER | FLOAT_NUMBER => HlTag::NumericLiteral.into(),
+ BYTE => HlTag::ByteLiteral.into(),
+ CHAR => HlTag::CharLiteral.into(),
+ IDENT if parent_matches::<ast::TokenTree>(&token) => {
+ match token.ancestors().nth(2).and_then(ast::Attr::cast) {
+ Some(attr) => {
match try_resolve_derive_input(sema, &attr, &ast::Ident::cast(token).unwrap()) {
Some(res) => highlight_def(sema, krate, Definition::from(res)),
None => HlTag::None.into(),
}
- } else {
- HlTag::None.into()
- }
- }
- p if p.is_punct() => match p {
- T![&] if parent_matches::<ast::BinExpr>(&token) => HlOperator::Bitwise.into(),
- T![&] => {
- let h = HlTag::Operator(HlOperator::Other).into();
- let is_unsafe = token
- .parent()
- .and_then(ast::RefExpr::cast)
- .map_or(false, |ref_expr| sema.is_unsafe_ref_expr(&ref_expr));
- if is_unsafe {
- h | HlMod::Unsafe
- } else {
- h
- }
- }
- T![::] | T![->] | T![=>] | T![..] | T![=] | T![@] | T![.] => {
- HlOperator::Other.into()
- }
- T![!] if parent_matches::<ast::MacroCall>(&token) => HlPunct::MacroBang.into(),
- T![!] if parent_matches::<ast::NeverType>(&token) => HlTag::BuiltinType.into(),
- T![!] if parent_matches::<ast::PrefixExpr>(&token) => HlOperator::Logical.into(),
- T![*] if parent_matches::<ast::PtrType>(&token) => HlTag::Keyword.into(),
- T![*] if parent_matches::<ast::PrefixExpr>(&token) => {
- let prefix_expr = token.parent().and_then(ast::PrefixExpr::cast)?;
-
- let expr = prefix_expr.expr()?;
- let ty = sema.type_of_expr(&expr)?.original;
- if ty.is_raw_ptr() {
- HlTag::Operator(HlOperator::Other) | HlMod::Unsafe
- } else if let Some(ast::UnaryOp::Deref) = prefix_expr.op_kind() {
- HlOperator::Other.into()
- } else {
- HlPunct::Other.into()
- }
- }
- T![-] if parent_matches::<ast::PrefixExpr>(&token) => {
- let prefix_expr = token.parent().and_then(ast::PrefixExpr::cast)?;
-
- let expr = prefix_expr.expr()?;
- match expr {
- ast::Expr::Literal(_) => HlTag::NumericLiteral,
- _ => HlTag::Operator(HlOperator::Other),
- }
- .into()
- }
- _ if parent_matches::<ast::PrefixExpr>(&token) => HlOperator::Other.into(),
- T![+] | T![-] | T![*] | T![/] if parent_matches::<ast::BinExpr>(&token) => {
- HlOperator::Arithmetic.into()
- }
- T![+=] | T![-=] | T![*=] | T![/=] if parent_matches::<ast::BinExpr>(&token) => {
- Highlight::from(HlOperator::Arithmetic) | HlMod::Mutable
- }
- T![|] | T![&] | T![!] | T![^] if parent_matches::<ast::BinExpr>(&token) => {
- HlOperator::Bitwise.into()
- }
- T![|=] | T![&=] | T![^=] if parent_matches::<ast::BinExpr>(&token) => {
- Highlight::from(HlOperator::Bitwise) | HlMod::Mutable
- }
- T![&&] | T![||] if parent_matches::<ast::BinExpr>(&token) => {
- HlOperator::Logical.into()
- }
- T![>] | T![<] | T![==] | T![>=] | T![<=] | T![!=]
- if parent_matches::<ast::BinExpr>(&token) =>
- {
- HlOperator::Comparison.into()
- }
- _ if parent_matches::<ast::BinExpr>(&token) => HlOperator::Other.into(),
- _ if parent_matches::<ast::RangeExpr>(&token) => HlOperator::Other.into(),
- _ if parent_matches::<ast::RangePat>(&token) => HlOperator::Other.into(),
- _ if parent_matches::<ast::RestPat>(&token) => HlOperator::Other.into(),
- _ if parent_matches::<ast::Attr>(&token) => HlTag::AttributeBracket.into(),
- kind => match kind {
- T!['['] | T![']'] => HlPunct::Bracket,
- T!['{'] | T!['}'] => HlPunct::Brace,
- T!['('] | T![')'] => HlPunct::Parenthesis,
- T![<] | T![>] => HlPunct::Angle,
- T![,] => HlPunct::Comma,
- T![:] => HlPunct::Colon,
- T![;] => HlPunct::Semi,
- T![.] => HlPunct::Dot,
- _ => HlPunct::Other,
- }
- .into(),
- },
- k if k.is_keyword() => {
- let h = Highlight::new(HlTag::Keyword);
- match k {
- T![await] => h | HlMod::Async | HlMod::ControlFlow,
- T![break]
- | T![continue]
- | T![else]
- | T![if]
- | T![in]
- | T![loop]
- | T![match]
- | T![return]
- | T![while]
- | T![yield] => h | HlMod::ControlFlow,
- T![for] if !is_child_of_impl(&token) => h | HlMod::ControlFlow,
- T![unsafe] => h | HlMod::Unsafe,
- T![true] | T![false] => HlTag::BoolLiteral.into(),
- // crate is handled just as a token if it's in an `extern crate`
- T![crate] if parent_matches::<ast::ExternCrate>(&token) => h,
- // self, crate and super are handled as either a Name or NameRef already
- T![self] | T![crate] | T![super] => return None,
- T![ref] => token
- .parent()
- .and_then(ast::IdentPat::cast)
- .and_then(|ident_pat| {
- (sema.is_unsafe_ident_pat(&ident_pat)).then(|| HlMod::Unsafe)
- })
- .map_or(h, |modifier| h | modifier),
- T![async] => h | HlMod::Async,
- _ => h,
}
+ None => HlTag::None.into(),
}
- _ => return None,
}
+ p if p.is_punct() => punctuation(sema, token, p),
+ k if k.is_keyword() => keyword(sema, token, k)?,
+ _ => return None,
};
Some(highlight)
}
-fn node(
+pub(super) fn node(
sema: &Semantics<RootDatabase>,
krate: Option<hir::Crate>,
bindings_shadow_count: &mut FxHashMap<hir::Name, u32>,
@@ -195,18 +67,6 @@ fn node(
let mut binding_hash = None;
let highlight = match_ast! {
match node {
- ast::Fn(__) => {
- bindings_shadow_count.clear();
- return None;
- },
- ast::Attr(__) => {
- HlTag::AttributeBracket.into()
- },
- // Highlight definitions depending on the "type" of the definition.
- ast::Name(name) => {
- highlight_name(sema, bindings_shadow_count, &mut binding_hash, krate, name)
- },
- // Highlight references like the definitions they resolve to
ast::NameRef(name_ref) => {
highlight_name_ref(
sema,
@@ -217,6 +77,9 @@ fn node(
name_ref,
)
},
+ ast::Name(name) => {
+ highlight_name(sema, bindings_shadow_count, &mut binding_hash, krate, name)
+ },
ast::Lifetime(lifetime) => {
match NameClass::classify_lifetime(sema, &lifetime) {
Some(NameClass::Definition(def)) => {
@@ -229,12 +92,129 @@ fn node(
_ => Highlight::from(SymbolKind::LifetimeParam) | HlMod::Definition,
}
},
- _ => return None,
+ ast::Fn(_) => {
+ bindings_shadow_count.clear();
+ return None;
+ },
+ _ => {
+ if [FN, CONST, STATIC].contains(&node.kind()) {
+ bindings_shadow_count.clear();
+ }
+ return None
+ },
}
};
Some((highlight, binding_hash))
}
+fn punctuation(sema: &Semantics<RootDatabase>, token: SyntaxToken, kind: SyntaxKind) -> Highlight {
+ let parent = token.parent();
+ let parent_kind = parent.as_ref().map_or(EOF, SyntaxNode::kind);
+ match (kind, parent_kind) {
+ (T![?], _) => HlTag::Operator(HlOperator::Other) | HlMod::ControlFlow,
+ (T![&], BIN_EXPR) => HlOperator::Bitwise.into(),
+ (T![&], _) => {
+ let h = HlTag::Operator(HlOperator::Other).into();
+ let is_unsafe = parent
+ .and_then(ast::RefExpr::cast)
+ .map(|ref_expr| sema.is_unsafe_ref_expr(&ref_expr));
+ if let Some(true) = is_unsafe {
+ h | HlMod::Unsafe
+ } else {
+ h
+ }
+ }
+ (T![::] | T![->] | T![=>] | T![..] | T![=] | T![@] | T![.], _) => HlOperator::Other.into(),
+ (T![!], MACRO_CALL) => HlPunct::MacroBang.into(),
+ (T![!], NEVER_TYPE) => HlTag::BuiltinType.into(),
+ (T![!], PREFIX_EXPR) => HlOperator::Logical.into(),
+ (T![*], PTR_TYPE) => HlTag::Keyword.into(),
+ (T![*], PREFIX_EXPR) => {
+ let is_raw_ptr = (|| {
+ let prefix_expr = parent.and_then(ast::PrefixExpr::cast)?;
+ let expr = prefix_expr.expr()?;
+ sema.type_of_expr(&expr)?.original.is_raw_ptr().then(|| ())
+ })();
+ if let Some(()) = is_raw_ptr {
+ HlTag::Operator(HlOperator::Other) | HlMod::Unsafe
+ } else {
+ HlOperator::Other.into()
+ }
+ }
+ (T![-], PREFIX_EXPR) => {
+ let prefix_expr = parent.and_then(ast::PrefixExpr::cast).and_then(|e| e.expr());
+ match prefix_expr {
+ Some(ast::Expr::Literal(_)) => HlTag::NumericLiteral,
+ _ => HlTag::Operator(HlOperator::Other),
+ }
+ .into()
+ }
+ (T![+] | T![-] | T![*] | T![/], BIN_EXPR) => HlOperator::Arithmetic.into(),
+ (T![+=] | T![-=] | T![*=] | T![/=], BIN_EXPR) => {
+ Highlight::from(HlOperator::Arithmetic) | HlMod::Mutable
+ }
+ (T![|] | T![&] | T![!] | T![^], BIN_EXPR) => HlOperator::Bitwise.into(),
+ (T![|=] | T![&=] | T![^=], BIN_EXPR) => {
+ Highlight::from(HlOperator::Bitwise) | HlMod::Mutable
+ }
+ (T![&&] | T![||], BIN_EXPR) => HlOperator::Logical.into(),
+ (T![>] | T![<] | T![==] | T![>=] | T![<=] | T![!=], BIN_EXPR) => {
+ HlOperator::Comparison.into()
+ }
+ (_, PREFIX_EXPR | BIN_EXPR | RANGE_EXPR | RANGE_PAT | REST_PAT) => HlOperator::Other.into(),
+ (_, ATTR) => HlTag::AttributeBracket.into(),
+ (kind, _) => match kind {
+ T!['['] | T![']'] => HlPunct::Bracket,
+ T!['{'] | T!['}'] => HlPunct::Brace,
+ T!['('] | T![')'] => HlPunct::Parenthesis,
+ T![<] | T![>] => HlPunct::Angle,
+ T![,] => HlPunct::Comma,
+ T![:] => HlPunct::Colon,
+ T![;] => HlPunct::Semi,
+ T![.] => HlPunct::Dot,
+ _ => HlPunct::Other,
+ }
+ .into(),
+ }
+}
+
+fn keyword(
+ sema: &Semantics<RootDatabase>,
+ token: SyntaxToken,
+ kind: SyntaxKind,
+) -> Option<Highlight> {
+ let h = Highlight::new(HlTag::Keyword);
+ let h = match kind {
+ T![await] => h | HlMod::Async | HlMod::ControlFlow,
+ T![async] => h | HlMod::Async,
+ T![break]
+ | T![continue]
+ | T![else]
+ | T![if]
+ | T![in]
+ | T![loop]
+ | T![match]
+ | T![return]
+ | T![while]
+ | T![yield] => h | HlMod::ControlFlow,
+ T![for] if parent_matches::<ast::ForExpr>(&token) => h | HlMod::ControlFlow,
+ T![unsafe] => h | HlMod::Unsafe,
+ T![true] | T![false] => HlTag::BoolLiteral.into(),
+ // crate is handled just as a token if it's in an `extern crate`
+ T![crate] if parent_matches::<ast::ExternCrate>(&token) => h,
+ // self, crate and super are handled as either a Name or NameRef already, unless they
+ // are inside unmapped token trees
+ T![self] | T![crate] | T![super] if parent_matches::<ast::NameRef>(&token) => return None,
+ T![self] if parent_matches::<ast::Name>(&token) => return None,
+ T![ref] => match token.parent().and_then(ast::IdentPat::cast) {
+ Some(ident) if sema.is_unsafe_ident_pat(&ident) => h | HlMod::Unsafe,
+ _ => h,
+ },
+ _ => h,
+ };
+ Some(h)
+}
+
fn highlight_name_ref(
sema: &Semantics<RootDatabase>,
krate: Option<hir::Crate>,
@@ -325,10 +305,9 @@ fn highlight_name(
krate: Option<hir::Crate>,
name: ast::Name,
) -> Highlight {
- let db = sema.db;
let name_kind = NameClass::classify(sema, &name);
if let Some(NameClass::Definition(Definition::Local(local))) = &name_kind {
- if let Some(name) = local.name(db) {
+ if let Some(name) = local.name(sema.db) {
let shadow_count = bindings_shadow_count.entry(name.clone()).or_default();
*shadow_count += 1;
*binding_hash = Some(calc_binding_hash(&name, *shadow_count))
@@ -338,7 +317,7 @@ fn highlight_name(
Some(NameClass::Definition(def)) => {
let mut h = highlight_def(sema, krate, def) | HlMod::Definition;
if let Definition::Trait(trait_) = &def {
- if trait_.is_unsafe(db) {
+ if trait_.is_unsafe(sema.db) {
h |= HlMod::Unsafe;
}
}
@@ -347,7 +326,7 @@ fn highlight_name(
Some(NameClass::ConstReference(def)) => highlight_def(sema, krate, def),
Some(NameClass::PatFieldShorthand { field_ref, .. }) => {
let mut h = HlTag::Symbol(SymbolKind::Field).into();
- if let hir::VariantDef::Union(_) = field_ref.parent_def(db) {
+ if let hir::VariantDef::Union(_) = field_ref.parent_def(sema.db) {
h |= HlMod::Unsafe;
}
h
@@ -729,21 +708,14 @@ fn parents_match(mut node: NodeOrToken<SyntaxNode, SyntaxToken>, mut kinds: &[Sy
kinds.is_empty()
}
-#[inline]
fn parent_matches<N: AstNode>(token: &SyntaxToken) -> bool {
token.parent().map_or(false, |it| N::can_cast(it.kind()))
}
-fn is_child_of_impl(token: &SyntaxToken) -> bool {
- match token.parent() {
- Some(e) => e.kind() == IMPL,
- _ => false,
- }
-}
-
fn is_in_fn_with_self_param<N: AstNode>(node: &N) -> bool {
node.syntax()
.ancestors()
+ .take_while(|node| ast::Expr::can_cast(node.kind()) || ast::Fn::can_cast(node.kind()))
.find_map(ast::Fn::cast)
.and_then(|s| s.param_list()?.self_param())
.is_some()
diff --git a/crates/ide/src/syntax_highlighting/test_data/highlight_doctest.html b/crates/ide/src/syntax_highlighting/test_data/highlight_doctest.html
index 7dae4a1a14..50013c06a7 100644
--- a/crates/ide/src/syntax_highlighting/test_data/highlight_doctest.html
+++ b/crates/ide/src/syntax_highlighting/test_data/highlight_doctest.html
@@ -143,12 +143,12 @@ pre { color: #DCDCCC; background: #3F3F3F; font-size: 22px; padd
<span class="comment documentation">///</span>
<span class="comment documentation">/// ```</span>
<span class="comment documentation">/// </span><span class="keyword control injected">loop</span><span class="none injected"> </span><span class="brace injected">{</span><span class="brace injected">}</span>
-<span class="attribute_bracket attribute">#</span><span class="attribute_bracket attribute">[</span><span class="builtin_attr attribute library">cfg_attr</span><span class="parenthesis attribute">(</span><span class="none attribute">not</span><span class="parenthesis attribute">(</span><span class="none attribute">feature</span><span class="attribute_bracket attribute"> </span><span class="operator attribute">=</span><span class="attribute_bracket attribute"> </span><span class="string_literal attribute">"false"</span><span class="parenthesis attribute">)</span><span class="comma attribute">,</span><span class="attribute_bracket attribute"> </span><span class="none attribute">doc</span><span class="attribute_bracket attribute"> </span><span class="operator attribute">=</span><span class="attribute_bracket attribute"> </span><span class="string_literal attribute">"</span><span class="keyword control injected">loop</span><span class="none injected"> </span><span class="brace injected">{</span><span class="brace injected">}</span><span class="string_literal attribute">"</span><span class="parenthesis attribute">)</span><span class="attribute_bracket attribute">]</span>
-<span class="attribute_bracket attribute">#</span><span class="attribute_bracket attribute">[</span><span class="builtin_attr attribute library">doc</span><span class="attribute_bracket attribute"> </span><span class="operator attribute">=</span><span class="attribute_bracket attribute"> </span><span class="string_literal attribute">"</span><span class="keyword control injected">loop</span><span class="none injected"> </span><span class="brace injected">{</span><span class="brace injected">}</span><span class="string_literal attribute">"</span><span class="attribute_bracket attribute">]</span>
+<span class="attribute_bracket attribute">#</span><span class="attribute_bracket attribute">[</span><span class="builtin_attr attribute library">cfg_attr</span><span class="parenthesis attribute">(</span><span class="none attribute">not</span><span class="parenthesis attribute">(</span><span class="none attribute">feature</span> <span class="operator attribute">=</span> <span class="string_literal attribute">"false"</span><span class="parenthesis attribute">)</span><span class="comma attribute">,</span> <span class="none attribute">doc</span> <span class="operator attribute">=</span> <span class="string_literal attribute">"</span><span class="keyword control injected">loop</span><span class="none injected"> </span><span class="brace injected">{</span><span class="brace injected">}</span><span class="string_literal attribute">"</span><span class="parenthesis attribute">)</span><span class="attribute_bracket attribute">]</span>
+<span class="attribute_bracket attribute">#</span><span class="attribute_bracket attribute">[</span><span class="builtin_attr attribute library">doc</span> <span class="operator attribute">=</span> <span class="string_literal attribute">"</span><span class="keyword control injected">loop</span><span class="none injected"> </span><span class="brace injected">{</span><span class="brace injected">}</span><span class="string_literal attribute">"</span><span class="attribute_bracket attribute">]</span>
<span class="comment documentation">/// ```</span>
<span class="comment documentation">///</span>
-<span class="attribute_bracket attribute">#</span><span class="attribute_bracket attribute">[</span><span class="builtin_attr attribute library">cfg_attr</span><span class="parenthesis attribute">(</span><span class="none attribute">feature</span><span class="attribute_bracket attribute"> </span><span class="operator attribute">=</span><span class="attribute_bracket attribute"> </span><span class="string_literal attribute">"alloc"</span><span class="comma attribute">,</span><span class="attribute_bracket attribute"> </span><span class="none attribute">doc</span><span class="attribute_bracket attribute"> </span><span class="operator attribute">=</span><span class="attribute_bracket attribute"> </span><span class="string_literal attribute">"```rust"</span><span class="parenthesis attribute">)</span><span class="attribute_bracket attribute">]</span>
-<span class="attribute_bracket attribute">#</span><span class="attribute_bracket attribute">[</span><span class="builtin_attr attribute library">cfg_attr</span><span class="parenthesis attribute">(</span><span class="none attribute">not</span><span class="parenthesis attribute">(</span><span class="none attribute">feature</span><span class="attribute_bracket attribute"> </span><span class="operator attribute">=</span><span class="attribute_bracket attribute"> </span><span class="string_literal attribute">"alloc"</span><span class="parenthesis attribute">)</span><span class="comma attribute">,</span><span class="attribute_bracket attribute"> </span><span class="none attribute">doc</span><span class="attribute_bracket attribute"> </span><span class="operator attribute">=</span><span class="attribute_bracket attribute"> </span><span class="string_literal attribute">"```ignore"</span><span class="parenthesis attribute">)</span><span class="attribute_bracket attribute">]</span>
+<span class="attribute_bracket attribute">#</span><span class="attribute_bracket attribute">[</span><span class="builtin_attr attribute library">cfg_attr</span><span class="parenthesis attribute">(</span><span class="none attribute">feature</span> <span class="operator attribute">=</span> <span class="string_literal attribute">"alloc"</span><span class="comma attribute">,</span> <span class="none attribute">doc</span> <span class="operator attribute">=</span> <span class="string_literal attribute">"```rust"</span><span class="parenthesis attribute">)</span><span class="attribute_bracket attribute">]</span>
+<span class="attribute_bracket attribute">#</span><span class="attribute_bracket attribute">[</span><span class="builtin_attr attribute library">cfg_attr</span><span class="parenthesis attribute">(</span><span class="none attribute">not</span><span class="parenthesis attribute">(</span><span class="none attribute">feature</span> <span class="operator attribute">=</span> <span class="string_literal attribute">"alloc"</span><span class="parenthesis attribute">)</span><span class="comma attribute">,</span> <span class="none attribute">doc</span> <span class="operator attribute">=</span> <span class="string_literal attribute">"```ignore"</span><span class="parenthesis attribute">)</span><span class="attribute_bracket attribute">]</span>
<span class="comment documentation">/// </span><span class="keyword injected">let</span><span class="none injected"> </span><span class="punctuation injected">_</span><span class="none injected"> </span><span class="operator injected">=</span><span class="none injected"> </span><span class="function injected">example</span><span class="parenthesis injected">(</span><span class="operator injected">&</span><span class="module injected">alloc</span><span class="operator injected">::</span><span class="macro injected">vec</span><span class="macro_bang injected">!</span><span class="bracket injected">[</span><span class="numeric_literal injected">1</span><span class="comma injected">,</span><span class="none injected"> </span><span class="numeric_literal injected">2</span><span class="comma injected">,</span><span class="none injected"> </span><span class="numeric_literal injected">3</span><span class="bracket injected">]</span><span class="parenthesis injected">)</span><span class="semicolon injected">;</span>
<span class="comment documentation">/// ```</span>
<span class="keyword">pub</span> <span class="keyword">fn</span> <span class="function declaration public">mix_and_match</span><span class="parenthesis">(</span><span class="parenthesis">)</span> <span class="brace">{</span><span class="brace">}</span>
diff --git a/crates/ide/src/syntax_highlighting/test_data/highlighting.html b/crates/ide/src/syntax_highlighting/test_data/highlighting.html
index d7f996fba9..1ae07a125f 100644
--- a/crates/ide/src/syntax_highlighting/test_data/highlighting.html
+++ b/crates/ide/src/syntax_highlighting/test_data/highlighting.html
@@ -262,7 +262,7 @@ pre { color: #DCDCCC; background: #3F3F3F; font-size: 22px; padd
<span class="keyword">impl</span> <span class="enum public">Bool</span> <span class="brace">{</span>
<span class="keyword">pub</span> <span class="keyword">const</span> <span class="keyword">fn</span> <span class="function associated consuming declaration public">to_primitive</span><span class="parenthesis">(</span><span class="self_keyword declaration">self</span><span class="parenthesis">)</span> <span class="operator">-&gt;</span> <span class="builtin_type">bool</span> <span class="brace">{</span>
- <span class="unresolved_reference">matches</span><span class="macro_bang">!</span><span class="parenthesis">(</span>self<span class="comma">,</span> Self<span class="colon">:</span><span class="colon">:</span>True<span class="parenthesis">)</span>
+ <span class="unresolved_reference">matches</span><span class="macro_bang">!</span><span class="parenthesis">(</span><span class="keyword">self</span><span class="comma">,</span> Self<span class="colon">:</span><span class="colon">:</span>True<span class="parenthesis">)</span>
<span class="brace">}</span>
<span class="brace">}</span>
<span class="keyword">const</span> <span class="constant declaration">USAGE_OF_BOOL</span><span class="colon">:</span><span class="builtin_type">bool</span> <span class="operator">=</span> <span class="enum public">Bool</span><span class="operator">::</span><span class="enum_variant public">True</span><span class="operator">.</span><span class="function associated consuming public">to_primitive</span><span class="parenthesis">(</span><span class="parenthesis">)</span><span class="semicolon">;</span>
diff --git a/crates/ide_db/src/defs.rs b/crates/ide_db/src/defs.rs
index ffef33b6c8..c9b8c94205 100644
--- a/crates/ide_db/src/defs.rs
+++ b/crates/ide_db/src/defs.rs
@@ -359,18 +359,6 @@ impl NameRefClass {
let parent = name_ref.syntax().parent()?;
- if let Some(method_call) = ast::MethodCallExpr::cast(parent.clone()) {
- if let Some(func) = sema.resolve_method_call(&method_call) {
- return Some(NameRefClass::Definition(Definition::Function(func)));
- }
- }
-
- if let Some(field_expr) = ast::FieldExpr::cast(parent.clone()) {
- if let Some(field) = sema.resolve_field(&field_expr) {
- return Some(NameRefClass::Definition(Definition::Field(field)));
- }
- }
-
if let Some(record_field) = ast::RecordExprField::for_field_name(name_ref) {
if let Some((field, local, _)) = sema.resolve_record_field(&record_field) {
let res = match local {
@@ -383,38 +371,6 @@ impl NameRefClass {
}
}
- if let Some(record_pat_field) = ast::RecordPatField::cast(parent.clone()) {
- if let Some(field) = sema.resolve_record_pat_field(&record_pat_field) {
- let field = Definition::Field(field);
- return Some(NameRefClass::Definition(field));
- }
- }
-
- if let Some(assoc_type_arg) = ast::AssocTypeArg::cast(parent.clone()) {
- if assoc_type_arg.name_ref().as_ref() == Some(name_ref) {
- // `Trait<Assoc = Ty>`
- // ^^^^^
- let path = name_ref.syntax().ancestors().find_map(ast::Path::cast)?;
- let resolved = sema.resolve_path(&path)?;
- if let PathResolution::Def(ModuleDef::Trait(tr)) = resolved {
- // FIXME: resolve in supertraits
- if let Some(ty) = tr
- .items(sema.db)
- .iter()
- .filter_map(|assoc| match assoc {
- hir::AssocItem::TypeAlias(it) => Some(*it),
- _ => None,
- })
- .find(|alias| alias.name(sema.db).to_smol_str() == name_ref.text().as_str())
- {
- return Some(NameRefClass::Definition(Definition::TypeAlias(ty)));
- }
- }
-
- return None;
- }
- }
-
if let Some(path) = ast::PathSegment::cast(parent.clone()).map(|it| it.parent_path()) {
if path.parent_path().is_none() {
if let Some(macro_call) = path.syntax().parent().and_then(ast::MacroCall::cast) {
@@ -428,10 +384,52 @@ impl NameRefClass {
return sema.resolve_path(&path).map(Into::into).map(NameRefClass::Definition);
}
- let extern_crate = ast::ExternCrate::cast(parent)?;
- let krate = sema.resolve_extern_crate(&extern_crate)?;
- let root_module = krate.root_module(sema.db);
- Some(NameRefClass::Definition(Definition::Module(root_module)))
+ match_ast! {
+ match parent {
+ ast::MethodCallExpr(method_call) => {
+ sema.resolve_method_call(&method_call)
+ .map(Definition::Function)
+ .map(NameRefClass::Definition)
+ },
+ ast::FieldExpr(field_expr) => {
+ sema.resolve_field(&field_expr)
+ .map(Definition::Field)
+ .map(NameRefClass::Definition)
+ },
+ ast::RecordPatField(record_pat_field) => {
+ sema.resolve_record_pat_field(&record_pat_field)
+ .map(Definition::Field)
+ .map(NameRefClass::Definition)
+ },
+ ast::AssocTypeArg(_) => {
+ // `Trait<Assoc = Ty>`
+ // ^^^^^
+ let containing_path = name_ref.syntax().ancestors().find_map(ast::Path::cast)?;
+ let resolved = sema.resolve_path(&containing_path)?;
+ if let PathResolution::Def(ModuleDef::Trait(tr)) = resolved {
+ // FIXME: resolve in supertraits
+ if let Some(ty) = tr
+ .items(sema.db)
+ .iter()
+ .filter_map(|&assoc| match assoc {
+ hir::AssocItem::TypeAlias(it) => Some(it),
+ _ => None,
+ })
+ .find(|alias| alias.name(sema.db).to_smol_str() == name_ref.text().as_str())
+ {
+ return Some(NameRefClass::Definition(Definition::TypeAlias(ty)));
+ }
+ }
+ None
+ },
+ ast::ExternCrate(extern_crate) => {
+ let krate = sema.resolve_extern_crate(&extern_crate)?;
+ let root_module = krate.root_module(sema.db);
+ Some(NameRefClass::Definition(Definition::Module(root_module)))
+ },
+ _ => None
+ }
+ }
}
pub fn classify_lifetime(