Merge #1551

1551: several highlighting cleanups r=matklad a=matklad



Co-authored-by: Aleksey Kladov <[email protected]>

1 files changed, 11 insertions, 0 deletions

diff --git a/crates/ra_syntax/src/ptr.rs b/crates/ra_syntax/src/ptr.rs
index 016256075..25824722f 100644
--- a/crates/ra_syntax/src/ptr.rs
+++ b/crates/ra_syntax/src/ptr.rs
@@ -60,6 +60,17 @@ impl<N: AstNode> AstPtr<N> {
     pub fn syntax_node_ptr(self) -> SyntaxNodePtr {
         self.raw
     }
+
+    // FIXME: extend AstNode to do this safely
+    pub fn cast_checking_kind<U: AstNode>(
+        self,
+        cond: impl FnOnce(SyntaxKind) -> bool,
+    ) -> Option<AstPtr<U>> {
+        if !cond(self.raw.kind()) {
+            return None;
+        }
+        Some(AstPtr { raw: self.raw, _ty: PhantomData })
+    }
 }
 
 impl<N: AstNode> From<AstPtr<N>> for SyntaxNodePtr {