From 2009ed9619a4d818c286151b1b65c63e7fcb8134 Mon Sep 17 00:00:00 2001
From: Vinicius Stock <vinicius.stock@shopify.com>
Date: Thu, 26 Mar 2026 10:14:08 -0400
Subject: [PATCH] Remove reference diagnostics from documents

---
 rust/rubydex/src/model/document.rs | 30 ------------------------------
 rust/rubydex/src/model/graph.rs    |  2 +-
 2 files changed, 1 insertion(+), 31 deletions(-)

diff --git a/rust/rubydex/src/model/document.rs b/rust/rubydex/src/model/document.rs
index 37f32a7c..b8c84a4f 100644
--- a/rust/rubydex/src/model/document.rs
+++ b/rust/rubydex/src/model/document.rs
@@ -4,7 +4,6 @@ use line_index::LineIndex;
 use url::Url;
 
 use crate::diagnostic::Diagnostic;
-use crate::model::identity_maps::IdentityHashMap;
 use crate::model::ids::{DefinitionId, ReferenceId};
 
 // Represents a document currently loaded into memory. Identified by its unique URI, it holds the edges to all
@@ -17,8 +16,6 @@ pub struct Document {
     method_reference_ids: Vec<ReferenceId>,
     constant_reference_ids: Vec<ReferenceId>,
     diagnostics: Vec<Diagnostic>,
-    definition_diagnostics: IdentityHashMap<DefinitionId, Vec<Diagnostic>>,
-    reference_diagnostics: IdentityHashMap<ReferenceId, Vec<Diagnostic>>,
 }
 
 impl Document {
@@ -31,8 +28,6 @@ impl Document {
             method_reference_ids: Vec::new(),
             constant_reference_ids: Vec::new(),
             diagnostics: Vec::new(),
-            definition_diagnostics: IdentityHashMap::default(),
-            reference_diagnostics: IdentityHashMap::default(),
         }
     }
 
@@ -87,31 +82,6 @@ impl Document {
         self.diagnostics.push(diagnostic);
     }
 
-    #[must_use]
-    pub fn diagnostics_for_definition(&self, id: DefinitionId) -> &[Diagnostic] {
-        self.definition_diagnostics.get(&id).map_or(&[], Vec::as_slice)
-    }
-
-    pub fn add_diagnostic_for_definition(&mut self, id: DefinitionId, diagnostic: Diagnostic) {
-        self.definition_diagnostics.entry(id).or_default().push(diagnostic);
-    }
-
-    #[must_use]
-    pub fn diagnostics_for_reference(&self, id: ReferenceId) -> &[Diagnostic] {
-        self.reference_diagnostics.get(&id).map_or(&[], Vec::as_slice)
-    }
-
-    pub fn add_diagnostic_for_reference(&mut self, id: ReferenceId, diagnostic: Diagnostic) {
-        self.reference_diagnostics.entry(id).or_default().push(diagnostic);
-    }
-
-    pub fn all_diagnostics(&self) -> impl Iterator<Item = &Diagnostic> {
-        self.diagnostics
-            .iter()
-            .chain(self.definition_diagnostics.values().flatten())
-            .chain(self.reference_diagnostics.values().flatten())
-    }
-
     /// Computes the require path for this document given load paths.
     ///
     /// Returns `None` if:
diff --git a/rust/rubydex/src/model/graph.rs b/rust/rubydex/src/model/graph.rs
index d65fb14d..bbf40a5b 100644
--- a/rust/rubydex/src/model/graph.rs
+++ b/rust/rubydex/src/model/graph.rs
@@ -407,7 +407,7 @@ impl Graph {
 
     #[must_use]
     pub fn all_diagnostics(&self) -> Vec<&Diagnostic> {
-        let document_diagnostics = self.documents.values().flat_map(Document::all_diagnostics);
+        let document_diagnostics = self.documents.values().flat_map(Document::diagnostics);
         let declaration_diagnostics = self.declarations.values().flat_map(Declaration::diagnostics);
 
         document_diagnostics.chain(declaration_diagnostics).collect()