diff options
Diffstat (limited to 'crates/ra_hir_def/src/nameres.rs')
-rw-r--r-- | crates/ra_hir_def/src/nameres.rs | 108 |
1 files changed, 3 insertions, 105 deletions
diff --git a/crates/ra_hir_def/src/nameres.rs b/crates/ra_hir_def/src/nameres.rs index af52fa36e..0b486ce16 100644 --- a/crates/ra_hir_def/src/nameres.rs +++ b/crates/ra_hir_def/src/nameres.rs | |||
@@ -57,8 +57,7 @@ mod tests; | |||
57 | 57 | ||
58 | use std::sync::Arc; | 58 | use std::sync::Arc; |
59 | 59 | ||
60 | use hir_expand::{diagnostics::DiagnosticSink, name::Name, InFile, MacroDefId}; | 60 | use hir_expand::{diagnostics::DiagnosticSink, name::Name, InFile}; |
61 | use once_cell::sync::Lazy; | ||
62 | use ra_arena::Arena; | 61 | use ra_arena::Arena; |
63 | use ra_db::{CrateId, Edition, FileId, FilePosition}; | 62 | use ra_db::{CrateId, Edition, FileId, FilePosition}; |
64 | use ra_prof::profile; | 63 | use ra_prof::profile; |
@@ -69,12 +68,12 @@ use ra_syntax::{ | |||
69 | use rustc_hash::FxHashMap; | 68 | use rustc_hash::FxHashMap; |
70 | 69 | ||
71 | use crate::{ | 70 | use crate::{ |
72 | builtin_type::BuiltinType, | ||
73 | db::DefDatabase, | 71 | db::DefDatabase, |
72 | item_scope::{BuiltinShadowMode, ModuleScope}, | ||
74 | nameres::{diagnostics::DefDiagnostic, path_resolution::ResolveMode}, | 73 | nameres::{diagnostics::DefDiagnostic, path_resolution::ResolveMode}, |
75 | path::ModPath, | 74 | path::ModPath, |
76 | per_ns::PerNs, | 75 | per_ns::PerNs, |
77 | AstId, ImplId, LocalImportId, LocalModuleId, ModuleDefId, ModuleId, TraitId, | 76 | AstId, ImplId, LocalModuleId, ModuleDefId, ModuleId, |
78 | }; | 77 | }; |
79 | 78 | ||
80 | /// Contains all top-level defs from a macro-expanded crate | 79 | /// Contains all top-level defs from a macro-expanded crate |
@@ -174,107 +173,6 @@ pub struct ModuleData { | |||
174 | pub impls: Vec<ImplId>, | 173 | pub impls: Vec<ImplId>, |
175 | } | 174 | } |
176 | 175 | ||
177 | #[derive(Debug, Default, PartialEq, Eq)] | ||
178 | pub struct ModuleScope { | ||
179 | items: FxHashMap<Name, Resolution>, | ||
180 | /// Macros visable in current module in legacy textual scope | ||
181 | /// | ||
182 | /// For macros invoked by an unquatified identifier like `bar!()`, `legacy_macros` will be searched in first. | ||
183 | /// If it yields no result, then it turns to module scoped `macros`. | ||
184 | /// It macros with name quatified with a path like `crate::foo::bar!()`, `legacy_macros` will be skipped, | ||
185 | /// and only normal scoped `macros` will be searched in. | ||
186 | /// | ||
187 | /// Note that this automatically inherit macros defined textually before the definition of module itself. | ||
188 | /// | ||
189 | /// Module scoped macros will be inserted into `items` instead of here. | ||
190 | // FIXME: Macro shadowing in one module is not properly handled. Non-item place macros will | ||
191 | // be all resolved to the last one defined if shadowing happens. | ||
192 | legacy_macros: FxHashMap<Name, MacroDefId>, | ||
193 | } | ||
194 | |||
195 | static BUILTIN_SCOPE: Lazy<FxHashMap<Name, Resolution>> = Lazy::new(|| { | ||
196 | BuiltinType::ALL | ||
197 | .iter() | ||
198 | .map(|(name, ty)| { | ||
199 | (name.clone(), Resolution { def: PerNs::types(ty.clone().into()), import: None }) | ||
200 | }) | ||
201 | .collect() | ||
202 | }); | ||
203 | |||
204 | /// Shadow mode for builtin type which can be shadowed by module. | ||
205 | #[derive(Debug, Copy, Clone, PartialEq, Eq)] | ||
206 | pub enum BuiltinShadowMode { | ||
207 | // Prefer Module | ||
208 | Module, | ||
209 | // Prefer Other Types | ||
210 | Other, | ||
211 | } | ||
212 | |||
213 | /// Legacy macros can only be accessed through special methods like `get_legacy_macros`. | ||
214 | /// Other methods will only resolve values, types and module scoped macros only. | ||
215 | impl ModuleScope { | ||
216 | pub fn entries<'a>(&'a self) -> impl Iterator<Item = (&'a Name, &'a Resolution)> + 'a { | ||
217 | //FIXME: shadowing | ||
218 | self.items.iter().chain(BUILTIN_SCOPE.iter()) | ||
219 | } | ||
220 | |||
221 | pub fn declarations(&self) -> impl Iterator<Item = ModuleDefId> + '_ { | ||
222 | self.entries() | ||
223 | .filter_map(|(_name, res)| if res.import.is_none() { Some(res.def) } else { None }) | ||
224 | .flat_map(|per_ns| { | ||
225 | per_ns.take_types().into_iter().chain(per_ns.take_values().into_iter()) | ||
226 | }) | ||
227 | } | ||
228 | |||
229 | /// Iterate over all module scoped macros | ||
230 | pub fn macros<'a>(&'a self) -> impl Iterator<Item = (&'a Name, MacroDefId)> + 'a { | ||
231 | self.items | ||
232 | .iter() | ||
233 | .filter_map(|(name, res)| res.def.take_macros().map(|macro_| (name, macro_))) | ||
234 | } | ||
235 | |||
236 | /// Iterate over all legacy textual scoped macros visable at the end of the module | ||
237 | pub fn legacy_macros<'a>(&'a self) -> impl Iterator<Item = (&'a Name, MacroDefId)> + 'a { | ||
238 | self.legacy_macros.iter().map(|(name, def)| (name, *def)) | ||
239 | } | ||
240 | |||
241 | /// Get a name from current module scope, legacy macros are not included | ||
242 | pub fn get(&self, name: &Name, shadow: BuiltinShadowMode) -> Option<&Resolution> { | ||
243 | match shadow { | ||
244 | BuiltinShadowMode::Module => self.items.get(name).or_else(|| BUILTIN_SCOPE.get(name)), | ||
245 | BuiltinShadowMode::Other => { | ||
246 | let item = self.items.get(name); | ||
247 | if let Some(res) = item { | ||
248 | if let Some(ModuleDefId::ModuleId(_)) = res.def.take_types() { | ||
249 | return BUILTIN_SCOPE.get(name).or(item); | ||
250 | } | ||
251 | } | ||
252 | |||
253 | item.or_else(|| BUILTIN_SCOPE.get(name)) | ||
254 | } | ||
255 | } | ||
256 | } | ||
257 | |||
258 | pub fn traits<'a>(&'a self) -> impl Iterator<Item = TraitId> + 'a { | ||
259 | self.items.values().filter_map(|r| match r.def.take_types() { | ||
260 | Some(ModuleDefId::TraitId(t)) => Some(t), | ||
261 | _ => None, | ||
262 | }) | ||
263 | } | ||
264 | |||
265 | fn get_legacy_macro(&self, name: &Name) -> Option<MacroDefId> { | ||
266 | self.legacy_macros.get(name).copied() | ||
267 | } | ||
268 | } | ||
269 | |||
270 | #[derive(Debug, Clone, PartialEq, Eq, Default)] | ||
271 | pub struct Resolution { | ||
272 | /// None for unresolved | ||
273 | pub def: PerNs, | ||
274 | /// ident by which this is imported into local scope. | ||
275 | pub import: Option<LocalImportId>, | ||
276 | } | ||
277 | |||
278 | impl CrateDefMap { | 176 | impl CrateDefMap { |
279 | pub(crate) fn crate_def_map_query( | 177 | pub(crate) fn crate_def_map_query( |
280 | // Note that this doesn't have `+ AstDatabase`! | 178 | // Note that this doesn't have `+ AstDatabase`! |