1 files changed, 8 insertions, 74 deletions

diff --git a/crates/ra_hir_def/src/src.rs b/crates/ra_hir_def/src/src.rs
index 20200d1db..499375b80 100644
--- a/crates/ra_hir_def/src/src.rs
+++ b/crates/ra_hir_def/src/src.rs
@@ -2,94 +2,28 @@
 
 use hir_expand::InFile;
 use ra_arena::map::ArenaMap;
-use ra_syntax::ast;
+use ra_syntax::AstNode;
 
-use crate::{
-    db::DefDatabase, ConstLoc, EnumLoc, FunctionLoc, ImplLoc, StaticLoc, StructLoc, TraitLoc,
-    TypeAliasLoc, UnionLoc,
-};
+use crate::{db::DefDatabase, AssocItemLoc, ItemLoc};
 
 pub trait HasSource {
     type Value;
     fn source(&self, db: &impl DefDatabase) -> InFile<Self::Value>;
 }
 
-impl HasSource for FunctionLoc {
-    type Value = ast::FnDef;
+impl<N: AstNode> HasSource for AssocItemLoc<N> {
+    type Value = N;
 
-    fn source(&self, db: &impl DefDatabase) -> InFile<ast::FnDef> {
+    fn source(&self, db: &impl DefDatabase) -> InFile<N> {
         let node = self.ast_id.to_node(db);
         InFile::new(self.ast_id.file_id, node)
     }
 }
 
-impl HasSource for TypeAliasLoc {
-    type Value = ast::TypeAliasDef;
+impl<N: AstNode> HasSource for ItemLoc<N> {
+    type Value = N;
 
-    fn source(&self, db: &impl DefDatabase) -> InFile<ast::TypeAliasDef> {
-        let node = self.ast_id.to_node(db);
-        InFile::new(self.ast_id.file_id, node)
-    }
-}
-
-impl HasSource for ConstLoc {
-    type Value = ast::ConstDef;
-
-    fn source(&self, db: &impl DefDatabase) -> InFile<ast::ConstDef> {
-        let node = self.ast_id.to_node(db);
-        InFile::new(self.ast_id.file_id, node)
-    }
-}
-
-impl HasSource for StaticLoc {
-    type Value = ast::StaticDef;
-
-    fn source(&self, db: &impl DefDatabase) -> InFile<ast::StaticDef> {
-        let node = self.ast_id.to_node(db);
-        InFile::new(self.ast_id.file_id, node)
-    }
-}
-
-impl HasSource for ImplLoc {
-    type Value = ast::ImplBlock;
-
-    fn source(&self, db: &impl DefDatabase) -> InFile<ast::ImplBlock> {
-        let node = self.ast_id.to_node(db);
-        InFile::new(self.ast_id.file_id, node)
-    }
-}
-
-impl HasSource for TraitLoc {
-    type Value = ast::TraitDef;
-
-    fn source(&self, db: &impl DefDatabase) -> InFile<ast::TraitDef> {
-        let node = self.ast_id.to_node(db);
-        InFile::new(self.ast_id.file_id, node)
-    }
-}
-
-impl HasSource for StructLoc {
-    type Value = ast::StructDef;
-
-    fn source(&self, db: &impl DefDatabase) -> InFile<ast::StructDef> {
-        let node = self.ast_id.to_node(db);
-        InFile::new(self.ast_id.file_id, node)
-    }
-}
-
-impl HasSource for UnionLoc {
-    type Value = ast::UnionDef;
-
-    fn source(&self, db: &impl DefDatabase) -> InFile<ast::UnionDef> {
-        let node = self.ast_id.to_node(db);
-        InFile::new(self.ast_id.file_id, node)
-    }
-}
-
-impl HasSource for EnumLoc {
-    type Value = ast::EnumDef;
-
-    fn source(&self, db: &impl DefDatabase) -> InFile<ast::EnumDef> {
+    fn source(&self, db: &impl DefDatabase) -> InFile<N> {
         let node = self.ast_id.to_node(db);
         InFile::new(self.ast_id.file_id, node)
     }