mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify highlight token match guards

Browse Source
This commit is contained in:
Lukas Wirth 2021-04-23 17:19:46 +02:00
parent ef6f596b4b
commit 1e88f5ec8e
2 changed files with 92 additions and 102 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

176
crates/ide/src/syntax_highlighting/highlight.rs
View File

@ -1,6 +1,6 @@
//! Computes color for a single element.
use hir::{AsAssocItem, AssocItemContainer, Semantics, VariantDef};
use hir::{AsAssocItem, Semantics};
use ide_db::{
defs::{Definition, NameClass, NameRefClass},
RootDatabase, SymbolKind,
@ -45,28 +45,26 @@ pub(super) fn element(
};
match name_kind {
Some(NameClass::ExternCrate(_)) => HlTag::Symbol(SymbolKind::Module).into(),
Some(NameClass::ExternCrate(_)) => SymbolKind::Module.into(),
Some(NameClass::Definition(def)) => highlight_def(db, def) | HlMod::Definition,
Some(NameClass::ConstReference(def)) => highlight_def(db, def),
Some(NameClass::PatFieldShorthand { field_ref, .. }) => {
let mut h = HlTag::Symbol(SymbolKind::Field).into();
if let Definition::Field(field) = field_ref {
if let VariantDef::Union(_) = field.parent_def(db) {
if let hir::VariantDef::Union(_) = field.parent_def(db) {
h |= HlMod::Unsafe;
}
}
h
}
None => highlight_name_by_syntax(name) | HlMod::Definition,
}
}
// Highlight references like the definitions they resolve to
NAME_REF if element.ancestors().any(|it| it.kind() == ATTR) => {
// even though we track whether we are in an attribute or not we still need this special case
// as otherwise we would emit unresolved references for name refs inside attributes
Highlight::from(HlTag::Symbol(SymbolKind::Function))
SymbolKind::Function.into()
}
NAME_REF => {
let name_ref = element.into_node().and_then(ast::NameRef::cast).unwrap();
@ -74,7 +72,7 @@ pub(super) fn element(
let is_self = name_ref.self_token().is_some();
let h = match NameRefClass::classify(sema, &name_ref) {
Some(name_kind) => match name_kind {
NameRefClass::ExternCrate(_) => HlTag::Symbol(SymbolKind::Module).into(),
NameRefClass::ExternCrate(_) => SymbolKind::Module.into(),
NameRefClass::Definition(def) => {
if let Definition::Local(local) = &def {
if let Some(name) = local.name(db) {
@ -95,7 +93,7 @@ pub(super) fn element(
if let Some(parent) = name_ref.syntax().parent() {
if matches!(parent.kind(), FIELD_EXPR | RECORD_PAT_FIELD) {
if let Definition::Field(field) = def {
if let VariantDef::Union(_) = field.parent_def(db) {
if let hir::VariantDef::Union(_) = field.parent_def(db) {
h |= HlMod::Unsafe;
}
}
@ -104,9 +102,7 @@ pub(super) fn element(
h
}
NameRefClass::FieldShorthand { .. } => {
HlTag::Symbol(SymbolKind::Field).into()
}
NameRefClass::FieldShorthand { .. } => SymbolKind::Field.into(),
},
None if syntactic_name_ref_highlighting => {
highlight_name_ref_by_syntax(name_ref, sema)
@ -114,7 +110,7 @@ pub(super) fn element(
None => HlTag::UnresolvedReference.into(),
};
if h.tag == HlTag::Symbol(SymbolKind::Module) && is_self {
HlTag::Symbol(SymbolKind::SelfParam).into()
SymbolKind::SelfParam.into()
} else {
h
}
@ -135,7 +131,7 @@ pub(super) fn element(
INT_NUMBER | FLOAT_NUMBER => HlTag::NumericLiteral.into(),
BYTE => HlTag::ByteLiteral.into(),
CHAR => HlTag::CharLiteral.into(),
QUESTION => Highlight::new(HlTag::Operator(HlOperator::Other)) | HlMod::ControlFlow,
QUESTION => HlTag::Operator(HlOperator::Other) | HlMod::ControlFlow,
LIFETIME => {
let lifetime = element.into_node().and_then(ast::Lifetime::cast).unwrap();
@ -143,44 +139,31 @@ pub(super) fn element(
Some(NameClass::Definition(def)) => highlight_def(db, def) | HlMod::Definition,
None => match NameRefClass::classify_lifetime(sema, &lifetime) {
Some(NameRefClass::Definition(def)) => highlight_def(db, def),
_ => Highlight::new(HlTag::Symbol(SymbolKind::LifetimeParam)),
_ => SymbolKind::LifetimeParam.into(),
},
_ => Highlight::new(HlTag::Symbol(SymbolKind::LifetimeParam)) | HlMod::Definition,
_ => Highlight::from(SymbolKind::LifetimeParam) | HlMod::Definition,
}
}
p if p.is_punct() => match p {
T![&] if element.parent().and_then(ast::BinExpr::cast).is_some() => {
HlTag::Operator(HlOperator::Bitwise).into()
}
T![&] if parent_matches::<ast::BinExpr>(&element) => HlOperator::Bitwise.into(),
T![&] => {
let h = HlTag::Operator(HlOperator::Other).into();
let is_unsafe = element
.parent()
.and_then(ast::RefExpr::cast)
.map(|ref_expr| sema.is_unsafe_ref_expr(&ref_expr))
.unwrap_or(false);
.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![.] => {
HlTag::Operator(HlOperator::Other).into()
}
T![!] if element.parent().and_then(ast::MacroCall::cast).is_some() => {
HlTag::Symbol(SymbolKind::Macro).into()
}
T![!] if element.parent().and_then(ast::NeverType::cast).is_some() => {
HlTag::BuiltinType.into()
}
T![!] if element.parent().and_then(ast::PrefixExpr::cast).is_some() => {
HlTag::Operator(HlOperator::Logical).into()
}
T![*] if element.parent().and_then(ast::PtrType::cast).is_some() => {
HlTag::Keyword.into()
}
T![*] if element.parent().and_then(ast::PrefixExpr::cast).is_some() => {
T![::] | T![->] | T![=>] | T![..] | T![=] | T![@] | T![.] => HlOperator::Other.into(),
T![!] if parent_matches::<ast::MacroCall>(&element) => SymbolKind::Macro.into(),
T![!] if parent_matches::<ast::NeverType>(&element) => HlTag::BuiltinType.into(),
T![!] if parent_matches::<ast::PrefixExpr>(&element) => HlOperator::Logical.into(),
T![*] if parent_matches::<ast::PtrType>(&element) => HlTag::Keyword.into(),
T![*] if parent_matches::<ast::PrefixExpr>(&element) => {
let prefix_expr = element.parent().and_then(ast::PrefixExpr::cast)?;
let expr = prefix_expr.expr()?;
@ -188,12 +171,12 @@ pub(super) fn element(
if ty.is_raw_ptr() {
HlTag::Operator(HlOperator::Other) | HlMod::Unsafe
} else if let Some(ast::PrefixOp::Deref) = prefix_expr.op_kind() {
HlTag::Operator(HlOperator::Other).into()
HlOperator::Other.into()
} else {
HlTag::Punctuation(HlPunct::Other).into()
HlPunct::Other.into()
}
}
T![-] if element.parent().and_then(ast::PrefixExpr::cast).is_some() => {
T![-] if parent_matches::<ast::PrefixExpr>(&element) => {
let prefix_expr = element.parent().and_then(ast::PrefixExpr::cast)?;
let expr = prefix_expr.expr()?;
@ -203,41 +186,31 @@ pub(super) fn element(
}
.into()
}
_ if element.parent().and_then(ast::PrefixExpr::cast).is_some() => {
HlTag::Operator(HlOperator::Other).into()
}
_ if parent_matches::<ast::PrefixExpr>(&element) => HlOperator::Other.into(),
T![+] | T![-] | T![*] | T![/] | T![+=] | T![-=] | T![*=] | T![/=]
if element.parent().and_then(ast::BinExpr::cast).is_some() =>
if parent_matches::<ast::BinExpr>(&element) =>
{
HlTag::Operator(HlOperator::Arithmetic).into()
HlOperator::Arithmetic.into()
}
T![|] | T![&] | T![!] | T![^] | T![|=] | T![&=] | T![^=]
if element.parent().and_then(ast::BinExpr::cast).is_some() =>
if parent_matches::<ast::BinExpr>(&element) =>
{
HlTag::Operator(HlOperator::Bitwise).into()
HlOperator::Bitwise.into()
}
T![&&] | T![||] if element.parent().and_then(ast::BinExpr::cast).is_some() => {
HlTag::Operator(HlOperator::Logical).into()
T![&&] | T![||] if parent_matches::<ast::BinExpr>(&element) => {
HlOperator::Logical.into()
}
T![>] | T![<] | T![==] | T![>=] | T![<=] | T![!=]
if element.parent().and_then(ast::BinExpr::cast).is_some() =>
if parent_matches::<ast::BinExpr>(&element) =>
{
HlTag::Operator(HlOperator::Comparison).into()
HlOperator::Comparison.into()
}
_ if element.parent().and_then(ast::BinExpr::cast).is_some() => {
HlTag::Operator(HlOperator::Other).into()
}
_ if element.parent().and_then(ast::RangeExpr::cast).is_some() => {
HlTag::Operator(HlOperator::Other).into()
}
_ if element.parent().and_then(ast::RangePat::cast).is_some() => {
HlTag::Operator(HlOperator::Other).into()
}
_ if element.parent().and_then(ast::RestPat::cast).is_some() => {
HlTag::Operator(HlOperator::Other).into()
}
_ if element.parent().and_then(ast::Attr::cast).is_some() => HlTag::Attribute.into(),
kind => HlTag::Punctuation(match kind {
_ if parent_matches::<ast::BinExpr>(&element) => HlOperator::Other.into(),
_ if parent_matches::<ast::RangeExpr>(&element) => HlOperator::Other.into(),
_ if parent_matches::<ast::RangePat>(&element) => HlOperator::Other.into(),
_ if parent_matches::<ast::RestPat>(&element) => HlOperator::Other.into(),
_ if parent_matches::<ast::Attr>(&element) => HlTag::Attribute.into(),
kind => match kind {
T!['['] | T![']'] => HlPunct::Bracket,
T!['{'] | T!['}'] => HlPunct::Brace,
T!['('] | T![')'] => HlPunct::Parenthesis,
@ -247,7 +220,7 @@ pub(super) fn element(
T![;] => HlPunct::Semi,
T![.] => HlPunct::Dot,
_ => HlPunct::Other,
})
}
.into(),
},
@ -303,7 +276,6 @@ fn hash<T: std::hash::Hash + std::fmt::Debug>(x: T) -> u64 {
hash((name, shadow_count))
}
}
fn highlight_def(db: &RootDatabase, def: Definition) -> Highlight {
match def {
Definition::Macro(_) => HlTag::Symbol(SymbolKind::Macro),
@ -319,12 +291,12 @@ fn highlight_def(db: &RootDatabase, def: Definition) -> Highlight {
}
match item.container(db) {
AssocItemContainer::Impl(i) => {
hir::AssocItemContainer::Impl(i) => {
if i.trait_(db).is_some() {
h |= HlMod::Trait;
}
}
AssocItemContainer::Trait(_t) => {
hir::AssocItemContainer::Trait(_t) => {
h |= HlMod::Trait;
}
}
@ -344,12 +316,12 @@ fn highlight_def(db: &RootDatabase, def: Definition) -> Highlight {
if let Some(item) = konst.as_assoc_item(db) {
h |= HlMod::Associated;
match item.container(db) {
AssocItemContainer::Impl(i) => {
hir::AssocItemContainer::Impl(i) => {
if i.trait_(db).is_some() {
h |= HlMod::Trait;
}
}
AssocItemContainer::Trait(_t) => {
hir::AssocItemContainer::Trait(_t) => {
h |= HlMod::Trait;
}
}
@ -363,12 +335,12 @@ fn highlight_def(db: &RootDatabase, def: Definition) -> Highlight {
if let Some(item) = type_.as_assoc_item(db) {
h |= HlMod::Associated;
match item.container(db) {
AssocItemContainer::Impl(i) => {
hir::AssocItemContainer::Impl(i) => {
if i.trait_(db).is_some() {
h |= HlMod::Trait;
}
}
AssocItemContainer::Trait(_t) => {
hir::AssocItemContainer::Trait(_t) => {
h |= HlMod::Trait;
}
}
@ -427,7 +399,7 @@ fn highlight_method_call(
method_call: &ast::MethodCallExpr,
) -> Option<Highlight> {
let func = sema.resolve_method_call(&method_call)?;
let mut h = HlTag::Symbol(SymbolKind::Function).into();
let mut h = SymbolKind::Function.into();
h |= HlMod::Associated;
if func.is_unsafe(sema.db) || sema.is_unsafe_method_call(&method_call) {
h |= HlMod::Unsafe;
@ -463,20 +435,20 @@ fn highlight_name_by_syntax(name: ast::Name) -> Highlight {
};
let tag = match parent.kind() {
STRUCT => HlTag::Symbol(SymbolKind::Struct),
ENUM => HlTag::Symbol(SymbolKind::Enum),
VARIANT => HlTag::Symbol(SymbolKind::Variant),
UNION => HlTag::Symbol(SymbolKind::Union),
TRAIT => HlTag::Symbol(SymbolKind::Trait),
TYPE_ALIAS => HlTag::Symbol(SymbolKind::TypeAlias),
TYPE_PARAM => HlTag::Symbol(SymbolKind::TypeParam),
RECORD_FIELD => HlTag::Symbol(SymbolKind::Field),
MODULE => HlTag::Symbol(SymbolKind::Module),
FN => HlTag::Symbol(SymbolKind::Function),
CONST => HlTag::Symbol(SymbolKind::Const),
STATIC => HlTag::Symbol(SymbolKind::Static),
IDENT_PAT => HlTag::Symbol(SymbolKind::Local),
_ => default,
STRUCT => SymbolKind::Struct,
ENUM => SymbolKind::Enum,
VARIANT => SymbolKind::Variant,
UNION => SymbolKind::Union,
TRAIT => SymbolKind::Trait,
TYPE_ALIAS => SymbolKind::TypeAlias,
TYPE_PARAM => SymbolKind::TypeParam,
RECORD_FIELD => SymbolKind::Field,
MODULE => SymbolKind::Module,
FN => SymbolKind::Function,
CONST => SymbolKind::Const,
STATIC => SymbolKind::Static,
IDENT_PAT => SymbolKind::Local,
_ => return default.into(),
};
tag.into()
@ -494,20 +466,15 @@ fn highlight_name_ref_by_syntax(name: ast::NameRef, sema: &Semantics<RootDatabas
METHOD_CALL_EXPR => {
return ast::MethodCallExpr::cast(parent)
.and_then(|it| highlight_method_call(sema, &it))
.unwrap_or_else(|| HlTag::Symbol(SymbolKind::Function).into());
.unwrap_or_else(|| SymbolKind::Function.into());
}
FIELD_EXPR => {
let h = HlTag::Symbol(SymbolKind::Field);
let is_union = ast::FieldExpr::cast(parent)
.and_then(|field_expr| {
let field = sema.resolve_field(&field_expr)?;
Some(if let VariantDef::Union(_) = field.parent_def(sema.db) {
true
} else {
false
})
})
.unwrap_or(false);
.and_then(|field_expr| sema.resolve_field(&field_expr))
.map_or(false, |field| {
matches!(field.parent_def(sema.db), hir::VariantDef::Union(_))
});
if is_union {
h | HlMod::Unsafe
} else {
@ -524,9 +491,9 @@ fn highlight_name_ref_by_syntax(name: ast::NameRef, sema: &Semantics<RootDatabas
_ => {
// within path, decide whether it is module or adt by checking for uppercase name
return if name.text().chars().next().unwrap_or_default().is_uppercase() {
HlTag::Symbol(SymbolKind::Struct)
SymbolKind::Struct
} else {
HlTag::Symbol(SymbolKind::Module)
SymbolKind::Module
}
.into();
}
@ -537,11 +504,11 @@ fn highlight_name_ref_by_syntax(name: ast::NameRef, sema: &Semantics<RootDatabas
};
match parent.kind() {
CALL_EXPR => HlTag::Symbol(SymbolKind::Function).into(),
CALL_EXPR => SymbolKind::Function.into(),
_ => if name.text().chars().next().unwrap_or_default().is_uppercase() {
HlTag::Symbol(SymbolKind::Struct)
SymbolKind::Struct
} else {
HlTag::Symbol(SymbolKind::Const)
SymbolKind::Const
}
.into(),
}
@ -576,6 +543,11 @@ fn parents_match(mut node: NodeOrToken<SyntaxNode, SyntaxToken>, mut kinds: &[Sy
kinds.len() == 0
}
#[inline]
fn parent_matches<N: AstNode>(element: &SyntaxElement) -> bool {
element.parent().map_or(false, |it| N::can_cast(it.kind()))
}
fn is_child_of_impl(element: &SyntaxElement) -> bool {
match element.parent() {
Some(e) => e.kind() == IMPL,

18
crates/ide/src/syntax_highlighting/tags.rs
View File

@ -234,6 +234,24 @@ fn from(tag: HlTag) -> Highlight {
}
}
impl From<HlOperator> for Highlight {
fn from(op: HlOperator) -> Highlight {
Highlight::new(HlTag::Operator(op))
}
}
impl From<HlPunct> for Highlight {
fn from(punct: HlPunct) -> Highlight {
Highlight::new(HlTag::Punctuation(punct))
}
}
impl From<SymbolKind> for Highlight {
fn from(sym: SymbolKind) -> Highlight {
Highlight::new(HlTag::Symbol(sym))
}
}
impl Highlight {
pub(crate) fn new(tag: HlTag) -> Highlight {
Highlight { tag, mods: HlMods::default() }