diff options
Diffstat (limited to 'crates/ra_hir/src/code_model')
-rw-r--r-- | crates/ra_hir/src/code_model/src.rs | 128 |
1 files changed, 0 insertions, 128 deletions
diff --git a/crates/ra_hir/src/code_model/src.rs b/crates/ra_hir/src/code_model/src.rs deleted file mode 100644 index bf3ee0834..000000000 --- a/crates/ra_hir/src/code_model/src.rs +++ /dev/null | |||
@@ -1,128 +0,0 @@ | |||
1 | //! FIXME: write short doc here | ||
2 | |||
3 | use hir_def::{AstItemDef, HasChildSource, HasSource as _, Lookup, VariantId}; | ||
4 | use hir_expand::either::Either; | ||
5 | use ra_syntax::ast; | ||
6 | |||
7 | use crate::{ | ||
8 | db::DefDatabase, Const, Enum, EnumVariant, FieldSource, Function, ImplBlock, Import, MacroDef, | ||
9 | Module, ModuleSource, Static, Struct, StructField, Trait, TypeAlias, Union, | ||
10 | }; | ||
11 | |||
12 | pub use hir_expand::Source; | ||
13 | |||
14 | pub trait HasSource { | ||
15 | type Ast; | ||
16 | fn source(self, db: &impl DefDatabase) -> Source<Self::Ast>; | ||
17 | } | ||
18 | |||
19 | /// NB: Module is !HasSource, because it has two source nodes at the same time: | ||
20 | /// definition and declaration. | ||
21 | impl Module { | ||
22 | /// Returns a node which defines this module. That is, a file or a `mod foo {}` with items. | ||
23 | pub fn definition_source(self, db: &impl DefDatabase) -> Source<ModuleSource> { | ||
24 | let def_map = db.crate_def_map(self.id.krate); | ||
25 | let src = def_map[self.id.local_id].definition_source(db); | ||
26 | src.map(|it| match it { | ||
27 | Either::A(it) => ModuleSource::SourceFile(it), | ||
28 | Either::B(it) => ModuleSource::Module(it), | ||
29 | }) | ||
30 | } | ||
31 | |||
32 | /// Returns a node which declares this module, either a `mod foo;` or a `mod foo {}`. | ||
33 | /// `None` for the crate root. | ||
34 | pub fn declaration_source(self, db: &impl DefDatabase) -> Option<Source<ast::Module>> { | ||
35 | let def_map = db.crate_def_map(self.id.krate); | ||
36 | def_map[self.id.local_id].declaration_source(db) | ||
37 | } | ||
38 | } | ||
39 | |||
40 | impl HasSource for StructField { | ||
41 | type Ast = FieldSource; | ||
42 | fn source(self, db: &impl DefDatabase) -> Source<FieldSource> { | ||
43 | let var = VariantId::from(self.parent); | ||
44 | let src = var.child_source(db); | ||
45 | src.map(|it| match it[self.id].clone() { | ||
46 | Either::A(it) => FieldSource::Pos(it), | ||
47 | Either::B(it) => FieldSource::Named(it), | ||
48 | }) | ||
49 | } | ||
50 | } | ||
51 | impl HasSource for Struct { | ||
52 | type Ast = ast::StructDef; | ||
53 | fn source(self, db: &impl DefDatabase) -> Source<ast::StructDef> { | ||
54 | self.id.source(db) | ||
55 | } | ||
56 | } | ||
57 | impl HasSource for Union { | ||
58 | type Ast = ast::UnionDef; | ||
59 | fn source(self, db: &impl DefDatabase) -> Source<ast::UnionDef> { | ||
60 | self.id.source(db) | ||
61 | } | ||
62 | } | ||
63 | impl HasSource for Enum { | ||
64 | type Ast = ast::EnumDef; | ||
65 | fn source(self, db: &impl DefDatabase) -> Source<ast::EnumDef> { | ||
66 | self.id.source(db) | ||
67 | } | ||
68 | } | ||
69 | impl HasSource for EnumVariant { | ||
70 | type Ast = ast::EnumVariant; | ||
71 | fn source(self, db: &impl DefDatabase) -> Source<ast::EnumVariant> { | ||
72 | self.parent.id.child_source(db).map(|map| map[self.id].clone()) | ||
73 | } | ||
74 | } | ||
75 | impl HasSource for Function { | ||
76 | type Ast = ast::FnDef; | ||
77 | fn source(self, db: &impl DefDatabase) -> Source<ast::FnDef> { | ||
78 | self.id.lookup(db).source(db) | ||
79 | } | ||
80 | } | ||
81 | impl HasSource for Const { | ||
82 | type Ast = ast::ConstDef; | ||
83 | fn source(self, db: &impl DefDatabase) -> Source<ast::ConstDef> { | ||
84 | self.id.lookup(db).source(db) | ||
85 | } | ||
86 | } | ||
87 | impl HasSource for Static { | ||
88 | type Ast = ast::StaticDef; | ||
89 | fn source(self, db: &impl DefDatabase) -> Source<ast::StaticDef> { | ||
90 | self.id.lookup(db).source(db) | ||
91 | } | ||
92 | } | ||
93 | impl HasSource for Trait { | ||
94 | type Ast = ast::TraitDef; | ||
95 | fn source(self, db: &impl DefDatabase) -> Source<ast::TraitDef> { | ||
96 | self.id.source(db) | ||
97 | } | ||
98 | } | ||
99 | impl HasSource for TypeAlias { | ||
100 | type Ast = ast::TypeAliasDef; | ||
101 | fn source(self, db: &impl DefDatabase) -> Source<ast::TypeAliasDef> { | ||
102 | self.id.lookup(db).source(db) | ||
103 | } | ||
104 | } | ||
105 | impl HasSource for MacroDef { | ||
106 | type Ast = ast::MacroCall; | ||
107 | fn source(self, db: &impl DefDatabase) -> Source<ast::MacroCall> { | ||
108 | Source { file_id: self.id.ast_id.file_id(), value: self.id.ast_id.to_node(db) } | ||
109 | } | ||
110 | } | ||
111 | impl HasSource for ImplBlock { | ||
112 | type Ast = ast::ImplBlock; | ||
113 | fn source(self, db: &impl DefDatabase) -> Source<ast::ImplBlock> { | ||
114 | self.id.source(db) | ||
115 | } | ||
116 | } | ||
117 | impl HasSource for Import { | ||
118 | type Ast = Either<ast::UseTree, ast::ExternCrateItem>; | ||
119 | |||
120 | /// Returns the syntax of the last path segment corresponding to this import | ||
121 | fn source(self, db: &impl DefDatabase) -> Source<Self::Ast> { | ||
122 | let src = self.parent.definition_source(db); | ||
123 | let (_, source_map) = db.raw_items_with_source_map(src.file_id); | ||
124 | let root = db.parse_or_expand(src.file_id).unwrap(); | ||
125 | let ptr = source_map.get(self.id); | ||
126 | src.with_value(ptr.map(|it| it.to_node(&root), |it| it.to_node(&root))) | ||
127 | } | ||
128 | } | ||