refactor: expose expression dependencies from hir

Author: tchupp

compiler/hir/hir_def/src/hir.rs

@@ -148,7 +148,7 @@ impl<'db> LoweringCtx<'db> {
             value_definitions.insert(expression_id, value_definition);
         }
 
-        let dep_graph = dependency_graph::DependencyGraph::new(self.expression_deps);
+        let dep_graph = dependency_graph::DependencyGraph::new(self.expression_deps.clone());
         let expression_groups = dep_graph.topological_order();
 
         HirModule::new(
@@ -162,6 +162,7 @@ impl<'db> LoweringCtx<'db> {
             self.behaviors,
             value_definitions,
             expression_groups,
+            self.expression_deps,
             self.scopes,
             warnings,
             self.errors,

compiler/hir/hir_def/src/hir_module.rs

@@ -1,7 +1,7 @@
 use super::*;
 use crate::index::{Index, IndexItem};
 use alloy_scope::{ScopeIdx, Scopes};
-use rustc_hash::FxHashMap;
+use rustc_hash::{FxHashMap, FxHashSet};
 use text_size::TextRange;
 
 #[derive(Debug, Clone, PartialEq)]
@@ -23,6 +23,7 @@ pub struct HirModule {
     behaviors: Index<Behavior, (TypeIdx, TypeIdx)>,
     value_definitions: FxHashMap<ExpressionIdx, ValueDefinition>,
     expression_groups: Vec<Vec<ExpressionIdx>>,
+    expression_dependencies: FxHashMap<ExpressionIdx, FxHashSet<ExpressionIdx>>,
     scopes: Scopes,
     warnings: Vec<LoweringWarning>,
     errors: Vec<LoweringError>,
@@ -76,6 +77,7 @@ impl HirModule {
             behaviors: Default::default(),
             value_definitions: Default::default(),
             expression_groups: Vec::new(),
+            expression_dependencies: Default::default(),
             scopes: Default::default(),
             warnings: Vec::new(),
             errors: Vec::new(),
@@ -94,6 +96,7 @@ impl HirModule {
         behaviors: Index<Behavior, (TypeIdx, TypeIdx)>,
         value_definitions: FxHashMap<ExpressionIdx, ValueDefinition>,
         expression_groups: Vec<Vec<ExpressionIdx>>,
+        expression_dependencies: FxHashMap<ExpressionIdx, FxHashSet<ExpressionIdx>>,
         scopes: Scopes,
         warnings: Vec<LoweringWarning>,
         errors: Vec<LoweringError>,
@@ -109,6 +112,7 @@ impl HirModule {
             behaviors,
             value_definitions,
             expression_groups,
+            expression_dependencies,
             scopes,
             warnings,
             errors,
@@ -224,6 +228,13 @@ impl HirModule {
         self.expressions.get(idx)
     }
 
+    pub fn get_expression_dependencies(&self, idx: ExpressionIdx) -> FxHashSet<ExpressionIdx> {
+        self.expression_dependencies
+            .get(&idx)
+            .cloned()
+            .unwrap_or_default()
+    }
+
     pub fn get_expression_range(&self, idx: ExpressionIdx) -> TextRange {
         self.expressions.get_range(idx)
     }