From 087122ad1a67c13fda313b30bf5e0817a7bfd79e Mon Sep 17 00:00:00 2001
From: Lukas Wirth <lukastw97@gmail.com>
Date: Mon, 3 Jan 2022 01:32:03 +0100
Subject: [PATCH] fix: Fix highlighting hack for self-params

---
 crates/ide/src/syntax_highlighting/highlight.rs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/crates/ide/src/syntax_highlighting/highlight.rs b/crates/ide/src/syntax_highlighting/highlight.rs
index 4a09d341f7..41b13170bb 100644
--- a/crates/ide/src/syntax_highlighting/highlight.rs
+++ b/crates/ide/src/syntax_highlighting/highlight.rs
@@ -712,11 +712,13 @@ fn parent_matches<N: AstNode>(token: &SyntaxToken) -> bool {
     token.parent().map_or(false, |it| N::can_cast(it.kind()))
 }
 
-fn is_in_fn_with_self_param<N: AstNode>(node: &N) -> bool {
+fn is_in_fn_with_self_param(node: &ast::NameRef) -> 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())
+        .find_map(ast::Item::cast)
+        .and_then(|item| match item {
+            ast::Item::Fn(fn_) => fn_.param_list()?.self_param(),
+            _ => None,
+        })
         .is_some()
 }