Add Body::range_for_any() function

Summary: As title Reviewed By: robertoaloi Differential Revision: D55746858 fbshipit-source-id: e980d335919e757ebce6a72c3d60153b71cb359c
WhatsApp · Apr 5, 2024 · 261ca49 · 261ca49
1 parent bb97a02
commit 261ca49
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/crates/hir/src/body.rs b/crates/hir/src/body.rs
@@ -223,6 +223,12 @@ impl Body {
         }
     }
 
+    pub fn range_for_any(&self, sema: &Semantic, id: AnyExprId) -> Option<TextRange> {
+        let body_map = self.get_body_map(sema)?;
+        let ast = body_map.any(id)?;
+        Some(ast.range())
+    }
+
     pub fn print_any_expr(&self, db: &dyn InternDatabase, expr: AnyExprId) -> String {
         match expr {
             AnyExprId::Expr(expr_id) => pretty::print_expr(db, self, expr_id),